#include <fstream>
#include <iostream>
#include <TString.h>
#include <TH1D.h>
#include <TH1F.h>
#include <TF1.h>
#include <TFile.h>
#include <TLatex.h>
#include <TStyle.h>
#include <TProfile.h>

#include <fstream>
#include <stdio.h>
#include <cmath>
#include <map>

using namespace std;

TString fres_name = "../Output/masses.dat";


void FillProb(){
  
  TString work_dir = "../Output/";
  
    
  const int n_histo_prob = 6;
  
  const int n_histo_nhits = 2;
    
  string title[n_histo_prob]= {"Prob_reco.dat","Prob_ghost.dat","Prob_reco_3.dat","Prob_ghost_3.dat", "Prob_reco_4.dat",
                                          "Prob_ghost_4.dat"};
					  

  TH1F* hProb[n_histo_prob];
  float xmax=1.01, xmin=0.0;
  int Nbins=250;
    
  TString mProbHistoName[n_histo_prob] = {"Prob_reco","Prob_ghost","Prob_reco_3","Prob_ghost_3", "Prob_reco_4",
                                           "Prob_ghost_4"};
					   
   TH1F* hHits[n_histo_nhits];
  float xmax_h=10.1, xmin_h=-0.1;
  int Nbins_h=51;
    
  TString mNHitsHistoName[n_histo_nhits] = {"NHits_reco","NHits_ghost"};
					   
					 
  for(int i=0; i<n_histo_prob; i++)
  { 
    fstream fin;
    char fname[100];
    strcpy(fname, work_dir.Data());
    strcat(fname, title[i].data());
    fin.open(fname,fstream::in);
      
    std::cout << fname << std::endl;
    hProb[i] = new TH1F(mProbHistoName[i].Data(),mProbHistoName[i].Data(),Nbins,xmin,xmax);
   
    if ( !(fin.is_open()) )
    {
      cout << "File " << fname << " does not exist!!!!!!!!! " << endl;
      continue;
    }
    
    if (i<2) hHits[i] = new TH1F(mNHitsHistoName[i].Data(),mNHitsHistoName[i].Data(),Nbins_h,xmin_h,xmax_h);
    
    float prob;
    string line;
    int nhits;
    while(std::getline(fin, line, '\n'))
    {
      
      if (i<2) {
	sscanf(line.data(),"%f %i",&prob, &nhits);
	hHits[i]->Fill(nhits);
      }
      else
	sscanf(line.data(),"%f",&prob);
      hProb[i]->Fill(prob);
    }
    fin.close(); 
  }
  TFile fRoot("Prob.root","RECREATE");

  for(int i=0; i<n_histo_prob; i++)
  { 
   if (i<2) {
	hHits[i]->Write();
	hProb[i]->Write();
      }
      else
	hProb[i]->Write(); 
  }
}
void FillMassHistoFromAscii(){

  fstream fpres; 
  fpres.open (fres_name.Data(), fstream::in);

  TH1F* hMass[30];
  float xmax[30], xmin[30];

  map<int,int> pdgToHisto;

  const int nParticles = 11;
  int mPartPDG[nParticles] = {310,3122,3312,-3312,3334,
                              3224,3114,
                              3000,
                              421,411,-411};
  TString mPartName[nParticles] = {"ks","lambda","ksi-","ksi+","omega-",
                                   "sigma*+","sigma*-",
                                   "Hdb",
                                   "D0","D+","D-"};

  for(int i=0; i<nParticles; i++)
    pdgToHisto[mPartPDG[i]] = i;

  for(int i=0; i<30; i++)
  {
    xmin[i] = 1.;
    xmax[i] = 3.;
  }
  xmin[0] = 0.3;
  xmax[0] = 1.3;

  xmin[1] = 1.;
  xmax[1] = 2.;

  for(int i=0; i<nParticles; i++)
  {
    hMass[i] = new TH1F(mPartName[i].Data(),mPartName[i].Data(),1000,xmin[i],xmax[i]);
  }

  for (int i = 0; !(fpres.eof()); i++){

    int num, pdg;
    float mass;

    fpres >> num >> mass >> pdg;
    hMass[pdgToHisto[pdg]]->Fill(mass);

    if (fpres.eof()) cout << " Data from file " << fres_name << " read succesfully " << endl; 
  } // read results

  fpres.close();

  TFile fRoot("Masses.root","RECREATE");

  for(int i=0; i<nParticles; i++)
    hMass[i]->Write();
}

void FillHistoFromAscii(){
  FillMassHistoFromAscii();
  FillProb();
}