#include "Renderer.h"
#include "AnimationStrategy.cpp"
#include "TransitionStrategy.cpp"
#include "NoneAnimation.h"
#include "PageTurnAnimation.h"
#include "PanAnimation.h"
#include "PanOneWayAnimation.h"
#include <android/log.h>
#include <algorithm>
#include <cmath>
#include <thread>
#include <chrono>

extern void callNextMediaCallback();

#define LOG_TAG "Renderer"
#define LOGI(...) __android_log_print(ANDROID_LOG_INFO, LOG_TAG, __VA_ARGS__)
#define LOGE(...) __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, __VA_ARGS__)

// ====================================================================
// 생성자, 소멸자 및 설정(Setter) 함수들
// ====================================================================

Renderer::Renderer() {
    randomEngine_.seed(std::chrono::high_resolution_clock::now().time_since_epoch().count());
    setAnimationMode(static_cast<int>(AnimationMode::PAN));
    setTransitionMode(static_cast<int>(TransitionMode::FADE));
}

Renderer::~Renderer() {
    release();
}

void Renderer::release() {
    std::lock_guard<std::mutex> lock(renderMutex_);
    currentMedia_.release();
    nextMedia_.release();
}

void Renderer::setNextMedia(int fd) {
    preloader_.startNextPreload(fd);
}

void Renderer::setAnimationSpeed(float speed) {
    animationSpeed_ = speed > 0 ? speed : 1.0f;
}

void Renderer::setFadeDuration(int durationMs) {
    fadeDurationMs_ = durationMs > 0 ? durationMs : 3000;
}

void Renderer::setPageTurnDelay(int delayMs) {
    pageTurnDelayMs_ = delayMs > 0 ? delayMs : 5000;
}

void Renderer::setAnimationMode(int mode) {
    configuredAnimationMode_ = (mode >= 0 && mode <= static_cast<int>(AnimationMode::PAGE_TURN))
                               ? static_cast<AnimationMode>(mode)
                               : AnimationMode::PAN;
    determineActiveAnimationMode();
}

void Renderer::setTransitionMode(int mode) {
    configuredTransitionMode_ = (mode >= 0 && mode <= static_cast<int>(TransitionMode::MOSAIC))
                                ? static_cast<TransitionMode>(mode)
                                : TransitionMode::FADE;
}

void Renderer::determineActiveAnimationMode() {
    AnimationMode modeToSetActive;
    if (configuredAnimationMode_ == AnimationMode::RANDOM) {
        static const std::vector<AnimationMode> availableModes = {
                AnimationMode::PAN, AnimationMode::ZOOM, AnimationMode::NONE,
                AnimationMode::PAN_ONE_WAY, AnimationMode::PAGE_TURN
        };
        std::uniform_int_distribution<size_t> dist(0, availableModes.size() - 1);
        modeToSetActive = availableModes[dist(randomEngine_)];
    } else {
        modeToSetActive = configuredAnimationMode_;
    }
    activeAnimationMode_ = modeToSetActive;

    switch (activeAnimationMode_) {
        case AnimationMode::PAN:
            animationStrategy_ = std::make_unique<PanAnimation>(animationSpeed_);
            break;
        case AnimationMode::PAN_ONE_WAY:
            animationStrategy_ = std::make_unique<PanOneWayAnimation>(animationSpeed_);
            break;
        case AnimationMode::ZOOM:
            animationStrategy_ = std::make_unique<NoneAnimation>(animationSpeed_);
            break;
        case AnimationMode::PAGE_TURN:
            animationStrategy_ = std::make_unique<PageTurnAnimation>(animationSpeed_, pageTurnDelayMs_);
            break;
        case AnimationMode::NONE:
        default:
            animationStrategy_ = std::make_unique<NoneAnimation>(animationSpeed_);
            break;
        case AnimationMode::RANDOM:
            break;
    }
}

// ====================================================================
// 상태 머신(State Machine) 핸들러 함수들
// ====================================================================

