// macro for analys bkg particles PDG id and drawing results
// in cool histogram
// author: a.karavdina
// date: 07 January 2014

#include "TString.h"
#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TChain.h"
#include "TClonesArray.h"
#include "/panda/pandaroot/lmd/LmdQA/PndLmdTrackQ.h"
#include "TPie.h"
#include "TLegend.h"
#include <iostream>
using namespace std;
void PDGcodes(TString path="/panda/myResults/DPMoutputDec2013/FTF/NoDipole/mom_15/"){
 // //  // color table: barp = 2, pi- = 8, e- = 4, e+ = 43, pi+ = 7, K- = 11, mu- = 13, p = 6, mu+ = 38, K+ = 9
// // //   ///list all id's and count each contribution ------------------------------
//   TString filename1 = path+"/TOT_Lumi_BkgSig_el_inel.root";
//   TFile *fin1 = new TFile(filename1,"READ");
//   TTree *tbkg1 = (TTree *)fin1->Get("tBkg"); //Bkg data
//   //  Int_t nevent = tbkg1->GetEntries();
//   Int_t nevent = tbkg1->GetEntries();
//   int f_id;
//   int f_issec;
//   TBranch *bid = tbkg1->GetBranch("id");
//   bid->SetAddress(&f_id);
//   //  TBranch *bissec = tbkg1->GetBranch("issecond");
//   //  bissec->SetAddress(&f_issec);
//   vector<pair<int,int> > v_id; //vector of particles: first=id,second=counter
//   for (Int_t i=0;i<nevent;i++){// loop over all trks
//     //    double Nnorm = tbkg1->GetEntries();
//     tbkg1->GetEvent(i);
//     //  if(f_issec<0) continue;//secondaries only
//     //   if(f_issec>=0) continue;//promaries only
//     bool addit=true;
//     for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
// 	if((*it).first==f_id){
// 	  addit=false;// add only ids not listed before;
// 	  (*it).second++;
// 	}
//     }
//     if(addit) v_id.push_back(std::make_pair(f_id,1));
//   }// end  loop over all trks

//   std::cout<<"List of particles: "<<endl;
//   double contrSum = 0;
//   //  double ids[10];
//   double contr[10];
//   int ncontr = 0;
//   for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
//     //ids[ncontr]=(*it).first;
//     if((*it).first>1e4) continue;
//     contr[ncontr]=(*it).second;
//     ncontr++;
//     contrSum +=100*((*it).second/double(nevent));
//     std::cout << ' ' << (*it).first <<" with contribution "<<(*it).second<<" "<< 100*((*it).second/double(nevent))<<" %  from tot.bkg";
//     std::cout << '\n';
//   }
//   cout<<"TOTAL: "<<contrSum<<endl;
//   //  Int_t colors[] = {2,8,4,43,31,11,13,6,38,9}; //15 GeV
//   // Int_t colors[] = {2,4,43,8,3,7,9,6,11,38}; //15 GeV
//   //  double rangcont[] = {38309, 7278, 6519, 34492, 1943, 18413, 1040, 6388, 1546, 934}; //15 GeV

//   // Int_t colors[] = {2,8,7,4,43,6,13,11,9,38}; //15 GeV no Cuts
//   //  double rangcont[] = {38309, 34492, 18413, 7278, 6519, 6388, 1943, 1546, 1040, 934}; //15 GeV no Cuts
//   Int_t colors[] = {2,8,4,7,43,6,11,13,38}; //15 GeV no Cuts
//   double rangcont[] = {2949,234,22,21,16,6,4,3,1}; //15 GeV X&Y Cuts
//   //barp, pi-, e-, pi+, e+, p, K-, mu-,mu+
//  // color table: barp = 2, pi- = 8, e- = 4, e+ = 43, pi+ = 7, K- = 11, mu- = 13, p = 6, mu+ = 38, K+ = 9

//   // Int_t colors[] = {8,2,13,31,6,4,43,38}; //1.5 GeV
//   // double rangcont[] = {1206,88,55,32,19,16,13,7}; //1.5 GeV
//   TPie *pie1 = new TPie("pie1",
// 			"Particles contributions",ncontr,rangcont,colors);
//  pie1->SetEntryLabel(0,"#bar{p}");
//  pie1->SetEntryLabel(1,"#pi^{-}");
//  pie1->SetEntryLabel(2,"#pi^{+}");
//  pie1->SetEntryLabel(3,"e^{-}");
//  pie1->SetEntryLabel(4,"e^{+}");
//  pie1->SetEntryLabel(5,"p");
//  pie1->SetEntryLabel(6,"#mu^{-}");
// pie1->SetEntryLabel(8,"K^{+}");
// pie1->SetEntryLabel(7,"K^{-}");
// // pie1->SetEntryLabel(9,"#mu^{+}");


//  //pie1->SetLabelFormat("%txt #scale[0.6]{(%perc)}");
//  pie1->SetAngularOffset(160.);
//  pie1->SetLabelsOffset(1);

 

//  pie1->SetEntryRadiusOffset(0, 0.04);
//  //  pie1->SetEntryRadiusOffset(8, 0.1);
//  // pie1->SetEntryRadiusOffset(7, 0.16);
//  // pie1->SetEntryRadiusOffset(6, 0.14);
//  // pie1->SetEntryRadiusOffset(5, 0.12);
//  // pie1->SetEntryRadiusOffset(4, 0.09);
//  // pie1->SetEntryRadiusOffset(3, 0.07);
//  // pie1->SetEntryRadiusOffset(2, 0.04);
//  // pie1->SetEntryRadiusOffset(1, 0.02);

//  pie1->SetRadius(.35);
//  //  pie1->Draw("r");
//  pie1->Draw("3d");
// TLegend *leg = new TLegend(0.1,0.7,0.48,0.9);
//  //   leg->SetHeader("The Legend Title");
//  for(int i=0;i<9;i++){
//    TString entry = pie1->GetEntryLabel(i);
//    entry +="   ";
//    char buf[256]  = {0};
//    snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf) - 1, "%c%0.02f",1,float(100.*(rangcont[i])/nevent));
//    entry += buf;
//    entry += " %";
//    leg->AddEntry(entry,entry,"");
//  }
//    // leg->AddEntry("f1","Function abs(#frac{sin(x)}{x})","l");
//    // leg->AddEntry("gr","Graph with error bars","lep");
//  leg->Draw();
// //   ///[END] list all id's and count each contribution ------------------------------

 ///primaries ------------------------------
  TString filename1 = path+"/TOT_Lumi_BkgSig_el_inel.root";
  TFile *fin1 = new TFile(filename1,"READ");
  TTree *tbkg1 = (TTree *)fin1->Get("tBkg"); //Bkg data
  //  Int_t nevent = tbkg1->GetEntries();
  Int_t neventGL = tbkg1->GetEntries("issecond<0");
  int f_id;
  int f_issec;
  TBranch *bid = tbkg1->GetBranch("id");
  bid->SetAddress(&f_id);
  TBranch *bissec = tbkg1->GetBranch("issecond");
  bissec->SetAddress(&f_issec);
  double f_prec;
  TBranch *bismom = tbkg1->GetBranch("prec");
  bismom->SetAddress(&f_prec);

  vector<pair<int,int> > v_id; //vector of particles: first=id,second=counter
  int nevent=0;
  for (Int_t i=0;i<neventGL;i++){// loop over all trks
    //    double Nnorm = tbkg1->GetEntries();
    tbkg1->GetEvent(i);
    if(f_issec>=0) continue;//primaries only
    //    if(fabs(f_prec-15.0)>5e-4) continue; // mom cut
    nevent++;
    bool addit=true;
    for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
 	if((*it).first==f_id){
 	  addit=false;// add only ids not listed before;
 	  (*it).second++;
 	}
    }
    if(addit) v_id.push_back(std::make_pair(f_id,1));
  }// end  loop over all trks

  std::cout<<"List of particles: "<<endl;
  double contrSum = 0;
  //  double ids[10];
  double contr[10];
  int ncontr = 0;
  for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
    //ids[ncontr]=(*it).first;
    contr[ncontr]=(*it).second;
    ncontr++;
    contrSum +=100*((*it).second/double(nevent));
    std::cout << ' ' << (*it).first <<" with contribution "<<(*it).second<<" "<< 100*((*it).second/double(nevent))<<" %  from tot.bkg";
    std::cout << '\n';
  }
  cout<<"TOTAL: "<<contrSum<<" = "<<nevent<<" trks"<<endl;
  //  Int_t colors[] = {2,8,4,43,31,11,3,6,38,9};
