FADCTraceView.cc

Go to the documentation of this file.

#include <sevt/PMT.h>
#include <utl/Accumulator.h>

#include "AddText.h"
#include "FADCTraceView.h"

#include <TCanvas.h>
#include <TH1D.h>
#include <TGraph.h>
#include <TGraphErrors.h>
#include <TArrow.h>
#include <TBox.h>

#include <boost/format.hpp>

#include <algorithm>
#include <sstream>

using namespace utl;
using namespace sevt;
using namespace std;
using namespace boost;
using namespace SdCalibPlotterOG;


void
FADCTraceView::DividePage(TCanvas& c)
{
  const double xmargin = 0.003;
  const double ymargin = 0.003;
  const double vertDivision = 0.2;

  TPad* const savePad = (TPad*)gPad;
  c.cd();
  const string name = c.GetName();
  ostringstream os;
  int n = 1;
  os.str(""); os << name << "_1";
  const double x1 = xmargin;
  const double x2 = vertDivision - 0.5*xmargin;
  const double x3 = vertDivision + 0.5*xmargin;
  const double x4 = 1 - xmargin;
  const double y11 = ymargin;
  const double y12 = 1 - ymargin;
  TPad* const pad1 = new TPad(os.str().c_str(), os.str().c_str(), x1, y11, x2, y12, 0);
  pad1->SetNumber(n++);
  pad1->Draw();
  const int nRows = 6;
  for (int row = 0; row < nRows; ++row) {
    const double y1 = 1 - (row+1)*(1 - ymargin)/nRows;
    const double y2 = 1 - ymargin - row*(1 - ymargin)/nRows;
    os.str(""); os << name << '_' << n;
    TPad* const pad = new TPad(os.str().c_str(), os.str().c_str(), x3, y1, x4, y2, 0);
    pad->SetNumber(n++);
    pad->SetLeftMargin(0.5*pad->GetLeftMargin());
    pad->SetRightMargin(0);
    pad->Draw();
  }
  savePad->cd();
}


