fragmentAngularDistribution.C

Go to the documentation of this file.

#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 "THStack.h"
#include "TCut.h"
#include "TString.h"
#include "TMath.h"

/***************************
 * Root script that produces a graph of the angular distribution of
 * a certain type of charged fragments itneractively.
 * 
 * Results are not stored in a histogram.
 * 
 * This can be compared to measurements made with a square 
 * detector that is being moved around. Such as that of E.Haettner[1].
 * 
 * Results are normalized to the 0-angle because documentation on E.Haettner's 
 * normalization is not found.
 * 
 * @author Gillis Danielsen
 * **************************/


void fragmentAngularDistribution() {

  gStyle->SetOptStat(0000000000); //remove the for this graphs totally redundant statbox
  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;

   int ZnumInt;
   TString pDepth, fragment, Znum, normToOneAtZeroAngle;
   cout << "Enter phantom depth (eg. 27.9, see experimentalData directory for choices): ";
   cin >> pDepth;
   cout << "Enter fragment Z-number (eg. 1): ";
   cin >> ZnumInt;
   //cout << "Enter fragment name (Znum 1 -> H,Znum 2->He...): ";
   //cin >> fragment;
   TString fragmentNameChoices[6] = {"H","He","Li","Be","B","C"};
   TString fragment = fragmentNameChoices[ZnumInt - 1];
   Znum = Form("%i",ZnumInt);
   cout << "Normalize to 1 at zero angle? (Y/N): ";
   cin >> normToOneAtZeroAngle;   

   TString experimentalDataPath = "experimentalData/iaeaBenchmark/angularDistributions/" + pDepth + "/" + fragment + "" + pDepth +".dat";
   TString simulationDataPath = "IAEA_" + pDepth + ".root";

   TCanvas *c1 = new TCanvas("AngularDistribution", "Angular distribution with discrete measurement annuluses");

   //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;

   //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.
   
        //good to keep for ref. G4 might give weird units due to change.
        metadata->Scan();
        
        //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 r;
        Double_t degrees;
        Double_t rMin;
        Double_t rMax;
        TString rMinString, rMaxString, experimentalNorm;
        Double_t maxValue = 0.0; //When normalizing to 1 this will allways end up being 1)
        
        int i = 0; //so that the degree steps can be varied to unevenly spaced values separate counter is used
        TNtuple* distrib = new TNtuple("angularDistrib","FragmentAngularDistrib","angle:particleAmount:normalized");
        std::cout << "Fragments comparison to the graphs in appendices of E.Haettner\n";
        std::cout << "Scattering distance: " << scatteringDistance << " cm" << endl ;
        std::cout << "(scattering distance may vary with data-files too, see haettner A.1." << endl << endl;
        //This will norm it to the zero degree entry to get rid of Emma's weird normalization
        //fixme detectorsidelengthstring
        rMinString = "0.00";
        rMaxString = Form("%f", detectorSideLength/2);
        //SA of a square Double_t zeroSAsquare = 4 * TMath::ASin(pow(detectorSideLength,2.0) / (4*pow(scatteringDistance,2) + pow(detectorSideLength,2)) );
        //SA with square approx squareApprox = (4*4) / (4*3.14*scatteringDistance*scatteringDistance);
        
        //First calculates the normalization from the zero position
        //Normalization by events becomes redundant but is left in place for future needs

        Double_t deltaPhi = TMath::ATan((detectorSideLength/2)/scatteringDistance); //Angle where side of detector is found
        /*
        //Alternative normalization, here zero position is also done with annulus where rMin=0, the actual detector is a square though
        //Difference with this approach and the other is very small
        Double_t normEntries = fragments->GetEntries("(Z == " + Znum + " && energy > 0 && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")");
        Double_t zeroSA = 2*TMath::Pi()*(TMath::Cos(0) - TMath::Cos(deltaPhi));
        */
        //fragments->Scan();
        //Results are normalized by a square detector mimicing H1 with center at 0 degrees.
        Double_t normEntries = fragments->GetEntries("(Z == " + Znum + " && posY < " + rMaxString + " && posY > -" + rMaxString + " &&  posZ > -" + rMaxString + " && posZ < " + rMaxString + ")");
        Double_t zeroSA = 4 * TMath::ASin(pow(detectorSideLength,2.0) / (4*pow(scatteringDistance,2) + pow(detectorSideLength,2)) );
        Double_t zeroYieldNormed = normEntries / (events * zeroSA);
        if(normToOneAtZeroAngle == "Y"){
                Double_t zeroNorm = zeroYieldNormed; //values normalized to one at zero
        }else{
                Double_t zeroNorm = 1.0; //non-zeronormalized values
        }

        distrib->Fill(0,normEntries,zeroYieldNormed/zeroNorm); //< degrees, entyamount, normalized result for graph
        //fragments->Scan(); //debug
        std::cout << "Norming events: " << normEntries << endl;
        //Loop through all other wanted angles, too large angles will fall outside reach of phantom window.
        for(Double_t j = deltaPhi*TMath::RadToDeg(); j <= 15.0; j=j+.05){
                i++;
                degrees = j * TMath::DegToRad();
                //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 outer and inner radiuses
                //this will give a bit of approximation at small angles and at 0 degrees this gives a completely round sensor.

                
                /*
                 * deltaPhi calculated so that phi+deltaphi points to one side of the detector and phi-deltaphi the other side
                 * 
                 * Alternative 1: detector is moved but the normal is not pointed towards the scattering source.
                 * Alternative 2: detector is moved and pointed towards scattering source. (E.Haettner seems to use this)
                 * 
                 * The difference in results is very minute though. (3% at largest angles)
                 */
                //Alternative 1
                //Double_t deltaPhi = degrees - TMath::ATan(TMath::Tan(degrees) - (detectorSideLength/(2*scatteringDistance)));
                //rMin = TMath::Max(0.0,r - (detectorSideLength/2));
                //rMax = r + (detectorSideLength/2);
                //Alternative 2
                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);
                /*
                * From Gunzert-marx. Solid angle of annulus with rmin trmax, 
                * a bit of an aproximation especially at small phi.
                */
                Double_t deltaOmega = 2*TMath::Pi()*(TMath::Cos(TMath::Max(0.0,degrees-deltaPhi)) - TMath::Cos(degrees+deltaPhi));
                int numEntries = fragments->GetEntries("(Z == " + Znum + "  && sqrt(posY^2 + posZ^2) < " + rMaxString + "&& sqrt(posY*posY + posZ*posZ) > " + rMinString + ")");
                distrib->Fill(j,numEntries,numEntries/(deltaOmega * events * zeroNorm)); //< degrees, entyamount, normalized result for graph
                distrib->Fill(-j,numEntries,numEntries/(deltaOmega * events * zeroNorm)); //< To get gaussian shape better visible
                maxValue = TMath::Max(maxValue, numEntries/(deltaOmega * events * zeroNorm)); //< for calculation of FWHM
                }
        distrib->SetMarkerStyle(2); //filled dot
        distrib->SetMarkerColor(kBlue);
        ntuple->SetMarkerStyle(22); //triangle
    ntuple->SetMarkerColor(kRed);
        
        TH1F* dummHisto = new TH1F("dummyHisto", fragment + ", " + Form("%.1f", waterThickness) + " cm",100, -3.0,14); //Dummyhisto fix for missing TNtuple methods.
        dummyHisto->SetXTitle("Angle (degrees)");
        dummyHisto->SetYTitle("(N/N0) [sr^-1]");
        if(normToOneAtZeroAngle == "Y"){
                dummyHisto->SetMaximum(1.1);
                dummyHisto->SetYTitle("[sr^-1]");
                Float_t zeroPosData; //This is where we store what we norm the experimental data with
                Float_t zeroPosAngle; //okay, so this should be zero, but regrettably is not allways that
                ntuple->SetBranchAddress("y",&zeroPosData);
                ntuple->SetBranchAddress("x",&zeroPosAngle);
                int row = 0;    
                ntuple->GetEntry(row); //Pull the first row, usually is the right one
                while(zeroPosAngle*zeroPosAngle > .01){
                        row++;
                        ntuple->GetEntry(row);
                        if(row == ntuple->GetEntries()){
                                std::cerr << "Could not find zero angle data in imported experimental data. Change normalization or relax exactness of this check." << endl;
                                exit();
                                }
                        }
                
                std::cout << "For zero-position of experimental data using angle " << zeroPosAngle << " with amount " << zeroPosData << " on row " << row << endl;
                experimentalNorm = Form("(1/%f)*", zeroPosData);
    }else{
                //nor normalization to 1 of data
                dummyHisto->SetMaximum(ntuple->GetMaximum("y")+ ntuple->GetMaximum("y")*.1);
                experimentalNorm = ""; //no norming of experimental resilts
        }
        dummyHisto->Draw();
        ntuple->Draw(experimentalNorm + "y:x","","p,same");
        distrib->Draw("normalized:angle","angle > -3 && angle < 14","p,same"); //similar axises to e.haettner

        //Calculate closest-point-FWHM.
        Float_t fwhm = 0.0, middle = 0.0, currentX, currentY;
        distrib->SetBranchAddress("normalized",&currentY);
        distrib->SetBranchAddress("angle",&currentX);
        for(int i = 0; i < distrib->GetEntries(); i++){
                distrib->GetEntry(i);
                if(pow(maxValue/2 - middle, 2.0) > pow(maxValue/2 - currentY, 2.0)){
                                fwhm = 2*currentX;
                                middle = currentY;
                        }else{
                                }
                }
        std::cout << "Calculated (closest point) FWHM of Monte-Carlo simulation to be: " << fwhm << " degrees" << endl; 
        
        /*
         * This code is left here because it allows to calcualte the values without using annuluses
         * this of course is a bit more like the experimental data but statistically less precise.
        for(Double_t p = 0.0;p < 14.0; p = p + 1.0){
        rMinString = "0.00";
        rMaxString = "2.00";
        TString plusY = Form("(posY - %f)", p);
        TString plusZ = Form("(posZ - %f)", p);
                Double_t deltaPhi = TMath::ATan((detectorSideLength/2)/scatteringDistance);
        Double_t normEntries = fragments->GetEntries("(Z == " + Znum + " && energy > 0 && sqrt(" + plusY + "^2 + " + plusZ + "^2) < " + rMaxString + "&& sqrt("+plusY + "^2 + " + plusZ + "^2) > " + rMinString + ")");
        //std::cout << "(Z == " + Znum + " && energy > 0 && sqrt(" + plusY + "^2 + " + plusZ + "^2) < " + rMaxString + "&& sqrt("+plusY + "^2 + " + plusZ + "^2) > " + rMinString + ")";
        Double_t zeroSA = 2*TMath::Pi()*(TMath::Cos(0) - TMath::Cos(deltaPhi));
        std::cout << "with " << p << "cm the amount is " << normEntries << " / " << normEntries /(events*zeroSA) << endl;
                }
                */
        
        
        //c1->SaveAs("angularDistrib_depth_" + pDepth + "_Z_" + Znum + "_normedToZero_" + normToOneAtZeroAngle + "_ComparedToEHaettner.png");
    pDepth.ReplaceAll(".","");
        c1->SaveAs("AD_" + pDepth + "_" + Znum + "_" + normToOneAtZeroAngle + ".png");
        in.close();
        f->Write();
}