#include <fmt/core.h>
#include <string>
#include <fuc2/testing.hpp>
#include "stats.hpp"
#include "data.hpp"

using namespace fuc2;

namespace perf_tests {
  struct Samples {
    std::vector<Stats> soa_big_for;
    std::vector<Stats> soa_many_for;
    std::vector<Stats> ls_big_for;
    std::vector<Stats> ls_many_for;
    std::vector<Stats> maos_big_for;
    std::vector<Stats> maos_many_for;
    std::vector<Stats> maos_many_for_2;
  };

  void test_large_soa(Samples& samples, int sample_size, int sample_number) {
    auto soa = new LargeSOA();

    for(int i = 0; i < sample_size; i++) {
      soa->push_back(i);
    }

    Stats big_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = big_for.time_start();
      for(int i = 0; i < sample_size; i++) {
        soa->a01[i] += soa->a02[i] + sample;
        soa->a02[i] += soa->a03[i] + sample;
        soa->a03[i] += soa->a04[i] + sample;
        soa->a04[i] += soa->a05[i] + sample;
        soa->a05[i] += soa->a06[i] + sample;
        soa->a06[i] += soa->a07[i] + sample;
        soa->a07[i] += soa->a08[i] + sample;
        soa->a08[i] += soa->a09[i] + sample;
        soa->a09[i] += soa->a10[i] + sample;
        soa->a10[i] += soa->a11[i] + sample;
        soa->a11[i] += soa->a12[i] + sample;
        soa->a12[i] += soa->a13[i] + sample;
        soa->a13[i] += soa->a14[i] + sample;
        soa->a14[i] += soa->a15[i] + sample;
        soa->a15[i] += soa->a16[i] + sample;
        soa->a16[i] += soa->a17[i] + sample;
        soa->a17[i] += soa->a18[i] + sample;
        soa->a18[i] += soa->a19[i] + sample;
        soa->a19[i] += soa->a20[i] + sample;
      }
      big_for.sample_time(start);
    }
    samples.soa_big_for.push_back(big_for);