void
FADCTraceView::Draw(const PMT& pmt)
{
  // raw trace
  if (!pmt.HasFADCTrace()) {
    Missing("No FADC Trace");
    return;
  }

  const auto& trace = pmt.GetFADCTrace(fGain);
  const auto traceLength = trace.GetSize();
  ostringstream os;
  os << "PMT" << pmt.GetId() << '_'
     << (fGain == sdet::PMTConstants::eLowGain ? "LowGain" : "HighGain")
     << "_Trace";
  auto& th = Add(new TH1D(os.str().c_str(), "", traceLength, 0, traceLength));
  for (unsigned int i = 0; i < traceLength; ++i)
    th.SetBinContent(i+1, trace[i]);
  th.SetStats(false);
  th.SetLineColor(kRed);

  if (!pmt.HasRecData()) {
    Missing("No PMT Rec Data");
    return;
  }

  const auto& pmtRec = pmt.GetRecData();

  // baseline
  TH1D* bh = nullptr;
  if (!pmtRec.HasFADCBaseline(fGain))
    Missing("No FADC Baseline");
  else {
    const auto& baseline = pmtRec.GetFADCBaseline(fGain);
    bh = &Add(new TH1D((os.str() + "2").c_str(), os.str().c_str(), traceLength, 0, traceLength));
    for (unsigned int i = 0; i < traceLength; ++i)
      bh->SetBinContent(i+1, baseline[i]);
    bh->SetStats(false);
    bh->SetLineColor(kBlue);
    bh->GetYaxis()->SetTitle("[FADC]");
    bh->GetYaxis()->SetTitleOffset(0.23);
    bh->GetYaxis()->SetTickLength(0.01);
    bh->Draw();
  }

  // min/max
  const auto& signals = pmtRec.GetSignals();
  const bool plotSignals =
    (fGain == sdet::PMTConstants::eHighGain || fGain == pmtRec.GetGainUsed());
  if (plotSignals) {
    for (unsigned int s = 0, ns = signals.size(); s < ns; ++s) {
      Accumulator::MinMax<int> mm(trace[signals[s].fStart]);
      for (int i = signals[s].fStart+1, n = signals[s].fStop; i < n; ++i)
        mm(trace[i]);
      const double diff = 0.075*(mm.GetMax() - mm.GetMin());
      const double newMin = mm.GetMin() - diff;
      const double newMax = mm.GetMax() + diff;
      if (th.GetMinimum() > newMin)
        th.SetMinimum(newMin);
      if (th.GetMaximum() < newMax)
        th.SetMaximum(newMax);
    }
  }

  if (bh) {
    bh->SetMinimum(th.GetMinimum());
    bh->SetMaximum(th.GetMaximum());
  }

  if (plotSignals) {

    // raw signals
    const auto& rawSignals = pmtRec.GetRawSignals();
    for (unsigned int s = 0, ns = rawSignals.size(); s < ns; ++s) {
      Accumulator::MinMax<int> mm(trace[rawSignals[s].fStart]);
      for (int i = rawSignals[s].fStart+1, n = rawSignals[s].fStop; i < n; ++i)
        mm(trace[i]);
      const double diff = 0.025*(mm.GetMax() - mm.GetMin());
      auto& sb = Add(new TBox(rawSignals[s].fStart, mm.GetMin() - diff,
                              rawSignals[s].fStop, mm.GetMax() + diff));
      sb.SetFillStyle(0);
      sb.SetLineColor(kGreen);
      sb.Draw();
    }

    // signals
    for (unsigned int s = 0, ns = signals.size(); s < ns; ++s) {
      Accumulator::MinMax<int> mm(trace[signals[s].fStart]);
      for (int i = signals[s].fStart+1, n = signals[s].fStop; i < n; ++i)
        mm(trace[i]);
      const int diff = mm.GetMax() - mm.GetMin();
      const double eps = 0.05*diff;
      auto& sb = Add(new TBox(signals[s].fStart, mm.GetMin() - eps,
                              signals[s].fStop, mm.GetMax() + eps));
      sb.SetFillStyle(0);
      sb.SetLineColor(8); // dark green
      sb.Draw();
      const double plotSpan = th.GetMaximum() - th.GetMinimum();
      if (diff < 0.3*plotSpan) {
        const double x = 0.5*(signals[s].fStop + signals[s].fStart);
        auto& a = Add(new TArrow(x, mm.GetMax() + 0.4*plotSpan,
                                 x, mm.GetMax() + 0.1*plotSpan, 0.01));
        a.SetAngle(20);
        a.SetLineColor(8); // dark green
        a.Draw();
      }
    }

  }

  th.Draw("same");

  // gain used
  if (fGain == pmtRec.GetGainUsed()) {
    auto& pt = Add(new TPaveText(0.95,0.77,0.99,0.87, "ndc"));
    pt.SetBorderSize(2);
    pt.SetTextAlign(12);
    pt.SetFillColor(kWhite);
    pt.AddText("Used");
    pt.Draw();
  }

  // sturated bins
  if (pmtRec.GetFADCSaturatedBins(fGain)) {
    auto& pt = Add(new TPaveText(0.94,0.62,0.99,0.74, "ndc"));
    pt.SetBorderSize(2);
    pt.SetTextAlign(12);
    pt.SetFillColor(kCyan);
    pt.AddText((format("N_{sat} = %d ") % pmtRec.GetFADCSaturatedBins(fGain)).str().c_str());
    pt.Draw();
  }

  if (pmt.HasCalibData()) {

    const auto& pmtCalib = pmt.GetCalibData();

    // tube not OK
    if (!pmtCalib.IsTubeOk()) {
      auto& pt = Add(new TPaveText(0.69,0.73,0.8,0.87, "ndc"));
      pt.SetBorderSize(2);
      pt.SetTextAlign(12);
      pt.SetFillColor(kRed);
      pt.AddText("Tube not OK");
      pt.Draw();
    }

    // D/A ratio
    if (fGain == sdet::PMTConstants::eLowGain) {
      auto& pt = Add(new TPaveText(0.92,0.90,0.99,1, "ndc"));
      pt.SetBorderSize(2);
      pt.SetTextAlign(12);
      pt.SetFillColor(kWhite);
      pt.AddText((format("D/A = %.2f (%.2f)") % pmtRec.GetGainRatio() % pmtCalib.GetGainRatio()).str().c_str());
      pt.Draw();
    }

  }
}


void
FADCTraceView::Missing(const string& what)
{
  auto& pt = Add(new TPaveText(0.3,0.4,0.7,0.6, "ndc"));
  pt.SetBorderSize(2);
  pt.SetFillColor(kRed);
  pt.SetTextAlign(12);
  pt.AddText(what.c_str());
  pt.Draw();
}