ProjectionReverse.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ProjectionReverse.C

#include <cmath>
#include <TFile.h>
#include <TString.h>
#include <TLine.h>
#include <TTree.h>
#include <TLatex.h>
#include <TGraphErrors.h>
#include <cassert>
#include <iostream>
#include <fstream>
using namespace std;



using std::cout;
using std::endl;
#endif

void ProjectionReverse()
{

         gStyle->SetOptFit(0);
   gStyle->SetOptStat(0);
        
        TFile *fin = new TFile("EnergyPosition3.root");

double y1 = -90;
double y2 = -117;
double y3 = -143;
double y4 = -167;
double y5 = -192;

//Set Tower Vertical Lines//

double x1 = -205;
double x2 = -228;
double x3 = -253;
double x4 = -277;
double x5 = -301;



int binx;

int forward = 2;
int backward = 2;


int biny;
double step = 1;
/*
double Xmin = 170;
double Xmax = 340.0;

double Ymin = 60;
double Ymax = 245.0;
*/
double Xmax = -170;
double Xmin = -340.0;

double Ymax = -60;
double Ymin = -245.0;
cout << Ymax << endl;

int NumX = (Xmax - Xmin)/step;
int NumY = (Ymax - Ymin)/step;

const int XBins = NumX;

const int YBins = NumY;
double x;
double y;

cout << "XBins = " << XBins << endl;
cout << "YBins = " << YBins << endl;


double value[XBins][YBins];

double finalvalue[XBins][YBins];

TH2D *Inter = new TH2D("Inter","",XBins,Xmin,Xmax,YBins,Ymin,Ymax);

/*
double ylower = 35;
double yhigher = 188;
double xlower = 178;
double xhigher = 301; 
*/
int k;

int NumberofValue;
double Sum;
int Xnow;

int Ynow;
TCanvas *c1 = new TCanvas("c1", "c1",0,0,800,600);

c1->cd();

EnPo->GetXaxis()->SetTitle("Horizontal Axis (mm)");
EnPo->GetYaxis()->SetTitle("Vertical Axis (mm)");

EnPo->SetTitle("Energy vs Horizontal and Vertical Positions Before Interpolation - 0 Degree");

EnPo->Draw("colz");
c1->Update();

c1->SaveAs("Before0Degree.png");

for(int i = 0; i < XBins; i++)
{

//x = Xmin + step*i;


for(int j = 0; j < YBins; j++)
{

//      TCanvas *c1 = new TCanvas("c1", "c1",0,0,800,600);
//      y = Ymin + step*j;




value[i][j] = EnPo->GetBinContent(i,j);


}
}

//Horizontal Interpolation Begins//
for(int i = backward; i < XBins - backward; i++)
{

        for(int j = backward; j < YBins - backward; j++)
{

//cout << "OK" << endl;

NumberofValue = 0;
Sum = 0;

Xnow = XBins - i;
Ynow = YBins - j;


for(int k = i - backward; k < i + forward +1; k++)
{

        for(int m = j - backward; m < j+backward+1; m++)
        {
        
                if(value[k][m] > 0) NumberofValue = NumberofValue + 1;
                
                Sum = Sum + value[k][m];
        
                        }

}

cout << "Number of Value = " << NumberofValue << endl;

cout << "Sum = " << Sum << endl;

if(NumberofValue > 0) finalvalue[i][j] = Sum/NumberofValue;

cout << " i = " << i <<  "  j = " << j << endl;
cout << " Xnow = " << Xnow <<  "  YNow = " << Ynow << endl;

cout << "Final Value is " << finalvalue[i][j]  << endl;


//Inter->SetBinContent(i,j,finalvalue[i][j]);

Inter->SetBinContent(Xnow,Ynow,finalvalue[i][j]);

}


}

/*
cout << "Horizontal Done" << endl;


//Vertical Interpolation Begins//

cout << "DONE Value" << endl;

for(int i = 1; i < XBins; i++)
{

        for(int j = 1; j < YBins; j++)
{

        cout << "Before Updating: Binx =  " << i << "   " << "Bin y = " << j << " Value =  " << value[i][j] << endl;


if (value[i][j] < 6.8 && value[i][j-1] >0 && i < XBins && j < YBins)
{


k = j;
do {
        k = k + 1;
        
//      cout << "k value is = " << k << endl;
}
while(value[i][k] == 0 && k < YBins); 

//cout << "neW k " << k  << endl;
                
value[i][j] =value[i][j-1] + (value[i][k] - value[i][j-1])/(k+1-j);

        cout << "After Updating: Binx =  " << i << "   " << "  Bin y = " << j << "  Value =  " <<value[i][j] << endl;

}


}


}

cout << "Vk =" << k << endl;

//Vertical Interpolation Ends//

cout << "VERTICAL DONE" << endl;

*/
/*
for(int i =0; i < XBins; i++)
{



        for(int j = 1; j < YBins; j++)
{


}


}
*/
Inter->GetXaxis()->SetTitle("Horizontal Axis (mm)");
Inter->GetYaxis()->SetTitle("Vertical Axis (mm)");
Inter->SetTitle("Energy vs Horizontal and Vertical Positions After Interpolation - 0 Degree");

  TFile *fout = new TFile("Interpolated2.root","RECREATE");
Inter->Write();







Inter->Draw("colz");


char Outname[512];

const int All = 7;

double left;
double right;
double initial = -191;
double inteval = -12.5;

for(int i=0; i < All; i++)
{
TCanvas *c22 = new TCanvas("c22", "c22",0,0,800,600);

Inter->SetMinimum(3.5);
Inter->SetMaximum(9.0);
c22->cd();
        Inter->Draw("colz");

        left  = initial + inteval * i - 5;
        right  = initial + inteval * i + 5;

   TLine *l1 = new TLine(left,Ymin,left,Ymax);

   l1->SetLineWidth(2);
   l1->Draw("same");

   TLine *l2 = new TLine(right,Ymin,right,Ymax);
   l2->SetLineWidth(2);
   l2->Draw("same");
sprintf(Outname,"Result/VerticalPlot%d.png",i);

c22->Update();

c22->SaveAs(Outname);

}

initial = -77;
int HAll = All + 5;
for(int i=0; i < HAll; i++)
{
TCanvas *c22 = new TCanvas("c22", "c22",0,0,800,600);

Inter->SetMinimum(3.5);
Inter->SetMaximum(9.0);
c22->cd();
        Inter->Draw("colz");

        left  = initial + inteval * i - 5;
        right  = initial + inteval * i + 5;

   TLine *l1 = new TLine(Xmin,left,Xmax,left);

   l1->SetLineWidth(2);
   l1->Draw("same");

   TLine *l2 = new TLine(Xmin,right,Xmax,right);
   l2->SetLineWidth(2);
   l2->Draw("same");
sprintf(Outname,"Result/HorizontalPlot%d.png",i);

c22->Update();

c22->SaveAs(Outname);

}


/*
// Horizontal line Tower 1 //
   TLine *l1 = new TLine(Xmin,y1,Xmax,y1);
   l1->SetLineWidth(1);
   l1->Draw("same");


 TLatex* texCol;
texCol= new TLatex(Xmin,y1,"Tower 1");
  texCol->SetTextAlign(13);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();


// Horizontal line Tower 2 //


   TLine *l2 = new TLine(Xmin,y2,Xmax,y2);
   l2->SetLineWidth(1);
   l2->Draw("same");



texCol= new TLatex(Xmin,y2,"Tower 2");
  texCol->SetTextAlign(13);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();

// Horizontal line Tower 3//


   TLine *l3 = new TLine(Xmin,y3,Xmax,y3);
   l3->SetLineWidth(1);
   l3->Draw("same");



texCol= new TLatex(Xmin,y3,"Tower 3");
  texCol->SetTextAlign(13);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();



// Horizontal line Tower 4 //


   TLine *l4 = new TLine(Xmin,y4,Xmax,y4);
   l4->SetLineWidth(1);
   l4->Draw("same");



texCol= new TLatex(Xmin,y4,"Tower 4");
  texCol->SetTextAlign(13);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();

  // Horizontal line Tower 5 //


   TLine *l4 = new TLine(Xmin,y5,Xmax,y5);
   l4->SetLineWidth(1);
   l4->Draw("same");




texCol= new TLatex(Xmin,y5,"Tower 5");
  texCol->SetTextAlign(13);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();




// Vertical line Tower 1 //
   TLine *l6 = new TLine(x1,Ymin,x1,Ymax);
   l6->SetLineWidth(1);
   l6->Draw("same");


 TLatex* texCol;
texCol= new TLatex(x1,Ymax,"Tower 1");
  texCol->SetTextAlign(33);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();


// Horizontal line Tower 2 //


   TLine *l7 = new TLine(x2,Ymin,x2,Ymax);
   l7->SetLineWidth(1);
   l7->Draw("same");


 TLatex* texCol;
texCol= new TLatex(x2,Ymax,"Tower 2");
  texCol->SetTextAlign(33);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();

// Horizontal line Tower 3//


   TLine *l8 = new TLine(x3,Ymin,x3,Ymax);
   l8->SetLineWidth(1);
   l8->Draw("same");


 TLatex* texCol;
texCol= new TLatex(x3,Ymax,"Tower 3");
  texCol->SetTextAlign(33);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();


// Horizontal line Tower 4 //

   TLine *l9 = new TLine(x4,Ymin,x4,Ymax);
   l9->SetLineWidth(1);
   l9->Draw("same");


 TLatex* texCol;
texCol= new TLatex(x4,Ymax,"Tower 4");
  texCol->SetTextAlign(33);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();


  // Horizontal line Tower 5 //


   TLine *l10 = new TLine(x5,Ymin,x5,Ymax);
   l10->SetLineWidth(1);
   l10->Draw("same");


 TLatex* texCol;
texCol= new TLatex(x5,Ymax,"Tower 5");
  texCol->SetTextAlign(33);
  texCol->SetTextSize(0.03);
 texCol->SetTextFont(42);
  texCol->Draw();
*/








}