MakeMEUPerCell.C

Go to the documentation of this file.

#include "CalibAnaPlot.h"
#include "CalibAnaTypes.h"
#include "TexBuilder.h"
#include <iomanip>
#include <numeric>
#include "TROOT.h"

class MEUPerCell : public CalibAnaPlot{

  public:

    MEUPerCell(std::string outpath, std::string group);
    // vvv CalibAnaPlot implimentations
    void SchedulePlots();
    void ScheduleSamples();
    void ScheduleHitVars();
    void FillHitPlots();
    // ^^^ CalibAnaPlot implimentations
    void DoAna();
    void DriftTex(std::string plotdir);

  private:
    std::vector< std::string > fEpochList;
    TexBuilder * fTex;
};


MEUPerCell::MEUPerCell(std::string outpath, std::string group)
: CalibAnaPlot(outpath, group)
{
}

void MakeMEUPerCell(std::string outpath)
{
  MEUPerCell meupercell(outpath, "nd_mp5");
  meupercell.Initialize();

  meupercell.DoAna();


  std::cout << "I HAVE FINISHED, KILL ME NOW!" << std::endl;

}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::DoAna()
{
  //WRITE YOUR ANALYSIS CODE HERE!

  for(auto sample : fSamples){
    ProcessTrees(sample.first);
  }

  std::vector<std::string> allsamples;
  for( auto& samp : fSamples ) allsamples.push_back(samp.first);

  fTex = new TexBuilder( fOutPath+"/MEUPerCell.tex", 
                         "Drift by FEB", 
                         "Tyler Alion, Alexander Booth, Yibing Zhang",
                         "Miniproduction 5 Data vs. MC. Tricell hits from Stopping Cosmics populate all plots, with Absolute Calibration cuts applied. Showing PE/cm and PECorr/cm broken down by cell group and view then further by FEB, then further by individual cell.");
  std::string plotdir = "nd_all_samples";
  //FINAL ARGUMENT, BOOL FOR RATIO PLOTS.
  Draw(plotdir, allsamples, false);
  DriftTex(plotdir);

  delete fTex;

}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::DriftTex(std::string plotdir)
{
  fTex->SectionNoNum("By View and Cell Group");
  fTex->AddFigureRow({plotdir+"/legend"}, {""}, {""});
  fTex->AddFigureRow({plotdir+"/pecm_time_x_0-31",  plotdir+"/pecorrcm_time_y_0-31" }, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecm_time_x_32-63", plotdir+"/pecorrcm_time_y_32-63"}, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecm_time_x_64-95", plotdir+"/pecorrcm_time_y_64-95"}, {"", ""}, {"", ""}); 
  fTex->NewPage();

  for(int p = 0; p < fDet->NPlanes(); p++)
  {
    fTex->SectionNoNum(Form("Plane %i", p));
    std::string pstr = (p % 2 == 0 ? "y_" : "x_") + std::to_string(p);
    fTex->AddFigureRow({plotdir+"/legend"}, {""}, {""});
    fTex->AddFigureRow({plotdir+"/pecm_time_"+pstr+"_0-31",  plotdir+"/pecorrcm_time_"+pstr+"_0-31" }, {"", ""}, {"", ""}); 
    fTex->AddFigureRow({plotdir+"/pecm_time_"+pstr+"_32-63", plotdir+"/pecorrcm_time_"+pstr+"_32-63"}, {"", ""}, {"", ""}); 
    fTex->AddFigureRow({plotdir+"/pecm_time_"+pstr+"_64-95", plotdir+"/pecorrcm_time_"+pstr+"_64-95"}, {"", ""}, {"", ""}); 
    fTex->NewPage();
    fTex->SubSectionNoNum(Form("Plane %i By Cell", p));
    for(int c = 0; c < fDet->NCells(); c++)
    {
      if(c % 3 == 0 && c!=0)
      {
        fTex->NewPage(); 
      }
      fTex->AddFigureRow({plotdir+"/pecm_time_"    +std::to_string(p)+"_"+std::to_string(c),  
                          plotdir+"/pecorrcm_time_"+std::to_string(p)+"_"+std::to_string(c) }, {"", ""}, {"", ""}); 
    }
    if(p!=fDet->NPlanes()-1) fTex->NewPage();
  }
}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::FillHitPlots()
{
  // The filled branches are protected members of
  // CalibAnaPlot, so can be used here

  bool abscal_hit = ( w > -100 && w < 100 && 
                      path > 0. &&
                      pecorr > 0. && pecorr/path < 100. &&
                      pe > 0. &&
                      cmFromEnd > 100 && cmFromEnd < 200 );
  if( !abscal_hit ) return;
  if(  plane > 191) return;

  std::string vstr, pstr;
  if(view==0){ vstr="x"; }
  if(view==1){ vstr="y"; }  
  pstr = std::to_string(plane);

  std::string end      = vstr + "_" + pstr + "_";
  std::string endGroup = vstr + "_";
  std::string endCell  = vstr + "_" + pstr + "_" + std::to_string(cell);

  float pecm     = pe/path;
  float pecorrcm = pecorr/path;
  if(cell <= 31){
    end      += "0-31";
    endGroup += "0-31";
  }
  else if(cell > 31 && cell <= 63){
    end      += "32-63";
    endGroup += "32-63";
  }
  else{
    end      += "64-95";
    endGroup += "64-95";
  }
  fPlots["pecm_time_"+end]    ->Fill(evt_time,pecm);
  fPlots["pecorrcm_time_"+end]->Fill(evt_time,pecorrcm);      

  fPlots["pecm_time_"+endGroup]    ->Fill(evt_time,pecm);
  fPlots["pecorrcm_time_"+endGroup]->Fill(evt_time,pecorrcm);      

  fPlots["pecm_time_"+endCell]    ->Fill(evt_time,pecm);
  fPlots["pecorrcm_time_"+endCell]->Fill(evt_time,pecorrcm);      
}