void Renderer::handleAnimationState(ANativeWindow_Buffer& buffer, int surfaceWidth, int surfaceHeight) {
    AnimationState animState;
    if (animationStrategy_) {
        float overflowX = 0.0f, overflowY = 0.0f;
        if (activeAnimationMode_ == AnimationMode::PAN || activeAnimationMode_ == AnimationMode::PAN_ONE_WAY) {
            float mediaW = static_cast<float>(currentMedia_.getWidth());
            float mediaH = static_cast<float>(currentMedia_.getHeight());
            if ((mediaW / mediaH) > (static_cast<float>(surfaceWidth) / surfaceHeight)) {
                float scale = static_cast<float>(surfaceHeight) / mediaH;
                overflowX = std::max(0.0f, mediaW * scale - surfaceWidth);
            } else {
                float scale = static_cast<float>(surfaceWidth) / mediaW;
                overflowY = std::max(0.0f, mediaH * scale - surfaceHeight);
            }
        }
        animState = animationStrategy_->update(overflowX, overflowY);
    }

    memset(buffer.bits, 0, buffer.stride * buffer.height * sizeof(uint32_t));
    float finalOffsetX, finalOffsetY, finalScale;
    if (activeAnimationMode_ == AnimationMode::PAN || activeAnimationMode_ == AnimationMode::PAN_ONE_WAY) {
        float scale;
        if ((static_cast<float>(currentMedia_.getWidth()) / currentMedia_.getHeight()) > (static_cast<float>(surfaceWidth) / surfaceHeight)) {
            scale = static_cast<float>(surfaceHeight) / currentMedia_.getHeight();
        } else {
            scale = static_cast<float>(surfaceWidth) / currentMedia_.getWidth();
        }
        finalScale = scale * animState.scale;
        finalOffsetX = animState.offsetX;
        finalOffsetY = animState.offsetY;
    } else {
        float baseScale, baseOffsetX, baseOffsetY;
        calculateFitScaleAndOffset(currentMedia_, surfaceWidth, surfaceHeight, baseScale, baseOffsetX, baseOffsetY);
        finalScale = baseScale * animState.scale;
        finalOffsetX = baseOffsetX + animState.offsetX;
        finalOffsetY = baseOffsetY + animState.offsetY;
    }
    drawMedia(buffer, currentMedia_, 1.0f, finalOffsetX, finalOffsetY, finalScale);

    if (animState.cycleComplete && nextMedia_.isValid()) {
        currentState_ = RenderState::TRANSITIONING;
        transitionStartTime_ = std::chrono::steady_clock::now();

        TransitionMode transModeToUse = configuredTransitionMode_;
        if (transModeToUse == TransitionMode::RANDOM) {
            std::uniform_int_distribution<int> dist(0, 2);
            transModeToUse = static_cast<TransitionMode>(dist(randomEngine_));
        }

        if (transModeToUse == TransitionMode::SLIDE) {
            transitionStrategy_ = std::make_unique<SlideTransition>(fadeDurationMs_, surfaceWidth);
        } else if (transModeToUse == TransitionMode::MOSAIC) {
            transitionStrategy_ = std::make_unique<MosaicTransition>(fadeDurationMs_, 20, 32, randomEngine_);
        } else {
            transitionStrategy_ = std::make_unique<FadeTransition>(fadeDurationMs_);
        }
        if(transitionStrategy_) transitionStrategy_->reset();
    }
}

void Renderer::handleTransitionState(ANativeWindow_Buffer& buffer, int surfaceWidth, int surfaceHeight) {
    if (!transitionStrategy_ || !nextMedia_.isValid()) {
        currentState_ = RenderState::ANIMATING;
        if(animationStrategy_) animationStrategy_->reset();
        return;
    }

    auto now = std::chrono::steady_clock::now();
    long long elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(now - transitionStartTime_).count();
    // 1. 상태 보고: isComplete()를 호출해서 끝났는지 "물어보기만" 함
    bool isComplete = transitionStrategy_->isComplete(elapsed);

    transitionStrategy_->execute(this, buffer, currentMedia_, nextMedia_, elapsed);

    if (isComplete) {
        currentMedia_ = std::move(nextMedia_);
        currentState_ = RenderState::ANIMATING;
        determineActiveAnimationMode();
        if(animationStrategy_) animationStrategy_->reset();
    }
}

