COLZ.cxx

Go to the documentation of this file.

////////////////////////////////////////////////////////////////
//                                                            //
//    Simple COLZ plot using Draw (and basic 1D and 2D plots) //
//    Because cool 3D COLZ do not work using fill + profCOLZ  //
//    GRRRrrr!                                                //
//                                                            //
//    Created by: Diana Patricia Mendez                       //
//    03 March 2016                                           //
////////////////////////////////////////////////////////////////

#include <iostream>
#include <TFile.h>
#include <TPad.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TH1F.h>
#include <TH1.h>
#include <TH2.h>
#include <TH2F.h>
#include <TH1D.h>
#include <TH2D.h>
#include <TH3D.h>
#include <TF1.h>
#include <TROOT.h>
#include <TFile.h>
#include <TTree.h>
#include <TSystem.h>
#include "TChain.h"

// ND specifications
#define ND_Planes  214
#define ND_Cells   344044
#define ND_XMin   -191
#define ND_XMax    192
#define ND_YMin   -187
#define ND_YMax    194
#define ND_Z       1587
#define ND_ZActive 1270

// FD specifications
#define FD_Planes  896
#define FD_Cells   20192
#define FD_XMin   -758
#define FD_XMax    765
#define FD_YMin   -749
#define FD_YMax    765
#define FD_Z       5962
#define FD_ZActive 1270

// canvas size
#define c1_x 700
#define c1_y 500 

#define c2_x 1000
#define c2_y 500



int main(){


  gROOT->ForceStyle();

  TFile* fOut = new TFile("mc.root", "RECREATE");

  std::cout << "Loading files..." << std::endl;
  TChain* tInChain = new TChain("muondedxana/fTree");

  tInChain->Add("mc/*.root");

  TH1D *hPECellPlaneX;
  TH1D *hPECellPlaneY;
  TH1D *hPECorrCellPlaneX;
  TH1D *hPECorrCellPlaneY;
  TH1D *hTrueCellPlaneX;
  TH1D *hTrueCellPlaneY;

  tInChain  ->Draw("PE/path:cell:plane >> hPECellPlaneX(214/2,0,214,96,0,96)","view==0 && (x>100 && x<200)","profcolz");
  tInChain  ->Draw("PE/path:cell:plane >> hPECellPlaneY(214/2,0,214,96,0,96)","view==1 && (x>100 && x<200)","profcolz");
  tInChain  ->Draw("PECorr/path:cell:plane >> hPECorrCellPlaneX(214/2,0,214,96,0,96)","view==0 && (x>100 && x<200)","profcolz");
  tInChain  ->Draw("PECorr/path:cell:plane >> hPECorrCellPlaneY(214/2,0,214,96,0,96)","view==1 && (x>100 && x<200)","profcolz");
  tInChain  ->Draw("trueE/path:cell:plane >> hTrueCellPlaneX(214/2,0,214,96,0,96)","view==0 && (x>100 && x<200)","profcolz");
  tInChain  ->Draw("trueE/path:cell:plane >> hTrueCellPlaneY(214/2,0,214,96,0,96)","view==1 && (x>100 && x<200)","profcolz");

  hPECellPlaneX -> SetTitle("");
  hPECellPlaneY -> SetTitle("");
  hPECorrCellPlaneX -> SetTitle("");
  hPECorrCellPlaneY -> SetTitle("");
  hTrueCellPlaneX -> SetTitle("");
  hTrueCellPlaneY -> SetTitle("");
  

  double PE, PECorr, path, w, cell, plane, x, trueE;
  int hitId, view, totalPlanes;

  std::cout << "Setting branches" << std::endl;
  tInChain->SetBranchAddress("PE", &PE);
  tInChain->SetBranchAddress("PECorr", &PECorr);
  tInChain->SetBranchAddress("path", &path);
  tInChain->SetBranchAddress("w", &w);
  tInChain->SetBranchAddress("cell", &cell);
  tInChain->SetBranchAddress("plane", &plane);
  tInChain->SetBranchAddress("view", &view);
  tInChain->SetBranchAddress("x", &x);
  tInChain->SetBranchAddress("trueE", &trueE);

  
  unsigned int nEntries = tInChain->GetEntries();

  std::cout << "Defining histograms" << std::endl;
  TH2D* hPEWX     = new TH2D("hPEWX","hPEWX",40,-200,200,50,0,100); 
  TH2D* hPEWY     = new TH2D("hPEWY","hPEWY",40,-200,200,50,0,100); 
  TH2D* hPECorrWX = new TH2D("hPECorrWX","hPECorrWX",40,-200,200,50,0,100); 
  TH2D* hPECorrWY = new TH2D("hPECorrWY","hPECorrWY",40,-200,200,50,0,100); 
  TH2D* hTrueWX   = new TH2D("hTrueWX","hTrueWX",40,-200,200,100,0,10); 
  TH2D* hTrueWY   = new TH2D("hTrueWY","hTrueWY",40,-200,200,100,0,10); 
  TH1D* hWX = new TH1D("hWX","hWX",40,-200,200); 
  TH1D* hWY = new TH1D("hWY","hWY",40,-200,200); 



  TH3D* hPECorrCellPlaneX_test = new TH3D("hPECorrCellPlaneX_test","hPECorrCellPlaneX_test",214/2,0,214,96,0,96,100,0,100); 
  TH3D* hPECorrCellPlaneY_test = new TH3D("hPECorrCellPlaneY_test","hPECorrCellPlaneY_test",214/2,0,214,96,0,96,100,0,100); 


  
  std::cout << "Filling histograms" << std::endl;
  for(unsigned int i=0; i<nEntries; i++){
    tInChain->GetEntry(i);

    if(x>100 && x<200){

      if(view==0){
        hPEWX ->Fill(w,PE/path);
        hPECorrWX ->Fill(w,PECorr/path);
        hTrueWX ->Fill(w,trueE/path);
        hWX ->Fill(w);

        hPECorrCellPlaneX_test->Fill(plane,cell,PECorr/path);
        
      }
      else if(view==1){
        hPEWY ->Fill(w,PE/path);
        hPECorrWY ->Fill(w,PECorr/path);
        hTrueWY   ->Fill(w,trueE/path);
        hWY ->Fill(w);

        hPECorrCellPlaneY_test->Fill(plane,cell,PECorr/path);

      }

    } // loop over tricells
    
  } // loop over entries

  hPEWX ->SetTitle("");
  hPEWY ->SetTitle("");
  hPECorrWX ->SetTitle("");
  hPECorrWY ->SetTitle("");
  hTrueWX ->SetTitle("");
  hTrueWY ->SetTitle("");
  hWX ->SetTitle("");
  hWY ->SetTitle("");

  hPECorrCellPlaneX_test->SetTitle("");
  hPECorrCellPlaneY_test->SetTitle("");

  
  fOut->Write();
  fOut->Close();

}

int COLZ(){
  return main();
}