    Stats many_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = many_for.time_start();
      for(int i = 0; i < sample_size; i++) soa->a01[i] += soa->a02[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a02[i] += soa->a03[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a03[i] += soa->a04[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a04[i] += soa->a05[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a05[i] += soa->a06[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a06[i] += soa->a07[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a07[i] += soa->a08[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a08[i] += soa->a09[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a09[i] += soa->a10[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a10[i] += soa->a11[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a11[i] += soa->a12[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a12[i] += soa->a13[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a13[i] += soa->a14[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a14[i] += soa->a15[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a15[i] += soa->a16[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a16[i] += soa->a17[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a17[i] += soa->a18[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a18[i] += soa->a19[i] + sample;
      for(int i = 0; i < sample_size; i++) soa->a19[i] += soa->a20[i] + sample;
      many_for.sample_time(start);
    }
    samples.soa_many_for.push_back(many_for);
  }

  void test_large_struct(Samples& samples, int sample_size, int sample_number) {
    auto ls = new std::vector<LargeStruct>;

    for(int i = 0; i < sample_size; i++) {
      ls->push_back({i});
    }

    Stats ls_big_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = ls_big_for.time_start();
      for(auto& el : *ls) {
        el.a01 += el.a02 + sample;
        el.a02 += el.a03 + sample;
        el.a03 += el.a04 + sample;
        el.a04 += el.a05 + sample;
        el.a05 += el.a06 + sample;
        el.a06 += el.a07 + sample;
        el.a07 += el.a08 + sample;
        el.a08 += el.a09 + sample;
        el.a09 += el.a10 + sample;
        el.a10 += el.a11 + sample;
        el.a11 += el.a12 + sample;
        el.a12 += el.a13 + sample;
        el.a13 += el.a14 + sample;
        el.a14 += el.a15 + sample;
        el.a15 += el.a16 + sample;
        el.a16 += el.a17 + sample;
        el.a17 += el.a18 + sample;
        el.a18 += el.a19 + sample;
        el.a19 += el.a20 + sample;
      }
      ls_big_for.sample_time(start);
    }

    samples.ls_big_for.push_back(ls_big_for);

    Stats ls_many_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = ls_many_for.time_start();
      for(auto& el : *ls) el.a01 += el.a02 + sample;
      for(auto& el : *ls) el.a02 += el.a03 + sample;
      for(auto& el : *ls) el.a03 += el.a04 + sample;
      for(auto& el : *ls) el.a04 += el.a05 + sample;
      for(auto& el : *ls) el.a05 += el.a06 + sample;
      for(auto& el : *ls) el.a06 += el.a07 + sample;
      for(auto& el : *ls) el.a07 += el.a08 + sample;
      for(auto& el : *ls) el.a08 += el.a09 + sample;
      for(auto& el : *ls) el.a09 += el.a10 + sample;
      for(auto& el : *ls) el.a10 += el.a11 + sample;
      for(auto& el : *ls) el.a11 += el.a12 + sample;
      for(auto& el : *ls) el.a12 += el.a13 + sample;
      for(auto& el : *ls) el.a13 += el.a14 + sample;
      for(auto& el : *ls) el.a14 += el.a15 + sample;
      for(auto& el : *ls) el.a15 += el.a16 + sample;
      for(auto& el : *ls) el.a16 += el.a17 + sample;
      for(auto& el : *ls) el.a17 += el.a18 + sample;
      for(auto& el : *ls) el.a18 += el.a19 + sample;
      for(auto& el : *ls) el.a19 += el.a20 + sample;
      ls_many_for.sample_time(start);
    }
    samples.ls_many_for.push_back(ls_many_for);
  }


  void test_maos_style(Samples& samples, int sample_size, int sample_number) {
    auto maos = new MultipleAOS();

    for(int i = 0; i < sample_size; i++) {
      maos->push_back(i);
    }

    Stats maos_big_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = maos_big_for.time_start();
      for(int i = 0; i < sample_size; i++) {
        maos->s1[i].a01 += maos->s1[i].a02 + sample;
        maos->s1[i].a02 += maos->s1[i].a03 + sample;
        maos->s1[i].a03 += maos->s1[i].a04 + sample;
        maos->s1[i].a04 += maos->s2[i].a05 + sample;

        maos->s2[i].a05 += maos->s2[i].a06 + sample;
        maos->s2[i].a06 += maos->s2[i].a07 + sample;
        maos->s2[i].a07 += maos->s2[i].a08 + sample;
        maos->s2[i].a08 += maos->s3[i].a09 + sample;

        maos->s3[i].a09 += maos->s3[i].a10 + sample;
        maos->s3[i].a10 += maos->s3[i].a11 + sample;
        maos->s3[i].a11 += maos->s3[i].a12 + sample;
        maos->s3[i].a12 += maos->s4[i].a13 + sample;

        maos->s4[i].a13 += maos->s4[i].a14 + sample;
        maos->s4[i].a14 += maos->s4[i].a15 + sample;
        maos->s4[i].a15 += maos->s4[i].a16 + sample;
        maos->s4[i].a16 += maos->s5[i].a17 + sample;

        maos->s5[i].a17 += maos->s5[i].a18 + sample;
        maos->s5[i].a18 += maos->s5[i].a19 + sample;
        maos->s5[i].a19 += maos->s5[i].a20 + sample;
      }
      maos_big_for.sample_time(start);
    }
    samples.maos_big_for.push_back(maos_big_for);

    Stats maos_many_for;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = maos_many_for.time_start();

      for(auto& el : maos->s1) {
        el.a01 += el.a02 + sample;
        el.a02 += el.a03 + sample;
        el.a03 += el.a04 + sample;
      }

      for(int i = 0; i < sample_size; i++) {
        maos->s1[i].a04 += maos->s2[i].a05 + sample;
      }

      for(auto& el : maos->s2) {
        el.a05 += el.a06 + sample;
        el.a06 += el.a07 + sample;
        el.a07 += el.a08 + sample;
      }

      for(int i = 0; i < sample_size; i++) {
        maos->s2[i].a08 += maos->s3[i].a09 + i;
      }

      for(auto& el : maos->s3) {
        el.a09 += el.a10 + sample;
        el.a10 += el.a11 + sample;
        el.a11 += el.a12 + sample;
      }

      for(int i = 0; i < sample_size; i++) {
        maos->s3[i].a12 += maos->s4[i].a13 + sample;
      }

      for(auto& el : maos->s4) {
        el.a13 += el.a14 + sample;
        el.a14 += el.a15 + sample;
        el.a15 += el.a16 + sample;
      }

      for(int i = 0; i < sample_size; i++) {
        maos->s4[i].a16 += maos->s5[i].a17 + sample;
      }

      for(auto& el : maos->s5) {
        el.a17 += el.a18 + sample;
        el.a18 += el.a19 + sample;
        el.a19 += el.a20 + sample;
      }

      maos_many_for.sample_time(start);
    }

    samples.maos_many_for.push_back(maos_many_for);

    Stats maos_many_for_2;
    for(int sample = 0; sample < sample_number; sample++) {
      auto start = maos_many_for.time_start();

      for(auto& el : maos->s1) {
        el.a01 += el.a02 + sample;
        el.a02 += el.a03 + sample;
        el.a03 += el.a04 + sample;
      }

      for(auto& el : maos->s2) {
        el.a05 += el.a06 + sample;
        el.a06 += el.a07 + sample;
        el.a07 += el.a08 + sample;
      }

      for(auto& el : maos->s3) {
        el.a09 += el.a10 + sample;
        el.a10 += el.a11 + sample;
        el.a11 += el.a12 + sample;
      }

      for(auto& el : maos->s4) {
        el.a13 += el.a14 + sample;
        el.a14 += el.a15 + sample;
        el.a15 += el.a16 + sample;
      }

      for(auto& el : maos->s5) {
        el.a17 += el.a18 + sample;
        el.a18 += el.a19 + sample;
        el.a19 += el.a20 + sample;
      }

      maos_many_for.sample_time(start);
    }

    samples.maos_many_for_2.push_back(maos_many_for);
  }

  void test_all_performance() {
    Samples samples;

    int sample_size = 100;
    int sample_count = 100;

    for(int i = 0; i < 10; i++) {
      // quick and dirty "randomness"
      if(i % 2 == 0) {
        test_large_soa(samples, sample_size, sample_count);
        test_large_struct(samples, sample_size, sample_count);
        test_maos_style(samples, sample_count, sample_count);
      } else if(i % 3 == 0) {
        test_large_struct(samples, sample_size, sample_count);
        test_large_soa(samples, sample_size, sample_count);
        test_maos_style(samples, sample_count, sample_count);
      } else {
        test_maos_style(samples, sample_count, sample_count);
        test_large_struct(samples, sample_size, sample_count);
        test_large_soa(samples, sample_size, sample_count);
      }
    }

    fmt::println("soa_bf soa_mf ls_bf ls_mf maos_bf maos_mf maos_mf2");

    Samples mofm;

    Stats soa_big_for;
    Stats soa_many_for;
    Stats ls_big_for;
    Stats ls_many_for;
    Stats maos_big_for;
    Stats maos_many_for;
    Stats maos_many_for_2;

    for(size_t i = 0; i < samples.soa_big_for.size(); i++) {
      fmt::println("{:.2f} {:.2f} {:.2f} {:.2f} {:.2f} {:.2f} {:.2f}",
          samples.soa_big_for[i].mean(),
          samples.soa_many_for[i].mean(),
          samples.ls_big_for[i].mean(),
          samples.ls_many_for[i].mean(),
          samples.maos_big_for[i].mean(),
          samples.maos_many_for[i].mean(),
          samples.maos_many_for_2[i].mean());

      // this is potentially total bullshit
      soa_big_for.sample(samples.soa_big_for[i].mean());
      soa_many_for.sample(samples.soa_many_for[i].mean());
      ls_big_for.sample(samples.ls_big_for[i].mean());
      ls_many_for.sample(samples.ls_many_for[i].mean());
      maos_big_for.sample(samples.maos_big_for[i].mean());
      maos_many_for.sample(samples.maos_many_for[i].mean());
      maos_many_for_2.sample(samples.maos_many_for_2[i].mean());
    }

    soa_big_for.dump("soa_big_for");
    soa_many_for.dump("soa_many_for");
    ls_big_for.dump("ls_big_for");
    ls_many_for.dump("ls_many_for");
    maos_big_for.dump("maos_big_for");
    maos_many_for.dump("maos_many_for");
    maos_many_for_2.dump("maos_many_for_2");

    // totally not sure if this t-test is legit, must study more
    soa_big_for.t_test(ls_big_for);
    soa_big_for.t_test(maos_big_for);

    // soa_many_for.t_test(ls_many_for);
    // soa_many_for.t_test(maos_many_for);

    // ls_many_for.t_test(maos_many_for);
    // ls_big_for.t_test(maos_big_for);

    // // the most pathologic case
    // fmt::println("---- {}", soa_big_for.mean() - ls_many_for.mean());
    // soa_big_for.t_test(ls_many_for);

    // fmt::println("--- MAOS many_for vs many_for_2");
    maos_many_for.t_test(maos_many_for_2);

    // fmt::println("--- MAOS big_for vs many_for_2");
    maos_big_for.t_test(maos_many_for_2);
  }

  fuc2::Set TESTS{
    .name="perf",
      .tests={
        TEST(test_all_performance),
      }
  };
}