MakeMEUPerDiblock.C

Go to the documentation of this file.

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

class MEUPerDiblock : public CalibAnaPlot{

  public:

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



  private:
    std::vector< std::string > fEpochList;
    TexBuilder * fTex;
    std::map< int, std::map < int, bool > > fBadFeb;//[plane][module]
};


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

void MakeMEUPerDiblock(std::string outpath, std::string group, unsigned int cap = 0, unsigned int stride = 0)
{
  //group WOULD BE fd_mp5 FOR EXAMPLE.
  MEUPerDiblock meuperdiblock(outpath, group);
  meuperdiblock.Initialize();
  if(cap)   meuperdiblock.SetTreeLoopCap   (cap);
  if(stride)meuperdiblock.SetTreeLoopStride(stride);

  meuperdiblock.DoAna();

}

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

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

  fTex = new TexBuilder( fOutPath+"/MEUPerDiblock.tex", 
      "Drift by Diblock", 
      "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 Diblock.");

  std::string plotdir = "fd_all_samples";

  std::vector< std::string > views = {"x", "y"};
  std::vector< std::string > yaxes = {"pecm", "pecorrcm"};
  std::vector< std::string > drawplots;
  for(auto& view : views){
    for(auto& yax : yaxes){
      drawplots.push_back(yax+"_diblock_"+view);
    }
  }

  Draw(plotdir,
      allsamples, // numerator samples  
      "fd_mc_p3", // denom sample: data over all time
      "",         // denom plot: mean for whole view-module group
      drawplots,  // plots to make in numerator samples
      "" );       // don't draw ratio for now

  //DriftTex(plotdir);

  delete fTex;
}

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

  fTex->SectionNoNum("MEU with FEB Cut");
  fTex->AddFigureRow({plotdir+"/legend"}, {""}, {""});
  fTex->AddFigureRow({plotdir+"/pecm_time_x_0-31_febcut",  plotdir+"/pecm_time_y_0-31_febcut" }, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecm_time_x_32-63_febcut", plotdir+"/pecm_time_y_32-63_febcut"}, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecm_time_x_64-95_febcut", plotdir+"/pecm_time_y_64-95_febcut"}, {"", ""}, {"", ""}); 
  fTex->NewPage();
  fTex->AddFigureRow({plotdir+"/legend"}, {""}, {""});
  fTex->AddFigureRow({plotdir+"/pecorrcm_time_x_0-31_febcut",  plotdir+"/pecorrcm_time_y_0-31_febcut" }, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecorrcm_time_x_32-63_febcut", plotdir+"/pecorrcm_time_y_32-63_febcut"}, {"", ""}, {"", ""}); 
  fTex->AddFigureRow({plotdir+"/pecorrcm_time_x_64-95_febcut", plotdir+"/pecorrcm_time_y_64-95_febcut"}, {"", ""}, {"", ""}); 
  fTex->NewPage();


  unsigned int nPlanes = fDet->NPlanes();
  for(unsigned int p = 0; p < 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"}, {"", ""}, {"", ""}); 
    if(p!=nPlanes-1) fTex->NewPage();
  }
}*/

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerDiblock::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 = "";
  if(view==0){ vstr="x"; }
  if(view==1){ vstr="y"; }  


  float pecm     = pe/path;
  float pecorrcm = pecorr/path;

  fPlots["pecm_diblock_"+vstr]    ->Fill(diblock, pecm);
  fPlots["pecorrcm_diblock_"+vstr]->Fill(diblock, pecorrcm);
}

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

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerDiblock::SchedulePlots()
{
  unsigned int nDiblocks = fDet->NDiblocks();

  std::vector< PlotAxis > xaxes = { 
    {"diblock",  "Diblock Number",  (int)nDiblocks, 0., (float)nDiblocks}
  };

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

  std::vector< std::string > views = {"x", "y"};
  std::string vstr = "";
  for( auto& v : views){
    if(v=="x") vstr = "X";
    if(v=="y") vstr = "Y";
    for(auto& xax : xaxes){
      for(auto& yax : yaxes){
        std::string name = yax.label +"_"+ xax.label +"_"+ v;
        std::cout << "Schedule: " << name << std::endl;

        fPlots[name] = new TProfile(name.c_str(), (vstr+" View;" + xax.title +";"+ yax.title).c_str(),
                                    xax.nbins, xax.min, xax.max);
      } // yaxes
    } // xaxes
  }
} // SchedulePlots

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerDiblock::ScheduleAllDataPlots()
{
  return;
}

//^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^~^
void MEUPerDiblock::ScheduleSamples()
{
  std::vector< std::string > epoch_list = {
    "3b", "5a"};
    /*
    "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(fDet->IsND()){
      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";
      if(!fSamples.count("nd_mc")){
        //fSamples["nd_mc"]="/nova/ana/users/abooth/miniprod5.b/nd_mc/merged.root";
      }
    }
    else if(fDet->IsFD()){
      fSamples["fd_data_ep"+ep]="/nova/ana/users/talion/miniprod5_calibana/fd_data_epoch"+ep+"/merged.root";
      if(!fSamples.count("fd_mc_p3")){
        //fSamples["fd_mc_p2"]="/nova/ana/users/talion/miniprod5_calibana/fd_mc_lowgain/merged.p2.root";
        fSamples["fd_mc_p3"]="/nova/ana/users/talion/miniprod5_calibana/fd_mc_highgain/merged.p3.root";
        //fSamples["fd_mc_p5"]="/nova/ana/users/talion/miniprod5_calibana/fd_mc_highgain/merged.p5.root";
      }
    }
  }

}