// Vita3K emulator project
// Copyright (C) 2021 Vita3K team
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

#include "SceJpegUser.h"

#include <codec/state.h>

typedef std::shared_ptr<DecoderState> DecoderPtr;

struct MJpegState {
    bool initialized = false;
    DecoderPtr decoder;
};

struct SceJpegMJpegInitInfo {
    uint32_t size;
    int32_t decoder_count;
    int32_t options;
};

struct SceJpegPitch {
    uint32_t x;
    uint32_t y;
};

struct SceJpegOutputInfo {
    int32_t color_space;
    uint16_t width;
    uint16_t height;
    uint32_t output_size;
    uint32_t unknown1;
    uint32_t unknown2;
    SceJpegPitch pitch[4];
};

EXPORT(int, sceJpegCreateSplitDecoder) {
    return UNIMPLEMENTED();
}

EXPORT(int, sceJpegCsc) {
    return UNIMPLEMENTED();
}

EXPORT(int, sceJpegDecodeMJpeg) {
    return UNIMPLEMENTED();
}

EXPORT(int, sceJpegDecodeMJpegYCbCr, const uint8_t *jpeg_data, uint32_t jpeg_size,
    uint8_t *output, uint32_t output_size, int mode, void *buffer, uint32_t buffer_size) {
    const auto state = host.kernel.obj_store.get<MJpegState>();

    DecoderSize size = {};

    state->decoder->send(jpeg_data, jpeg_size);
    state->decoder->receive(output, &size);

    // Top 16 bits = width, bottom 16 bits = height.
    return (size.width << 16u) | size.height;
}

EXPORT(int, sceJpegDeleteSplitDecoder) {
    return UNIMPLEMENTED();
}

EXPORT(int, sceJpegFinishMJpeg) {
    host.kernel.obj_store.erase<MJpegState>();

    return 0;
}

EXPORT(int, sceJpegGetOutputInfo, const uint8_t *jpeg_data, uint32_t jpeg_size,
    int32_t format, int32_t mode, SceJpegOutputInfo *output) {
    const auto state = host.kernel.obj_store.get<MJpegState>();

    DecoderSize size = {};

    state->decoder->send(jpeg_data, jpeg_size);
    state->decoder->receive(nullptr, &size);

    output->width = size.width;
    output->height = size.height;
    output->color_space = 2u << 16u;
    output->output_size = size.width * size.height * 3;

    return 0;
}

// TODO: Decoder options are ignored for the time being.
EXPORT(int, sceJpegInitMJpeg, int32_t decoder_count) {
    host.kernel.obj_store.create<MJpegState>();
    const auto state = host.kernel.obj_store.get<MJpegState>();
    state->decoder = std::make_shared<MjpegDecoderState>();

    return 0;
}

EXPORT(int, sceJpegInitMJpegWithParam, const SceJpegMJpegInitInfo *info) {
    host.kernel.obj_store.create<MJpegState>();
    const auto state = host.kernel.obj_store.get<MJpegState>();
    state->decoder = std::make_shared<MjpegDecoderState>();

    return 0;
}

EXPORT(int, sceJpegMJpegCsc, uint8_t *rgba, const uint8_t *yuv,
    int32_t size, int32_t image_width, int32_t format, int32_t sampling) {
    uint32_t width = size >> 16u;
    uint32_t height = size & (~0u >> 16u);

    convert_yuv_to_rgb(yuv, rgba, width, height);

    return 0;
}

EXPORT(int, sceJpegSplitDecodeMJpeg) {
    return UNIMPLEMENTED();
}

BRIDGE_IMPL(sceJpegCreateSplitDecoder)
BRIDGE_IMPL(sceJpegCsc)
BRIDGE_IMPL(sceJpegDecodeMJpeg)
BRIDGE_IMPL(sceJpegDecodeMJpegYCbCr)
BRIDGE_IMPL(sceJpegDeleteSplitDecoder)
BRIDGE_IMPL(sceJpegFinishMJpeg)
BRIDGE_IMPL(sceJpegGetOutputInfo)
BRIDGE_IMPL(sceJpegInitMJpeg)
BRIDGE_IMPL(sceJpegInitMJpegWithParam)
BRIDGE_IMPL(sceJpegMJpegCsc)
BRIDGE_IMPL(sceJpegSplitDecodeMJpeg)