dna/dnaphysics/plotRadioactive.C

0001 // -------------------------------------------------------------------
0002 // -------------------------------------------------------------------
0003 //
0004 // *********************************************************************
0005 // To execute this macro under ROOT after your simulation ended,
0006 //   1 - launch ROOT (usually type 'root' at your machine's prompt)
0007 //   2 - type '.X plot.C' at the ROOT session prompt
0008 // *********************************************************************
0009
0010 void SetLeafAddress(TNtuple* ntuple, const char* name, void* address);
0011
0012 void plotRadioactive()
0013 {
0014   gROOT->Reset();
0015   gStyle->SetPalette(1);
0016   gROOT->SetStyle("Plain");
0017
0018   TCanvas* c1 = new TCanvas ("c1","",20,20,1000,500);
0019   c1->Divide(2,1);
0020
0021   // Uncomment if merging should be done
0022   //system ("rm -rf dna.root");
0023   //system ("hadd dna.root dna_*.root");
0024
0025   TFile* f = new TFile("dna.root");
0026
0027   TNtuple* ntuple;
0028   ntuple = (TNtuple*)f->Get("step");
0029   bool rowWise = true;
0030   TBranch* eventBranch = ntuple->FindBranch("row_wise_branch");
0031   if ( ! eventBranch ) rowWise = false;
0032   // std::cout <<  "rowWise: " << rowWise << std::endl;
0033
0034   //*********************************************************************
0035   // canvas tab 1
0036   //*********************************************************************
0037
0038   c1->cd(1);
0039   gStyle->SetOptStat(000000);
0040
0041   // All
0042   ntuple->SetFillStyle(1001);
0043   ntuple->SetFillColor(2);
0044   ntuple->Draw("flagProcess","","B");
0045
0046   // Excitation
0047   ntuple->SetFillStyle(1001);
0048   ntuple->SetFillColor(3);
0049   ntuple->Draw("flagProcess","flagProcess==12||flagProcess==15||flagProcess==22||flagProcess==32||flagProcess==42||flagProcess==52||flagProcess==62","Bsame");
0050
0051   // Elastic
0052   ntuple->SetFillStyle(1001);
0053   ntuple->SetFillColor(4);
0054   ntuple->Draw("flagProcess","flagProcess==11||flagProcess==21||flagProcess==31||flagProcess==41||flagProcess==51||flagProcess==61||flagProcess==110||flagProcess==210||flagProcess==410||flagProcess==510||flagProcess==710||flagProcess==120||flagProcess==220||flagProcess==420||flagProcess==520||flagProcess==720","Bsame");
0055
0056   // Ionisation
0057   ntuple->SetFillStyle(1001);
0058   ntuple->SetFillColor(5);
0059   ntuple->Draw("flagProcess","flagProcess==13||flagProcess==23||flagProcess==33||flagProcess==43||flagProcess==53||flagProcess==63||flagProcess==73||flagProcess==130||flagProcess==230||flagProcess==430||flagProcess==530||flagProcess==730","Bsame");
0060
0061   // Charge decrease
0062   //ntuple->SetFillStyle(1001);
0063   //ntuple->SetFillColor(6);
0064   //ntuple->Draw("flagProcess","flagProcess==24||flagProcess==44||flagProcess==54","Bsame");
0065
0066   // Charge increase
0067   //ntuple->SetFillStyle(1001);
0068   //ntuple->SetFillColor(7);
0069   //ntuple->Draw("flagProcess","flagProcess==35||flagProcess==55||flagProcess==65","Bsame");
0070
0071   gPad->SetLogy();
0072
0073   //*********************************************************************
0074   // canvas tab 2
0075   //*********************************************************************
0076
0077   c1->cd(2);
0078
0079   ntuple->SetMarkerColor(2);
0080
0081   ntuple->Draw("x:y:z","flagParticle==1");
0082
0083   //ntuple->SetMarkerColor(4);
0084   //ntuple->SetMarkerSize(4);
0085   //ntuple->Draw("x:y:z/1000","flagParticle==4 || flagParticle==5 || flagParticle==6","same");
0086 }
0087
0088 void SetLeafAddress(TNtuple* ntuple, const char* name, void* address) {
0089   TLeaf* leaf = ntuple->FindLeaf(name);
0090   if ( ! leaf ) {
0091     std::cerr << "Error in <SetLeafAddress>: unknown leaf --> " << name << std::endl;
0092     return;
0093   }
0094   leaf->SetAddress(address);
0095 }