// ====================================================================
// 메인 렌더링 루프 (교통정리 담당)
// ====================================================================
void Renderer::renderFrame(ANativeWindow* window) {
    if (!window) return;
    std::lock_guard<std::mutex> lock(renderMutex_);

    if (!currentMedia_.isValid() && preloader_.isPreloadedDataReady()) {
        currentMedia_ = preloader_.swapAndRelease();
        if (currentMedia_.isValid()) {
            currentState_ = RenderState::ANIMATING;
            determineActiveAnimationMode();
            if(animationStrategy_) animationStrategy_->reset();
            callNextMediaCallback();
        }
    }
    if (!currentMedia_.isValid()) {
        ANativeWindow_Buffer buffer;
        if (ANativeWindow_lock(window, &buffer, nullptr) == 0) {
            memset(buffer.bits, 0, buffer.stride * buffer.height * sizeof(uint32_t));
            ANativeWindow_unlockAndPost(window);
        }
        return;
    }
    if (!nextMedia_.isValid() && preloader_.isPreloadedDataReady()) {
        nextMedia_ = preloader_.swapAndRelease();
        if(nextMedia_.isValid()) { callNextMediaCallback(); }
    }

    ANativeWindow_Buffer buffer;
    if (ANativeWindow_lock(window, &buffer, nullptr) != 0) return;
    int surfaceWidth = ANativeWindow_getWidth(window);
    int surfaceHeight = ANativeWindow_getHeight(window);

    switch (currentState_) {
        case RenderState::ANIMATING:
            handleAnimationState(buffer, surfaceWidth, surfaceHeight);
            break;
        case RenderState::TRANSITIONING:
            handleTransitionState(buffer, surfaceWidth, surfaceHeight);
            break;
    }

    ANativeWindow_unlockAndPost(window);
}

// ====================================================================
// 하위 그리기 함수 및 헬퍼 함수들
// ====================================================================

void Renderer::drawMedia(ANativeWindow_Buffer& buffer, MediaAsset& media, float alpha, float finalOffsetX, float finalOffsetY, float finalScale) {
    if (!media.isValid() || alpha <= 0.0f) return;
    if (media.getType() == MediaAsset::Type::IMAGE) {
        renderImageFrame(media, buffer, finalScale, finalOffsetX, finalOffsetY, alpha);
    } else {
        renderVideoFrame(media, buffer, finalScale, finalOffsetX, finalOffsetY, alpha);
    }
}

void Renderer::renderImageFrame(const MediaAsset& media, ANativeWindow_Buffer& buffer, float scale, float offsetX, float offsetY, float alpha) {
    uint32_t* dstPixels = (uint32_t*)buffer.bits;
    int dstStride = buffer.stride;
    const uint8_t* pixelData = media.getType() == MediaAsset::Type::IMAGE ? media.getImageData() : media.getRgbBuffer().data();
    if (!pixelData) return;

    int imgW = media.getWidth();
    int imgH = media.getHeight();
    uint8_t alphaByte = static_cast<uint8_t>(alpha * 255.0f);

    for (int y = 0; y < buffer.height; ++y) {
        int srcY = static_cast<int>((y + offsetY) / scale);
        if (srcY < 0 || srcY >= imgH) continue;
        uint32_t* dstRow = dstPixels + y * dstStride;
        for (int x = 0; x < buffer.width; ++x) {
            int srcX = static_cast<int>((x + offsetX) / scale);
            if (srcX < 0 || srcX >= imgW) continue;
            const uint8_t* srcPixel = &pixelData[(srcY * imgW + srcX) * 4];
            uint32_t dstPixelValue = dstRow[x];
            uint8_t dstB = (dstPixelValue >> 0) & 0xFF;
            uint8_t dstG = (dstPixelValue >> 8) & 0xFF;
            uint8_t dstR = (dstPixelValue >> 16) & 0xFF;
            uint8_t finalR = (srcPixel[0] * alphaByte + dstR * (255 - alphaByte)) / 255;
            uint8_t finalG = (srcPixel[1] * alphaByte + dstG * (255 - alphaByte)) / 255;
            uint8_t finalB = (srcPixel[2] * alphaByte + dstB * (255 - alphaByte)) / 255;
            dstRow[x] = (0xFF << 24) | (finalR << 16) | (finalG << 8) | finalB;
        }
    }
}

