void PixelMergeSmallFiles(){
TFile * oFilebDist = new TFile("bDistr2.root", "RECREATE");
TNtuple* SimEventsGlobal = new TNtuple("SimEventGlob", "SimEventGlob", "EvN:b");
TFile * oFileLinks = new TFile("Links2.root", "RECREATE");
TNtuple* LinksGlobal = new TNtuple("LinksGlob", "LinksGlob", "EvN:fedid:linkn:nHits");
char FileInNumber[5];
string FileInPath = "/net/pstore01/d00/scratch/icali/CMSSW_2_1_11/PixelAnalysis/PixelNTuple_hydjet_x2_mb_oldPL_d20081106/";
string FileInNameRoot= "hydjet_x2_mb_oldPL_d20081106_r0";
Float_t j = 0;
for(int FileN = 901; FileN <1801; ++FileN){
sprintf(FileInNumber, "%.5d", FileN);
string FileInName = FileInPath+FileInNameRoot+FileInNumber+".root";
TFile *iFile = new TFile((const char*)FileInName.c_str());
if(!iFile->IsZombie()){
PixelAnalyzer->cd();
TNtuple *FEDLinks = (TNtuple*)iFile->FindObjectAny("Links");
TNtuple *SimEvents = (TNtuple*)iFile->FindObjectAny("SimEvent");
Float_t FEDEvN, fedid, linkn, nHits;
FEDLinks->SetBranchAddress("EventN",&FEDEvN);
FEDLinks->SetBranchAddress("fedid",&fedid);
FEDLinks->SetBranchAddress("linkn",&linkn);
FEDLinks->SetBranchAddress("nHits",&nHits);
Long64_t FEDLinkLenght =FEDLinks->GetEntries();
Float_t SimEvN, b;
SimEvents->SetBranchAddress("EventN",&SimEvN);
SimEvents->SetBranchAddress("mult",&b);
Long64_t SimEventLenght =SimEvents->GetEntries();
Long64_t i;
for(i=0; i < SimEventLenght; ++i){
SimEvents->GetEntry(i);
SimEventsGlobal->Fill(i, b);
}
Float_t OldEvN= -1;
for(i=0; i < FEDLinkLenght; ++i){
FEDLinks->GetEntry(i);
if(OldEvN != FEDEvN){
OldEvN= FEDEvN;
++j;
}
LinksGlobal->Fill(j, fedid, linkn, nHits);
}
}
}
oFilebDist->Write();
oFileLinks->Write();
}
void read_ntuple_from_file(){
int i,j,k,n;
TFile *in = new TFile("ntupleoutputsample.root");
TNtuple *data = (TNtuple*) in->GetObjectChecked("data","TNtuple");
double pot,cur,temp,pres;
float *row_content; //Must necessarily be float and not a double... WHY?
TH1D *histo = new TH1D("histo","HISTO",100,0,10);
cout << "Potential\tCurrent\tTemperature\tPressure" << endl;
for(i=0;i<data->GetEntries();++i){
data->GetEntry(i);
row_content = data->GetArgs();
pot = row_content[0];
cur = row_content[1];
temp = row_content[2];
pres = row_content[3];
cout << pot << "\t" << cur << "\t" << temp << "\t" << pres << endl;
histo->Fill(pot);
}
histo->Draw();
}
void ROCOccupancy(Int_t StartRun, Int_t EndRun){
using namespace std;
//string FileRoot = "/home/l_tester/log/bt05r";
string FileInName = "ProcessedData.root";
char RunNumber[6];
Float_t j = 0;
for(int RunN = StartRun; RunN <=EndRun; ++RunN){
string FileRoot = "/d00/icali/PixelHwStudies/PSIStuff/log/bt05r";
sprintf(RunNumber, "%.6d", RunN);
FileRoot = FileRoot + RunNumber + "/";
FileInName = FileRoot + FileInName;
TFile *iFile = new TFile((const char*)FileInName.c_str());
if(!iFile->IsZombie()){
TNtuple *Digis = (TNtuple*)iFile->FindObjectAny("events");
Int_t EvN, roc, ROCHits, MeanPh, NCols;
Digis->SetBranchAddress("ROCHits", &ROCHits);
Digis->SetBranchAddress("MeanPh", &MeanPh);
Digis->SetBranchAddress("roc", &roc);
Digis->SetBranchAddress("NCols", &NCols);
Digis->SetBranchAddress("EvN", &EvN);
Int_t iNTlenght = Digis->GetEntries();
Int_t MaxROCHits =0;
for(int i =0 ; i < iNTlenght; ++i){
Digis->GetEntry(i);
if(ROCHits > MaxROCHits) MaxROCHits = ROCHits;
}
cout << "Run: " << RunN << " MaxROCHits: " << MaxROCHits << endl;
}
}
}
void fitBjetJES(int ppPbPb=1, int cbinlo=12, int cbinhi=40){
if(!ppPbPb){
cbinlo=0;
cbinhi=40;
}
gStyle->SetOptTitle(0);
gStyle->SetOptStat(0);
TFile *fL;
if(!ppPbPb)fL=new TFile("histos/ppMC_hiReco_jetTrig_highPurity_JEC.root");
else fL=new TFile("histos/PbPbQCDMC_pt30by3_ipHICalibCentWeight.root");
// these are dummy files for pp
TFile *fB=new TFile("histos/PbPbBMC_pt30by3_ipHICalibCentWeight.root");
TFile *fC=new TFile("histos/PbPbCMC_pt30by3_ipHICalibCentWeight.root");
TNtuple *tL = (TNtuple*) fL->Get("nt");
TNtuple *tB = (TNtuple*) fB->Get("nt");
TNtuple *tC = (TNtuple*) fC->Get("nt");
float jtptL, refptL, jtetaL, weightL, refparton_flavorForBL, binL;
tL->SetBranchAddress("jtpt",&jtptL);
tL->SetBranchAddress("jteta",&jtetaL);
tL->SetBranchAddress("refpt",&refptL);
tL->SetBranchAddress("weight",&weightL);
if(ppPbPb)tL->SetBranchAddress("bin",&binL);
tL->SetBranchAddress("refparton_flavorForB",&refparton_flavorForBL);
float jtptB, refptB, jtetaB, weightB, refparton_flavorForBB, binB;
tB->SetBranchAddress("jtpt",&jtptB);
tB->SetBranchAddress("jteta",&jtetaB);
tB->SetBranchAddress("refpt",&refptB);
tB->SetBranchAddress("weight",&weightB);
if(ppPbPb)tB->SetBranchAddress("bin",&binB);
tB->SetBranchAddress("refparton_flavorForB",&refparton_flavorForBB);
float jtptC, refptC, jtetaC, weightC, refparton_flavorForBC, binC;
tC->SetBranchAddress("jtpt",&jtptC);
tC->SetBranchAddress("jteta",&jtetaC);
tC->SetBranchAddress("refpt",&refptC);
tC->SetBranchAddress("weight",&weightC);
if(ppPbPb)tC->SetBranchAddress("bin",&binC);
tC->SetBranchAddress("refparton_flavorForB",&refparton_flavorForBC);
TProfile *hL = new TProfile("hL","hL",250,50,300,0,10);
TProfile *hB = new TProfile("hB","hB",250,50,300,0,10);
TProfile *hC = new TProfile("hC","hC",250,50,300,0,10);
hL->Sumw2(),hB->Sumw2(),hC->Sumw2();
for(int i=0;i<tL->GetEntries();i++){
tL->GetEntry(i);
if(!ppPbPb) binL=39;
if(fabs(jtetaL)<2 && binL>=cbinlo && binL<cbinhi)
hL->Fill(refptL,jtptL/refptL,weightL);
if(!ppPbPb){
if(fabs(jtetaL)<2 && binL>=cbinlo && binL<cbinhi && abs(refparton_flavorForBL)==5)
hB->Fill(refptL,jtptL/refptL,weightL);
if(fabs(jtetaL)<2 && binL>=cbinlo && binL<cbinhi && abs(refparton_flavorForBL)==4)
hC->Fill(refptL,jtptL/refptL,weightL);
}
}
if(ppPbPb){
for(int i=0;i<tB->GetEntries();i++){
tB->GetEntry(i);
if(fabs(jtetaB)<2 && binB>=cbinlo && binB<cbinhi && abs(refparton_flavorForBB)==5)
hB->Fill(refptB,jtptB/refptB,weightB);
}
for(int i=0;i<tC->GetEntries();i++){
tC->GetEntry(i);
if(fabs(jtetaC)<2 && binC>=cbinlo && binC<cbinhi && abs(refparton_flavorForBC)==4)
hC->Fill(refptC,jtptC/refptC,weightC);
}
}
hL->SetMinimum(0.);
hL->SetLineColor(kBlue);
hB->SetLineColor(kRed);
hC->SetLineColor(kGreen);
hL->SetMarkerColor(kBlue);
hB->SetMarkerColor(kRed);
hC->SetMarkerColor(kGreen);
//hL->SetMarkerStyle(4);
//hB->SetMarkerStyle(4);
//hC->SetMarkerStyle(4);
hL->SetXTitle("genJet p_{T} (GeV/c)");
hL->SetYTitle("<reco p_{T} / gen p_{T} >");
hL->GetXaxis()->SetRangeUser(50.,199.999);
hL->GetYaxis()->SetRangeUser(0.5,1.05);
TCanvas *c1=new TCanvas("c1","c1",800,600);
c1->SetGridx(1);
c1->SetGridy(1);
hL->Draw("e1");
hB->Draw("e1,same");
hC->Draw("e1,same");
TLegend *leg=new TLegend(0.4,0.15,0.9,0.45);
leg->SetBorderSize(0);
leg->SetFillStyle(0);
if(ppPbPb&&cbinlo==0&&cbinhi==40)leg->SetHeader("Pythia+Hydjet, 0-100%");
leg->AddEntry(hL,"Inclusive jets","pl");
leg->AddEntry(hC,"c-jets","pl");
leg->AddEntry(hB,"b-jets","pl");
leg->Draw();
TCanvas *c2=new TCanvas("c2","c2",1);
/*
TH1F *hL2 = (TH1F*)hL->Clone("hL2");
TH1F *hB2 = (TH1F*)hB->Clone("hB2");
hL2->Add(hB2,-1);
hL2->Draw();
*/
TH1F *hcorr = new TH1F("hcorr","hcorr",250,50,300);
hcorr->Sumw2();
for(int i=0;i<hL->GetNbinsX();i++){
cout<<" b resp "<<hB->GetBinContent(i+1)<<endl;
cout<<" l resp "<<hL->GetBinContent(i+1)<<endl;
cout<<" l offset "<<1.-hL->GetBinContent(i+1)<<endl;
cout<<" corrected b resp "<<hB->GetBinContent(i+1)+1.-hL->GetBinContent(i+1)<<endl;
float jesOffset = 1.-hL->GetBinContent(i+1);
hcorr->SetBinContent(i+1,hB->GetBinContent(i+1)+jesOffset);
hcorr->SetBinError(i+1,sqrt(hB->GetBinError(i+1)*hB->GetBinError(i+1)+hL->GetBinError(i+1)*hL->GetBinError(i+1)));
}
hcorr->SetMinimum(0.5);
hcorr->SetMaximum(1.1);
hcorr->SetLineColor(kRed);
hcorr->SetMarkerColor(kRed);
hcorr->SetMarkerStyle(4);
hcorr->Draw();
TF1 *fCorr = new TF1("fCorr","[0]+[1]*log(x)+[2]*log(x)*log(x)",50,300);
fCorr->SetLineWidth(1);
fCorr->SetLineColor(kBlue);
hcorr->Fit(fCorr);
TFile *fout;
if(ppPbPb) fout =new TFile(Form("bJEShistos/bJetScale_PbPb_Cent_fineBin_%d_%d.root",cbinlo,cbinhi),"recreate");
else fout =new TFile("bJEShistos/bJetScale_PP_fineBin.root","recreate");
hcorr->Write();
fCorr->Write();
fout->Close();
}
int main(int argc, char* argv[]){
TApplication* a = new TApplication("a", 0, 0);
TStyle* style= DrawTools::setStyle();
style->cd();
std::string outputdir = "CalibComparisonPlots/";
std::string mkdir_command = "mkdir -p " + outputdir;
system( mkdir_command.c_str() );
std::string runName0 = "BTF_259_20140502-012847_beam";
std::string runName1 = "BTF_259_20140502-012847_beam";
std::string runName2 = "BTF_259_20140502-012847_beam";
std::string runName3 = "BTF_259_20140502-012847_beam";
std::string runName4 = "BTF_259_20140502-012847_beam";
//std::string tag = "default";
std::string tag = "V03";
std::string inputDir = "./CeF3Calibration";
if( argc == 5 ) {
std::string runName_str0(argv[1]);
runName0 = runName_str0;
std::string runName_str1(argv[2]);
runName1 = runName_str1;
std::string runName_str2(argv[3]);
runName2 = runName_str2;
std::string tag_str(argv[4]);
tag = tag_str;
} else if(argc == 7){
std::string runName_str0(argv[1]);
runName0 = runName_str0;
std::string runName_str1(argv[2]);
runName1 = runName_str1;
std::string runName_str2(argv[3]);
runName2 = runName_str2;
std::string runName_str3(argv[4]);
runName3 = runName_str3;
std::string runName_str4(argv[5]);
runName4 = runName_str4;
std::string tag_str(argv[6]);
tag = tag_str;
} else{
std::cout<<"Usage:"<<std::endl;
std::cout<<"./calibrateCef3 BTF_XX BTF_XX BTF_XX tag "<<std::endl;
exit(12345);
}
TCanvas* canny = new TCanvas("canny", "",200,200);
canny->cd();
ifstream in;
float x;
int nlines;
TNtuple *ntuple = new TNtuple("ntuple","data from .txt file", "x");
double corr0[4];
double corr_uncert0[4];
double corr1[4];
double corr_uncert1[4];
double corr2[4];
double corr_uncert2[4];
double corr3[4];
double corr_uncert3[4];
double corr4[4];
double corr_uncert4[4];
int const ntot = 10;
TString openname[ntot] = {
Form("%s/constants_%s_%s.txt",inputDir.c_str(),runName0.c_str(), tag.c_str() ), Form("%s/constants_uncert_%s_%s.txt", inputDir.c_str(),runName0.c_str(), tag.c_str()), Form("%s/constants_%s_%s.txt", inputDir.c_str(),runName1.c_str(), tag.c_str()), Form("%s/constants_uncert_%s_%s.txt", inputDir.c_str(),runName1.c_str(), tag.c_str()), Form("%s/constants_%s_%s.txt",inputDir.c_str(), runName2.c_str(), tag.c_str() ),Form("%s/constants_uncert_%s_%s.txt", inputDir.c_str(), runName2.c_str(), tag.c_str() ), Form("%s/constants_%s_%s.txt", inputDir.c_str(), runName3.c_str(), tag.c_str() ), Form("%s/constants_uncert_%s_%s.txt", inputDir.c_str(), runName3.c_str(), tag.c_str() ), Form("%s/constants_%s_%s.txt", inputDir.c_str(), runName4.c_str(), tag.c_str() ),Form("%s/constants_uncert_%s_%s.txt", inputDir.c_str(), runName4.c_str(), tag.c_str() ) };
for(int i=0; i<ntot+1; ++i){
in.open(openname[i-1]);
while (nlines<4) {
in>>x;
if(!in.good()) break;
ntuple->Fill(x);
nlines++;
}
in.close();
nlines=0;
}
ntuple->SetBranchAddress("x",&x);
for(int j=0; j<4; ++j){
ntuple->GetEntry(j);
corr0[j]= x;
ntuple->GetEntry(j+4);
corr_uncert0[j]=x;
ntuple->GetEntry(j+8);
corr1[j]= x;
ntuple->GetEntry(j+12);
corr_uncert1[j]=x;
ntuple->GetEntry(j+16);
corr2[j]= x;
ntuple->GetEntry(j+20);
corr_uncert2[j]=x;
if(argc==7){
ntuple->GetEntry(j+24);
corr3[j]= x;
ntuple->GetEntry(j+28);
corr_uncert3[j]=x;
ntuple->GetEntry(j+32);
corr4[j]= x;
ntuple->GetEntry(j+36);
corr_uncert4[j]=x;
}
}
double ch0[4] = {-0.1,0.9,1.9,2.9};
double ch1[4]={-0.05,0.95,1.95, 2.95};
double ch2[4] = {0,1,2,3};
double ch3[4] = {0.05,1.05,2.05,3.05};
double ch4[4] ={0.1,1.1,2.1,3.1};
double cherr[4]={0,0,0,0};
TGraphErrors* graf0 = new TGraphErrors(4,ch0,corr0,cherr,corr_uncert0);
graf0->SetMarkerStyle(9);
graf0->SetMarkerSize(0.5);
graf0->SetMarkerColor(kGreen+1);
TGraphErrors* graf1 = new TGraphErrors(4,ch1,corr1,cherr,corr_uncert1);
graf1->SetMarkerStyle(9);
graf1->SetMarkerSize(0.5);
graf1->SetMarkerColor(kBlue);
TGraphErrors* graf2 = new TGraphErrors(4,ch2,corr2,cherr,corr_uncert2);
graf2->SetMarkerStyle(9);
graf2->SetMarkerSize(0.5);
graf2->SetMarkerColor(kMagenta);
TMultiGraph *multi=new TMultiGraph();
multi->Add(graf0); multi->Add(graf1); multi->Add(graf2);
TLegend* leg = new TLegend(0.7, 0.7, 0.9, 0.9);
if(argc==5){
leg->AddEntry(graf0,"Run 92","P");
leg->AddEntry(graf1,"Run 246","P");
leg->AddEntry(graf2,"Run 259","P");
multi->SetTitle(";Channel Nr.;Correction Factor");
multi->Draw("AP");
multi->GetYaxis()->SetRangeUser(0.975,1.025);
multi->GetXaxis()->SetNdivisions(4);
}
if(argc==7){
graf1->SetMarkerColor(kCyan+1);
graf2->SetMarkerColor(kBlue);
TGraphErrors* graf3 = new TGraphErrors(4,ch3,corr3,cherr,corr_uncert3);
graf3->SetMarkerStyle(9);
graf3->SetMarkerSize(0.5);
graf3->SetMarkerColor(kViolet-1);
TGraphErrors* graf4 = new TGraphErrors(4,ch4,corr4,cherr,corr_uncert4);
graf4->SetMarkerStyle(9);
graf4->SetMarkerSize(0.5);
graf4->SetMarkerColor(kPink+7);
multi->Add(graf3);
multi->Add(graf4);
multi->SetTitle(";Channel Nr.;Correction Factor");
multi->Draw("AP");
leg->AddEntry(graf0,"98.3 MeV","P");
leg->AddEntry(graf1,"147.4 MeV","P");
leg->AddEntry(graf2,"196.5 MeV","P");
leg->AddEntry(graf3,"294.8 MeV","P");
leg->AddEntry(graf4,"491.4 MeV","P");
//multi->GetYaxis()->SetRangeUser(0.9,1.1);
multi->GetXaxis()->SetNdivisions(4);
}
multi->SetTitle(";Channel Nr.;Correction Factor");
multi->Draw("AP");
canny->Update();
leg->SetFillColor(0);
leg->Draw("same");
TLine* lin = new TLine(-0.25,1.,3.25,1.);
lin->SetLineColor(kRed);
lin->Draw();
if(argc==5){canny->SaveAs( Form( "%s/comparison_%s.pdf", outputdir.c_str(), tag.c_str() ));
} else if(argc==7){ canny->SaveAs( Form( "%s/comparison_energy_%s.pdf", outputdir.c_str(), tag.c_str() ));}
delete canny;
return 0;
}
Int_t mt102_readNtuplesFillHistosAndFit()
{
// No nuisance for batch execution
gROOT->SetBatch();
// Perform the operation sequentially ---------------------------------------
TChain inputChain("multiCore");
inputChain.Add("mc101_multiCore_*.root");
TH1F outHisto("outHisto", "Random Numbers", 128, -4, 4);
{
TimerRAII t("Sequential read and fit");
inputChain.Draw("r >> outHisto");
outHisto.Fit("gaus");
}
// We now go MT! ------------------------------------------------------------
// The first, fundamental operation to be performed in order to make ROOT
// thread-aware.
ROOT::EnableMT();
// We adapt our parallelisation to the number of input files
const auto nFiles = inputChain.GetListOfFiles()->GetEntries();
std::forward_list<UInt_t> workerIDs(nFiles);
std::iota(std::begin(workerIDs), std::end(workerIDs), 0);
// We define the histograms we'll fill
std::vector<TH1F> histograms;
histograms.reserve(nFiles);
for (auto workerID : workerIDs){
histograms.emplace_back(TH1F(Form("outHisto_%u", workerID), "Random Numbers", 128, -4, 4));
}
// We define our work item
auto workItem = [&histograms](UInt_t workerID) {
TFile f(Form("mc101_multiCore_%u.root", workerID));
TNtuple *ntuple = nullptr;
f.GetObject("multiCore", ntuple);
auto &histo = histograms.at(workerID);
for (UInt_t index = 0; index < ntuple->GetEntriesFast(); ++index) {
ntuple->GetEntry(index);
histo.Fill(ntuple->GetArgs()[0]);
}
};
TH1F sumHistogram("SumHisto", "Random Numbers", 128, -4, 4);
// Create the collection which will hold the threads, our "pool"
std::vector<std::thread> workers;
// We measure time here as well
{
TimerRAII t("Parallel execution");
// Spawn workers
// Fill the "pool" with workers
for (auto workerID : workerIDs) {
workers.emplace_back(workItem, workerID);
}
// Now join them
for (auto&& worker : workers) worker.join();
// And reduce
std::for_each(std::begin(histograms), std::end(histograms),
[&sumHistogram](const TH1F & h) {
sumHistogram.Add(&h);
});
sumHistogram.Fit("gaus",0);
}
return 0;
}
int main(int argc, char **argv)
{
TFile *f;
TFile *newf;
TNtuple *ntuple;
TNtuple *newntuple;
TH1F *h1;
TF1 *func;
TString Metal;
TString dataLoc;
Int_t dataSL;
if(argc < 5 || argc > 6){
std::cout << "The number of arguments is incorrect" << std::endl;
return 0;
}
dataLoc = argv[1];
PHI_MIN = (Double_t) std::stod(argv[2]);
PHI_MAX = (Double_t) std::stod(argv[3]);
N_PHI = (Int_t) std::stoi(argv[4]);
if(argc >= 6) Metal = argv[5];
dataSL = dataLoc.Length();
if(dataLoc(dataSL-1, 1) != "/"){
dataLoc = dataLoc + "/";
}
std::cout << "The following settings are being used" << std::endl;
std::cout << "Data Directory = " << dataLoc << std::endl;
std::cout << PHI_MIN << " < PHI < " << PHI_MAX << ", N_PHI = " << N_PHI << std::endl;
if(argc == 6) std::cout << "Running only for Metal "<< Metal << std::endl;
else std::cout << "Running all Metals" << std::endl;
Float_t Q2, Xb, Zh, Pt, Phi, Val, Err;
Float_t A, Ac, Acc;
Float_t AErr, AcErr, AccErr;
Float_t ChiSQ;
Int_t nentries, empty;
if(Metal == "") N_METAL = 6;
else N_METAL = 1;
for(Int_t met = 0; met < N_METAL; met++){
if(Metal == ""){
if(met == 0) Metal = "C";
else if(met == 1) Metal = "Fe";
else if(met == 2) Metal = "Pb";
else if(met == 3) Metal = "D_C";
else if(met == 4) Metal = "D_Fe";
else if(met == 5) Metal = "D_Pb";
}
if(dataLoc == "")
f = new TFile("../Gen5DimData/" + Metal + "_5_dim_dist.root", "READ");
else
f = new TFile(dataLoc + Metal + "_5_dim_dist.root", "READ");
ntuple = (TNtuple*) f->Get("fit_data");
nentries = ntuple->GetEntries();
ntuple->SetBranchAddress("Xb", &Xb);
ntuple->SetBranchAddress("Q2", &Q2);
ntuple->SetBranchAddress("Xb", &Xb);
ntuple->SetBranchAddress("Zh", &Zh);
ntuple->SetBranchAddress("Pt", &Pt);
ntuple->SetBranchAddress("Phi", &Phi);
ntuple->SetBranchAddress("Val", &Val);
ntuple->SetBranchAddress("Err", &Err);
newntuple = new TNtuple("AAcAcc_data", "AAcAcc_data", "Q2:Xb:Zh:Pt:A:AErr:Ac:AcErr:Acc:AccErr:ChiSQ");
func = new TF1("fit", "[0]+TMath::Cos(x*TMath::Pi()/180)*[1]+TMath::Cos(2*x*TMath::Pi()/180)*[2]");
newf = new TFile(Metal + "newphihist.root", "RECREATE");
newf->cd();
for(Int_t i = 0; i < nentries; i = i + N_PHI){
ntuple->GetEntry(i);
h1 = new TH1F((const char*) Form("PhiDist Q2=%.3f Xb=%.3f Zh=%.3f Pt=%.3f", Q2, Xb, Zh, Pt),
(const char*) Form("PhiDist Q2=%.3f Xb=%.3f Zh=%.3f Pt=%.3f", Q2, Xb, Zh, Pt), N_PHI, PHI_MIN, PHI_MAX);
empty = 0;
for(Int_t j = 1; j <= N_PHI; j++){
ntuple->GetEntry(i+j-1);
h1->SetBinContent(j, Val);
if(h1->GetBinContent(j) == 0)
empty++;
h1->SetBinError(j, Err*1.04);
}
if(empty <= (N_PHI - MIN_BIN)){
h1->Fit(func, "q");
h1->Write();
if(func->GetNDF() != 0){
ChiSQ = func->GetChisquare();
A = func->GetParameter(0);
AErr = func->GetParError(0);
Ac = func->GetParameter(1);
AcErr = func->GetParError(1);
Acc = func->GetParameter(2);
AccErr = func->GetParError(2);
newntuple->Fill(Q2, Xb, Zh, Pt, A, AErr, Ac, AcErr, Acc, AccErr, ChiSQ);
}
}
h1->Delete();
}
Metal = "";
delete ntuple;
newf->cd();
newntuple->Write();
newntuple->Delete();
newf->Close();
delete newf;
f->Close();
delete f;
}
return 0;
}
int main( int argc, char* argv[] )
{
cout << endl;
cout << "\t8888888b. d8b 888 8888888b. 888 888 " << endl;
cout << "\t888 Y88b Y8P 888 888 Y88b 888 888 " << endl;
cout << "\t888 888 888 888 888 888 888 " << endl;
cout << "\t888 d88P 8888b. 88888b. 888 .d88888 888 d88P 888 .d88b. 888888 " << endl;
cout << "\t8888888P\" \"88b 888 \"88b 888 d88\" 888 8888888P\" 888 d88\"\"88b 888 " << endl;
cout << "\t888 T88b .d888888 888 888 888 888 888 888 888 888 888 888 " << endl;
cout << "\t888 T88b 888 888 888 d88P 888 Y88b 888 888 888 Y88..88P Y88b. " << endl;
cout << "\t888 T88b \"Y888888 88888P\" 888 \"Y88888 888 888 \"Y88P\" \"Y888 " << endl;
cout << "\t 888 " << endl;
cout << "\t 888 " << endl;
cout << "\t 888 " << endl;
cout << endl << endl;
cout << "Usage:" << endl;
cout << "\t" << argv[0] << "\tSomeFile.root\t" << "phys_param1\tphys_param2\toutput_directory" << endl;
cout << endl << "\teg:\t" << argv[0] << "\tLLcontourData.root\tdeltaGamma\tPhi_s\toutputdir"<<endl;
cout << endl;
cout << "6th (Optional) Option:"<<endl;
cout << "\t\t\tCV\t\t(Plot the CV of 'nuisence' Physics Parameters)"<<endl;
cout << "\t\t\tRelCV\t\t(Plot the variation in the CV of the 'nuisence' Physics Parameters)"<<endl;
cout << "\t\t\tExtraCV\t\t(Plot all of the CV, errors and pulls for each 'nuisence' Physics Parameter)"<<endl;
cout << "\t\t\tNOFC\t\t(Don't process FC data in a file if any is there)"<<endl;
cout << endl;
if( (argc != 5) && (argc != 6) ) exit(-1);
// Use UUID based seed from ROOT, just used for unique identification of ROOT objects
TRandom3* rand_gen = new TRandom3(0);
// Setup the Canvas and such
EdStyle* RapidFit_Style = new EdStyle();
RapidFit_Style->SetStyle();
// By Design
TString outputdir = argv[4];
TString param1string = argv[2];
TString param2string = argv[3];
// Make OutputDir
gSystem->mkdir( outputdir );
// Open the File
TFile * input = TFile::Open( argv[1] );
TNtuple* allresults;
input->GetObject("RapidFitResult", allresults);
if(!allresults){
input->GetObject("RapidFitResult/RapidFitResult",allresults);
if(!allresults){
cout << "Couldn't find ntuple RapidFitResult in TFile" << endl;
exit(1);
}
}
// Switches for different plotting
bool CV_Drift = false;
bool Want_Physics_Params=false;
bool Want_Extra_Physics_Param_Info=false;
bool want_FC = true;
bool high_res = true; // False method may only call plot fully once... still in development
if( argc == 6 )
{
if( string(argv[5]) == "CV" ) Want_Physics_Params = true;
if( string(argv[5]) == "RelCV" ) { Want_Physics_Params = true; CV_Drift = true; }
if( string(argv[5]) == "ExtraCV" ) { Want_Physics_Params = true; Want_Extra_Physics_Param_Info = true; }
if( string(argv[5]) == "NOFC" ) want_FC = false;
}
// Strings that are universally defined
TString Double_Tolerance="0.000001";
TString Fit_Status = "Fit_Status";
TString NLL = "NLL";
TString notgen = "-9999.";
TString error_suffix = "_error";
TString value_suffix = "_value";
TString pull_suffix = "_pull";
TString gen_suffix = "_gen";
TString Copy_Option = "fast";
// Strings that are relevent to this plot
// This is currently given by the user, but in time we can write an algorithm to determine this
TString param1_gen = param1string + gen_suffix;
TString param1_val = param1string + value_suffix;
TString param1_err = param1string + error_suffix;
TString param2_gen = param2string + gen_suffix;
TString param2_val = param2string + value_suffix;
TString param2_err = param2string + error_suffix;
// Get a list of ALL parameters within the file, this is given for free in the algorithms I wrote above
// Get a list of all branches in allresults
vector<TString> all_parameters = get_branch_names( allresults );
// Get a list of all branches in allresults with '_value' in their name
vector<TString> all_parameter_values = filter_names( all_parameters, value_suffix );
vector<TString> all_parameter_errors = filter_names( all_parameters, error_suffix );
vector<TString> all_parameter_pulls = filter_names( all_parameters, pull_suffix );
// Now to read in the Global Minima
// Store the Global fit corrdinate and value
Float_t nll=0, Global_Best_NLL=0, x_point=0, y_point=0;
// Store the value of the 'nuisence' parameters as well
Float_t* best_fit_values = new Float_t[all_parameter_values.size()];
// Construct an array to hold the best values for the rest of the parameters
Float_t* best_fit_temp_values = new Float_t[all_parameter_values.size()];
allresults->SetBranchAddress(NLL,&nll);
// Tell ROOT where to store the values
for( unsigned short int i=0; i < all_parameter_values.size(); ++i )
{
allresults->SetBranchAddress(all_parameter_values[i],&best_fit_temp_values[i]);
}
// Actually store the values in resident memory of the program
cout << endl << "BEST FIT RESULTS AS DETERMINED FROM THE STANDARD FIT:" << endl;
allresults->GetEntry(0);
Global_Best_NLL = nll;
for( unsigned short int i=0; i < all_parameter_values.size(); ++i )
{
cout << all_parameter_values[i] << ":\t" << best_fit_temp_values[i] << endl;
best_fit_values[i] = best_fit_temp_values[i];
if( string(all_parameter_values[i].Data()).compare(param1_val.Data()) == 0 ) x_point = best_fit_temp_values[i];
if( string(all_parameter_values[i].Data()).compare(param2_val.Data()) == 0 ) y_point = best_fit_temp_values[i];
}
cout << endl;
// General Cuts to be applied for various plots
// Fit_Status == 3
TString Fit_Cut = "(abs(" + Fit_Status + "-3.0)<"+Double_Tolerance+")";
// param1_gen == notgen && param1_err == 0
TString Param_1_Cut = "(abs(" + param1_gen + "-" + notgen +")<" + Double_Tolerance + ")&&(abs("+ param1_err + "-0.0)<"+ Double_Tolerance + ")";
// param2_gen == notgen && param2_err == 0
TString Param_2_Cut = "(abs(" + param2_gen + "-" + notgen + ")<"+ Double_Tolerance + ")&&(abs(" +param2_err +"-0.0)<" + Double_Tolerance + ")";
// Combine the individual Cuts
TString Fit_Cut_String = Param_1_Cut + "&&" + Param_2_Cut + "&&" + Fit_Cut;
// Toys have a defined generation Value, Fits to Data DO NOT
// param1_gen != notgen
TString p1isatoy = "(abs(" + param1_gen + "-" + notgen + ")>" + Double_Tolerance + ")";
// param2_gen != notgen
TString p2isatoy = "(abs(" + param2_gen + "-" + notgen + ")>" + Double_Tolerance + ")";
// Combine the individual Cuts
TString Toy_Cut_String = p1isatoy + "&&" + p2isatoy;
// Check for Toys in the file and wether I should run the FC code
allresults->Draw( NLL, Toy_Cut_String, "goff" );
bool Has_Toys = allresults->GetSelectedRows() > 0;
cout << endl << "NUMBER OF TOYS IN FILE:\t" << allresults->GetSelectedRows() << endl;
if( int(allresults->GetEntries() - allresults->GetSelectedRows()) == 0 )
{
cerr << "SERIOUS ERROR:\tSOMETHING HAS REALLY GOTTEN SCREWED UP!" << endl;
exit(-3498);
}
// Fit values for the global fit are now stored in:
//
// Global_Best_NLL, best_fit_values, x_point, y_point
// Tell the user
cout << "GLOBAL DATA BEST FIT NLL:\t" << setprecision(10) << Global_Best_NLL << "\tAT:\tX:" << setprecision(10) << x_point << "\tY:\t" <<setprecision(10)<< y_point << endl;
// Check wether the minima as defined from the Global fit is the true minima within the phase-space
//Check_Minima( allresults, Fit_Cut_String, &Global_Best_NLL, NLL, Double_Tolerance, param1_val, param2_val );
TString NLL_Min; // Of course ROOT doesn't have USEFUL constructors!
NLL_Min+=Global_Best_NLL;
// Plot the distribution of successfully fitted grid points for the PLL scan
// NB: For FC this will likely saturate due to multiple layers of fits
cout << endl << "FOUND UNIQUE GRID POINTS, PLOTTING" << endl;
LL2D_Grid( allresults, Fit_Cut_String, param1string, param2string, rand_gen, "LL", outputdir );
// Construct a plot string for the NLL plot and plot it
// PLL part of Draw_String
TString PLL = "(" + NLL + "-" + NLL_Min + ")";
// Gridding part of Draw_String
TString Param1_Param2 = ":" + param1_val + ":" + param2_val;
// Draw String for PLL
TString NLL_DrawString = PLL + Param1_Param2;
cout << endl << "PLOTTING NLL VARIATION" << endl;
// PLL plot
TH2* pllhist=NULL;
TGraph2D* pllgraph=NULL;
if( high_res ) {
pllhist = Plot_From_Cut( allresults, NLL_DrawString, Fit_Cut_String, rand_gen, param1string, param2string );
} else {
pllgraph = Plot_From_Cut_lo( allresults, NLL_DrawString, Fit_Cut_String, rand_gen, param1string, param2string );
}
// Array storing the addresses of all of the Physics Plots still in memory
TH2** All_Physics_Plots = new TH2*[all_parameter_values.size()];
TH2** All_Physics_Errors = new TH2*[all_parameter_values.size()];
TH2** All_Physics_Pulls = new TH2*[all_parameter_values.size()];
// Making use of switch
if( Want_Physics_Params )
{
// Physics Plots
Physics_Plots( all_parameter_values, best_fit_values, allresults, rand_gen, Param1_Param2, CV_Drift, All_Physics_Plots, Fit_Cut_String);
cout << endl << "Finalising CV Plots" << endl << endl;
Finalize_Physics_Plots( All_Physics_Plots, all_parameter_values, param1string, param2string, outputdir, CV_Drift );
if( Want_Extra_Physics_Param_Info )
{
// Physics Plots
//
// Passing false as Highly unlikely that I will 'ever' want to watch error/pull drift rather than the absolute value
Float_t* null_pointer=NULL;
Physics_Plots( all_parameter_errors, null_pointer, allresults, rand_gen, Param1_Param2, false, All_Physics_Errors, Fit_Cut_String);
Physics_Plots( all_parameter_pulls, null_pointer, allresults, rand_gen, Param1_Param2, false, All_Physics_Pulls, Fit_Cut_String);
cout << endl << "Finalising Extra CV Plots" << endl << endl;
Finalize_Physics_Plots( All_Physics_Errors, all_parameter_errors, param1string, param2string, outputdir, false );
Finalize_Physics_Plots( All_Physics_Pulls, all_parameter_pulls, param1string, param2string, outputdir, false );
}
}
TFile* new_FC_Output = NULL;
TTree* FC_Output = new TTree( "FC_Output", "FC_Output" );;
TH2* FC_Plot = NULL;
// Making use of switch
if( Has_Toys && want_FC )
{
// FC Plot
cout << "FOUND TOYS IN FILE, PLOTTING FC" <<endl;
TString FC_Tuple_File( outputdir+ "/FC_Tuple.root" );
new_FC_Output = new TFile( FC_Tuple_File, "RECREATE" );
FC_Plot = FC_TOYS( allresults, Fit_Cut_String, param1string, param2string, NLL, Fit_Cut, Global_Best_NLL, FC_Output, Double_Tolerance, rand_gen );
FC_Output->Write();
//LL2D_Grid( FC_Output, Fit_Cut, param1_val, param2_val, rand_gen, "FC" );
}
// Contours to be used in plotting
int cont_num = 3;
double* pllconts = new double[unsigned(cont_num)];
pllconts[0] = 1.15; pllconts[1] = 2.36;
pllconts[2] = 3.0;// pllconts[3] = 4.61;
double* fcconts = new double[unsigned(cont_num)];
fcconts[0] = 0.68; fcconts[1] = 0.9;
fcconts[2] = 0.95;// fcconts[3] = 0.99;
double* confs = new double[unsigned(cont_num)];
confs[0] = 68.0; confs[1] = 90.0;
confs[2] = 95.0;// confs[3] = 99.0;
cout <<endl<< "SAVING GRAPHS" << endl;
if( high_res )
{
Plot_Styled_Contour( pllhist, cont_num, pllconts, confs, outputdir, "Likelihood Profile" );
} else {
//Plot_Styled_Contour2( pllgraph, cont_num, pllconts, confs, outputdir, "Likelihood Profile" );
}
if( Has_Toys && want_FC ) // If FC was generated
{
// Thinking of implementing a PLL Plot for the FC as it's easier to plot that (and looks more impressive)
//TH2* FC_PLL = Invert_PLL( FC_Plot );
Plot_Styled_Contour( FC_Plot, cont_num, fcconts, confs, outputdir, "FeldmanCousins Profile" );
//Plot_Styled_Contour( FC_PLL, cont_num, fcconts, confs, outputdir, "FeldmanCousins Profile PLL" );
//TString FC_Tuple_File( outputdir+ "FC_Tuple.root" );
//TFile* new_FC_Output = new TFile( FC_Tuple_File, "RECREATE" );
//FC_Output->Write();
Plot_Both( pllhist, FC_Plot, cont_num, fcconts, pllconts, confs, outputdir );
FC_Stats( FC_Output, param1string, param2string, rand_gen, outputdir );
if( new_FC_Output != NULL ) new_FC_Output->Close();
}
return 0;
}
//iteration code
void iterate(TrkSettings s,int iter, int stepType, bool doCondor, bool testErrors = false)
{
float pt, eta, phi, weight, centPU, rmin, maxJetPt,trkStatus,pNRec,mpt,mtrkQual,nEv;
TFile * histFile;
std::string ifPP = "";
if(s.nPb==0) ifPP = "pp_";
if(iter==0) histFile = TFile::Open(Form("%scorrHists_job%d.root",ifPP.c_str(),s.job),"recreate");
else histFile = TFile::Open(Form("%scorrHists_job%d.root",ifPP.c_str(),s.job),"update");
//make needed gen skim if it hasn't been done yet
if(iter==0)
{
TH1D *genPre[20], *mrecoPre[20];
TH2D *genPre2[20], *mrecoPre2[20];
std::cout << "Denominator info not yet calculated; calculating and saving it..." << std::endl;
for(int i = 0; i<8; i++)
{
if(i==6) continue;
if(i != 1 && i!=7)
{
genPre[i] = makeTH1(s,i,"gen");
mrecoPre[i] = makeTH1(s,i,"mreco");
}
else
{
genPre2[i] = makeTH2(s,i,"gen");
mrecoPre2[i]= makeTH2(s,i,"mreco");
}
}
TFile * skim;
if(doCondor)
{
if(s.reuseSkim) skim = TFile::Open(Form("/mnt/hadoop/cms/store/user/abaty/tracking_Efficiencies/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
else skim = TFile::Open(Form("%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
}
else skim = TFile::Open(Form("/export/d00/scratch/abaty/trackingEff/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
//for efficiency
std::cout << "Doing Efficiency denominator" << std::endl;
TNtuple * gen = (TNtuple*) skim->Get("Gen");
gen->SetBranchAddress("genPt",&pt);
gen->SetBranchAddress("genEta",&eta);
gen->SetBranchAddress("genPhi",&phi);
gen->SetBranchAddress("weight",&weight);
gen->SetBranchAddress("centPU",¢PU);
gen->SetBranchAddress("rmin",&rmin);
gen->SetBranchAddress("jtpt",&maxJetPt);
gen->SetBranchAddress("pNRec",&pNRec);
gen->SetBranchAddress("nEv",&nEv);
for(int i = 0; i<gen->GetEntries(); i++)
{
gen->GetEntry(i);
if(s.doSplit && nEv==1) continue;
genPre[0]->Fill(pt,weight);
genPre2[1]->Fill(eta,phi,weight);
genPre[2]->Fill(centPU,weight);
genPre[3]->Fill(maxJetPt,weight);
genPre[4]->Fill(eta,weight);
genPre[5]->Fill(rmin,weight);
genPre2[7]->Fill(eta,pt,weight);
}
//for fake
std::cout << "Doing Fake denominator" << std::endl;
TNtuple * reco = (TNtuple*) skim->Get("Reco");
reco->SetBranchAddress("trkPt",&pt);
reco->SetBranchAddress("trkEta",&eta);
reco->SetBranchAddress("trkPhi",&phi);
reco->SetBranchAddress("weight",&weight);
reco->SetBranchAddress("centPU",¢PU);
reco->SetBranchAddress("rmin",&rmin);
reco->SetBranchAddress("jtpt",&maxJetPt);
reco->SetBranchAddress("trkStatus",&trkStatus);
reco->SetBranchAddress("nEv",&nEv);
for(int i = 0; i<reco->GetEntries(); i++)
{
reco->GetEntry(i);
if(s.doSplit && nEv==1) continue;
if(trkStatus<-100) continue;
mrecoPre[0]->Fill(pt,weight);
mrecoPre2[1]->Fill(eta,phi,weight);
mrecoPre[2]->Fill(centPU,weight);
mrecoPre[3]->Fill(maxJetPt,weight);
mrecoPre[4]->Fill(eta,weight);
mrecoPre[5]->Fill(rmin,weight);
mrecoPre2[7]->Fill(eta,pt,weight);
}
//Secondary calculation (no iterations)
std::cout << "Quickly calculating the Secondary Rate from the reco tree (No further iterations needed)" << std::endl;
TH2D * Secondary_Matched = new TH2D("Secondary_Matched",";pt;",s.multiRecoBins.at(s.job/s.nPtBinCoarse),s.ptMin,s.ptMax,24,-2.4,2.4);
TH2D * Secondary_Secondaries = new TH2D("Secondary_Secondaries",";pt;",s.multiRecoBins.at(s.job/s.nPtBinCoarse),s.ptMin,s.ptMax,24,-2.4,2.4);
for(int i = 0; i<reco->GetEntries(); i++)
{
reco->GetEntry(i);
if(s.doSplit && nEv==1) continue;
if(trkStatus>-100)
{
Secondary_Matched->Fill(pt,eta,weight);
if(trkStatus==-99) Secondary_Secondaries->Fill(pt,eta,weight);
}
}
TH2D * Secondary = (TH2D*)Secondary_Secondaries->Clone("SecondaryRate");
Secondary->Divide(Secondary_Matched);
Secondary->SetDirectory(histFile);
Secondary_Matched->SetDirectory(histFile);
Secondary_Secondaries->SetDirectory(histFile);
//end Secondary Reco calculation
//multiReco calculation (no iterations)
std::cout << "Quickly calculating the Multiple Reconstruction Rate from the gen tree (No further iterations needed)" << std::endl;
TH1D * MultiGen = new TH1D("MultiGen",";pt;",s.multiRecoBins.at(s.job/s.nPtBinCoarse),s.ptMin,s.ptMax);
TH1D * MultiReco = new TH1D("MultiMatchedReco",";pt;",s.multiRecoBins.at(s.job/s.nPtBinCoarse),s.ptMin,s.ptMax);
for(int i = 0; i<gen->GetEntries(); i++)
{
gen->GetEntry(i);
if(s.doSplit && nEv==1) continue;
if(pNRec>-1)
{
MultiGen->Fill(pt,weight);
if(pNRec>1) MultiReco->Fill(pt,(pNRec-1)*weight);
}
}
TH1D * Multi = (TH1D*)MultiReco->Clone("MultipleRecoRate");
Multi->Divide(MultiGen);
Multi->SetDirectory(histFile);
MultiReco->SetDirectory(histFile);
MultiGen->SetDirectory(histFile);
//end Multiple Reco calculation
skim->Close();
histFile->Write();
}
//redundant for first step, but needed if the gen file was made and saved previously
std::cout << "Loading appropriate information for denominator (gen for efficiency, reco for fake)..." << std::endl;
TH1D * genHist[20], *recoHist[20];
TH2D * genHist2[20], *recoHist2[20];
genHist[0] = (TH1D*)histFile->Get("gen_pt");
genHist2[1] = (TH2D*)histFile->Get("gen_accept");
genHist[2] = (TH1D*)histFile->Get("gen_centPU");
genHist[3] = (TH1D*)histFile->Get("gen_maxJetPt");
genHist[4] = (TH1D*)histFile->Get("gen_eta");
genHist[5] = (TH1D*)histFile->Get("gen_rmin");
genHist2[7] = (TH2D*)histFile->Get("gen_etaPt");
std::cout << "Efficiency denominator histogram available now." << std::endl;
recoHist[0] = (TH1D*)histFile->Get("mreco_pt");
recoHist2[1] = (TH2D*)histFile->Get("mreco_accept");
recoHist[2] = (TH1D*)histFile->Get("mreco_centPU");
recoHist[3] = (TH1D*)histFile->Get("mreco_maxJetPt");
recoHist[4] = (TH1D*)histFile->Get("mreco_eta");
recoHist[5] = (TH1D*)histFile->Get("mreco_rmin");
recoHist2[7] = (TH2D*)histFile->Get("mreco_etaPt");
std::cout << "Fake denominator histogram available now." << std::endl;
//************************************************************************************************************
std::cout << "Calculating numerator for efficiency/fake calculation..." << std::endl;
TH1D *mrecoHist, *divHist, *frecoHist, *fdivHist;
TH2D *mrecoHist2,*divHist2, *frecoHist2, *fdivHist2;
//getting old eff histograms to calculate the updated efficiency (the number 30 is arbitrary, increase if more are needed)
TH1D * previousEff[30], *previousFake[30];
TH2D * previousEff2[30], *previousFake2[30];
for(int i=0; i<iter; i++)
{
int type = s.stepOrder.at(i%s.nStep);
if(type==0 || type==2 || type==3 || type==4 || type==5)
{
previousEff[i] = (TH1D*)histFile->Get(Form("eff_step%d",i));
previousFake[i] = (TH1D*)histFile->Get(Form("fake_step%d",i));
}
if(type==1 || type==7)
{
previousEff2[i] = (TH2D*)histFile->Get(Form("eff_step%d",i));
previousFake2[i] = (TH2D*)histFile->Get(Form("fake_step%d",i));
}
}
//setting up stuff for reading out of skim
TH1D * mrecoErr;
TH1D * mrecoW;
TH2D * mrecoErr2;
TH2D * mrecoW2;
if(stepType == 0 || stepType==2 || stepType==3 || stepType==4 || stepType==5)
{
mrecoHist = makeTH1(s,stepType,Form("mreco_eff_step%d",iter));
frecoHist = makeTH1(s,stepType,Form("reco_fake_step%d",iter));
if(testErrors)
{
mrecoErr = (TH1D*)mrecoHist->Clone("mrecoErr");
mrecoW = (TH1D*)mrecoHist->Clone("mrecoW");
}
}
if(stepType == 1 || stepType == 7)
{
mrecoHist2 = makeTH2(s,stepType,Form("mreco_eff_step%d",iter));
frecoHist2 = makeTH2(s,stepType,Form("reco_fake_step%d",iter));
if(testErrors)
{
mrecoErr2 = (TH2D*)mrecoHist2->Clone("mrecoErr");
mrecoW2 = (TH2D*)mrecoHist2->Clone("mrecoW");
}
}
TFile * skim;
if(doCondor)
{
if(s.reuseSkim) skim = TFile::Open(Form("/mnt/hadoop/cms/store/user/abaty/tracking_Efficiencies/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
else skim = TFile::Open(Form("%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
}
else skim = TFile::Open(Form("/export/d00/scratch/abaty/trackingEff/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
TNtuple * reco = (TNtuple*) skim->Get("Reco");
reco->SetBranchAddress("trkPt",&pt);
reco->SetBranchAddress("trkEta",&eta);
reco->SetBranchAddress("trkPhi",&phi);
reco->SetBranchAddress("weight",&weight);
reco->SetBranchAddress("centPU",¢PU);
reco->SetBranchAddress("rmin",&rmin);
reco->SetBranchAddress("jtpt",&maxJetPt);
reco->SetBranchAddress("trkStatus",&trkStatus);
reco->SetBranchAddress("nEv",&nEv);
//reading out of skim
for(int i = 0; i<reco->GetEntries(); i++)
{
//applying efficiencies from all previous steps
float previousFakeCorr = 1;
reco->GetEntry(i);
if(s.doSplit && nEv==1) continue;
//fake part
if(iter!=0)
{
for(int n = 0; n<iter; n++)
{
int type = s.stepOrder.at(n%s.nStep);
if(type==0) previousFakeCorr *= previousFake[n]->GetBinContent(previousFake[n]->FindBin(pt));
if(type==1) previousFakeCorr *= previousFake2[n]->GetBinContent(previousFake2[n]->GetXaxis()->FindBin(eta),previousFake2[n]->GetYaxis()->FindBin(phi));
if(type==2) previousFakeCorr *= previousFake[n]->GetBinContent(previousFake[n]->FindBin(centPU));
if(type==3) previousFakeCorr *= previousFake[n]->GetBinContent(previousFake[n]->FindBin(maxJetPt));
if(type==4) previousFakeCorr *= previousFake[n]->GetBinContent(previousFake[n]->FindBin(eta));
if(type==5) previousFakeCorr *= previousFake[n]->GetBinContent(previousFake[n]->FindBin(rmin));
if(type==7) previousFakeCorr *= previousFake2[n]->GetBinContent(previousFake2[n]->GetXaxis()->FindBin(eta),previousFake2[n]->GetYaxis()->FindBin(pt));
}
}
if(previousFakeCorr==0) std::cout << "\nWarning!!! A correction is going to infinity. This usually indicates an empty bin somewhere, try using a coarser binning or more events! \n" << std::endl;
//filling histograms
if(stepType==0) frecoHist->Fill(pt,weight/previousFakeCorr);
if(stepType==1) frecoHist2->Fill(eta,phi,weight/previousFakeCorr);
if(stepType==2) frecoHist->Fill(centPU,weight/previousFakeCorr);
if(stepType==3) frecoHist->Fill(maxJetPt,weight/previousFakeCorr);
if(stepType==4) frecoHist->Fill(eta,weight/previousFakeCorr);
if(stepType==5) frecoHist->Fill(rmin,weight/previousFakeCorr);
if(stepType==7) frecoHist2->Fill(eta,pt,weight/previousFakeCorr);
}
//eff part
TNtuple * gen = (TNtuple*) skim->Get("Gen");
gen->SetBranchAddress("genPt",&pt);
gen->SetBranchAddress("genEta",&eta);
gen->SetBranchAddress("genPhi",&phi);
gen->SetBranchAddress("weight",&weight);
gen->SetBranchAddress("centPU",¢PU);
gen->SetBranchAddress("rmin",&rmin);
gen->SetBranchAddress("jtpt",&maxJetPt);
gen->SetBranchAddress("pNRec",&pNRec);
gen->SetBranchAddress("mtrkPt",&mpt);
gen->SetBranchAddress("mtrkQual",&mtrkQual);
gen->SetBranchAddress("nEv",&nEv);
//reading out of skim
for(int i = 0; i<gen->GetEntries(); i++)
{
//applying efficiencies from all previous steps
float previousEffCorr = 1;
float previousEffCorrErr = 0;
gen->GetEntry(i);
if(s.doSplit && nEv==1) continue;
if(mtrkQual<1|| mpt<=0) continue;
if(iter!=0)
{
for(int n = 0; n<iter; n++)//calculating current efficiency correction
{
int type = s.stepOrder.at(n%s.nStep);
if(type==0){
previousEffCorr *= previousEff[n]->GetBinContent(previousEff[n]->FindBin(pt));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff[n]->GetBinError(previousEff[n]->FindBin(pt))/previousEff[n]->GetBinContent(previousEff[n]->FindBin(pt)),2);
}
if(type==1){
previousEffCorr *= previousEff2[n]->GetBinContent(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(phi));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff2[n]->GetBinError(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(phi))/previousEff2[n]->GetBinContent(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(phi)),2);
}
if(type==2){
previousEffCorr *= previousEff[n]->GetBinContent(previousEff[n]->FindBin(centPU));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff[n]->GetBinError(previousEff[n]->FindBin(centPU))/previousEff[n]->GetBinContent(previousEff[n]->FindBin(centPU)),2);
}
if(type==3){
previousEffCorr *= previousEff[n]->GetBinContent(previousEff[n]->FindBin(maxJetPt));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff[n]->GetBinError(previousEff[n]->FindBin(maxJetPt))/previousEff[n]->GetBinContent(previousEff[n]->FindBin(maxJetPt)),2);
}
if(type==4){
previousEffCorr *= previousEff[n]->GetBinContent(previousEff[n]->FindBin(eta));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff[n]->GetBinError(previousEff[n]->FindBin(eta))/previousEff[n]->GetBinContent(previousEff[n]->FindBin(eta)),2);
}
if(type==5){
previousEffCorr *= previousEff[n]->GetBinContent(previousEff[n]->FindBin(rmin));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff[n]->GetBinError(previousEff[n]->FindBin(rmin))/previousEff[n]->GetBinContent(previousEff[n]->FindBin(rmin)),2);
}
if(type==7){
previousEffCorr *= previousEff2[n]->GetBinContent(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(pt));
if(testErrors) previousEffCorrErr += TMath::Power(previousEff2[n]->GetBinError(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(pt))/previousEff2[n]->GetBinContent(previousEff2[n]->GetXaxis()->FindBin(eta),previousEff2[n]->GetYaxis()->FindBin(pt)),2);
}
}
if(testErrors) previousEffCorrErr = previousEffCorrErr/TMath::Power(previousEffCorr,2);
}
if(previousEffCorr==0) std::cout << "\nWarning!!! A correction is going to infinity. This usually indicates an empty bin somewhere, try using a coarser binning or more events! \n" << std::endl;
//filling histograms
float var1, var2;
if(stepType==0) var1 = pt;
if(stepType==1){var1 = eta; var2 = phi;}
if(stepType==2) var1 = centPU;
if(stepType==3) var1 = maxJetPt;
if(stepType==4) var1 = eta;
if(stepType==5) var1 = rmin;
if(stepType==7){var1 = eta; var2 = pt;}
if(stepType!=1 && stepType!=7){
mrecoHist->Fill(var1,weight/previousEffCorr);
if(testErrors){
mrecoErr->Fill(var1,weight*weight*previousEffCorrErr);
mrecoW->Fill(var1,weight);
}
}
else{
mrecoHist2->Fill(var1,var2,weight/previousEffCorr);
if(testErrors){
mrecoErr2->Fill(var1,var2,weight*weight*previousEffCorrErr);
mrecoW2->Fill(var1,var2,weight);
}
}
}
skim->Close();
//saving reco and efficiencies/fake rates
std::cout << "Calculating updated Efficiency/Fake Rate and saving histograms" << std::endl;
histFile->cd();
if(stepType==0 || stepType == 2 || stepType == 3 || stepType==4 || stepType==5)
{
if(testErrors)
{
mrecoErr->Divide(mrecoW);
mrecoErr->Divide(mrecoW);
for(int i = 1 ; i<mrecoErr->GetSize()-1; i++) mrecoHist->SetBinError(i,TMath::Power(mrecoErr->GetBinContent(i),0.5));
}
divHist = (TH1D*)mrecoHist->Clone(Form("eff_step%d",iter));
divHist->Divide(genHist[stepType]);
mrecoHist->Write();
divHist->Write();
fdivHist = (TH1D*)frecoHist->Clone(Form("fake_step%d",iter));
fdivHist->Divide(recoHist[stepType]);
frecoHist->Write();
fdivHist->Write();
}
if(stepType==1 || stepType==7)
{
if(testErrors)
{
mrecoErr2->Divide(mrecoW2);
mrecoErr2->Divide(mrecoW2);
for(int i = 1 ; i<mrecoErr2->GetNbinsX()+1; i++)
for(int j = 1; j<mrecoErr2->GetNbinsY()+1; j++) mrecoHist2->SetBinError(i,j,TMath::Power(mrecoErr2->GetBinContent(i,j),0.5));
}
divHist2 = (TH2D*)mrecoHist2->Clone(Form("eff_step%d",iter));
divHist2->Divide(genHist2[stepType]);
mrecoHist2->Write();
divHist2->Write();
fdivHist2 = (TH2D*)frecoHist2->Clone(Form("fake_step%d",iter));
fdivHist2->Divide(recoHist2[stepType]);
frecoHist2->Write();
fdivHist2->Write();
}
//*********************************************************************************************
//*********************************************************************************************
//Only executed on last step before exiting program
if(iter>=s.nStep*s.fullIterations && s.terminateStep==stepType)
{
//writing final correction tables (consolidating multiple steps of the same variable)
//once again 10 is an arbitrary number, increase if needed...
std::cout << "Consolidating into final histograms by multiplying out efficiencies/fake rates per variable" << std::endl;
TH1D * finalEff[10], *finalFake[10];
TH2D * finalEff2[10], *finalFake2[10];
for(int i=0; i<s.nStep; i++)
{
int type = s.stepOrder.at(i%s.nStep);
if(type == 0 || type==2 || type==3 || type==4 || type==5)
{
finalEff[i] = (TH1D*)previousEff[i]->Clone(Form("finalEff_type%d",i));
finalFake[i] = (TH1D*)previousFake[i]->Clone(Form("finalFake_type%d",i));
}
if(type == 1 || type==7)
{
finalEff2[i] = (TH2D*)previousEff2[i]->Clone(Form("finalEff_type%d",i));
finalFake2[i] = (TH2D*)previousFake2[i]->Clone(Form("finalFake_type%d",i));
}
}
for(int n = s.nStep; n<iter; n++)
{
int type2 = s.stepOrder.at(n%s.nStep);
if(type2==0 || type2==2 || type2==3 || type2==4 || type2==5)
{
finalEff[n%s.nStep]->Multiply(previousEff[n]);
finalFake[n%s.nStep]->Multiply(previousFake[n]);
}
if(type2==1 || type2==7)
{
finalEff2[n%s.nStep]->Multiply(previousEff2[n]);
finalFake2[n%s.nStep]->Multiply(previousFake2[n]);
}
}
if(stepType == 0 || stepType==2 || stepType==3 || stepType==4 || stepType==5)
{
finalEff[iter%s.nStep]->Multiply((TH1D*)histFile->Get(Form("eff_step%d",iter)));
finalFake[iter%s.nStep]->Multiply((TH1D*)histFile->Get(Form("fake_step%d",iter)));
}
if(stepType == 1 || stepType == 7)
{
finalEff2[iter%s.nStep]->Multiply((TH2D*)histFile->Get(Form("eff_step%d",iter)));
finalFake2[iter%s.nStep]->Multiply((TH2D*)histFile->Get(Form("fake_step%d",iter)));
}
for(int i=0; i<s.nStep; i++)
{
int type = s.stepOrder.at(i%s.nStep);
if(type == 0 || type==2 || type==3 || type==4 || type == 5){ finalEff[i]->Write(); finalFake[i]->Write();}
if(type == 1 || type == 7){ finalEff2[i]->Write(); finalFake2[i]->Write();}
}
//*******************************************************************************************************
//writing calculating final closures in each variable checked after applying corrections
std::cout << "Calculating Final Closures..." << std::endl;
TH1D * finalEffClosure[10], *finalFakeClosure[10];
TH2D * finalEffClosure2[10], *finalFakeClosure2[10];
for(int i=0; i<8; i++)
{
int type = i;
if(type==0 || type==2 || type==3 || type==4 || type==5)
{
finalEffClosure[i] = (TH1D*)genHist[i]->Clone(Form("finalEffClosure_var%d",i));
finalFakeClosure[i] = (TH1D*)recoHist[i]->Clone(Form("finalFakeClosure_var%d",i));
finalEffClosure[i]->Reset();
finalFakeClosure[i]->Reset();
}
if(type==1 || type==7)
{
finalEffClosure2[i] = (TH2D*)genHist2[i]->Clone(Form("finalEffClosure_var%d",i));
finalFakeClosure2[i] = (TH2D*)recoHist2[i]->Clone(Form("finalFakeClosure_var%d",i));
finalEffClosure2[i]->Reset();
finalFakeClosure2[i]->Reset();
}
}
if(doCondor)
{
if(s.reuseSkim) skim = TFile::Open(Form("/mnt/hadoop/cms/store/user/abaty/tracking_Efficiencies/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
else skim = TFile::Open(Form("%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
}
else skim = TFile::Open(Form("/export/d00/scratch/abaty/trackingEff/ntuples/%strackSkim_job%d.root",ifPP.c_str(),s.job),"read");
reco = (TNtuple*) skim->Get("Reco");
reco->SetBranchAddress("trkPt",&pt);
reco->SetBranchAddress("trkEta",&eta);
reco->SetBranchAddress("trkPhi",&phi);
reco->SetBranchAddress("weight",&weight);
reco->SetBranchAddress("centPU",¢PU);
reco->SetBranchAddress("rmin",&rmin);
reco->SetBranchAddress("jtpt",&maxJetPt);
reco->SetBranchAddress("trkStatus",&trkStatus);
reco->SetBranchAddress("nEv",&nEv);
//reading out of skim
for(int i = 0; i<reco->GetEntries(); i++)
{
//applying efficiencies from all previous steps
float previousFakeCorr = 1;
reco->GetEntry(i);
if(s.doSplit && nEv==1) continue;
for(int n=0; n<s.nStep; n++)//getting correction
{
int type = s.stepOrder.at(n%s.nStep);
if(type==0) previousFakeCorr *= finalFake[n]->GetBinContent(finalFake[n]->FindBin(pt));
if(type==1) previousFakeCorr *= finalFake2[n]->GetBinContent(finalFake2[n]->GetXaxis()->FindBin(eta),finalFake2[n]->GetYaxis()->FindBin(phi));
if(type==2) previousFakeCorr *= finalFake[n]->GetBinContent(finalFake[n]->FindBin(centPU));
if(type==3) previousFakeCorr *= finalFake[n]->GetBinContent(finalFake[n]->FindBin(maxJetPt));
if(type==4) previousFakeCorr *= finalFake[n]->GetBinContent(finalFake[n]->FindBin(eta));
if(type==5) previousFakeCorr *= finalFake[n]->GetBinContent(finalFake[n]->FindBin(rmin));
if(type==7) previousFakeCorr *= finalFake2[n]->GetBinContent(finalFake2[n]->GetXaxis()->FindBin(eta),finalFake2[n]->GetYaxis()->FindBin(pt));
}
if(previousFakeCorr<1) previousFakeCorr==1;
finalFakeClosure[0]->Fill(pt,weight/previousFakeCorr);
finalFakeClosure2[1]->Fill(eta,phi,weight/previousFakeCorr);
finalFakeClosure[2]->Fill(centPU,weight/previousFakeCorr);
finalFakeClosure[3]->Fill(maxJetPt,weight/previousFakeCorr);
finalFakeClosure[4]->Fill(eta,weight/previousFakeCorr);
finalFakeClosure[5]->Fill(rmin,weight/previousFakeCorr);
finalFakeClosure2[7]->Fill(eta,pt,weight/previousFakeCorr);
}
gen = (TNtuple*) skim->Get("Gen");
gen->SetBranchAddress("genPt",&pt);
gen->SetBranchAddress("genEta",&eta);
gen->SetBranchAddress("genPhi",&phi);
gen->SetBranchAddress("weight",&weight);
gen->SetBranchAddress("centPU",¢PU);
gen->SetBranchAddress("rmin",&rmin);
gen->SetBranchAddress("jtpt",&maxJetPt);
gen->SetBranchAddress("pNRec",&pNRec);
gen->SetBranchAddress("mtrkPt",&mpt);
gen->SetBranchAddress("mtrkQual",&mtrkQual);
gen->SetBranchAddress("nEv",&nEv);
//reading out of skim
for(int i = 0; i<gen->GetEntries(); i++)
{
//applying efficiencies from all previous steps
float previousEffCorr = 1;
gen->GetEntry(i);
if(s.doSplit && nEv==1) continue;
if(mtrkQual<1 || mpt<=0) continue;
for(int n=0; n<s.nStep; n++)//getting correction
{
int type = s.stepOrder.at(n%s.nStep);
if(type==0) previousEffCorr *= finalEff[n]->GetBinContent(finalEff[n]->FindBin(pt));
if(type==1) previousEffCorr *= finalEff2[n]->GetBinContent(finalEff2[n]->GetXaxis()->FindBin(eta),finalEff2[n]->GetYaxis()->FindBin(phi));
if(type==2) previousEffCorr *= finalEff[n]->GetBinContent(finalEff[n]->FindBin(centPU));
if(type==3) previousEffCorr *= finalEff[n]->GetBinContent(finalEff[n]->FindBin(maxJetPt));
if(type==4) previousEffCorr *= finalEff[n]->GetBinContent(finalEff[n]->FindBin(eta));
if(type==5) previousEffCorr *= finalEff[n]->GetBinContent(finalEff[n]->FindBin(rmin));
if(type==7) previousEffCorr *= finalEff2[n]->GetBinContent(finalEff2[n]->GetXaxis()->FindBin(eta), finalEff2[n]->GetYaxis()->FindBin(pt));
}
if(previousEffCorr>1) previousEffCorr==1;
finalEffClosure[0]->Fill(pt,weight/previousEffCorr);
finalEffClosure2[1]->Fill(eta,phi,weight/previousEffCorr);
finalEffClosure[2]->Fill(centPU,weight/previousEffCorr);
finalEffClosure[3]->Fill(maxJetPt,weight/previousEffCorr);
finalEffClosure[4]->Fill(eta,weight/previousEffCorr);
finalEffClosure[5]->Fill(rmin,weight/previousEffCorr);
finalEffClosure2[7]->Fill(eta,pt,weight/previousEffCorr);
}
histFile->cd();
for(int i=0; i<8; i++)
{
if(i==6) continue;
if(i!=1 && i!=7)
{
finalFakeClosure[i]->Divide(recoHist[i]);
finalEffClosure[i]->Divide(genHist[i]);
finalFakeClosure[i]->Write();
finalEffClosure[i]->Write();
}
else
{
finalFakeClosure2[i]->Divide(recoHist2[i]);
finalEffClosure2[i]->Divide(genHist2[i]);
finalFakeClosure2[i]->Write();
finalEffClosure2[i]->Write();
}
}
std::cout << "Calculating Final Closures..." << std::endl;
}
histFile->Close();
std::cout << "Finished with iteration \n" << std::endl;
return;
}
float n3HeSim(int nev_=0, float rfsf_phi_=300, int sev=0)
{
// commandline option '-var=val' to change any global
for (int i=1; i<gApplication->Argc(); ++i)
{
TString opt(gApplication->Argv(i));
if ((opt[0]=='-' || opt[0]=='+') && opt.Contains("="))
gROOT->ProcessLine(opt=opt.Append(";").Strip(TString::kLeading,'-'));
}
if (nev_)
nev = nev_;
if (rfsf_phi_!=0)
rfsf_phi=rfsf_phi_/360.;
TFile* f = new TFile("/home/siplu/GIT/n3He_Soft/Github/Simulation/data/mcstas.root");
TNtuple* ntp = (TNtuple*)f->Get("mcstas");
// set up ntuple
BRANCH(l);
BRANCH(pol);
BRANCH(p5);
BRANCH(p6);
BRANCH(t6);
BRANCH(x6);
BRANCH(y6);
BRANCH(vx6);
BRANCH(vy6);
BRANCH(vz6);
float chop[4];
TBranch* b_chop[4];
for (int j=0;j<nch;++j)
{
b_chop[j]=ntp->GetBranch(vch[j]);
ntp->SetBranchAddress(vch[j],&chop[j]);
}
// precalculate track projections
float dwc = lwc/nz; // WC parameters
float da = 2./na, a0=-1+da/2; // polar [cos(theta)] integration
float db = 2*PI/nb, b0=-PI+db/2; // azimuthal [phi] integration
float cxyz[3], &cx=cxyz[0],&cy=cxyz[1],&cz=cxyz[2]; // projection cosines
float ccz[NDIV], ccx[NDIV][NDIV], ccy[NDIV][NDIV], as; // cached values
float aa=a0;
for (int ia=0; ia<na; ++ia, aa+=da)
{
ccz[ia]=aa;
if (ccz[ia]==0)
ccz[ia]=1e-11;
as=sqrt(1-aa*aa);
float bb=b0;
for (int ib=0; ib<nb; ++ib, bb+=db)
{
ccx[ia][ib]=as*cos(bb);
if (ccx[ia][ib]==0)
ccx[ia][ib]=1e-11;
ccy[ia][ib]=as*sin(bb);
if (ccy[ia][ib]==0)
ccy[ia][ib]=1e-11;
}
}
// cumulative beta distribution
for (int i=0;i<nbet;++i)
beta_int[i+1]=beta_int[i]+beta_pt[i];
// initialize WC response matrices
double n5 = 0; // pre-polarizer neutrons
double N[NTOF]={0}; // chopped neutrons at end of SMpol
double P2N[NTOF]={0}; // " * P^2
double WN[NTOF]={0}; // captures in 3He target
double BN[NTOF][nwc]={{0}}; // N_i = N beta_i = \sum{n_i} ions
double GBN[NTOF][nwc]={{0}}; // M_i = N_i gamma_i = \sum{n_i gamma}
double D2N[NTOF][nwc][nwc]={{{0}}}; // delta^2 N_ij = covariance
double GW[NTOF][nwc]={{0}}; // weight of each element
// loop over events, fill "n","p2n" histograms, wc response matrices
if (ntp->GetEntries()<nev)
nev=ntp->GetEntries();
for (int iev=sev;iev<sev+nev;++iev)
{
if (verbose && cnt)
{ // :........:....
if (iev%(50*cnt)==0)
cout<<endl<<" :"<<flush;
else if (iev%(10*cnt)==0)
cout<<":"<<flush;
else if (iev%cnt==0) cout<<"."<<flush;
}
// read variables
ntp->GetEntry(iev);
hl5.Fill(l,p5*1.4/2*60);
n5 += p5;
hl6.Fill(l,p6*1.4/2*60);
// polarizer
float tx = 1; //Transmission
b_pol->GetEntry(iev);
if (pol!=pol)
pol=0; //nan
// choppers
for (int j=0;j<nch;++j)
{
float chw = chop[j]-ch_phi[j];
if (chw<0)
chw += 360;
if (chw>ch_open[j])
{
tx=0;
break;
}
}
if (tx==0)
continue;
hlc.Fill(l,p6*1.4/2*60);
// RFSF, averaged over spin sequence
float t_sf = (t6+(zsf-z6)/vz6)*freq;
static const float spin_seq_ave[]={1, -0.5, 0, 0.5, -1, 0.5, 0, -0.5};
if (psf==1 && (t_sf<rfsf_phi || t_sf>rfsf_phi+dsf))
continue;
if (t_sf!=0)
{
if (dsf<0)
{
pol*= spin_seq_ave[int(fabs(t_sf-rfsf_phi-0.5)+0.5)%8];
pol*=(1-cos(k_rfsf/vz6/(fmod(t_sf-rfsf_phi+10,1)+rfsf_phi)))/2.;
}
else
{
if (t_sf<rfsf_phi || t_sf>rfsf_phi+dsf) pol*=psf;
}
}
hls.Fill(l,p6*1.4/2*60);
// fill histograms for N,P2N
float dtev=(zwc-z6)/vz6, tev=fmod((t6+dtev)*freq, 1);
int it((tev*ntof+1)-1);
assert(it>=0 && it<ntof);
float n = p6 * tx;
N [it] += n;
P2N[it] += n * pol*pol;
// collimator
float xx=x6+vx6*(zwc-z6)/vz6;
if (xcol>0 && fabs(xx)>xcol/2)
continue;
float yy=y6+vy6*(zwc-z6)/vz6;
if (ycol>0 && fabs(yy)>ycol/2)
continue;
hlx.Fill(l,p6*1.4/2*60);
// simulate ion chamber response
float zev=dwc/2.;
for (int iz=0; iz<nz; ++iz, zev+=dwc)
{
dtev=(zev+zwc-z6)/vz6;
tev=fmod((t6+dtev)*freq, 1);
it = int(tev*ntof+1)-1;
assert(it>=0 && it<ntof);
float xev=x6+vx6*dtev, yev=y6+vy6*dtev;
float rhosig=rho3*ksig*l; // attenuation const and weight
float wza = (exp(-rhosig*zev)-exp(-rhosig*(zev+dwc)))*da/2*db/2/PI;
for (int ia=0; ia<na; ++ia)
{
cz=ccz[ia];
for (int ib=0; ib<nb; ++ib)
{
cx=ccx[ia][ib];
cy=ccy[ia][ib];
float d1=d1r, d2=d2r, d1t,d2t; // ionization range
// restrict ionization to active volume of wc
if (cz>0)
d1t=(z1w-zev)/cz, d2t=(z2w-zev)/cz;
else
d1t=(z2w-zev)/cz, d2t=(z1w-zev)/cz;
if (d1<d1t)
d1=d1t;
if (d2>d2t)
d2=d2t;
if (cx>0)
d1t=(x1w-xev)/cx, d2t=(x2w-xev)/cx;
else
d1t=(x2w-xev)/cx, d2t=(x1w-xev)/cx;
if (d1<d1t)
d1=d1t;
if (d2>d2t) d2=d2t;
if (cy>0)
d1t=(y1w-yev)/cy, d2t=(y2w-yev)/cy;
else
d1t=(y2w-yev)/cy, d2t=(y1w-yev)/cy;
if (d1<d1t)
d1=d1t;
if (d2>d2t)
d2=d2t;
if (d2<d1)
continue; // not in active volume
// iterate over ionized cells
float beta1, beta0=beta_fn(d1);
float wz=(z0w-zev)/cz, dwz=dzw/cz;
int jz((d1-wz)/dwz), djz=1;
if (jz==nwz)
--jz;
wz+=dwz*jz;
if (cz>0)
wz+=dwz;
else
dwz*=-1, djz*=-1;
float wx=(x0w-xev)/cx, dwx=dxw/cx;
int jx((d1-wx)/dwx), djx=1;
if (jx==nwx)
--jx;
wx+=dwx*jx;
if (cx>0)
wx+=dwx;
else
dwx*=-1, djx*=-1;
int k=0;
assert(0<=jz&&jz<nwz && 0<=jx&&jx<nwx);
while (0<=jz&&jz<nwz && 0<=jx&&jx<nwx)
{
kzx[k] = jz*nwx + jx;
if (d2<=wz && d2<=wx)
{
kbeta[k++]=beta_fn(d2)-beta0;
break;
}
if (wz<wx)
{
beta1=beta_fn(wz);
jz+=djz;
d1=wz;
wz+=dwz;
}
else
{
beta1=beta_fn(wx);
jx+=djx;
d1=wx;
wx+=dwx;
}
kbeta[k++]=beta1-beta0;
beta0=beta1;
}
// tabulate ion response
WN[it] += n*wza;
for (int i=0;i<k;++i)
{
float ni= n*wza*kbeta[i];
BN [it][kzx[i]] += ni;
GBN[it][kzx[i]] += ni*cxyz[geom_xyz];
for (int j=0;j<k;++j)
{
D2N[it][kzx[i]][kzx[j]] += ni*kbeta[j];
}
}
}
}
}
}
f->Close();
// calculate FOM=p2n, projection; draw both
double N_tot=0, P2N_tot=0, WN_tot=0, xd2a_tot=0;
for (int it=0;it<ntof;++it)
{
TMatrixDSym mat(nwc, &**D2N[it]);
if (verbose>3 && ntof==1)
{
cout<<"\n delta^2 N_ij :\n";
mat.Print();
}
// for (int i=0;i<nwc;++i) for (int j=0;j<nwc;++j) if (i!=j) mat(i,j)=0;
mat.Invert();
if (verbose>3 && ntof==1)
{
cout<<"\n delta^-2 N_ij :\n";
mat.Print();
}
double xd2a=0;
for (int i=0;i<nwc;++i)
{
for (int j=0;j<nwc;++j)
{
GW[it][i] += GBN[it][i]*GBN[it][j]*mat(i,j);
}
xd2a += GW[it][i];
}
N_tot += N[it];
P2N_tot+= P2N[it];
WN_tot += WN[it];
xd2a_tot += xd2a;
if (verbose > 2)
{
printf("\n beta_i:");
for (int i=0;i<nwc;++i)
{
if (!(i%nwx))
printf("\n ");
printf(" %9.5g",BN[it][i]/N[it]);
}
printf("\n\n gamma_i:");
for (int i=0;i<nwc;++i)
{
if (!(i%nwx))
printf("\n ");
printf(" %9.5g",GBN[it][i]/BN[it][i]);
}
printf("\n\n weight_i:");
for (int i=0;i<nwc;++i)
{
if (!(i%nwx))
printf("\n ");
printf(" %9.5g",GW[it][i]);
}
printf("\n\n");
}
if (verbose > 4)
{
printf("\n\n beta_i:");
for (int j=0;j<nwc;++j)
{
printf(" %9.5g",BN[it][j]/N[it]);
}
printf("\n\n beta_ij:");
for (int i=0;i<nwc;++i)
{
printf("\n ");
for (int j=0;j<nwc;++j)
{
printf(" %9.5g", D2N[it][i][j]/N[it]);
}
}
}
if (verbose>1 && ntof>1)
{
cout<<" N = "<< N[it] <<endl;
cout<<" <P^2> = "<< P2N[it] / N[it] <<endl;
cout<<" <W> = "<< WN[it] / N[it] <<endl;
cout<<" delta^-2 A = "<< xd2a*P2N[it]/N[it] <<endl;
cout<<" sigma_d = "<< sqrt(N[it] / xd2a) <<endl<<endl<<endl;
}
}
cout<<"All TOF bins combined"<<endl;
cout<<" N0 = "<< n5 <<endl;
cout<<" N = "<< N_tot <<endl;
cout<<" <P^2> = "<< P2N_tot/N_tot <<endl;
cout<<" <W> = "<< WN_tot / N_tot <<endl;
cout<<" delta^-2 A = "<< xd2a_tot*P2N_tot/N_tot <<endl;
cout<<" sigma_d = "<< sqrt(N_tot / xd2a_tot) <<endl;
return sqrt(N_tot / xd2a_tot);
}
void fragmentEnergyDistributionDifferentAngles() {
gStyle->SetOptStat(0000000000); //remove the for this graphs totally redundant statbox
// gROOT->SetStyle("clearRetro");
TString pDepth, fragment, Znum, normToOneAtZeroAngle;
cout << "Enter phantom depth (eg. 27.9, see experimentalData directory for choices): ";
cin >> pDepth;
TString simulationDataPath = "IAEA_" + pDepth + ".root";
TString dir = gSystem->UnixPathName(gInterpreter->GetCurrentMacroName());
dir.ReplaceAll("basic.C","");
dir.ReplaceAll("/./","/");
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("fragmentEnergyWithAngularDistribution.root","RECREATE");
TNtuple *ntuple = new TNtuple("ntuple","Data from ascii file","Energy:He:B:H:Li:Be");
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++;
}
//Let's pull in the simulation-data
TFile *MCData = TFile::Open("IAEA_200000.root");
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.
//good to keep for ref. G4 might give weird units due to change.
metadata->Scan();
std::cout << "Recieved metadata-row: " << events << " " << detectorDistance << " " << waterThickness << " " << beamEnergy << " " << energyError << " " << phantomCenterDistance;
//A lot of hardcoded histograms, ugly
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 *hist1 = new TH1F("hist1", "", binAmount, 0.0, maxEnergy);
TH1F *hist2 = new TH1F("hist2", "", binAmount, 0.0, maxEnergy);
TH1F *hist3 = new TH1F("hist3", "", binAmount, 0.0, maxEnergy);
TH1F *hist4 = new TH1F("hist4", "", binAmount, 0.0, maxEnergy);
TH1F *hist5 = new TH1F("hist5", "", binAmount, 0.0, maxEnergy);
TH1F *hist6 = new TH1F("hist6", "", binAmount, 0.0, maxEnergy);
TH1F *hist7 = new TH1F("hist7", "", binAmount, 0.0, maxEnergy);
TH1F *hist8 = new TH1F("hist8", "", binAmount, 0.0, maxEnergy);
TH1F *hist9 = new TH1F("hist9", "", binAmount, 0.0, maxEnergy);
for(int k = 1; k <= 6; k++){
TString Znum = Form("%i", k);
hist1->SetTitle("Z=" + Znum);
//ALL UNITS ARE cm!
Double_t detectorSideLength = 4; //40mm, as e.haettner H1 detector
Double_t scatteringDistance = detectorDistance - phantomCenterDistance; //temporarily hard-coded, should be distance from target-center to detector
Double_t degrees; //< actually radians
Double_t r, rMin, rMax, deltaOmega, normFloat;
TString rMinString, rMaxString, normString;
TString same = "";
TString histName;
TCanvas *c3 = new TCanvas("histograms", "Distribution (at different angles)");
int i = 0; //so that the degree steps can be varied to unevenly spaced values separate counter is used
std::cout << "The following numbers also make it possible to make number of fragments comparison to the graph in A1 of E.Haettner\n";
for(Double_t j = 0.0; j <= 8.0; j=j+1.0){
i++;
degrees = j * TMath::DegToRad();
//std::cout << "plotting for Z = " << Znum << " at " << j << " degrees\n";
//Distance from straight beam at the requested angle
r = scatteringDistance * TMath::Tan(degrees);
//now the "detector is rotated around all possible perpendicularlynangle values to beamline".
//This forms an annulus with rMin and RMax as otuer and inner radiuses
//Notice this will give a bit of approximation at small angles where at 0 degrees this gives a round sensor.
Double_t deltaPhi = TMath::ATan((TMath::Cos(degrees)*detectorSideLength)/(2*scatteringDistance));
rMin = TMath::Max(0.0,r - (detectorSideLength/(2*TMath::Cos(degrees))));
rMax = rMin + ((detectorSideLength*TMath::Sin(degrees))/TMath::Tan((TMath::Pi()/2) - degrees - deltaPhi)) + (detectorSideLength*TMath::Cos(degrees));
rMinString = Form("%f", rMin);
rMaxString = Form("%f", rMax);
//normalization of the bins.
deltaPhi = degrees - TMath::ATan(TMath::Tan(degrees) - detectorSideLength/(2*scatteringDistance)); // this should be around arctan(detectorsidelength/sd)
if(j != 0.0){
deltaOmega = 2*TMath::Pi()*(TMath::Cos(TMath::Max(0.0,degrees-deltaPhi)) - TMath::Cos(degrees+deltaPhi));
}else{
deltaOmega = 4 * TMath::ASin(pow(detectorSideLength,2.0) / (4*pow(scatteringDistance,2) + pow(detectorSideLength,2)) );
}
normFloat = deltaOmega * events * binWidth;
normString = Form("/%f", normFloat);
// The following is veryvery ugly relies on a bunch of hardcoded histograms because other solutions did not work
histName = Form("hist%i", i);
if(j != 0.0){
fragments->Project(histName,"energy", "(Z == " + Znum + " && energy > 0 && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")" + normString);
}else{
fragments->Project(histName,"energy", "(Z == " + Znum + " && energy > 0 && posZ < " + rMaxString + "&& posY < " + rMaxString + " && posY > 0 && posZ > 0)" + normString);
}
int numEntries = fragments->GetEntries("(Z == " + Znum + " && energy > 0 && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")");
std::cout << "\nj: "<< numEntries/(deltaOmega * events) << " entries for " << j;
}
//the ugly hardcoded histograms being plotted
//0 degrees
hist1->SetLineColor(kBlue);
hist1->Draw();
//1 degree
hist2->SetLineColor(kGreen);
hist2->Draw("same"); //add "same" when also plotting 0 degrees
//2 degrees
hist3->SetLineColor(kRed);
hist3->Draw("same");
//3 degrees
//hist4->SetLineColor(kGreen + 5);
//hist4->Draw("same");
//4 degrees
hist5->SetLineColor(kGreen + 3); //gives a darker shade of green
hist5->Draw("same");
//5 degrees
//hist6->SetLineColor(kRed);
//hist6->Draw("same");
//6 degrees
hist7->SetLineColor(kRed);
hist7->Draw("same");
//7 degrees
//hist8->SetLineColor(kRed);
//hist8->Draw("same");
//8 degrees
//hist9->SetLineColor(kRed);
//hist9->Draw("same");
// Legends for the data
leg = new TLegend(0.9,0.7,1,1); //coordinates are fractions
leg->SetHeader("Angles");
leg->AddEntry(hist1,"0","l");
leg->AddEntry(hist2,"1","l");
leg->AddEntry(hist3,"2","l");
leg->AddEntry(hist5,"4","l");
leg->AddEntry(hist7,"6","l");
leg->Draw();
c3->SaveAs("AEDistrib" + Znum + ".png");
}
in.close();
f->Write();
}
void DoTDeriMax1090Correction(TString SpectrumFileInput = "/lustre/miifs05/scratch/him-specf/hyp/steinen/COSYBeamtestAna/COSYnewMogon/June2014/COSYJune2014Dataset11_200,100,0,5339_SR1.root", TString FitFileInput = "/lustre/miifs05/scratch/him-specf/hyp/steinen/COSYBeamtestAna/COSYnewMogon/Fit/FitCOSYJune2014Dataset11_200,100,0,5339_SR1.root")
{
TH2D *hSpectrumTDeriMax1090_EnergyChannel;
TH2D *hSpectrumTDeriMax1090Rel_EnergyChannel;
TH2D *hSpectrumT1090_EnergyChannelCorr1;
TNtuple *DataNTuple;
TFile *SpectrumInput = new TFile(SpectrumFileInput.Data());
hSpectrumTDeriMax1090_EnergyChannel = (TH2D*) SpectrumInput->Get("Histograms/Energy_DeriMaxT90/Energy_DeriMaxT90_01");
hSpectrumTDeriMax1090_EnergyChannel->SetDirectory(0);
hSpectrumTDeriMax1090Rel_EnergyChannel = (TH2D*) SpectrumInput->Get("Histograms/Energy_DeriMaxT90Rel/Energy_DeriMaxT90Rel_01");
hSpectrumTDeriMax1090Rel_EnergyChannel->SetDirectory(0);
hSpectrumT1090_EnergyChannelCorr1 = (TH2D*) SpectrumInput->Get("Histograms/EnergyRt1090/EnergyRt1090CorrectionRt_01");
hSpectrumT1090_EnergyChannelCorr1->SetDirectory(0);
SpectrumInput->Close();
//hSpectrumTDeriMax1090_EnergyChannel->Draw("colz");
TFile *FitInput = new TFile(FitFileInput.Data(),"Update");
DataNTuple = (TNtuple*)FitInput->Get("DataNTuple");
DataNTuple->Scan();
Int_t entries = (Int_t)DataNTuple->GetEntries();
cout<<"Number of Entries: "<<entries<< endl;
const int entriesArrayValue =entries;
TF1 *FitFunc[entriesArrayValue];
float Energy=0;
float ChannelPeakPos=0;
float ChannelRangeMin=0;
float ChannelRangeMax=0;
DataNTuple->SetBranchAddress("Energy",&Energy);
DataNTuple->SetBranchAddress("ChannelPeakPos",&ChannelPeakPos);
DataNTuple->SetBranchAddress("ChannelRangeMin",&ChannelRangeMin);
DataNTuple->SetBranchAddress("ChannelRangeMax",&ChannelRangeMax);
TCanvas* can=new TCanvas();
gPad->SetLogz();
hSpectrumTDeriMax1090_EnergyChannel->GetXaxis()->SetRangeUser(-30,250);
hSpectrumTDeriMax1090Rel_EnergyChannel->GetXaxis()->SetRangeUser(-0.1,0.95);
for (Int_t ki=0;ki<entries;ki++)
{
DataNTuple->GetEntry(ki);
//if (int(Energy) == 1332)
//if (int(Energy) == 510)
//if (ki == entries-1)
{
cout << ChannelRangeMin << " " << ChannelRangeMax << endl;
//first correction via TDeriMaxT90Rel
hSpectrumTDeriMax1090_EnergyChannel->GetYaxis()->SetRangeUser(ChannelRangeMin,ChannelRangeMax);
hSpectrumTDeriMax1090Rel_EnergyChannel->GetYaxis()->SetRangeUser(ChannelRangeMin,ChannelRangeMax);
//TF1* FitFuncSlices = new TF1("FitFuncSlices","gaus(0)+pol0(3)",ChannelRangeMin,ChannelRangeMax);
//FitFuncSlices->SetParameters(1000,ChannelPeakPos-10,4,10);
//FitFuncSlices->SetParLimits(1,ChannelRangeMin,ChannelRangeMax);
//FitFuncSlices->SetParLimits(2,0,10);
//FitFuncSlices->SetParLimits(3,0,100);
//gDirectory->ls();
TH1D *hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually=new TH1D("hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually","",hSpectrumTDeriMax1090Rel_EnergyChannel->GetNbinsX(),-0.3,1.3);
//cout <<hSpectrumTDeriMax1090_EnergyChannel_MaxPos->GetEntries()<< endl;
for(int binX = hSpectrumTDeriMax1090Rel_EnergyChannel->GetXaxis()->FindBin(-0.1);binX <= hSpectrumTDeriMax1090Rel_EnergyChannel->GetXaxis()->FindBin(0.90);binX++)
{
cout << "binx " << binX << endl;
TH1D *hProfileY =hSpectrumTDeriMax1090Rel_EnergyChannel->ProjectionY("_py",binX,binX);
double MaxValue=hProfileY->GetBinCenter(hProfileY->GetMaximumBin());
TF1* FitFuncSlices = new TF1("FitFuncSlices","gaus(0)+[3]",MaxValue-20,MaxValue+20);
TF1* FitFuncGausSlices = new TF1("FitFuncGausSlices","gaus(0)",MaxValue-20,MaxValue+20);
FitFuncGausSlices->SetParameters(hProfileY->GetBinContent(hProfileY->GetMaximumBin()),MaxValue,4);
hProfileY->Fit(FitFuncGausSlices,"RNQ");
FitFuncSlices->SetParameters(FitFuncGausSlices->GetParameter(0),FitFuncGausSlices->GetParameter(1),FitFuncGausSlices->GetParameter(2),10);
FitFuncSlices->SetParLimits(0,0,10000);
FitFuncSlices->SetParLimits(1,MaxValue-10,MaxValue+10);
FitFuncSlices->SetParLimits(2,0,10);
FitFuncSlices->SetParLimits(3,0,100);
hProfileY->Fit(FitFuncSlices,"RNQ");
cout <<MaxValue<<" " << FitFuncSlices->GetParameter(1) << " " << FitFuncSlices->GetParError(1) <<endl;
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->SetBinContent(binX, FitFuncSlices->GetParameter(1));
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->SetBinError(binX, FitFuncSlices->GetParError(1));
}
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->GetYaxis()->SetRangeUser(ChannelPeakPos-100,ChannelPeakPos+50);
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->GetXaxis()->SetRangeUser(-0.05,0.9);
TF1 * funcCorrConst = new TF1("funcCorrConst","pol0",-0.03,0.03);
funcCorrConst->SetLineColor(kRed);
TF1 * funcCorrPol = new TF1("funcCorrPol",PolPiecewise,-0.05,0.9,6);
funcCorrPol->SetLineColor(kBlue);
funcCorrPol->SetParameter(0,0.04);
funcCorrPol->SetParameter(1,ChannelPeakPos);
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->Fit(funcCorrConst,"R");
hSpectrumTDeriMax1090Rel_EnergyChannel_MaxPosManually->Fit(funcCorrPol,"R+");
TCanvas* can2=new TCanvas();
gPad->SetLogz();
hSpectrumTDeriMax1090Rel_EnergyChannel->Draw("colz");
funcCorrPol->DrawCopy("same");
for(int i = 7;i>0;i--)
funcCorrPol->SetParameter(i,funcCorrPol->GetParameter(i)/funcCorrPol->GetParameter(1));
funcCorrPol->Print();
char buf[20];
sprintf(buf, "funcCorrNorm_%d",ki);
funcCorrPol->Write(buf,TObject::kOverwrite);
funcCorrConst->Delete();
funcCorrPol->Delete();
//second correction via T1090 (after first)
hSpectrumT1090_EnergyChannelCorr1->GetYaxis()->SetRangeUser(ChannelRangeMin,ChannelRangeMax);
//TF1* FitFuncSlices = new TF1("FitFuncSlices","gaus(0)+pol0(3)",ChannelRangeMin,ChannelRangeMax);
//FitFuncSlices->SetParameters(1000,ChannelPeakPos-10,4,10);
//FitFuncSlices->SetParLimits(1,ChannelRangeMin,ChannelRangeMax);
//FitFuncSlices->SetParLimits(2,0,10);
//FitFuncSlices->SetParLimits(3,0,100);
//gDirectory->ls();
TH1D *hSpectrumT1090_EnergyChannelCorr1_MaxPosManually=new TH1D("hSpectrumT1090_EnergyChannelCorr1_MaxPosManually","",hSpectrumT1090_EnergyChannelCorr1->GetNbinsX(),0,1000);
//cout <<hSpectrumTDeriMax1090_EnergyChannel_MaxPos->GetEntries()<< endl;
for(int binX = hSpectrumT1090_EnergyChannelCorr1->GetXaxis()->FindBin(30);binX <= hSpectrumT1090_EnergyChannelCorr1->GetXaxis()->FindBin(500);binX++)
{
cout << "binx " << binX << endl;
TH1D *hProfileY =hSpectrumT1090_EnergyChannelCorr1->ProjectionY("_py",binX,binX);
//hProfileY->Draw();
//return 0;
double MaxValue=hProfileY->GetBinCenter(hProfileY->GetMaximumBin());
FitFuncSlices = new TF1("FitFuncSlices","gaus(0)+[3]",MaxValue-20,MaxValue+20);
FitFuncGausSlices = new TF1("FitFuncGausSlices","gaus(0)",MaxValue-20,MaxValue+20);
FitFuncGausSlices->SetParameters(hProfileY->GetBinContent(hProfileY->GetMaximumBin()),MaxValue,4);
hProfileY->Fit(FitFuncGausSlices,"RNQ");
FitFuncSlices->SetParameters(FitFuncGausSlices->GetParameter(0),FitFuncGausSlices->GetParameter(1),FitFuncGausSlices->GetParameter(2),10);
FitFuncSlices->SetParLimits(0,0,10000);
FitFuncSlices->SetParLimits(1,MaxValue-10,MaxValue+10);
FitFuncSlices->SetParLimits(2,0,10);
FitFuncSlices->SetParLimits(3,0,100);
hProfileY->Fit(FitFuncSlices,"RNQ");
cout <<MaxValue<<" " << FitFuncSlices->GetParameter(1) << " " << FitFuncSlices->GetParError(1) <<endl;
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->SetBinContent(binX, FitFuncSlices->GetParameter(1));
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->SetBinError(binX, FitFuncSlices->GetParError(1));
}
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->GetYaxis()->SetRangeUser(ChannelPeakPos-100,ChannelPeakPos+50);
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->GetXaxis()->SetRangeUser(30,500);
new TCanvas();
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->Draw();
TF1 * funcCorrConst = new TF1("funcCorrConst","pol0",50,350);
funcCorrConst->SetLineColor(kRed);
TF1 * funcCorrPol = new TF1("funcCorrPol","pol6",50,300);
funcCorrPol->SetLineColor(kBlue);
funcCorrPol->SetParameter(0,0.04);
funcCorrPol->SetParameter(1,ChannelPeakPos);
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->Fit(funcCorrConst,"R");
hSpectrumT1090_EnergyChannelCorr1_MaxPosManually->Fit(funcCorrPol,"R+");
TCanvas* can2=new TCanvas();
gPad->SetLogz();
hSpectrumT1090_EnergyChannelCorr1->Draw("colz");
funcCorrPol->DrawCopy("same");
for(int i = 7;i>=0;i--)
funcCorrPol->SetParameter(i,funcCorrPol->GetParameter(i)/funcCorrConst->GetParameter(0));
funcCorrPol->Print();
char buf[20];
sprintf(buf, "funcCorr2Norm_%d",ki);
funcCorrPol->Write(buf,TObject::kOverwrite);
funcCorrConst->Delete();
funcCorrPol->Delete();
//return 0;
}
}
//can->ls();
gDirectory->Delete("ncCorrPol;1");
FitInput->ls();
//TCanvas* can3=new TCanvas();
//hSpectrumTDeriMax1090_EnergyChannel->ProfileX()->Draw();
//hSpectrumTDeriMax1090Rel_EnergyChannel->ProfileX()->Draw();
//hSpectrumTDeriMax1090Rel_EnergyChannel->Fit("pol3");
FitInput->Close();
}
void TakeDataAnalyzer(Int_t RunN){
//Int_t RunN = 4591;
//string FileRoot = "/d00/icali/PixelHwStudies/PSIStuff/log/bt05r";
string FileRoot = "/home/l_tester/log/bt05r";
string FileInName = "thistos.root";
string FileOutName = "ProcessedData.root";
char RunNumber[6];
sprintf(RunNumber, "%.6d", RunN);
FileRoot = FileRoot + RunNumber + "/";
FileInName = FileRoot + FileInName;
FileOutName = FileRoot + FileOutName;
TFile *iFile = new TFile((const char*)FileInName.c_str());
TNtuple *Digis = (TNtuple*)iFile->FindObjectAny("events");
Int_t EvN, row, col, roc, ph;
Float_t vcal;
Digis->SetBranchAddress("row", &row);
Digis->SetBranchAddress("col", &col);
Digis->SetBranchAddress("roc", &roc);
Digis->SetBranchAddress("ph", &ph);
Digis->SetBranchAddress("vcal", &vcal);
Digis->SetBranchAddress("eventNr", &EvN);
Int_t iNTlenght = Digis->GetEntries();
TFile *oFile = new TFile((const char*)FileOutName.c_str(), "RECREATE");
TNtuple *Multiplicity = new TNtuple("ModMult", "ModMult", "EvN:roc:NModHits:NROCHits:NRows:NCols:MeanPh:MeanPhPuls",10000000);
//******************Calculating Number of Events**************
Digis->GetEntry(iNTlenght -1);
std::cout << " MaxEvent " << EvN << std::endl;
//*****************Creating Multiplicity NTuple***************
ModuleMultiplicity ModMult;
Int_t OldEvN = 1;
for(Int_t i =0; i< iNTlenght; ++i){
Digis->GetEntry(i);
if(OldEvN != EvN ||( i == iNTlenght -1) ){
for(Int_t j=0; j < 16; ++j){
Float_t NModHits = ModMult.ModMultiplicity();
Float_t NROCHits = ModMult.ROCMultiplicity(j);
Float_t NRows = ModMult.NumberOfRows(j);
Float_t NCols = ModMult.NumberOfColumns(j);
Float_t MeanPh = ModMult.MeanPh(j);
Float_t MeanPhPuls = ModMult.MeanPhPuls(j);
if(NROCHits >0) Multiplicity->Fill(EvN-1, j,NModHits ,NROCHits ,NRows ,NCols ,MeanPh, MeanPhPuls);
}
ModMult.Clean();
OldEvN = EvN;
}
bool SelPixel = 0;
if(row ==5 && col ==5)SelPixel = 1;
if(roc>-1&&col>-1)ModMult.Add(roc, row, col, ph, SelPixel);
}
oFile->Write();
oFile->Close();
}
void jettrigger()
{
TFile *f = new TFile("jettrig.root");
TNtuple *nt = (TNtuple *)f->Get("nt");
//Declaration of leaves types
Float_t pt;
Float_t eta;
Float_t phi;
Float_t mass;
Float_t jt40;
Float_t jt60;
Float_t jt80;
Float_t jt100;
Float_t pscl40;
Float_t pscl60;
Float_t pscl80;
Float_t pscl100;
// Set branch addresses.
nt->SetBranchAddress("pt",&pt);
nt->SetBranchAddress("eta",&eta);
nt->SetBranchAddress("phi",&phi);
nt->SetBranchAddress("mass",&mass);
nt->SetBranchAddress("jt40",&jt40);
nt->SetBranchAddress("jt60",&jt60);
nt->SetBranchAddress("jt80",&jt80);
nt->SetBranchAddress("jt100",&jt100);
nt->SetBranchAddress("pscl40",&pscl40);
nt->SetBranchAddress("pscl60",&pscl60);
nt->SetBranchAddress("pscl80",&pscl80);
nt->SetBranchAddress("pscl100",&pscl100);
TFile *fout = new TFile("jettrig_withweight.root","recreate");
TNtuple *ntout = new TNtuple("ntweight","ntweight","pt:eta:phi:mass:jt40:jt60:jt80:jt100:pscl40:pscl60:pscl80:weightJet:weight12003");
//TTree *ntout=new TTree("ntweight","ntweight");
ntout->SetDirectory(fout);
/* ntout->Branch("pt",&pt);
ntout->Branch("eta",&eta);
ntout->Branch("phi",&phi);
ntout->Branch("mass",&mass);
ntout->Branch("jt40",&jt40);
ntout->Branch("jt60",&jt60);
ntout->Branch("jt80",&jt80);
ntout->Branch("jt100",&jt100);
ntout->Branch("pscl40",&pscl40);
ntout->Branch("pscl60",&pscl60);
ntout->Branch("pscl80",&pscl80);
ntout->Branch("pscl100",&pscl100);
ntout->Branch("weight",&weight);
*/
Long64_t nentries = nt->GetEntries();
cout<<nentries<<endl;
int oneperc = nentries/100;
Long64_t nbytes = 0;
//#ifndef __CINT__
//#pragma omp parallel for ordered schedule(dynamic)
//#endif
for (Long64_t i=0; i<nentries;i++) {
nbytes += nt->GetEntry(i);
if (i % oneperc == 0) cout<<"\r"<<i/oneperc<<"% "<<flush;
float weightJet = 0, weight12003 = 0;
if (jt40 && !jt60 && !jt80 && !jt100) weight12003 = 1/pscl40;
if (jt60 && !jt80 && !jt100) weight12003 = 1;
if (jt80 && !jt100) weight12003 = 1;
if (jt100) weight12003 = 1;
if (jt40 && pt>40 && pt<60) weightJet = pscl40;
if (jt60 && pt>60 && pt<80) weightJet = pscl60;
if (jt80 && pt>80 && pt<100) weightJet = pscl80;
if (jt100 && pt>100) weightJet = pscl100;
//#ifndef __CINT__
//#pragma omp ordered
//#endif
ntout->Fill(pt,eta,phi,mass,jt40,jt60,jt80,jt100,pscl40,pscl60,pscl80,weightJet,weight12003);
}
cout<<endl;
cout<<ntout->GetEntries()<<endl;
ntout->Write();
fout->Close();
f->Close();
f = new TFile("jettrig_withweight.root");
nt = (TNtuple *)f->Get("ntweight");
cout<<nt->GetEntries()<<endl;
f->Close();
}
void fragmentYieldsPlot() {
gStyle->SetOptStat(0000000000); //remove the for this graphs totally redundant statbox
int ZNumGiven;
cout << "Enter fragment Z-number (eg. 1): ";
cin >> ZNumGiven;
TCanvas *c1 = new TCanvas("fragmentYieldsPlot", "Total yield of fragments zero to ten degrees as function of depth");
TString fragmentNameChoices[6] = {"H","He","Li","Be","B","C"};
TString fragmentName = fragmentNameChoices[ZNumGiven-1];
std::cout << fragmentName << endl;
TH1F* dummyHisto = new TH1F("dummyHisto", fragmentName + " yields 0-10 degrees" ,100, 0.0,40); //Dummyhisto fix for missing TNtuple methods.
dummyHisto->SetXTitle("Depth (cm)");
dummyHisto->SetYTitle("N/N0");
ifstream in;
TString experimentalDataPath = "experimentalData/iaeaBenchmark/yields/TDK" + fragmentName + ".dat";
ifstream in;
//Pull in ascii/exfor-style data
in.open(experimentalDataPath);
Float_t f1,f2;
Int_t nlines = 0;
TFile *f = new TFile("fragmentAngularDistribution.root","RECREATE");
TNtuple *ntuple = new TNtuple("ntuple","Data from ascii file","x:y");
Char_t DATAFLAG[4];
Int_t NDATA;
Char_t n1[15], n2[15];
in >> DATAFLAG >> NDATA ; // Read EXFOR line: 'DATA 6'
in >> n1 >> n2; // Read column titles: 'Energy He B [...]'
cout <<n1<<" "<<n2<<"\n";
while (1) {
in >> f1 >> f2;
if (!in.good()) break;
if (nlines < 500 ) printf("%f %f\n",f1,f2);
ntuple->Fill(f1,f2);
nlines++;
}
std::cout << "Imported " << nlines << " lines from data-file" << endl;
TNtuple *simData = new TNtuple("ntuple","Data from ascii file","depth:H:He:Li:Be:B:C");
// gROOT->SetStyle("clearRetro");
//this will be used as base for pulling the experimental data
TString dir = gSystem->UnixPathName(gInterpreter->GetCurrentMacroName());
dir.ReplaceAll("fragmentAngularDistribution.C","");
dir.ReplaceAll("/./","/");
ifstream in;
simData->Fill(0.0,0.0,0.0,0.0,0.0,0.0,1.0);
for(int j = 1; j <= 40;j=j=j+1){
TString pDepth, fragment, Znum, normToOneAtZeroAngle;
pDepth = Form("%i",j);
/*
cout << "Enter phantom depth (eg. 27.9, see experimentalData directory for choices): ";
cin >> pDepth;
*/
TString simulationDataPath = "IAEA_" + pDepth + ".root";
//Let's pull in the simulation-data
//TFile *MCData = TFile::Open("IAEA_" + pDepth + ".root");
TFile *MCData = TFile::Open(simulationDataPath);
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.
//ALL UNITS ARE cm!
Double_t scatteringDistance = detectorDistance - phantomCenterDistance; //temporarily hard-coded, should be distance from target-center to detector
Double_t degrees = 10.0;
Double_t r, rMin, rMax, graphMaximum = 0.0;
Double_t norming = events*.999;
TString rMinString;
TString rMaxString;
rMinString = "0.00";
rMaxString = Form("%f", scatteringDistance*TMath::ATan(degrees*TMath::DegToRad()));
Double_t H = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",1) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
Double_t He = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",2) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
Double_t Li = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",3) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
Double_t Be = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",4) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
Double_t B = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",5) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
Double_t C = ((Double_t*) fragments->GetEntries("(Z == " + TString::Format("%i",6) + " && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")")) / norming;
simData->Fill(waterThickness,H,He,Li,Be,B,C);
}
simData->Scan();
simData->SetMarkerStyle(2); //filled dot
simData->SetMarkerColor(kBlue);
graphMaximum = TMath::Max(graphMaximum, simData->GetMaximum(fragmentName));
graphMaximum = TMath::Max(graphMaximum, ntuple->GetMaximum("y"));
dummyHisto->SetMaximum(graphMaximum + .05*graphMaximum);
dummyHisto->Draw();
simData->Draw(fragmentName + ":depth","","p,same");
ntuple->SetMarkerStyle(22); //triangle
ntuple->SetMarkerColor(kRed);
ntuple->Draw("y:x","","p,same");
c1->SaveAs("fragmentYieldsFor" + fragmentName + ".png");
}