CalorPad.cxx

Go to the documentation of this file.

#include "lareventdisplay/EventDisplay/CalorPad.h"
#include "lareventdisplay/EventDisplay/AnalysisBaseDrawer.h"
#include "lareventdisplay/EventDisplay/AnalysisDrawingOptions.h"
#include "nuevdb/EventDisplayBase/EventHolder.h"
#include "nuevdb/EventDisplayBase/View2D.h"

#include "TFile.h"
#include "TGraph.h"
#include "TH1F.h"
#include "TPad.h"

#include "art/Framework/Principal/fwd.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "cetlib/search_path.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

namespace {
  // Utility function to make uniform error messages.
  void writeErrMsg(const char* fcn, cet::exception const& e)
  {
    mf::LogWarning("CalorPad") << "CalorPad::" << fcn << " failed with message:\n" << e;
  }
}

evd::CalorPad::CalorPad(const char* name,
                        const char* title,
                        double x1,
                        double y1,
                        double x2,
                        double y2,
                        int curvetype)
  : DrawingPad(name, title, x1, y1, x2, y2), fcurvetype(curvetype)
{

  // Set up pad.
  this->Pad()->cd();
  this->Pad()->SetBit(kCannotPick);
  this->Pad()->SetBit(TPad::kCannotMove);
  this->Pad()->SetFillColor(kWhite);
  this->Pad()->SetLeftMargin(0.10);
  this->Pad()->SetRightMargin(0.025);
  this->Pad()->SetTopMargin(0.025);
  this->Pad()->SetBottomMargin(0.10);
  this->Pad()->Draw();

  dedx_range_pro = 0;
  dedx_range_ka = 0;
  dedx_range_pi = 0;
  dedx_range_mu = 0;
  ke_range_pro = 0;
  ke_range_ka = 0;
  ke_range_pi = 0;
  ke_range_mu = 0;

  fView = new evdb::View2D();
}

//......................................................................
// Destructor.
evd::CalorPad::~CalorPad()
{
  if (dedx_range_pro) {
    delete dedx_range_pro;
    dedx_range_pro = 0;
  }
  if (dedx_range_ka) {
    delete dedx_range_ka;
    dedx_range_ka = 0;
  }
  if (dedx_range_pi) {
    delete dedx_range_pi;
    dedx_range_pi = 0;
  }
  if (dedx_range_mu) {
    delete dedx_range_mu;
    dedx_range_mu = 0;
  }
  if (ke_range_pro) {
    delete ke_range_pro;
    ke_range_pro = 0;
  }
  if (ke_range_ka) {
    delete ke_range_ka;
    ke_range_ka = 0;
  }
  if (ke_range_pi) {
    delete ke_range_pi;
    ke_range_pi = 0;
  }
  if (ke_range_mu) {
    delete ke_range_mu;
    ke_range_mu = 0;
  }
  if (fView) {
    delete fView;
    fView = 0;
  }
}

//......................................................................
// Draw selected objects.

void evd::CalorPad::Draw(const char* /*opt*/)
{

  this->Pad()->cd();

  //Remove all previous objects from Pad's primitive list
  this->Pad()->Clear();

  //Remove all previous TPolyMarkers, TLatexs, etc... from list of such objects
  fView->Clear();

  //Draw coordinate axis and also GEANT based dE/dx vs. Range, or KE vs. Range, curves.
  DrawRefCurves();

  // grab the event from the singleton
  const art::Event* evt = evdb::EventHolder::Instance()->GetEvent();

  // Insert graphic objects into fView collection.
  if (evt) {
    try {
      if (fcurvetype == 1)
        AnalysisBaseDraw()->DrawDeDx(*evt, fView);
      else if (fcurvetype == 0)
        AnalysisBaseDraw()->DrawKineticEnergy(*evt, fView);
      else if (fcurvetype == 2)
        AnalysisBaseDraw()->CalorShower(*evt, fView);
    }
    catch (cet::exception const& e) {
      if (fcurvetype == 1)
        writeErrMsg("Draw->DrawDeDx", e);
      else if (fcurvetype == 0)
        writeErrMsg("Draw->DrawKineticEnergy", e);
      else if (fcurvetype == 2)
        writeErrMsg("Draw->CalorShower", e);
    }
  }

  // Draw objects on pad.
  fView->Draw();
  fPad->Modified();
  fPad->Update();
}

