#include "mpeg-ps.h"
#include "mpeg-ts.h"
#include "mpeg-ts-proto.h"
#include <assert.h>
#include <string.h>
#include <stdio.h>
#include <map>

static void *ts_alloc(void * /*param*/, size_t bytes)
{
    static char s_buffer[188];
    assert(bytes <= sizeof(s_buffer));
    return s_buffer;
}

static void ts_free(void * /*param*/, void * /*packet*/)
{
    return;
}

static void ts_write(void *param, const void *packet, size_t bytes)
{
    fwrite(packet, bytes, 1, (FILE *)param);
}

inline const char *ts_type(int type)
{
    switch (type) {
    case PSI_STREAM_MP3:
        return "MP3";
    case PSI_STREAM_AAC:
        return "AAC";
    case PSI_STREAM_H264:
        return "H264";
    case PSI_STREAM_H265:
        return "H265";
    default:
        return "*";
    }
}

static int ts_stream(void *ts, int codecid)
{
    static std::map<int, int> streams;
    std::map<int, int>::const_iterator it = streams.find(codecid);
    if (streams.end() != it) {
        return it->second;
    }

    int i = mpeg_ts_add_stream(ts, codecid, NULL, 0);
    streams[codecid] = i;
    return i;
}

static void on_ts_packet(void *ts, int stream, int avtype, int flags, int64_t pts, int64_t dts, const void *data, size_t bytes)
{
    printf("[%s] pts: %08lu, dts: %08lu%s\n", ts_type(avtype), (unsigned long)pts, (unsigned long)dts, flags ? " [I]" : "");

    mpeg_ts_write(ts, ts_stream(ts, avtype), flags, pts, dts, data, bytes);
}

static void mpeg_ts_file(const char *file, void *muxer)
{
    unsigned char ptr[188];
    struct ts_demuxer_t *ts;
    FILE *fp = fopen(file, "rb");

    ts = ts_demuxer_create(on_ts_packet, muxer);
    while (1 == fread(ptr, sizeof(ptr), 1, fp)) {
        ts_demuxer_input(ts, ptr, sizeof(ptr));
    }
    ts_demuxer_flush(ts);
    ts_demuxer_destroy(ts);
    fclose(fp);
}

//mpeg_ts_test("test/fileSequence0.ts", "test/apple.ts")
void mpeg_ts_test(const char *input)
{
    char output[256] = { 0 };
    snprintf(output, sizeof(output), "%s.ts", input);

    struct mpeg_ts_func_t tshandler;
    tshandler.alloc = ts_alloc;
    tshandler.write = ts_write;
    tshandler.free = ts_free;

    FILE *fp = fopen(output, "wb");
    void *ts = mpeg_ts_create(&tshandler, fp);

    mpeg_ts_file(input, ts);

    mpeg_ts_destroy(ts);
    fclose(fp);
}