#include "common.h"
#include "codecfactory.h"
#include "caltime.h"
#include "lr.h"
#include "piecewise_fix_integer_template.h"
#include "piecewise_fix_integer_template_float.h"
#include "piecewise_cost_merge_integer_template_double.h"
#include "FOR_integer_template.h"
#include "delta_integer_template.h"
#include "delta_cost_integer_template.h"
#include "delta_cost_merge_integer_template.h"
#include "piecewise_cost_merge_integer_template_test.h"

typedef uint64_t leco_type;

int random(int m)
{
    return rand() % m;
}

template <typename T>
static std::vector<T> load_data_binary(const std::string &filename,
                                       bool print = true)
{
    std::vector<T> data;

    std::ifstream in(filename, std::ios::binary);
    if (!in.is_open())
    {
        std::cerr << "unable to open " << filename << std::endl;
        exit(EXIT_FAILURE);
    }
    // Read size.
    uint64_t size;
    in.read(reinterpret_cast<char *>(&size), sizeof(uint64_t));
    data.resize(size);
    // Read values.
    in.read(reinterpret_cast<char *>(data.data()), size * sizeof(T));
    in.close();

    return data;
}

template <typename T>
static std::vector<T> load_data(const std::string &filename)
{
    std::vector<T> data;
    std::ifstream srcFile(filename, std::ios::in);
    if (!srcFile)
    {
        std::cout << "error opening source file." << std::endl;
        return data;
    }

    while (srcFile.good())
    {
        T next;
        srcFile >> next;
        if (!srcFile.good())
        {
            break;
        }
        data.emplace_back(next);
    }
    srcFile.close();

    return data;
}