//  double rangcont[] = {22687, 4365, 1182, 720, 619, 133, 115, 92, 30, 27};

  // Int_t colors[] = {2,8,7,11,6,9};  
  // double rangcont[] = {13744,5157,652,255,44,23};
  // //barp,pi-,pi+,K-,p,K+

  // Int_t colors[] = {2,8,11};  
  // double rangcont[] = {2768,156,3};
 //barp,pi-,K-
  Int_t colors[] = {8,2,7,6,11,9};  
  double rangcont[] = {54182,49681,41967,2536,2083,933};

  //color table: barp = 2, pi- = 8, e- = 4, e+ = 43, pi+ = 7, K- = 11, mu- = 13, p = 6, mu+ = 38, K+ = 9
  TPie *pie1 = new TPie("pie1",
 			"Particles contributions",ncontr,rangcont,colors);
 pie1->SetEntryLabel(0,"#pi^{-}");
 pie1->SetEntryLabel(1,"#bar{p}");
 pie1->SetEntryLabel(2,"#pi^{+}");
 pie1->SetEntryLabel(3,"p");
 pie1->SetEntryLabel(4,"K^{-}");
 pie1->SetEntryLabel(5,"K^{+}");

 //pie1->SetLabelFormat("%txt #scale[0.6]{(%perc)}");
 pie1->SetAngularOffset(160.);
 pie1->SetLabelsOffset(1);
 pie1->SetEntryRadiusOffset(0, 0.04);
 //  pie1->SetEntryRadiusOffset(8, 0.1);
 // pie1->SetEntryRadiusOffset(7, 0.16);
 // pie1->SetEntryRadiusOffset(6, 0.14);
 // pie1->SetEntryRadiusOffset(5, 0.12);
 // pie1->SetEntryRadiusOffset(4, 0.05);
 // pie1->SetEntryRadiusOffset(3, 0.03);
 // pie1->SetEntryRadiusOffset(2, 0.01);
 // pie1->SetEntryRadiusOffset(2, 0.04);
 // pie1->SetEntryRadiusOffset(3, 0.03);
 // pie1->SetEntryRadiusOffset(4, 0.02);

 pie1->SetRadius(.35);
 //  pie1->Draw("r");
 pie1->Draw("3d");

  ///[END] primaries ------------------------------

 // // // ///secondaries ------------------------------
 //  TString filename1 = path+"/TOT_Lumi_BkgSig_el_inel.root";
 //  TFile *fin1 = new TFile(filename1,"READ");
 //  TTree *tbkg1 = (TTree *)fin1->Get("tBkg"); //Bkg data
 //  //  Int_t nevent = tbkg1->GetEntries();
 //  Int_t neventGL = tbkg1->GetEntries("issecond<0");
 //  int f_id;
 //  int f_issec;
 //  TBranch *bid = tbkg1->GetBranch("id");
 //  bid->SetAddress(&f_id);
 //  TBranch *bissec = tbkg1->GetBranch("issecond");
 //  bissec->SetAddress(&f_issec);
 //  vector<pair<int,int> > v_id; //vector of particles: first=id,second=counter
 //  int nevent=0;
 //  for (Int_t i=0;i<neventGL;i++){// loop over all trks
 //    //    double Nnorm = tbkg1->GetEntries();
 //    tbkg1->GetEvent(i);
 //    if(f_issec<0) continue;//secondaries only
 //    nevent++;
 //    bool addit=true;
 //    for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
 // 	if((*it).first==f_id){
 // 	  addit=false;// add only ids not listed before;
 // 	  (*it).second++;
 // 	}
 //    }
 //    if(addit) v_id.push_back(std::make_pair(f_id,1));
 //  }// end  loop over all trks

 //  std::cout<<"List of particles: "<<endl;
 //  double contrSum = 0;
 //  //  double ids[10];
 //  double contr[10];
 //  int ncontr = 0;
 //  for (std::vector<pair<int,int> >::iterator it = v_id.begin(); it != v_id.end(); ++it){
 //    //ids[ncontr]=(*it).first;
 //    contr[ncontr]=(*it).second;
 //    ncontr++;
 //    contrSum +=100*((*it).second/double(nevent));
 //    std::cout << ' ' << (*it).first <<" with contribution "<<(*it).second<<" "<< 100*((*it).second/double(nevent))<<" %  from tot.bkg";
 //    std::cout << '\n';
 //  }
 //  cout<<"TOTAL: "<<contrSum<<" = "<<nevent<<" trks"<<endl;
 //  // //  Int_t colors[] = {8,7,4,43,6,2,13,11,9,38};
 //  // //  double rangcont[] = {9182,6698,3061,2722,2513,2068,799,417,396,374};
 //  //  //pi-, pi+, e-,e+, p, barp, mu-, K-, K+, mu+
 //  //color table: barp = 2, pi- = 8, e- = 4, e+ = 43, pi+ = 7, K- = 11, mu- = 13, p = 6, mu+ = 38, K+ = 9

 //  Int_t colors[] = {8,2,4,7,43,6,13,38,11};
 //  double rangcont[] = {63,30,22,19,15,5,3,1,1};
 //  //pi-, barp, e-, pi+, e+, p, mu-, mu+,K-

 //  Int_t nvals = sizeof(rangcont)/sizeof(rangcont[0]);
 //  TPie *pie1 = new TPie("pie1",
 // 			"Particles contributions",nvals,rangcont,colors);
 //  // pie1->SetEntryLabel(0,"#pi^{-}");
 // //  pie1->SetEntryLabel(1,"e^{-}"); 
 // //  pie1->SetEntryLabel(2,"e^{+}");
 // //  pie1->SetEntryLabel(3,"#pi^{+}");
 // //  pie1->SetEntryLabel(4,"#bar{p}");
 // //  pie1->SetEntryLabel(5,"#mu^{-}"); 



 // // pie1->SetEntryLabel(2,"K^{-}");
 // // pie1->SetEntryLabel(3,"p");
 // // pie1->SetEntryLabel(4,"#pi^{+}");

 // //pie1->SetLabelFormat("%txt #scale[0.6]{(%perc)}");
 // pie1->SetAngularOffset(160.);
 // pie1->SetLabelsOffset(1);
 // pie1->SetEntryRadiusOffset(0, 0.02);
 // //  pie1->SetEntryRadiusOffset(8, 0.1);
 // // pie1->SetEntryRadiusOffset(7, 0.16);
 // // pie1->SetEntryRadiusOffset(6, 0.14);
 // // pie1->SetEntryRadiusOffset(5, 0.12);
 // // pie1->SetEntryRadiusOffset(4, 0.05);
 // // pie1->SetEntryRadiusOffset(3, 0.03);
 // // pie1->SetEntryRadiusOffset(2, 0.01);
 // // pie1->SetEntryRadiusOffset(2, 0.04);
 // // pie1->SetEntryRadiusOffset(3, 0.03);
 // // pie1->SetEntryRadiusOffset(4, 0.02);

 // pie1->SetRadius(.35);
 // //  pie1->Draw("r");
 // pie1->Draw("3d");

 // // //  ///[END] secondaries ------------------------------

