lars/geant4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
fe04dcb406aa54fc7ca9d45d3d69b5990fc67342
geant4/examples/advanced/hadrontherapy/RootScripts/iaeaBenchmark/fragmentEnergy.C
T
Gabriele Cosmo 74cad5e589 Import Geant4 9.3.0 source tree
2016-06-09 16:15:05 +02:00

201 lines
8.9 KiB
C++
Raw Blame History

#include "Riostream.h"
#include "TSystem.h"
#include "TInterpreter.h"
#include "TROOT.h"
#include "TApplication.h"
#include "TFile.h"
#include "TNtuple.h"
#include "TCanvas.h"
#include "TH1F.h"
#include "TCut.h"
#include "TString.h"
#include "TMath.h"
/**
* Macro for plotting the fragment energy distributions.
*
* Usage:
* root -l RootScripts/iaeaBenchmark/fragmentEnergy.C++
*/
void fragmentEnergy() {
////////////////////////////////////////
////// Importing data /////////
////////////////////////////////////////
TString dir = gSystem->UnixPathName(gInterpreter->GetCurrentMacroName());
dir.ReplaceAll("fragmentEnergy.C","");
dir.ReplaceAll("/./","/");
gStyle->SetOptStat(0000000000); //remove the for this graphs totally redundant statbox
TString macroPath(gROOT->GetMacroPath());
gROOT->SetMacroPath(macroPath + ":RootScripts/iaeaBenchmark");
//gROOT->LoadMacro("rootlogon.C");
//gROOT->SetStyle("clearRetro"); //For stylesheet
TString pDepth;
cout << "Enter phantom depth (eg. 27.9, see experimentalData directory for choices): " << endl;
cout << "Entering 27.9 will make script look for IAEA_27.9.root:";
cin >> pDepth;
TString simulationDataPath = "IAEA_" + pDepth + ".root";
TNtuple *ntuple = new TNtuple("ntuple","Data from ascii file","Energy:He:B:H:Li:Be");
if(pDepth == "27.9"){
//there is only hand.read data for this depth, fixme, is ugly
ifstream in;
in.open(Form("experimentalData/iaeaBenchmark/fragmentEnergySpctra279mmWater0deg.dat",dir.Data()));
Float_t f1,f2,f3, f4,f5,f6;
Int_t nlines = 0;
TFile *f = new TFile("fragmentEnergy.root","RECREATE");
Char_t DATAFLAG[4];
Int_t NDATA;
Char_t n1[6], n2[2], n3[2], n4[2], n5[2], n6[2];
in >> DATAFLAG >> NDATA ; // Read EXFOR line: 'DATA 6'
in >> n1 >> n2 >> n3 >> n4 >> n5 >> n6; // Read column titles: 'Energy He B [...]'
cout <<n1<<" "<<n2<<" "<<n3<<" "<<n4<<" "<<n5<<" "<<n6<<"\n";
while (1) {
in >> f1 >> f2 >> f3 >>f4 >> f5 >> f6;
if (!in.good()) break;
if (nlines < 500 ) printf("%f %0.2f %0.2f %0.2f %0.2f %0.2f \n",f1,f2,f3,f4,f5,f6);
ntuple->Fill(f1,f2,f3,f4,f5,f6);
nlines++;
}
ntuple->SetMarkerStyle(5);
ntuple->Draw("He:Energy","","l");
ntuple->Draw("B:Energy","","l,Same");
ntuple->Draw("H:Energy","","l,Same");
ntuple->Draw("Li:Energy","","l,Same");
ntuple->Draw("Be:Energy","","l,Same");
printf(" found %d points\n",nlines);
}
//Let's pull in the monte carlo simulation results
TCanvas *mc = new TCanvas("mc", "Simulation");
TFile *MCData = TFile::Open(simulationDataPath);
TH1F* MC_helium = (TH1F*)MCData->Get("heliumEnergyAfterPhantom");
TH1F* MC_hydrogen = (TH1F*)MCData->Get("hydrogenEnergyAfterPhantom");
//scale and plot
TNtuple *fragments = (TNtuple*) MCData->Get("fragmentNtuple");
//Block bellow pulls out the simulation's metadata from the metadata ntuple.
TNtuple *metadata = (TNtuple*) MCData->Get("metaData");
Float_t events, detectorDistance,waterThickness,beamEnergy,energyError,phantomCenterDistance;
metadata->SetBranchAddress("events",&events);
metadata->SetBranchAddress("waterThickness",&waterThickness);
metadata->SetBranchAddress("detectorDistance",&detectorDistance);
metadata->SetBranchAddress("beamEnergy",&beamEnergy);
metadata->SetBranchAddress("energyError",&energyError);
metadata->SetBranchAddress("phantomCenterDistance",&phantomCenterDistance);
metadata->GetEntry(0); //there is just one row to consider.
//analysis numbers based on metadata
Double_t scatteringDistance = detectorDistance - phantomCenterDistance;
Double_t detectorSideLength = 4; //hardcoded, we have a zero angle square detector
//good to keep for ref. G4 might give weird units due to change.
metadata->Scan();
Double_t binAmount = 50.0; //casting from int failed somehow, so in float temporarily, fixme
Double_t maxEnergy = 450.0;
Double_t binWidth = maxEnergy / binAmount;
TH1F *histH = new TH1F("histH", "Hydrogen", binAmount, 0.0, maxEnergy);
//HistH will be black
TH1F *histHe = new TH1F("histHe", "Helium", binAmount, 0.0, maxEnergy);
histHe->SetLineColor(kRed);
TH1F *histLi = new TH1F("histLi", "Lithium", binAmount, 0.0, maxEnergy);
histLi->SetLineColor(kBlue);
TH1F *histBe = new TH1F("histBe", "Beryllium", binAmount, 0.0, maxEnergy);
histBe->SetLineColor(kGreen);
TH1F *histB = new TH1F("histB", "Boron", binAmount, 0.0, maxEnergy);
histB->SetLineColor(kYellow);
TH1F *histC = new TH1F("histC", "Carbon", binAmount, 0.0, maxEnergy);
TH1F* histPos = new TH1F("histPos", "check position",100,-2000,2000);
//Solid angle according to \Omega = 4 \arcsin \frac {\alpha\beta} {\sqrt{(4d^2+\alpha^2)(4d^2+\beta^2)}}
Double_t steradians = 4 * TMath::ASin(pow(detectorSideLength,2.0) / (4*pow(scatteringDistance,2) + pow(detectorSideLength,2)) );
std::cout << "Detector seen at solid angle: " << steradians << endl;
TString normalization(Form("/%f", steradians*events*binWidth));
fragments->SetLineColor(kRed);
fragments->SetMarkerStyle(22);
TString halfSideLengthString(Form("%f", detectorSideLength/2));
fragments->SetLineColor(kGreen);
fragments->Project("histHe", "energy", "(Z == 2 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histB", "energy", "(Z == 5 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histH", "energy", "(Z == 1 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histLi", "energy", "(Z == 3 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histBe", "energy", "(Z == 4 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histB", "energy", "(Z == 5 && energy > 45 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
histH->SetMaximum(0.27);
/*
* This is cludgy but the previous plots can't contain <45MeV stuff but hte following calcualtions need them
* Thus there is another projection done with different selections, this is error-prone,
* se to that changes are made in both places.
* */
fragments->Project("histHe", "energy", "(Z == 2 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization);
fragments->Project("histB", "energy", "(Z == 5 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization, "same");
fragments->Project("histH", "energy", "(Z == 1 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization, "same");
fragments->Project("histLi", "energy", "(Z == 3 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization, "same");
fragments->Project("histBe", "energy", "(Z == 4 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization, "same");
fragments->Project("histB", "energy", "(Z == 5 && energy > 0 && abs(posY) < " + halfSideLengthString + " && abs(posZ) < " + halfSideLengthString + " )" + normalization, "same");
histH->SetXTitle("Energy per nucleon (MeV/u)");
histH->SetYTitle("(N/N0) [1/(MeV*sr)]");
TCanvas *c3 = new TCanvas("histograms", "Energy distribution of charged fragments");
Int_t nEve = events; //redundant
cout <<"H : " << histH->GetEntries() / nEve << endl;
histH->Draw("");
cout <<"He : " << histHe->GetEntries() / nEve << endl;
histHe->Draw("same");
cout <<"Li : " << histLi->GetEntries() / nEve << endl;
histLi->Draw("same");
cout <<"Be : " << histBe->GetEntries() / nEve << endl;
histBe->Draw("same");
cout <<"B : " << histB->GetEntries() / nEve << endl;
histB->Draw("same");
// Legends for the data
leg = new TLegend(0.8,0.5,1,1); //coordinates are fractions
leg->SetHeader("Fragments");
leg->AddEntry(histH,"Hydrogen","l");
leg->AddEntry(histHe,"Helium","l");
leg->AddEntry(histLi,"Lithium","l");
leg->AddEntry(histBe,"Beryllium","l");
leg->AddEntry(histB,"Boron","l");
leg->Draw();
ntuple->SetMarkerStyle(22);
ntuple->Draw("H:Energy","","p,same");
ntuple->SetMarkerColor(kRed);
ntuple->Draw("He:Energy","","p,same");
ntuple->SetMarkerColor(kBlue);
ntuple->Draw("Li:Energy","","p,same");
ntuple->SetMarkerColor(kGreen);
ntuple->Draw("Be:Energy","","p,same");
ntuple->SetMarkerColor(kYellow);
ntuple->Draw("B:Energy","","p,same");
c3->SaveAs("fragmentEnergyDistr.png");
in.close();
f->Write();
}
Reference in New Issue View Git Blame Copy Permalink