void FirstContour() { TString dir = gSystem->UnixPathName(__FILE__); dir.ReplaceAll("FirstContour.C","../hsimple.C"); dir.ReplaceAll("/./","/"); if (!gInterpreter->IsLoaded(dir.Data())) gInterpreter->LoadMacro(dir.Data()); TFile *file = (TFile*)gROOT->ProcessLineFast("hsimple(1)"); if (!file) return; TTree *ntuple = (TTree*)file->Get("ntuple"); TCanvas *c1 = new TCanvas("c1","Contours",10,10,800,600); gStyle->SetPalette(1); ntuple->Draw("py:px","px*px+py*py < 20", "contz,list"); //we must call Update to force the canvas to be painted. When //painting the contour plot, the list of contours is generated //and a reference to it added to the Root list of special objects c1->Update(); TCanvas *c2 = new TCanvas("c2","First contour",100,100,800,600); TObjArray *contours = (TObjArray*)gROOT->GetListOfSpecials()->FindObject("contours"); if (!contours) return; TList *lcontour1 = (TList*)contours->At(0); if (!lcontour1) return; TGraph *gc1 = (TGraph*)lcontour1->First(); if (!gc1) return; if (gc1->GetN() < 10) return; gc1->SetMarkerStyle(21); gc1->Draw("alp"); //We make a TCutG object with the array obtained from this graph TCutG *cutg = new TCutG("cutg",gc1->GetN(),gc1->GetX(),gc1->GetY()); //We create a polymarker object with npmax points. const Int_t npmax = 50000; TPolyMarker *pm = new TPolyMarker(npmax); Int_t np = 0; while(1) { Double_t x = -4 +8*gRandom->Rndm(); Double_t y = -4 +8*gRandom->Rndm(); if (cutg->IsInside(x,y)) { pm->SetPoint(np,x,y); np++; if (np == npmax) break; } } pm->Draw(); }
void run_plot(TString FileNameHead = "output_proto",TString fileKine="../Kinematics/Decay_kinematics/10Be_4He_19MeV.txt",TString fileKine2="../Kinematics/Decay_kinematics/10Be_12C_19MeV.txt") { TString workdir = getenv("VMCWORKDIR"); TString FilePath = workdir + "/macro/Unpack_GETDecoder2/"; TString FileNameTail = ".root"; TString FileName = FilePath + FileNameHead + FileNameTail; std::cout<<" Opening File : "<<FileName.Data()<<std::endl; TFile* file = new TFile(FileName.Data(),"READ"); //TFile* file = new TFile(FileNameHead_chain.Data(),"READ"); TTree* tree = (TTree*) file -> Get("cbmsim"); Int_t nEvents = tree -> GetEntriesFast(); std::cout<<" Number of events : "<<nEvents<<std::endl; TCanvas *c2 = new TCanvas("c2","c2",200,10,700,700); c2->Divide(2,1); TCanvas *c3 = new TCanvas("c3","c3",200,10,700,700); //c3->Divide(2,1); TH2D* Q02_Kine = new TH2D("Q02_Kine","Q02_Kine",1000,0,180,1000,0,180); Q02_Kine->SetMarkerColor(2); Q02_Kine->SetMarkerStyle(20); Q02_Kine->SetMarkerSize(0.7); TH2D* Q13_Kine = new TH2D("Q13_Kine","Q13_Kine",1000,0,180,1000,0,180); Q13_Kine->SetMarkerColor(2); Q13_Kine->SetMarkerStyle(20); Q13_Kine->SetMarkerSize(0.7); TH1D* Vertex = new TH1D("Vertex","Vertex",100,0,1000); TH2D* Vertex_vs_Angle = new TH2D("Vertex_vs_Angle","Vertex_vs_Angle",1000,0,1000,200,0,180); TCutG *cutg = new TCutG("CUTG",27); cutg->SetVarX("Q02_Kine"); cutg->SetVarY(""); cutg->SetTitle("Graph"); cutg->SetFillColor(1); cutg->SetPoint(0,13.91253,83.41335); cutg->SetPoint(1,16.43826,70.76698); cutg->SetPoint(2,29.69832,47.3185); cutg->SetPoint(3,32.22404,36.51639); cutg->SetPoint(4,49.90412,29.13934); cutg->SetPoint(5,68.21562,18.86417); cutg->SetPoint(6,79.79186,15.70258); cutg->SetPoint(7,86.73761,10.96019); cutg->SetPoint(8,86.52713,6.217797); cutg->SetPoint(9,77.05566,5.427399); cutg->SetPoint(10,69.26801,8.325526); cutg->SetPoint(11,55.79747,15.17564); cutg->SetPoint(12,50.53555,20.9719); cutg->SetPoint(13,35.80215,25.97775); cutg->SetPoint(14,31.17165,21.23536); cutg->SetPoint(15,24.43639,19.3911); cutg->SetPoint(16,19.80589,24.39695); cutg->SetPoint(17,20.01637,30.98361); cutg->SetPoint(18,22.75257,36.25293); cutg->SetPoint(19,21.91066,46.79157); cutg->SetPoint(20,16.0173,54.16862); cutg->SetPoint(21,8.229653,66.02459); cutg->SetPoint(22,6.545836,73.13817); cutg->SetPoint(23,7.177267,86.04801); cutg->SetPoint(24,11.59729,88.15574); cutg->SetPoint(25,13.70206,83.41335); cutg->SetPoint(26,13.91253,83.41335); TCutG *cutg2 = new TCutG("CUTG2",6); cutg2->SetVarX("Q02_Kine"); cutg2->SetVarY(""); cutg2->SetTitle("Graph"); cutg2->SetFillColor(1); cutg2->SetPoint(0,54.85507,28.95589); cutg2->SetPoint(1,64.59005,23.76482); cutg2->SetPoint(2,64.52764,10.0358); cutg2->SetPoint(3,54.81763,12.22152); cutg2->SetPoint(4,54.86755,29.0242); cutg2->SetPoint(5,54.85507,28.95589); TCutG *cutg3 = new TCutG("CUTG2",6); cutg3->SetVarX("Q02_Kine"); cutg3->SetVarY(""); cutg3->SetTitle("Graph"); cutg3->SetFillColor(1); cutg3->SetPoint(0,27.46418,21.43491); cutg3->SetPoint(1,37.45702,16.44231); cutg3->SetPoint(2,37.13467,2.463016); cutg3->SetPoint(3,27.46418,4.792898); cutg3->SetPoint(4,27.78653,21.43491); cutg3->SetPoint(5,27.46418,21.43491); TTreeReader Reader1("cbmsim", file); TTreeReaderValue<TClonesArray> analysisArray(Reader1, "ATProtoEventAna"); Int_t evnt = 0; while (Reader1.Next()) { if(evnt%1000==0) std::cout<<" Event : "<<evnt<<std::endl; evnt++; ATProtoEventAna* analysis = (ATProtoEventAna*) analysisArray->At(0); //Double_t ATProtoAnalysis::*HoughDist = &ATProtoAnalysis::fHoughDist; std::vector<Double_t> *AngleFit = analysis->GetAngleFit(); std::vector<Double_t> *Par0 = analysis->GetPar0(); std::vector<Double_t> *vertex = analysis->GetVertex(); std::vector<Double_t> *Chi2 = analysis->GetChi2(); std::vector<Int_t> *NDF = analysis->GetNDF(); if( TMath::Abs(vertex->at(0) - vertex->at(2))<20 && (vertex->at(0)>0 && vertex->at(2)>0)) if( Chi2->at(0)/NDF->at(0)<10.0 && Chi2->at(2)/NDF->at(2)<10.0 ) Q02_Kine->Fill(AngleFit->at(0),AngleFit->at(2)); if( TMath::Abs(vertex->at(1) - vertex->at(3))<20 && (vertex->at(1)>0 && vertex->at(3)>0)) if( Chi2->at(1)/NDF->at(1)<10.0 && Chi2->at(3)/NDF->at(3)<10.0 ) Q02_Kine->Fill(AngleFit->at(1),AngleFit->at(3)); if(cutg->IsInside(AngleFit->at(0),AngleFit->at(2)) ) { Vertex->Fill(Par0->at(0)); Vertex_vs_Angle->Fill(Par0->at(0),AngleFit->at(0)); } if(cutg->IsInside(AngleFit->at(1),AngleFit->at(3)) ) { Vertex->Fill(Par0->at(1)); Vertex_vs_Angle->Fill(Par0->at(1),AngleFit->at(1)); } //if(cutg3->IsInside(AngleFit->at(0),AngleFit->at(2)) ) std::cout<<evnt<<std::endl; } Double_t *ThetaCMS = new Double_t[20000]; Double_t *ThetaLabRec = new Double_t[20000]; Double_t *EnerLabRec = new Double_t[20000]; Double_t *ThetaLabSca = new Double_t[20000]; Double_t *EnerLabSca = new Double_t[20000]; Double_t *ThetaCMS2 = new Double_t[20000]; Double_t *ThetaLabRec2 = new Double_t[20000]; Double_t *EnerLabRec2 = new Double_t[20000]; Double_t *ThetaLabSca2 = new Double_t[20000]; Double_t *EnerLabSca2 = new Double_t[20000]; std::ifstream *kineStr = new std::ifstream(fileKine.Data()); Int_t numKin=0; if(!kineStr->fail()) { while(!kineStr->eof()) { *kineStr>>ThetaCMS[numKin]>>ThetaLabRec[numKin]>>EnerLabRec[numKin]>>ThetaLabSca[numKin]>>EnerLabSca[numKin]; numKin++; } } else if(kineStr->fail()) std::cout<<" Warning : No Kinematics file found for this reaction! Please run the macro on $SIMPATH/macro/Kinematics/Decay_kinematics/Mainrel.cxx"<<std::endl;
//dEdx cut based on graphical cut defined in dEdxCut.cpp void RunDedxCut(TString inputfile, TString outputfile, TString system, Int_t energy) { cout << "Running dE/dx mode with energy " << energy << endl; TCutG* cutg = initialise_dedx_cutg(system, energy); cout << "Graphcut intialized" << endl; TFile *input_rootfile = new TFile(inputfile); TTree* input_tree = (TTree*)input_rootfile->Get("events"); ParticleTree output_tree(outputfile); Event *event = new Event(); Particle *particle; input_tree->SetBranchAddress("event",&event); const Long64_t treeNentries = input_tree->GetEntries(); Long64_t ev; UInt_t Npa; UInt_t part; Float_t local_dedx; Float_t dedx_uppercut = 3.; if(!(system.CompareTo("PbPb"))) { if((energy == 158 ) || (energy == 160)) dedx_uppercut = 1.65; else if(energy == 20) dedx_uppercut = 1.6; } float p; cout << "Cut dE/dx > " << dedx_uppercut << " applied" << endl; for(ev=0; ev<treeNentries; ++ev) { if(!(ev%500)) cout << "Event: " << ev << endl; input_tree->GetEntry(ev); Npa = event->GetNpa(); output_tree.BeginEvent(); for(part=0; part<Npa; part++) { particle = event->GetParticle(part); p = TMath::Sqrt(TMath::Power(particle->GetPx(),2)+TMath::Power(particle->GetPy(),2)+TMath::Power(particle->GetPz(),2)); local_dedx = choose_dedx(particle, system); if(cutg->IsInside(p,local_dedx)) continue; if(local_dedx > dedx_uppercut) continue; output_tree.AddParticle(particle->GetCharge(), particle->GetBx(), particle->GetBy(), particle->GetPx(), particle->GetPy(), particle->GetPz(), particle->GetdEdx(), particle->GetdEdxVtpc1(), particle->GetdEdxVtpc2(), particle->GetdEdxMtpc(), particle->GetNdEdx(), particle->GetNdEdxVtpc1(), particle->GetNdEdxVtpc2(), particle->GetNdEdxMtpc()); } output_tree.EndEvent(); } output_tree.Close(); input_rootfile->Close(); }