#include <iostream>
#include <vector>

#include <TString.h>
#include <TH1D.h>
#include <TH1I.h>
#include <TH2D.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TFile.h>
#include <TStopwatch.h>

#include <qaParticle.h>
#include <qaParticleLight.h>
#include <qaEvent.h>
#include <qaReader_manager.h>
#include <qaReader_smash_root.h>
#include <qaReader_mcpico.h>
#include <Utility.h>
#include <qaWriter_manager.h>
#include <qaWriter_analysistree2.h>

#ifdef _MCINI_
#include <qaReader_mcini.h>
#endif
#ifdef _PHQMD_
#include <qaReader_phqmd.h>
#endif
#ifdef _HSD_ROOT_
#include <qaReader_hsd_root.h>
#endif

// AnalysisTree headers
#include <AnalysisTree/Configuration.hpp>
#include <AnalysisTree/DataHeader.hpp>
#include <AnalysisTree/EventHeader.hpp>
#include <AnalysisTree/Detector.hpp>
#include <AnalysisTree/Matching.hpp>

int main(int argc, char **argv)
{
  TString iFileName, oFileName;
  TString outputFormat = "analysistree_v2";

  if (argc < 7)
  {
    std::cerr << "./matc -i input.list -o output.root -input_format [INPUT_FORMAT] [OPTIONAL: -output_format [OUTPUT_FORMAT]]" << std::endl;
    std::cerr << "Available input formats:" << std::endl;
    std::cerr << "\tmcpico - simple custom ROOT format to store model data" << std::endl;
    std::cerr << "\tparticle - ROOT format that is used by the SMASH model" << std::endl;
#ifdef _MCINI_
    std::cerr << "\tmcini - custom ROOT format to store both initial state and final state (UniGen data format) model data" << std::endl;
#endif
#ifdef _PHQMD_
    std::cerr << "\tphqmd - custom ROOT format to store PHQMD (with MST) model data" << std::endl;
#endif
#ifdef _HSD_ROOT_
    std::cerr << "\thsd - custom ROOT format to store HSD model data" << std::endl;
#endif
    std::cerr << "Available output formats:" << std::endl;
    std::cerr << "\tanalysistree_v2 - AnalysisTree ver. 2 data format" << std::endl;
    return 1;
  }
  for (int i = 1; i < argc; i++)
  {
    if (std::string(argv[i]) != "-i" &&
        std::string(argv[i]) != "-o" &&
        std::string(argv[i]) != "-input_format" &&
        std::string(argv[i]) != "-output_format")
    {
      std::cerr << "\n[ERROR]: Unknown parameter " << i << ": " << argv[i] << std::endl;
      return 2;
    }
    else
    {
      if (std::string(argv[i]) == "-i" && i != argc - 1)
      {
        iFileName = argv[++i];
        continue;
      }
      if (std::string(argv[i]) == "-i" && i == argc - 1)
      {
        std::cerr << "\n[ERROR]: Input file name was not specified " << std::endl;
        return 1;
      }
      if (std::string(argv[i]) == "-o" && i != argc - 1)
      {
        oFileName = argv[++i];
        continue;
      }
      if (std::string(argv[i]) == "-o" && i == argc - 1)
      {
        std::cerr << "\n[ERROR]: Output file name was not specified " << std::endl;
        return 1;
      }
      if (std::string(argv[i]) == "-input_format" && i != argc - 1)
      {
        qaUtility::GetInstance()->format = argv[++i];
        continue;
      }
      if (std::string(argv[i]) == "-input_format" && i == argc - 1)
      {
        std::cerr << "\n[ERROR]: Input file format was not specified " << std::endl;
        return 1;
      }
      if (std::string(argv[i]) == "-output_format" && i != argc - 1)
      {
        outputFormat = argv[++i];
        continue;
      }
      if (std::string(argv[i]) == "-output_format" && i == argc - 1)
      {
        std::cerr << "\n[ERROR]: Output file format was not specified " << std::endl;
        return 1;
      }
    }
  }

  TStopwatch timer;
  timer.Start();

  qaReader_manager *readerManager;
  if (qaUtility::GetInstance()->format == "mcpico")
  {
    readerManager = new qaReader_mcpico();
  }
#ifdef _MCINI_
  if (qaUtility::GetInstance()->format == "mcini")
  {
    readerManager = new qaReader_mcini();
  }
#endif
#ifdef _PHQMD_
  if (qaUtility::GetInstance()->format == "phqmd")
  {
    readerManager = new qaReader_phqmd();
  }
#endif
#ifdef _HSD_ROOT_
  if (qaUtility::GetInstance()->format == "hsd")
  {
    readerManager = new qaReader_hsd_root();
  }
#endif
  if (qaUtility::GetInstance()->format == "particles")
  {
    readerManager = new qaReader_smash_root();
  }

  if (!readerManager)
  {
    std::cerr << "This input format is not found!" << std::endl;
    return 20;
  }

  qaWriter_manager *writerManager;
  if (outputFormat == "analysistree_v2")
  {
    writerManager = new qaWriter_analysistree2();
  }

  if (!writerManager)
  {
    std::cerr << "This output format is not found!" << std::endl;
    return 30;
  }

  readerManager->SetChain(iFileName.Data());
  Long64_t Nentries = readerManager->GetEntries();
  Int_t Nparticles;

  qaEvent *event = nullptr;
  qaParticle *particle = nullptr;

  writerManager->Init(oFileName.Data(), "modelTree");

  for (Long64_t iev = 0; iev < Nentries; iev++)
  {
    if (iev % 1000 == 0)
      std::cout << "Event [" << Minbias_counter << "/" << Nentries << "]" << std::endl;

    event = (qaEvent *)readerManager->ReadEvent(iev);
    if (!event)
      continue;

    writerManager->WriteEvent(event);

    Nparticles = event->GetNparticles();

    for (int iparticle = 0; iparticle < Nparticles; iparticle++)
    {
      particle = readerManager->ReadParticle(iparticle);

      if (!particle)
        continue;

      writerManager->WriteParticle(particle);
      
      delete particle;
    }

    writerManager->FillTree();

    delete event;
  }

  writerManager->WriteTree();

  timer.Stop();
  timer.Print();

  return 0;
}