Files
geant4/examples/extended/medical/dna/dnaphysics/plotRadioactive.C
2025-06-26 09:17:29 +02:00

96 lines
3.3 KiB
C++

// -------------------------------------------------------------------
// -------------------------------------------------------------------
//
// *********************************************************************
// To execute this macro under ROOT after your simulation ended,
// 1 - launch ROOT (usually type 'root' at your machine's prompt)
// 2 - type '.X plot.C' at the ROOT session prompt
// *********************************************************************
void SetLeafAddress(TNtuple* ntuple, const char* name, void* address);
void plotRadioactive()
{
gROOT->Reset();
gStyle->SetPalette(1);
gROOT->SetStyle("Plain");
TCanvas* c1 = new TCanvas ("c1","",20,20,1000,500);
c1->Divide(2,1);
// Uncomment if merging should be done
//system ("rm -rf dna.root");
//system ("hadd dna.root dna_*.root");
TFile* f = new TFile("dna.root");
TNtuple* ntuple;
ntuple = (TNtuple*)f->Get("step");
bool rowWise = true;
TBranch* eventBranch = ntuple->FindBranch("row_wise_branch");
if ( ! eventBranch ) rowWise = false;
// std::cout << "rowWise: " << rowWise << std::endl;
//*********************************************************************
// canvas tab 1
//*********************************************************************
c1->cd(1);
gStyle->SetOptStat(000000);
// All
ntuple->SetFillStyle(1001);
ntuple->SetFillColor(2);
ntuple->Draw("flagProcess","","B");
// Excitation
ntuple->SetFillStyle(1001);
ntuple->SetFillColor(3);
ntuple->Draw("flagProcess","flagProcess==12||flagProcess==15||flagProcess==22||flagProcess==32||flagProcess==42||flagProcess==52||flagProcess==62","Bsame");
// Elastic
ntuple->SetFillStyle(1001);
ntuple->SetFillColor(4);
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");
// Ionisation
ntuple->SetFillStyle(1001);
ntuple->SetFillColor(5);
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");
// Charge decrease
//ntuple->SetFillStyle(1001);
//ntuple->SetFillColor(6);
//ntuple->Draw("flagProcess","flagProcess==24||flagProcess==44||flagProcess==54","Bsame");
// Charge increase
//ntuple->SetFillStyle(1001);
//ntuple->SetFillColor(7);
//ntuple->Draw("flagProcess","flagProcess==35||flagProcess==55||flagProcess==65","Bsame");
gPad->SetLogy();
//*********************************************************************
// canvas tab 2
//*********************************************************************
c1->cd(2);
ntuple->SetMarkerColor(2);
ntuple->Draw("x:y:z","flagParticle==1");
//ntuple->SetMarkerColor(4);
//ntuple->SetMarkerSize(4);
//ntuple->Draw("x:y:z/1000","flagParticle==4 || flagParticle==5 || flagParticle==6","same");
}
void SetLeafAddress(TNtuple* ntuple, const char* name, void* address) {
TLeaf* leaf = ntuple->FindLeaf(name);
if ( ! leaf ) {
std::cerr << "Error in <SetLeafAddress>: unknown leaf --> " << name << std::endl;
return;
}
leaf->SetAddress(address);
}