#!/bin/bash
#
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2026 AMLOGIC
#
# Media modules load script for AMLbian
# This script load media modules for AMLbian

set -e

MODULES=(
    # Media core modules
    media_clock
    firmware
    pts_server
    decoder_common
    stream_input

    # Video decoder modules
    amvdec_mmpeg12
    amvdec_mmpeg4
    amvdec_vc1
    amvdec_mh264
    amvdec_h265
    amvdec_vp9
    amvdec_mmjpeg
    amvdec_avs2
    amvdec_mavs
    amvdec_av1
    amvdec_h266
    amvdec_avs3

    # V4L2 decoder modules
    amvdec_mmpeg12_v4l
    amvdec_mmpeg4_v4l
    amvdec_mh264_v4l
    amvdec_h265_v4l
    amvdec_vp9_v4l
    amvdec_mmjpeg_v4l
    amvdec_avs2_v4l
    amvdec_mavs_v4l
    amvdec_av1_v4l
    amvdec_avs3_v4l
    amvdec_avs2_fb_v4l
    amvdec_vc1_v4l
    amvdec_h266_v4l
    amvdec_mvc1_v4l

    # Video encoder modules
    encoder_common
    jpegenc
    memalloc
    vc8000
    tahoe

    # Other media modules
    video_framerate_adapter
    media_sync
    amvdec_ports
)

echo "==== Start loading Amlogic media modules ===="

for mod in "${MODULES[@]}"; do
    echo "[INFO] Loading module: $mod"

    if lsmod | grep -q "^${mod}\b"; then
        echo "[SKIP] $mod already loaded"
        continue
    fi

    if modprobe "$mod"; then
        echo "[OK] $mod loaded"
    else
        echo "[FAIL] Failed to load $mod"
    fi

    echo
done

echo "==== Module loading finished ===="