//......................................................................
// Draw truth curves

void evd::CalorPad::DrawRefCurves()
{

  if (dedx_range_pro) {
    delete dedx_range_pro;
    dedx_range_pro = 0;
  }
  if (dedx_range_ka) {
    delete dedx_range_ka;
    dedx_range_ka = 0;
  }
  if (dedx_range_pi) {
    delete dedx_range_pi;
    dedx_range_pi = 0;
  }
  if (dedx_range_mu) {
    delete dedx_range_mu;
    dedx_range_mu = 0;
  }
  if (ke_range_pro) {
    delete ke_range_pro;
    ke_range_pro = 0;
  }
  if (ke_range_ka) {
    delete ke_range_ka;
    ke_range_ka = 0;
  }
  if (ke_range_pi) {
    delete ke_range_pi;
    ke_range_pi = 0;
  }
  if (ke_range_mu) {
    delete ke_range_mu;
    ke_range_mu = 0;
  }

  double ymax;
  if (fcurvetype == 1)
    ymax = 50.0;
  else
    ymax = 200.0;
  TH1F* h = this->Pad()->DrawFrame(0.0, 0.0, 25.0, ymax);
  h->GetXaxis()->SetLabelSize(0.04);
  h->GetXaxis()->SetTitleSize(0.04);
  h->GetXaxis()->CenterTitle();
  h->GetYaxis()->SetLabelSize(0.04);
  h->GetYaxis()->SetTitleSize(0.04);
  h->GetYaxis()->CenterTitle();

  if (fcurvetype == 1) {
    h->GetXaxis()->SetTitle("Residual Range (cm)");
    h->GetYaxis()->SetTitle("dE/dx (MeV/cm)");
  }
  else {
    h->GetXaxis()->SetTitle("Total Range (cm)");
    h->GetYaxis()->SetTitle("T (MeV)");
  }

  art::ServiceHandle<evd::AnalysisDrawingOptions const> anaOpt;

  cet::search_path sp("FW_SEARCH_PATH");
  if (!sp.find_file(anaOpt->fCalorTemplateFileName + ".root", fROOTfile))
    throw cet::exception("Chi2ParticleID")
      << "cannot find the root template file: \n"
      << anaOpt->fCalorTemplateFileName << "\n bail ungracefully.\n";

  TFile* file = TFile::Open(fROOTfile.c_str());
  if (fcurvetype == 1) {
    dedx_range_pro = (TGraph*)file->Get("dedx_range_pro");
    dedx_range_ka = (TGraph*)file->Get("dedx_range_ka");
    dedx_range_pi = (TGraph*)file->Get("dedx_range_pi");
    dedx_range_mu = (TGraph*)file->Get("dedx_range_mu");

    dedx_range_pro->SetMarkerStyle(7);
    dedx_range_ka->SetMarkerStyle(7);
    dedx_range_pi->SetMarkerStyle(7);
    dedx_range_mu->SetMarkerStyle(7);

    dedx_range_pro->SetMarkerColor(kBlack);
    dedx_range_ka->SetMarkerColor(kGray + 2);
    dedx_range_pi->SetMarkerColor(kGray + 1);
    dedx_range_mu->SetMarkerColor(kGray);

    dedx_range_mu->Draw("P,same");
    dedx_range_pi->Draw("P,same");
    dedx_range_ka->Draw("P,same");
    dedx_range_pro->Draw("P,same");
  }
  else {
    ke_range_pro = (TGraph*)file->Get("kinen_range_pro");
    ke_range_ka = (TGraph*)file->Get("kinen_range_ka");
    ke_range_pi = (TGraph*)file->Get("kinen_range_pi");
    ke_range_mu = (TGraph*)file->Get("kinen_range_mu");

    ke_range_pro->SetMarkerStyle(7);
    ke_range_ka->SetMarkerStyle(7);
    ke_range_pi->SetMarkerStyle(7);
    ke_range_mu->SetMarkerStyle(7);

    ke_range_pro->SetMarkerColor(kBlack);
    ke_range_ka->SetMarkerColor(kGray + 2);
    ke_range_pi->SetMarkerColor(kGray + 1);
    ke_range_mu->SetMarkerColor(kGray);

    ke_range_mu->Draw("P,same");
    ke_range_pi->Draw("P,same");
    ke_range_ka->Draw("P,same");
    ke_range_pro->Draw("P,same");
  }
  file->Close();
}