// // // // //  ///  primary/secondary ---------------------------------------------------------------------------
//   TString filename1 = path+"/TOT_Lumi_BkgSig_el_inel.root";
//   TFile *fin1 = new TFile(filename1,"READ");
//   TTree *tbkg1 = (TTree *)fin1->Get("tBkg"); //Bkg data

//  double n_prim = tbkg1->GetEntries("issecond<0"); //primaries
//  double n_sec = tbkg1->GetEntries("issecond>=0"); //secondaries

//   cout<<"n_prim = "<<n_prim<<endl;
//   cout<<"n_sec = "<<n_sec<<endl;


//   double vals[]={n_prim,n_sec};
//   Int_t colors[] = {41,45};

//   Int_t nvals = sizeof(vals)/sizeof(vals[0]);
//   TPie *pie1 = new TPie("pie1",
//  		       "Pie with offset and no colors",nvals,vals,colors);
//  pie1->SetEntryLabel(0,"primaries");
//  pie1->SetEntryLabel(1,"secondaries");

//  pie1->SetLabelFormat("%txt #scale[0.9]{(%perc)}");
//  pie1->SetAngularOffset(250.);
//  pie1->SetLabelsOffset(0.1);
//  // pie1->SetEntryRadiusOffset(5, 0.12);
//  // pie1->SetEntryRadiusOffset(4, 0.09);
//  // pie1->SetEntryRadiusOffset(3, 0.07);
//  pie1->SetEntryRadiusOffset(0, 0.1);
//  // pie1->SetEntryRadiusOffset(1, 0.02);

//  pie1->SetRadius(.35);
//  pie1->Draw("3d");
//  // // // // /// [END] primary/secondary ---------------------------------------------------------------------------

}