TrackPid/anaplots/PIDplots.h

0001 //
0002 //  PIDplots.h
0003 //
0004 //  Author :    Weihu Ma
0005 //  Created:    04/2021
0006 //  Version:    1.0
0007 //
0008
0009 #include <iostream>
0010 #include <fstream>
0011 #include <iomanip>
0012 #include <stdio.h>
0013 #include <stdlib.h>
0014 #include <string>
0015 #include <math.h>
0016
0017 #include "TROOT.h"
0018 #include "TApplication.h"
0019 #include "TDirectory.h"
0020 #include "TMath.h"
0021 #include "TGraph.h"
0022 #include "TGraph2D.h"
0023 #include "TRandom.h"
0024 #include "TStyle.h"
0025 #include "TCanvas.h"
0026 #include "TPad.h"
0027 #include "TF3.h"
0028 #include "TError.h"
0029 #include "Fit/LogLikelihoodFCN.h"
0030 #include "Fit/BasicFCN.h"
0031 #include "Fit/BinData.h"
0032 #include "Fit/UnBinData.h"
0033 #include "Fit/FcnAdapter.h"
0034 #include "Fit/FitConfig.h"
0035 #include "Fit/FitResult.h"
0036 #include "Fit/Fitter.h"
0037 #include "Fit/Chi2FCN.h"
0038 #include "Fit/PoissonLikelihoodFCN.h"
0039 #include "TVirtualFitter.h"
0040 #include "Math/Minimizer.h"
0041 #include "Math/MinimizerOptions.h"
0042 #include "Math/FitMethodFunction.h"
0043 #include "Math/WrappedMultiTF1.h"
0044 #include "Math/Error.h"
0045 #include "Math/VirtualIntegrator.h"
0046 #include "Math/GSLIntegrator.h"
0047 #include "HFitInterface.h"
0048 #include "Fit/FitExecutionPolicy.h"
0049 #include "TF2.h"
0050 #include "TF1.h"
0051 #include "TGraphErrors.h"
0052 #include "TGraph.h"
0053 #include "TGaxis.h"
0054 #include "TString.h"
0055 #include "TSystem.h"
0056 #include "TTree.h"
0057 #include "TLegend.h"
0058 #include "TText.h"
0059 #include "TLatex.h"
0060 #include "TFile.h"
0061 #include "TAxis.h"
0062 #include "TNtuple.h"
0063 #include "TH1F.h"
0064 #include "TH2F.h"
0065 #include "TFrame.h"
0066 #include "TVector3.h"
0067 #include "TFormula.h"
0068
0069 #include "TMVA/Factory.h"
0070 #include "TMVA/DataLoader.h"
0071 #include "TMVA/Tools.h"
0072 #include "TMVA/TMVAGui.h"
0073
0074
0075 #include "omp.h"
0076 using namespace std;
0077
0078 class PIDplots//A
0079 {
0080  private:
0081
0082   int Nfile;
0083   int Nevent;
0084   int Nmpnbin;
0085   int tag;
0086
0087   float me;
0088   float mep;
0089   float men;
0090   float pi;
0091   float rYs[3];
0092
0093   const char *input_file[400];
0094   char input_file_embed[400];
0095
0096   const char * output_eID;
0097   const char * output_plot_eID;
0098   const char * output_plot_eID_eta;
0099   const char * output_plot_eID_pt;
0100   const char * output_plot_eID_HOM;
0101   const char * output_plot_eID_HOP;
0102   const char * output_plot_eID_EOP;
0103
0104   //////////////////////////////////////////eID
0105   TFile *fHistogramFile_eID;
0106   TH1F *h1pt_e;
0107   TH1F *h1EOP;
0108   TH1F *h1EOP_e;
0109   TH1F *h1HOM;
0110   TH1F *h1HOP;
0111   TH1F *h1dR;//dR=sqrt(deta*deta+dphi*dphi)
0112   TH1F *h1gflavor;
0113
0114   int NEID_electron;
0115
0116   int NEID_electron_eta[50];
0117   int NEID_electron_pt[50];
0118   int NEID_electron_HOP[50];
0119   int NEID_electron_HOM[50];
0120   int NEID_electron_EOP[50];
0121
0122
0123   int N_electron_eta[50];
0124   int N_electron_pt[50];
0125   int N_electron_HOM[50];
0126   int N_electron_HOP[50];
0127   int N_electron_EOP[50];
0128
0129   int N_pion_eta[50];
0130   int N_K_eta[50];
0131   int N_antiproton_eta[50];
0132   int N_pion_pt[50];
0133   int N_K_pt[50];
0134   int N_antiproton_pt[50];
0135   int N_pion_HOM[50];
0136   int N_K_HOM[50];
0137   int N_antiproton_HOM[50];
0138   int N_pion_HOP[50];
0139   int N_K_HOP[50];
0140   int N_antiproton_HOP[50];
0141   int N_pion_EOP[50];
0142   int N_K_EOP[50];
0143   int N_antiproton_EOP[50];
0144
0145   int NEID_pion_eta[50];
0146   int NEID_K_eta[50];
0147   int NEID_antiproton_eta[50];
0148   int NEID_pion_pt[50];
0149   int NEID_K_pt[50];
0150   int NEID_antiproton_pt[50];
0151
0152
0153   int NEID_pion;
0154   int NEID_K;
0155   int NEID_antiproton;
0156   float eta_point[50];
0157   float pt_point[50];
0158   float HOM_point[50];
0159   float HOP_point[50];
0160   float EOP_point[50];
0161
0162 public:
0163
0164   void inputpars();
0165   void fhistogram_eID();
0166   void eID();
0167   void eIDoutput();
0168 };
0169