int main(int argc, const char *argv[])
{
    using namespace Codecset;
    FOR_int<leco_type> codec;
    std::string method = "FOR_my";
    std::string source_file = std::string(argv[1]);
    int blocks = atoi(argv[2]);
    int model_size = atoi(argv[3]);
    bool binary = atoi(argv[4]);
    leco_type filter1 = 0;
    leco_type filter2 = 0;
    leco_type base = 0;
    bool filter_experiment = false;
    bool filter_close_experiment = false;
    if (argc > 5)
    {
        filter1 = atoll(argv[5]);
        filter_experiment = true;
    }
    if (argc > 6)
    {
        filter2 = atoll(argv[6]);
        filter_experiment = false;
        filter_close_experiment = true;
        base = atoll(argv[7]);
    }
    // alternatives : Delta_int, Delta_cost, Delta_cost_merge, FOR_int, Leco_int, Leco_cost, Leco_cost_merge_hc,  Leco_cost_merge, Leco_cost_merge_double

    std::vector<leco_type> data;
    if (binary)
    {
        data = load_data_binary<leco_type>("../data/" + source_file);
    }
    else
    {
        data = load_data<leco_type>("../data/" + source_file);
    }

    int N = data.size();

    int block_size = data.size() / blocks;
    blocks = data.size() / block_size;
    if (blocks * block_size < N)
    {
        blocks++;
    } // handle with the last block, maybe < block_size
    codec.init(blocks, block_size);

    std::vector<uint8_t *> block_start_vec;

    uint64_t totalsize = 0;
    for (int i = 0; i < blocks; i++)
    {
        // std::cout<<"block "<<i<<std::endl;
        int block_length = block_size;
        if (i == blocks - 1)
        {
            block_length = N - (blocks - 1) * block_size;
        }

        uint8_t *descriptor = (uint8_t *)malloc(block_length * sizeof(leco_type) * 4);
        uint8_t *res = descriptor;
        res = codec.encodeArray8_int(data.data() + (i * block_size), block_length, descriptor, i);
        uint32_t segment_size = res - descriptor;
        descriptor = (uint8_t *)realloc(descriptor, segment_size);
        block_start_vec.push_back(descriptor);
        totalsize += segment_size;
    }

    double origin_size = (sizeof(leco_type) * N * 1.0);
    double total_model_size = model_size * blocks;
    double cr_wo_model = (totalsize - total_model_size) * 100.0 / origin_size;
    double cr_model = total_model_size * 100.0 / origin_size;
    double compressrate = (totalsize)*100.0 / origin_size;

    bool flag = true;
    std::vector<leco_type> recover(data.size());
    double totaltime = 0.0;
    // std::cout << "decompress all!" << std::endl;
    int repeat = 10;

    double start = getNow();
    for (int iter = 0; iter < repeat; iter++)
    {
        for (int i = 0; i < blocks; i++)
        {
            int block_length = block_size;
            if (i == blocks - 1)
            {
                block_length = N - (blocks - 1) * block_size;
            }
            codec.decodeArray8(block_start_vec[i], block_length, recover.data() + i * block_size, i);
        }
        for (auto index : codec.mul_add_diff_set)
        {
            recover[index.first] += index.second;
        }
#ifndef NDEBUG
        for (int j = 0; j < N; j++)
        {
            if (data[j] != recover[j])
            {
                std::cout << " num: " << j << " true is: " << data[j] << " predict is: " << recover[j] << std::endl;
                std::cout << "something wrong! decompress all failed" << std::endl;
                flag = false;
                break;
            }
        }
        if (!flag)
        {
            break;
        }
#endif
    }
    double end = getNow();
    totaltime += (end - start);
    double da_ns = totaltime / (N * repeat) * 1000000000;

    // std::cout << "random access decompress!" << std::endl;
    std::vector<uint32_t> ra_pos;
    repeat = 1;
    for (int i = 0; i < N * repeat; i++)
    {
        ra_pos.push_back(random(N));
    }
    flag = true;
    double randomaccesstime = 0.0;
    start = getNow();
    leco_type mark = 0;
    for (auto index : ra_pos)
    {

        leco_type tmpvalue = codec.randomdecodeArray8(block_start_vec[(int)index / block_size], index % block_size, NULL, N);
        mark += tmpvalue;
#ifndef NDEBUG
        if (data[index] != tmpvalue)
        {

            std::cout << "num: " << index << "true is: " << data[index] << " predict is: " << tmpvalue << std::endl;
            flag = false;
            std::cout << "something wrong! random access failed" << std::endl;
        }
        if (!flag)
        {
            break;
        }
#endif
    }
    end = getNow();
    randomaccesstime += (end - start);
    std::ofstream outfile("fix_log", std::ios::app);
    outfile << mark << std::endl;

    double ra_ns = randomaccesstime / (N * repeat) * 1000000000;

    int total_counter = 0;
    if (filter_experiment)
    {
        std::vector<std::pair<leco_type, leco_type>> zonemap;
        for (int i = 0; i < blocks; i++)
        {
            int block_length = block_size;
            if (i == blocks - 1)
            {
                block_length = N - (blocks - 1) * block_size;
            }
            leco_type block_min = UINT32_MAX;
            leco_type block_max = 0;
            for (int j = 0; j < block_length; j++)
            {
                if (data[i * block_size + j] < block_min)
                {
                    block_min = data[i * block_size + j];
                }
                if (data[i * block_size + j] > block_max)
                {
                    block_max = data[i * block_size + j];
                }
            }
            zonemap.emplace_back(std::make_pair(block_min, block_max));
        }
        std::vector<uint32_t> bit_pos(data.size());
        start = getNow();
        for (int iter = 0; iter < repeat; iter++)
        {
            for (int i = 0; i < blocks; i++)
            {
                if(zonemap[i].second<=filter1){
                    continue;
                }
                int block_length = block_size;
                if (i == blocks - 1)
                {
                    block_length = N - (blocks - 1) * block_size;
                }
                total_counter += codec.filter_range(block_start_vec[i], block_length, filter1, bit_pos.data() + total_counter, i);
            }
        }
        end = getNow();
    }

    double filter_totaltime = (end - start);

     
    if (filter_close_experiment)
    {
        total_counter = 0;
        std::vector<std::pair<leco_type, leco_type>> zonemap;
        for (int i = 0; i < blocks; i++)
        {
            int block_length = block_size;
            if (i == blocks - 1)
            {
                block_length = N - (blocks - 1) * block_size;
            }
            leco_type block_min = UINT32_MAX;
            leco_type block_max = 0;
            for (int j = 0; j < block_length; j++)
            {
                if (data[i * block_size + j] < block_min)
                {
                    block_min = data[i * block_size + j];
                }
                if (data[i * block_size + j] > block_max)
                {
                    block_max = data[i * block_size + j];
                }
            }
            zonemap.emplace_back(std::make_pair(block_min, block_max));
        }
        std::vector<uint32_t> bit_pos(data.size());
        start = getNow();
        for (int iter = 0; iter < repeat; iter++)
        {
            for (int i = 0; i < blocks; i++)
            {
                // if(zonemap[i].second<=filter1 || zonemap[i].first>=filter2){
                //     continue;
                // }
                int block_length = block_size;
                if (i == blocks - 1)
                {
                    block_length = N - (blocks - 1) * block_size;
                }
                int tmpcount= codec.filter_range_close(block_start_vec[i], block_length, bit_pos.data() + total_counter, i, filter1, filter2, base);
                total_counter+=tmpcount;

                int truecount = 0;
                for(int j = 0;j<block_length;j++){
                    if(data[i*block_size + j]%base>filter1 &&data[i*block_size + j]%base<filter2){
                        truecount++;
                    }
                }
                if(truecount!=tmpcount){
                    std::cout<<"wrong"<<i<<" "<<truecount<<" "<<tmpcount<<std::endl;
                }
            }
        }
        end = getNow();
    }
    double filter_close_totaltime = (end - start);


    std::cout << method << " " << source_file << " " << blocks << " " << compressrate << " " << cr_model << " " << cr_wo_model << " " << da_ns << " " << ra_ns << " " << 0 << " " << total_counter << " " << filter_totaltime<<" "<<filter_close_totaltime << std::endl;

    for (int i = 0; i < (int)block_start_vec.size(); i++)
    {
        free(block_start_vec[i]);
    }
}