#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 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();
}