void Renderer::renderVideoFrame(MediaAsset& media, ANativeWindow_Buffer& buffer, float scale, float offsetX, float offsetY, float alpha) {
    AVFormatContext* fmt_ctx = media.getFormatContext();
    AVCodecContext* codec_ctx = media.getCodecContext();
    AVFrame* frame = media.getFrame();
    AVPacket* pkt = media.getPacket();
    SwsContext* sws_ctx = media.getSwsContext();
    int video_stream_idx = media.getVideoStreamIndex();

    if (!fmt_ctx || !codec_ctx || !frame || !pkt || !sws_ctx) return;

    int ret = av_read_frame(fmt_ctx, pkt);
    if (ret >= 0) {
        if (pkt->stream_index == video_stream_idx) {
            if (avcodec_send_packet(codec_ctx, pkt) >= 0) {
                if (avcodec_receive_frame(codec_ctx, frame) == 0) {
                    std::vector<uint8_t>& rgbBuf = media.getRgbBuffer();
                    uint8_t* dst[4] = { rgbBuf.data(), nullptr, nullptr, nullptr };
                    int dstStride_arr[4] = { media.getWidth() * 4, 0, 0, 0 };
                    sws_scale(sws_ctx, frame->data, frame->linesize, 0, media.getHeight(), dst, dstStride_arr);
                }
            }
        }
        av_packet_unref(pkt);
    } else if (ret == AVERROR_EOF) {
        av_seek_frame(fmt_ctx, video_stream_idx, 0, AVSEEK_FLAG_BACKWARD);
    }

    renderImageFrame(media, buffer, scale, offsetX, offsetY, alpha);
}

void Renderer::calculateFitScaleAndOffset(const MediaAsset& media, int surfaceWidth, int surfaceHeight, float& outScale, float& outOffsetX, float& outOffsetY) const {
    if (!media.isValid() || media.getWidth() == 0 || media.getHeight() == 0) {
        outScale = 1.0f; outOffsetX = 0.0f; outOffsetY = 0.0f;
        return;
    }
    float mediaAspect = static_cast<float>(media.getWidth()) / media.getHeight();
    float surfaceAspect = static_cast<float>(surfaceWidth) / surfaceHeight;
    if (mediaAspect > surfaceAspect) {
        outScale = static_cast<float>(surfaceHeight) / media.getHeight();
        outOffsetX = (static_cast<float>(surfaceWidth) - (media.getWidth() * outScale)) / 2.0f;
        outOffsetY = 0.0f;
    } else {
        outScale = static_cast<float>(surfaceWidth) / media.getWidth();
        outOffsetX = 0.0f;
        outOffsetY = (static_cast<float>(surfaceHeight) - (media.getHeight() * outScale)) / 2.0f;
    }
}

/**
 * @brief 현재 렌더러의 내부 상태를 문자열로 반환합니다. (디버깅용)
 */
std::string Renderer::getDebugInfo() const {
    std::stringstream ss;
    ss << "========== Native State ==========\n";
    ss << " Configured Anim: " << static_cast<int>(configuredAnimationMode_) << "\n";
    ss << " Active Anim    : " << static_cast<int>(activeAnimationMode_) << "\n";
    ss << " Configured Trans: " << static_cast<int>(configuredTransitionMode_) << "\n";
    ss << " In CurrentState: " << static_cast<int>(currentState_) << "\n";

    if (currentMedia_.isValid()) {
        ss << " Current Media: VALID ["
           << (currentMedia_.getType() == MediaAsset::Type::IMAGE ? "Image" : "Video")
           << " " << currentMedia_.getWidth() << "x" << currentMedia_.getHeight() << "]\n";
    } else {
        ss << " Current Media: INVALID\n";
    }

    if (nextMedia_.isValid()) {
        ss << " Next Media   : VALID ["
           << (nextMedia_.getType() == MediaAsset::Type::IMAGE ? "Image" : "Video")
           << " " << nextMedia_.getWidth() << "x" << nextMedia_.getHeight() << "]";
    } else {
        ss << " Next Media   : INVALID";
    }
    return ss.str();
}