int main(){
size_t tnum = 200, nnum = 202;
std::vector<double> te(tnum), ne(nnum);
std::vector<std::vector<double>> rate_ii(tnum, std::vector<double>(nnum));
std::vector<std::vector<double>> rate_ie(tnum, std::vector<double>(nnum));
std::vector<std::vector<double>> rate_ei(tnum, std::vector<double>(nnum));
std::vector<std::vector<double>> rate_ee(tnum, std::vector<double>(nnum));
plasmaFormulary::IonProperty proton(myconst::pmass, 1);
for (size_t i = 0; i < tnum; ++i){
te[i] = 0.1 * exp(log(1.0e4 / 0.1) / (tnum - 1) * i);
for (size_t j = 0; j < nnum; ++j){
ne[j] = 1.0e17 * exp(log(1.0e22 / 1.0e17) / (nnum - 1) * j);
rate_ii[i][j] = plasmaFormulary::nu_ii_slow(ne[j], te[i], ne[j], te[i], proton, ne[j], te[i], proton)/ne[j];
rate_ie[i][j] = plasmaFormulary::nu_ie_slow(ne[j], te[i], ne[j], te[i], proton) / ne[j];
rate_ei[i][j] = plasmaFormulary::nu_ei_slow(ne[j], te[i], ne[j], te[i], proton) / ne[j];
rate_ee[i][j] = plasmaFormulary::nu_ee_slow(ne[j], te[i]) / ne[j];
}
}
IGORdata::write_itx(rate_ii, "rate_coef_ii.itx", "rate_coef_ii");
IGORdata::write_itx(rate_ie, "rate_coef_ie.itx", "rate_coef_ie");
IGORdata::write_itx(rate_ei, "rate_coef_ei.itx", "rate_coef_ei");
IGORdata::write_itx(rate_ee, "rate_coef_ee.itx", "rate_coef_ee");
IGORdata::write_edgeVector(te, "te.itx", "te");
IGORdata::write_edgeVector(ne, "ne.itx", "ne");
}
void Lorentz()
{
// gamma + p -> gamma ' + p '
TLorentzVector proton(0,0,0,0.938);
TLorentzVector photon(1,0,0,2);
proton.Print();
photon.Print();
TLorentzVector CM=proton+photon;
CM.Print();
CM.Vect().Print();
v= CM.Vect();
TLorentzVector protonCM;
TLorentzVector photonCM;
protonCM=proton;
protonCM.Boost(-v);
photonCM=photon;
photonCM.Boost(-v);
protonCM.Print();
photonCM.Print();
}
void Loop(Long64_t inEntries)
{
Long64_t nentries = 1;
TH1F *invMassPosNeg = new TH1F();
for (Long64_t jentry=0; jentry<nentries;jentry++) {
if(Stopmin!=1) continue;
TLorentzVector piMinus(3, 2, 1, 0);
TLorentzVector proton(3, 2, 1, 0);
TLorentzVector sum = piMinus+proton;
invMassPosNeg->Fill(3);
}
delete invMassPosNeg;
invMassPosNeg = 0;
}
void newReduceTree(){
// -- define tuple file name, tuple name and cuts to apply
// -- and also the name of the output file
const std::string filename = "/afs/cern.ch/work/a/apmorris/public/Lb2chicpK/signal_samples/Lb2chicpK_MC_2011_2012_signal_cut.root";//change
const std::string treename = "DecayTree";
const std::string cuts = "";
const std::string outFile = "/afs/cern.ch/work/a/apmorris/public/Lb2chicpK/signal_samples/reduced_Lb2chicpK_MC_2011_2012_signal.root");//change
TFile* file = TFile::Open( filename.c_str() );
if( !file ) std::cout << "file " << filename << " does not exist" << std::endl;
TTree* tree = (TTree*)file->Get( treename.c_str() );
if( !tree ) std::cout << "tree " << treename << " does not exist" << std::endl;
// -- activate the branches you need
tree->SetBranchStatus("*", 0);
tree->SetBranchStatus("chi_c_M",1);
tree->SetBranchStatus("chi_c_P",1);
tree->SetBranchStatus("chi_c_PE",1);
tree->SetBranchStatus("chi_c_PT",1);
tree->SetBranchStatus("chi_c_PX",1);
tree->SetBranchStatus("chi_c_PY",1);
tree->SetBranchStatus("chi_c_PZ",1);
tree->SetBranchStatus("chi_c_ETA",1);
tree->SetBranchStatus("kaon_M",1);
tree->SetBranchStatus("kaon_P",1);
tree->SetBranchStatus("kaon_PE",1);
tree->SetBranchStatus("kaon_PT",1);
tree->SetBranchStatus("kaon_PX",1);
tree->SetBranchStatus("kaon_PY",1);
tree->SetBranchStatus("kaon_PZ",1);
tree->SetBranchStatus("kaon_ETA",1);
tree->SetBranchStatus("kaon_IPCHI2_OWNPV",1);
tree->SetBranchStatus("kaon_TRACK_GhostProb",1);
tree->SetBranchStatus("kaon_ProbNNp",1);
tree->SetBranchStatus("kaon_ProbNNk",1);
tree->SetBranchStatus("proton_M",1);
tree->SetBranchStatus("proton_P",1);
tree->SetBranchStatus("proton_PE",1);
tree->SetBranchStatus("proton_PT",1);
tree->SetBranchStatus("proton_PX",1);
tree->SetBranchStatus("proton_PY",1);
tree->SetBranchStatus("proton_PZ",1);
tree->SetBranchStatus("proton_ETA",1);
tree->SetBranchStatus("proton_IPCHI2_OWNPV",1);
tree->SetBranchStatus("proton_TRACK_GhostProb",1);
tree->SetBranchStatus("proton_ProbNNp",1);
tree->SetBranchStatus("proton_ProbNNk",1);
tree->SetBranchStatus("Jpsi_M",1);
tree->SetBranchStatus("Jpsi_P",1);
tree->SetBranchStatus("Jpsi_PE",1);
tree->SetBranchStatus("Jpsi_PT",1);
tree->SetBranchStatus("Jpsi_PX",1);
tree->SetBranchStatus("Jpsi_PY",1);
tree->SetBranchStatus("Jpsi_PZ",1);
tree->SetBranchStatus("Jpsi_ETA",1);
tree->SetBranchStatus("gamma_M",1);
tree->SetBranchStatus("gamma_P",1);
tree->SetBranchStatus("gamma_PE",1);
tree->SetBranchStatus("gamma_PT",1);
tree->SetBranchStatus("gamma_PX",1);
tree->SetBranchStatus("gamma_PY",1);
tree->SetBranchStatus("gamma_PZ",1);
tree->SetBranchStatus("gamma_CL",1);
tree->SetBranchStatus("gamma_ETA",1);
tree->SetBranchStatus("muminus_M",1);
tree->SetBranchStatus("muminus_P",1);
tree->SetBranchStatus("muminus_PE",1);
tree->SetBranchStatus("muminus_PT",1);
tree->SetBranchStatus("muminus_PX",1);
tree->SetBranchStatus("muminus_PY",1);
tree->SetBranchStatus("muminus_PZ",1);
tree->SetBranchStatus("muminus_ETA",1);
tree->SetBranchStatus("muminus_ProbNNmu",1);
tree->SetBranchStatus("muminus_TRACK_GhostProb",1);
tree->SetBranchStatus("muplus_M",1);
tree->SetBranchStatus("muplus_P",1);
tree->SetBranchStatus("muplus_PE",1);
tree->SetBranchStatus("muplus_PT",1);
tree->SetBranchStatus("muplus_PX",1);
tree->SetBranchStatus("muplus_PY",1);
tree->SetBranchStatus("muplus_PZ",1);
tree->SetBranchStatus("muplus_ETA",1);
tree->SetBranchStatus("muplus_ProbNNmu",1);
tree->SetBranchStatus("muplus_TRACK_GhostProb",1);
tree->SetBranchStatus("Lambda_b0_DTF_MASS_constr1",1);
tree->SetBranchStatus("Lambda_b0_DTF_MASS_constr2",1);
tree->SetBranchStatus("Lambda_b0_DTF_CHI2NDOF",1);
tree->SetBranchStatus("Lambda_b0_IPCHI2_OWNPV",1);
tree->SetBranchStatus("Lambda_b0_FDS",1);
tree->SetBranchStatus("Lambda_b0_L0MuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_L0DiMuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1TrackMuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1TrackAllL0Decision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_pi0veto",1);
tree->SetBranchStatus("Lambda_b0_PT",1);
//for MC only //
tree->SetBranchStatus("chi_c_BKGCAT",1); //
tree->SetBranchStatus("Jpsi_BKGCAT",1); //
tree->SetBranchStatus("Lambda_b0_BKGCAT",1); //
// -- this file is just here to make the 'CopyTree' happy
TFile* dummyFile = new TFile("/afs/cern.ch/work/a/apmorris/private/cern/ntuples/dummy.root","RECREATE");
TTree* rTree1 = tree->CopyTree(cuts.c_str());
double chi_c_M, chi_c_P, chi_c_PE, chi_c_PT, chi_c_PX, chi_c_PY, chi_c_PZ, chi_c_ETA;
double kaon_M, kaon_P, kaon_PE, kaon_PX, kaon_PT, kaon_PY, kaon_PZ, kaon_ETA;
double proton_M, proton_P, proton_PE, proton_PT, proton_PX, proton_PY, proton_PZ, proton_ETA;
double Jpsi_M, Jpsi_P, Jpsi_PE, Jpsi_PT, Jpsi_PX, Jpsi_PY, Jpsi_PZ, Jpsi_ETA;
double gamma_M, gamma_P, gamma_PE, gamma_PT, gamma_PX, gamma_PY, gamma_PZ, gamma_ETA, gamma_CL;
double muminus_M, muminus_P, muminus_PE, muminus_PT, muminus_PX, muminus_PY, muminus_PZ, muminus_ETA, muminus_ProbNNmu, muminus_TRACK_GhostProb;
double muplus_M, muplus_P, muplus_PE, muplus_PT, muplus_PX, muplus_PY, muplus_PZ, muplus_ETA, muplus_ProbNNmu, muplus_TRACK_GhostProb;
double kaon_IPCHI2_OWNPV, kaon_TRACK_GhostProb, kaon_ProbNNp, kaon_ProbNNk;
double proton_IPCHI2_OWNPV, proton_TRACK_GhostProb, proton_ProbNNp, proton_ProbNNk;
double Lambda_b0_DTF_MASS_constr1, Lambda_b0_DTF_MASS_constr2, Lambda_b0_DTF_CHI2NDOF, Lambda_b0_IPCHI2_OWNPV;
double Lambda_b0_FDS, Lambda_b0_pi0veto, Lambda_b0_PT;
bool Lambda_b0_L0MuonDecision_TOS, Lambda_b0_L0DiMuonDecision_TOS;
bool Lambda_b0_Hlt1DiMuonHighMassDecision_TOS, Lambda_b0_Hlt1DiMuonLowMassDecision_TOS;
bool Lambda_b0_Hlt1TrackMuonDecision_TOS, Lambda_b0_Hlt1TrackAllL0Decision_TOS;
bool Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS, Lambda_b0_Hlt2DiMuonDetachedDecision_TOS;
bool Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS, Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS;
rTree1->SetBranchAddress("chi_c_M", &chi_c_M);
rTree1->SetBranchAddress("chi_c_P", &chi_c_P);
rTree1->SetBranchAddress("chi_c_PE", &chi_c_PE);
rTree1->SetBranchAddress("chi_c_PT", &chi_c_PT);
rTree1->SetBranchAddress("chi_c_PX", &chi_c_PX);
rTree1->SetBranchAddress("chi_c_PY", &chi_c_PY);
rTree1->SetBranchAddress("chi_c_PZ", &chi_c_PZ);
rTree1->SetBranchAddress("chi_c_ETA", &chi_c_ETA);
rTree1->SetBranchAddress("kaon_M", &kaon_M);
rTree1->SetBranchAddress("kaon_P", &kaon_P);
rTree1->SetBranchAddress("kaon_PE", &kaon_PE);
rTree1->SetBranchAddress("kaon_PX", &kaon_PX);
rTree1->SetBranchAddress("kaon_PT", &kaon_PT);
rTree1->SetBranchAddress("kaon_PY", &kaon_PY);
rTree1->SetBranchAddress("kaon_PZ", &kaon_PZ);
rTree1->SetBranchAddress("kaon_ETA", &kaon_ETA);
rTree1->SetBranchAddress("kaon_IPCHI2_OWNPV", &kaon_IPCHI2_OWNPV);
rTree1->SetBranchAddress("kaon_TRACK_GhostProb", &kaon_TRACK_GhostProb);
rTree1->SetBranchAddress("kaon_ProbNNp", &kaon_ProbNNp);
rTree1->SetBranchAddress("kaon_ProbNNk", &kaon_ProbNNk);
rTree1->SetBranchAddress("proton_M", &proton_M);
rTree1->SetBranchAddress("proton_P", &proton_P);
rTree1->SetBranchAddress("proton_PE", &proton_PE);
rTree1->SetBranchAddress("proton_PT", &proton_PT);
rTree1->SetBranchAddress("proton_PX", &proton_PX);
rTree1->SetBranchAddress("proton_PY", &proton_PY);
rTree1->SetBranchAddress("proton_PZ", &proton_PZ);
rTree1->SetBranchAddress("proton_ETA", &proton_ETA);
rTree1->SetBranchAddress("proton_IPCHI2_OWNPV", &proton_IPCHI2_OWNPV);
rTree1->SetBranchAddress("proton_TRACK_GhostProb", &proton_TRACK_GhostProb);
rTree1->SetBranchAddress("proton_ProbNNp", &proton_ProbNNp);
rTree1->SetBranchAddress("proton_ProbNNk", &proton_ProbNNk);
rTree1->SetBranchAddress("Jpsi_M", &Jpsi_M);
rTree1->SetBranchAddress("Jpsi_P", &Jpsi_P);
rTree1->SetBranchAddress("Jpsi_PE", &Jpsi_PE);
rTree1->SetBranchAddress("Jpsi_PT", &Jpsi_PT);
rTree1->SetBranchAddress("Jpsi_PX", &Jpsi_PX);
rTree1->SetBranchAddress("Jpsi_PY", &Jpsi_PY);
rTree1->SetBranchAddress("Jpsi_PZ", &Jpsi_PZ);
rTree1->SetBranchAddress("Jpsi_ETA", &Jpsi_ETA);
rTree1->SetBranchAddress("gamma_M", &gamma_M);
rTree1->SetBranchAddress("gamma_P", &gamma_P);
rTree1->SetBranchAddress("gamma_PE", &gamma_PE);
rTree1->SetBranchAddress("gamma_PT", &gamma_PT);
rTree1->SetBranchAddress("gamma_PX", &gamma_PX);
rTree1->SetBranchAddress("gamma_PY", &gamma_PY);
rTree1->SetBranchAddress("gamma_PZ", &gamma_PZ);
rTree1->SetBranchAddress("gamma_ETA", &gamma_ETA);
rTree1->SetBranchAddress("gamma_CL", &gamma_CL);
rTree1->SetBranchAddress("muminus_M", &muminus_M);
rTree1->SetBranchAddress("muminus_P", &muminus_P);
rTree1->SetBranchAddress("muminus_PE", &muminus_PE);
rTree1->SetBranchAddress("muminus_PT", &muminus_PT);
rTree1->SetBranchAddress("muminus_PX", &muminus_PX);
rTree1->SetBranchAddress("muminus_PY", &muminus_PY);
rTree1->SetBranchAddress("muminus_PZ", &muminus_PZ);
rTree1->SetBranchAddress("muminus_ETA", &muminus_ETA);
rTree1->SetBranchAddress("muminus_ProbNNmu", &muminus_ProbNNmu);
rTree1->SetBranchAddress("muminus_TRACK_GhostProb", &muminus_TRACK_GhostProb);
rTree1->SetBranchAddress("muplus_M", &muplus_M);
rTree1->SetBranchAddress("muplus_P", &muplus_P);
rTree1->SetBranchAddress("muplus_PE", &muplus_PE);
rTree1->SetBranchAddress("muplus_PT", &muplus_PT);
rTree1->SetBranchAddress("muplus_PX", &muplus_PX);
rTree1->SetBranchAddress("muplus_PY", &muplus_PY);
rTree1->SetBranchAddress("muplus_PZ", &muplus_PZ);
rTree1->SetBranchAddress("muplus_ETA", &muplus_ETA);
rTree1->SetBranchAddress("muplus_ProbNNmu", &muplus_ProbNNmu);
rTree1->SetBranchAddress("muplus_TRACK_GhostProb", &muplus_TRACK_GhostProb);
rTree1->SetBranchAddress("Lambda_b0_DTF_MASS_constr1", &Lambda_b0_DTF_MASS_constr1);
rTree1->SetBranchAddress("Lambda_b0_DTF_MASS_constr2", &Lambda_b0_DTF_MASS_constr2);
rTree1->SetBranchAddress("Lambda_b0_DTF_CHI2NDOF", &Lambda_b0_DTF_CHI2NDOF);
rTree1->SetBranchAddress("Lambda_b0_IPCHI2_OWNPV", &Lambda_b0_IPCHI2_OWNPV);
rTree1->SetBranchAddress("Lambda_b0_L0DiMuonDecision_TOS", &Lambda_b0_L0DiMuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_L0MuonDecision_TOS", &Lambda_b0_L0MuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_FDS", &Lambda_b0_FDS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS", &Lambda_b0_Hlt1DiMuonHighMassDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS", &Lambda_b0_Hlt1DiMuonLowMassDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1TrackMuonDecision_TOS", &Lambda_b0_Hlt1TrackMuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1TrackAllL0Decision_TOS", &Lambda_b0_Hlt1TrackAllL0Decision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS", &Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_pi0veto", &Lambda_b0_pi0veto);
rTree1->SetBranchAddress("Lambda_b0_PT", &Lambda_b0_PT);
//for MC only //
int chi_c_BKGCAT, Jpsi_BKGCAT, Lambda_b0_BKGCAT; //
rTree1->SetBranchAddress("chi_c_BKGCAT", &chi_c_BKGCAT); //
rTree1->SetBranchAddress("Jpsi_BKGCAT", &Jpsi_BKGCAT); //
rTree1->SetBranchAddress("Lambda_b0_BKGCAT", &Lambda_b0_BKGCAT); //
TFile* rFile = new TFile( outFile.c_str() ,"RECREATE");
TTree* rTree2 = new TTree();
rTree2->SetName("DecayTree");
rTree2->Branch("chi_c_M", &chi_c_M, "chi_c_M/D");
rTree2->Branch("chi_c_P", &chi_c_P, "chi_c_P/D");
rTree2->Branch("chi_c_PE", &chi_c_PE, "chi_c_PE/D");
rTree2->Branch("chi_c_PT", &chi_c_PT, "chi_c_PT/D");
rTree2->Branch("chi_c_PX", &chi_c_PX, "chi_c_PX/D");
rTree2->Branch("chi_c_PY", &chi_c_PY, "chi_c_PY/D");
rTree2->Branch("chi_c_PZ", &chi_c_PZ, "chi_c_PZ/D");
rTree2->Branch("chi_c_ETA", &chi_c_ETA, "chi_c_ETA/D");
rTree2->Branch("kaon_M", &kaon_M, "kaon_M/D");
rTree2->Branch("kaon_P", &kaon_P, "kaon_P/D");
rTree2->Branch("kaon_PE", &kaon_PE, "kaon_PE/D");
rTree2->Branch("kaon_PX", &kaon_PX, "kaon_PX/D");
rTree2->Branch("kaon_PT", &kaon_PT, "kaon_PT/D");
rTree2->Branch("kaon_PY", &kaon_PY, "kaon_PY/D");
rTree2->Branch("kaon_PZ", &kaon_PZ, "kaon_PZ/D");
rTree2->Branch("kaon_ETA", &kaon_ETA, "kaon_ETA/D");
rTree2->Branch("kaon_IPCHI2_OWNPV", &kaon_IPCHI2_OWNPV, "kaon_IPCHI2_OWNPV/D");
rTree2->Branch("kaon_TRACK_GhostProb", &kaon_TRACK_GhostProb, "kaon_TRACK_GhostProb/D");
rTree2->Branch("kaon_ProbNNp", &kaon_ProbNNp, "kaon_ProbNNp/D");
rTree2->Branch("kaon_ProbNNk", &kaon_ProbNNk, "kaon_ProbNNk/D");
rTree2->Branch("proton_M", &proton_M, "proton_M/D");
rTree2->Branch("proton_P", &proton_P, "proton_P/D");
rTree2->Branch("proton_PE", &proton_PE, "proton_PE/D");
rTree2->Branch("proton_PT", &proton_PT, "proton_PT/D");
rTree2->Branch("proton_PX", &proton_PX, "proton_PX/D");
rTree2->Branch("proton_PY", &proton_PY, "proton_PY/D");
rTree2->Branch("proton_PZ", &proton_PZ, "proton_PZ/D");
rTree2->Branch("proton_ETA", &proton_ETA, "proton_ETA/D");
rTree2->Branch("proton_IPCHI2_OWNPV", &proton_IPCHI2_OWNPV, "proton_IPCHI2_OWNPV/D");
rTree2->Branch("proton_TRACK_GhostProb", &proton_TRACK_GhostProb, "proton_TRACK_GhostProb/D");
rTree2->Branch("proton_ProbNNp", &proton_ProbNNp, "proton_ProbNNp/D");
rTree2->Branch("proton_ProbNNk", &proton_ProbNNk, "proton_ProbNNk/D");
rTree2->Branch("Jpsi_M", &Jpsi_M, "Jpsi_M/D");
rTree2->Branch("Jpsi_P", &Jpsi_P, "Jpsi_P/D");
rTree2->Branch("Jpsi_PE", &Jpsi_PE, "Jpsi_PE/D");
rTree2->Branch("Jpsi_PT", &Jpsi_PT, "Jpsi_PT/D");
rTree2->Branch("Jpsi_PX", &Jpsi_PX, "Jpsi_PX/D");
rTree2->Branch("Jpsi_PY", &Jpsi_PY, "Jpsi_PY/D");
rTree2->Branch("Jpsi_PZ", &Jpsi_PZ, "Jpsi_PZ/D");
rTree2->Branch("Jpsi_ETA", &Jpsi_ETA, "Jpsi_ETA/D");
rTree2->Branch("gamma_M", &gamma_M, "gamma_M/D");
rTree2->Branch("gamma_P", &gamma_P, "gamma_P/D");
rTree2->Branch("gamma_PE", &gamma_PE, "gamma_PE/D");
rTree2->Branch("gamma_PT", &gamma_PT, "gamma_PT/D");
rTree2->Branch("gamma_PX", &gamma_PX, "gamma_PX/D");
rTree2->Branch("gamma_PY", &gamma_PY, "gamma_PY/D");
rTree2->Branch("gamma_PZ", &gamma_PZ, "gamma_PZ/D");
rTree2->Branch("gamma_ETA", &gamma_ETA, "gamma_ETA/D");
rTree2->Branch("gamma_CL", &gamma_CL, "gamma_CL/D");
rTree2->Branch("muminus_M", &muminus_M, "muminus_M/D");
rTree2->Branch("muminus_P", &muminus_P, "muminus_P/D");
rTree2->Branch("muminus_PE", &muminus_PE, "muminus_PE/D");
rTree2->Branch("muminus_PT", &muminus_PT, "muminus_PT/D");
rTree2->Branch("muminus_PX", &muminus_PX, "muminus_PX/D");
rTree2->Branch("muminus_PY", &muminus_PY, "muminus_PY/D");
rTree2->Branch("muminus_PZ", &muminus_PZ, "muminus_PZ/D");
rTree2->Branch("muminus_ETA", &muminus_ETA, "muminus_ETA/D");
rTree2->Branch("muminus_ProbNNmu", &muminus_ProbNNmu, "muminus_ProbNNmu/D");
rTree2->Branch("muminus_TRACK_GhostProb", &muminus_TRACK_GhostProb, "muminus_TRACK_GhostProb/D");
rTree2->Branch("muplus_M", &muplus_M, "muplus_M/D");
rTree2->Branch("muplus_P", &muplus_P, "muplus_P/D");
rTree2->Branch("muplus_PE", &muplus_PE, "muplus_PE/D");
rTree2->Branch("muplus_PT", &muplus_PT, "muplus_PT/D");
rTree2->Branch("muplus_PX", &muplus_PX, "muplus_PX/D");
rTree2->Branch("muplus_PY", &muplus_PY, "muplus_PY/D");
rTree2->Branch("muplus_PZ", &muplus_PZ, "muplus_PZ/D");
rTree2->Branch("muplus_ETA", &muplus_ETA, "muplus_ETA/D");
rTree2->Branch("muplus_ProbNNmu", &muplus_ProbNNmu, "muplus_ProbNNmu/D");
rTree2->Branch("muplus_TRACK_GhostProb", &muplus_TRACK_GhostProb, "muplus_TRACK_GhostProb/D");
rTree2->Branch("Lambda_b0_DTF_MASS_constr1", &Lambda_b0_DTF_MASS_constr1, "Lambda_b0_DTF_MASS_constr1/D");
rTree2->Branch("Lambda_b0_DTF_MASS_constr2", &Lambda_b0_DTF_MASS_constr2, "Lambda_b0_DTF_MASS_constr2/D");
rTree2->Branch("Lambda_b0_DTF_CHI2NDOF", &Lambda_b0_DTF_CHI2NDOF, "Lambda_b0_DTF_CHI2NDOF/D");
rTree2->Branch("Lambda_b0_IPCHI2_OWNPV", &Lambda_b0_IPCHI2_OWNPV, "Lambda_b0_IPCHI2_OWNPV/D");
rTree2->Branch("Lambda_b0_L0DiMuonDecision_TOS",&Lambda_b0_L0DiMuonDecision_TOS,"Lambda_b0_L0DiMuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_L0MuonDecision_TOS", &Lambda_b0_L0MuonDecision_TOS, "Lambda_b0_L0MuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_FDS", &Lambda_b0_FDS, "Lambda_b0_FDS/D");
rTree2->Branch("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS", &Lambda_b0_Hlt1DiMuonHighMassDecision_TOS, "Lambda_b0_Hlt1DiMuonHighMassDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS", &Lambda_b0_Hlt1DiMuonLowMassDecision_TOS, "Lambda_b0_Hlt1DiMuonLowMassDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1TrackMuonDecision_TOS", &Lambda_b0_Hlt1TrackMuonDecision_TOS, "Lambda_b0_Hlt1TrackMuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1TrackAllL0Decision_TOS", &Lambda_b0_Hlt1TrackAllL0Decision_TOS, "Lambda_b0_Hlt1TrackAllL0Decision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS", &Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS, "Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS/B");
rTree2->Branch("Lambda_b0_pi0veto", &Lambda_b0_pi0veto, "Lambda_b0_pi0veto/D");
rTree2->Branch("Lambda_b0_PT", &Lambda_b0_PT, "Lambda_b0_PT/D");
//for MC only //
rTree2->Branch("chi_c_BKGCAT", &chi_c_BKGCAT, "chi_c_BKGCAT/I"); //
rTree2->Branch("Jpsi_BKGCAT", &Jpsi_BKGCAT, "Jpsi_BKGCAT/I"); //
rTree2->Branch("Lambda_b0_BKGCAT", &Lambda_b0_BKGCAT, "Lambda_b0_BKGCAT/I"); //
//-----------------------------------------------------------------------------------
const double mK(493.677);
const double mpi(139.57);
const double mp(938.27);
double m_pK, m_chicp;
double proton_as_kaon_M, proton_as_pion_M, proton_as_proton_M;
double m_chicpK_proton_as_kaon, m_chicpK_proton_as_pion, m_chicpK_proton_as_proton;
double m_JpsipK;
rTree2->Branch("m_JpsipK", &m_JpsipK, "m_JpsipK/D");
rTree2->Branch("m_chicp", &m_chicp, "m_chicp/D");
rTree2->Branch("m_pK", &m_pK, "m_pK/D");
rTree2->Branch("proton_as_kaon_M", &proton_as_kaon_M, "proton_as_kaon_M/D");
rTree2->Branch("proton_as_pion_M", &proton_as_pion_M, "proton_as_pion_M/D");
rTree2->Branch("proton_as_proton_M", &proton_as_proton_M, "proton_as_proton_M/D");
rTree2->Branch("m_chicpK_proton_as_kaon", &m_chicpK_proton_as_kaon, "m_chicpK_proton_as_kaon/D");
rTree2->Branch("m_chicpK_proton_as_pion", &m_chicpK_proton_as_pion, "m_chicpK_proton_as_pion/D");
rTree2->Branch("m_chicpK_proton_as_proton", &m_chicpK_proton_as_proton, "m_chicpK_proton_as_proton/D");
int percentCounter = 1;
for(int i = 0; i < rTree1->GetEntries(); ++i){
const int percent = (int)(rTree1->GetEntries()/100.0);
if( i == percent*percentCounter ){
std::cout << percentCounter << " %" << std::endl;
percentCounter++;
}
rTree1->GetEntry(i);
double proton_P = sqrt(proton_PX*proton_PX + proton_PY*proton_PY + proton_PZ*proton_PZ) ;
TLorentzVector proton_as_kaon( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + mK*mK));
TLorentzVector proton_as_pion( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + mpi*mpi));
TLorentzVector proton_as_proton( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + proton_M*proton_M));
TLorentzVector kaon( kaon_PX, kaon_PY, kaon_PZ, kaon_PE);
TLorentzVector chic( chi_c_PX, chi_c_PY, chi_c_PZ, chi_c_PE);
TLorentzVector proton(proton_PX, proton_PY, proton_PZ, proton_PE);
TLorentzVector Jpsi( Jpsi_PX, Jpsi_PY, Jpsi_PZ, Jpsi_PE);
proton_as_kaon_M = proton_as_kaon.M();
proton_as_pion_M = proton_as_pion.M();
proton_as_proton_M = proton_as_proton.M();
TLorentzVector chic_PasK_K = chic + kaon + proton_as_kaon;
TLorentzVector chic_PasPi_K = chic + kaon + proton_as_pion;
TLorentzVector chic_PasP_K = chic + kaon + proton_as_proton;
m_chicpK_proton_as_kaon = chic_PasK_K.M();
m_chicpK_proton_as_pion = chic_PasPi_K.M();
m_chicpK_proton_as_proton = chic_PasP_K.M();
TLorentzVector JpsipK = Jpsi + proton + kaon;
TLorentzVector chicp = chic + proton;
TLorentzVector pK = proton + kaon;
m_JpsipK = JpsipK.M();
m_chicp = chicp.M();
m_pK = pK.M();
rTree2->Fill();
}
rTree2->Print();
rTree2->Write();
rFile->Save();
}
void init_physics()
{
ions[0] = proton();
ions[1] = electron();
ions[0].location = (struct vector) {0,0,0};
ions[0].velocity=(struct vector){0,0,0};
ions[1].location = (struct vector) {0,20,0};
ions[1].velocity=(struct vector){0.15,0,0};
}
/*
void init_physics()
{
#define RANDO ((rand()%100)-50)
int i;
for(i=0;i<N_IONS;i++){
switch(rand()%8){
case 0:
ions[i] = nuclion(rand()%9);
break;
default:
ions[i] = electron();
break;
}
ions[i].location = (struct vector) {RANDO,RANDO,RANDO};
ions[i].velocity=(struct vector){0,0,0};
}
}
*/
void dump_state()
{
int i;
printf("--- IONS ---\n");
for(i=0;i<N_IONS;i++){
printf("[%i]\n",i);
printf("charge %Lf \n",ions[i].charge);
printf("mass %Lf \n",ions[i].mass);
v_print("location",ions[i].location);
v_printe("velocity",ions[i].velocity);
v_printe("acceleration",ions[i].accel);
}
printf("-----------\n");
}
void calculate_acceleration(struct particle * p,double dt)
{
struct vector fv = {0,0,0};
int i;
for(i=0;i<N_IONS;i++){
fv = v_add(fv,gravitation(ions[i],*p));
fv = v_add(fv,coulombs(ions[i],*p));
// fv = v_add(fv,biotsavart(ions[i],*p));
}
/* because f/m=a */
fv.x/=p->mass;
fv.y/=p->mass;
fv.z/=p->mass;
p->accel = v_scalar_mul(dt,fv);
}
void apply_vel(struct particle * p)
{
p->velocity = v_add(p->accel,p->velocity);
p->location = v_add(p->location,p->velocity);
}
void physics_tick(double dt)
{
int i;
for(i=0;i<N_IONS;i++)
calculate_acceleration(&ions[i],dt);
for(i=0;i<N_IONS;i++)
apply_vel(&ions[i]);
if(keys['P'])
dump_state();
}
