#include "ffcpp/Player.h"
#include <memory>
#include <SDL.h>
#include <SDL_thread.h>
#include <iostream>

#include <chrono>
#include <thread>

namespace ff = ffcpp;

#define WINDOW_WIDTH    640
#define WINDOW_HEIGHT   480

class SDLWindow: public ff::IVideoSink {
private:
    template<typename T> using SDLUniquePtr = std::unique_ptr<T, DECLSPEC void SDLCALL(*)(T*)>;
    using SDLWindowPtr = SDLUniquePtr<SDL_Window>;
    using SDLRendererPtr = SDLUniquePtr<SDL_Renderer>;
    using SDLTexturePtr = SDLUniquePtr<SDL_Texture>;

private:
    SDLWindowPtr _wnd;
    SDLRendererPtr _renderer;
    SDLTexturePtr _texture;

public:
    SDLWindow(): _wnd(nullptr, SDL_DestroyWindow), _renderer(nullptr, SDL_DestroyRenderer), _texture(nullptr, SDL_DestroyTexture) {
        int res = SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO | SDL_INIT_TIMER);
        if(res < 0) throw std::runtime_error("Error initializing SDL");

        _wnd.reset(SDL_CreateWindow("ffPlayer", SDL_WINDOWPOS_UNDEFINED, SDL_WINDOWPOS_UNDEFINED, WINDOW_WIDTH, WINDOW_HEIGHT, SDL_WINDOW_OPENGL));
        if(!_wnd) throw std::runtime_error("Error creating SDL window");

        _renderer.reset(SDL_CreateRenderer(_wnd.get(), -1, 0));
        if(!_renderer) throw std::runtime_error("Error creating SDL renderer");

        _texture.reset(SDL_CreateTexture(_renderer.get(), SDL_PIXELFORMAT_YV12, SDL_TEXTUREACCESS_STREAMING, WINDOW_WIDTH, WINDOW_HEIGHT));
        if(!_texture) throw std::runtime_error("Error creating SDL texture");
    }

public:
    virtual AVPixelFormat getPixelFormat() const noexcept override {
        return AV_PIX_FMT_YUV420P;
    }

    virtual int getWidth() const noexcept override {
        return WINDOW_WIDTH;
    }

    virtual int getHeight() const noexcept override {
        return WINDOW_HEIGHT;
    }

    virtual void drawFrame(void* pixelsData, int pitch) override {
        std::cout << "drawing frame" << std::endl;
        SDL_UpdateTexture(_texture.get(), nullptr, pixelsData, pitch);
        SDL_RenderClear(_renderer.get());
        SDL_RenderCopy(_renderer.get(), _texture.get(), nullptr, nullptr);
        SDL_RenderPresent(_renderer.get());
    }

    virtual void drawPlanarYUVFrame(const void *yPlane, const void *uPlane, const void *vPlane, int yPitch, int uPitch, int vPitch) override {
        SDL_UpdateYUVTexture(_texture.get(), nullptr, (const uint8_t*)yPlane, yPitch, (const uint8_t*)uPlane, uPitch, (const uint8_t*)vPlane, vPitch);
        SDL_RenderClear(_renderer.get());
        SDL_RenderCopy(_renderer.get(), _texture.get(), nullptr, nullptr);
        SDL_RenderPresent(_renderer.get());
        //std::this_thread::sleep_for(std::chrono::milliseconds(40));
    }
};

int main(int argc, char** argv) {
    auto wnd = std::make_shared<SDLWindow>();
    ff::Player player(wnd);
    player.setMedia(argv[1]);
    player.play();

    return 0;
}