//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::ScheduleHitVars()
{
  fHitVars="w, view, plane, cell, pe, pecorr, path, evt_time, cmFromEnd,";
}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::SchedulePlots()
{
  int nPlanes = fDet->NPlanes();
  int nCells  = fDet->NCells();

  std::vector< PlotAxis > xaxes = { 
    {"time",  "Unix Time (s)",  100, 14e8, 16e8}
  };

  std::vector< ContentAxis > yaxes = { 
    {"pecm",     "PE/cm"},
    {"pecorrcm", "PECorr/cm"},
  };

  std::vector< std::string > cells = {"0-31", "32-63", "64-95"};
  std::vector<int> planes(nPlanes);
  std::iota(planes.begin(), planes.end(), 0);
  std::vector<int> allcells(nCells);
  std::iota(allcells.begin(), allcells.end(), 0);
  for(auto& p : planes){
    std::string vstr = (p % 2 == 0 ? "Y" : "X");
    std::string v    = (p % 2 == 0 ? "y" : "x");
    for(auto& xax : xaxes){
      for(auto& yax : yaxes){
        for(auto& c : cells){

          std::string name = yax.label +"_"+ xax.label +"_"+ v + "_" + std::to_string(p) + "_" + c;

          fPlots[name] = new TProfile( name.c_str(),
              (vstr+" View, Plane "+std::to_string(p)+", Cells "+c+";" + xax.title +";"+ yax.title).c_str(),
              xax.nbins, xax.min, xax.max );


          if(p < 2){
            name = yax.label +"_"+ xax.label +"_"+ v + "_" + c;
            fPlots[name] = new TProfile( name.c_str(),
                (vstr+" View, Cells "+c+";" + xax.title +";"+ yax.title).c_str(),
                xax.nbins, xax.min, xax.max );
          }
        }
        for(auto& c : allcells){
          std::string name = yax.label +"_"+ xax.label +"_"+ v + "_" + std::to_string(p) + "_" + std::to_string(c);

          fPlots[name] = new TProfile( name.c_str(),
              (vstr+" View, Plane "+std::to_string(p)+", Cell "+std::to_string(c)+";" + xax.title +";"+ yax.title).c_str(),
              60, xax.min, xax.max );
        }
      }
    }
  }
}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerCell::ScheduleSamples()
{
  std::vector< std::string > epoch_list = {
    "2a", "2b","2c","2d","2e",
    "3b","3c","3d","3e",
    "4a", "4b.1", "4b.2", "4c", // split 4b up manually
    "5a","5b",
    "6a","6b",
    "7a","7b","7c","7d",
    "8b" };

  fEpochList = epoch_list;

  for( auto& ep : epoch_list ){
    if(ep=="4b.1" || ep=="4b.2") 
      fSamples["nd_data_ep"+ep]="/nova/ana/users/talion/miniprod5/pcliststop_miniprod5.b/nd_data_epoch4b/calibana/merged.root";
    else 
      fSamples["nd_data_ep"+ep]="/nova/ana/users/talion/miniprod5/pcliststop_miniprod5.b/nd_data_epoch"+ep+"/calibana/merged.root";
  }
  
  fSamples["nd_mc"]="/nova/ana/users/abooth/miniprod5.b/nd_mc/merged.root";
}