EcalAlignment_PayloadInspector.cc

Go to the documentation of this file.

#include "CondCore/Utilities/interface/PayloadInspectorModule.h"
#include "CondCore/Utilities/interface/PayloadInspector.h"
#include "CondCore/CondDB/interface/Time.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "CondCore/EcalPlugins/plugins/EcalDrawUtils.h"

// the data format of the condition to be inspected
#include "CondFormats/Alignment/interface/Alignments.h"

#include "TH2F.h" // a 2-D histogram with four bytes per cell (float)
#include "TCanvas.h"
#include "TLine.h"
#include "TStyle.h"
#include "TLatex.h"//write mathematical equations.
#include "TPave.h"
#include "TPaveStats.h"
#include <string>
#include <fstream>

namespace {
enum {
  kEBChannels = 61200, kEEChannels = 14648
};
enum {MIN_IETA = 1, MIN_IPHI = 1, MAX_IETA = 85, MAX_IPHI = 360};// barrel (EB) lower and upper bounds on eta and phi
enum {IX_MIN = 1, IY_MIN = 1, IX_MAX = 100, IY_MAX = 100};// endcaps (EE) lower and upper bounds on x and y

/*****************************************
 2d plot of ECAL Alignment of 1 IOV
 ******************************************/
class EcalAlignmentPlot: public cond::payloadInspector::PlotImage<Alignments> {
public:
  EcalAlignmentPlot() :
      cond::payloadInspector::PlotImage<Alignments>("ECAL Alignment - map ") {
    setSingleIov(true);
  }

  bool fill(const std::vector<std::tuple<cond::Time_t, cond::Hash> >& iovs)override {
    auto iov = iovs.front(); //get reference to 1st element in the vector iovs
    std::shared_ptr < Alignments > payload = fetchPayload(std::get < 1 > (iov)); //std::get<1>(iov) refers to the Hash in the tuple iov
    unsigned int run = std::get < 0 > (iov);  //referes to Time_t in iov.
    TH2F* align;  //pointer to align which is a 2D histogram
    std::string subdet;
    int NbRows;
    if (payload.get()) { //payload is an iov retrieved from payload using hash.
      NbRows = (*payload).m_align.size();
      if (NbRows == 36)
        subdet = "EB";
      else if (NbRows == 4)
        subdet = "EE";
      else if (NbRows == 8)
        subdet = "ES";
      else
        subdet = "unknown";
      //  align = new TH2F("Align",Form("Alignment %s", subdet.c_str()),6, 0, 6, NbRows, 0, NbRows);
      align =new TH2F("Align","x           y            z               Phi         Theta         Psi",
              6, 0, 6, NbRows, 0, NbRows);

      double row = NbRows - 0.5;
      for (std::vector<AlignTransform>::const_iterator it =
          (*payload).m_align.begin(); it != (*payload).m_align.end();it++) {

        align->Fill(0.5, row, (*it).translation().x());
        align->Fill(1.5, row, (*it).translation().y());
        align->Fill(2.5, row, (*it).translation().z());
        align->Fill(3.5, row, (*it).rotation().getPhi());
        align->Fill(4.5, row, (*it).rotation().getTheta());
        align->Fill(5.5, row, (*it).rotation().getPsi());
        row = row - 1.;
      }
    }   // if payload.get()
    else
      return false;

    gStyle->SetPalette(1);
    gStyle->SetOptStat(0);
    TCanvas canvas("CC map", "CC map", 1000, 1000);
    TLatex t1;
    t1.SetNDC();
    t1.SetTextAlign(26);
    t1.SetTextSize(0.05);
    t1.SetTextColor(2);
    t1.DrawLatex(0.5, 0.96,Form("Ecal %s Alignment, IOV %i", subdet.c_str(), run));
    //      t1.SetTextSize(0.03);
    //      t1.DrawLatex(0.3, 0.94, "x          y           z           Phi          Theta          Psi");

    TPad* pad = new TPad("pad", "pad", 0.0, 0.0, 1.0, 0.94);
    pad->Draw();
    pad->cd();
    align->Draw("TEXT");
    TLine* l = new TLine;
    l->SetLineWidth(1);
    for (int i = 1; i < NbRows; i++) {
      double y = (double) i;
      l = new TLine(0., y, 6., y);
      l->Draw();
    }

    for (int i = 1; i < 6; i++) {
      double x = (double) i;
      double y = (double) NbRows;
      l = new TLine(x, 0., x, y);
      l->Draw();
    }

    align->GetXaxis()->SetTickLength(0.);
    align->GetXaxis()->SetLabelSize(0.);
    align->GetYaxis()->SetTickLength(0.);
    align->GetYaxis()->SetLabelSize(0.);

    std::string ImageName(m_imageFileName);
    canvas.SaveAs(ImageName.c_str());
    return true;
  }      // fill method
};

/*********************************************************
 2d plot of ECAL Alignment difference between 2 IOVs
 **********************************************************/
class EcalAlignmentDiff: public cond::payloadInspector::PlotImage<Alignments> {

public:
  EcalAlignmentDiff() :
      cond::payloadInspector::PlotImage<Alignments>("ECAL Alignment difference") {
    setSingleIov(false);
  }

  bool fill(const std::vector<std::tuple<cond::Time_t, cond::Hash> >& iovs)override {

    unsigned int run[2], irun = 0;
    float val[6][36];
    TH2F* align = new TH2F("", "", 1, 0., 1., 1, 0., 1.); // pseudo creation
    std::string subdet;
    int NbRows = 0;

    for (auto const & iov : iovs) {
      std::shared_ptr < Alignments > payload = fetchPayload(std::get < 1 > (iov));
      run[irun] = std::get < 0 > (iov);

      if (payload.get()) {
        NbRows = (*payload).m_align.size();
        if (irun == 1) {
          if (NbRows == 36)
            subdet = "EB";
          else if (NbRows == 4)
            subdet = "EE";
          else if (NbRows == 8)
            subdet = "ES";
          else
            subdet = "unknown";
          delete align;
          align =
              new TH2F("Align","x           y            z               Phi         Theta         Psi",
                  6, 0, 6, NbRows, 0, NbRows);
        }

        double row = NbRows - 0.5;
        int irow = 0;
        for (std::vector<AlignTransform>::const_iterator it =(*payload).m_align.begin();
            it != (*payload).m_align.end(); it++) {

          if (irun == 0) {
            val[0][irow] = (*it).translation().x();
            val[1][irow] = (*it).translation().y();
            val[2][irow] = (*it).translation().z();
            val[3][irow] = (*it).rotation().getPhi();
            val[4][irow] = (*it).rotation().getTheta();
            val[5][irow] = (*it).rotation().getPsi();
          } else {
            align->Fill(0.5, row,(*it).translation().x() - val[0][irow]);
            align->Fill(1.5, row,(*it).translation().y() - val[1][irow]);
            align->Fill(2.5, row,(*it).translation().z() - val[2][irow]);
            align->Fill(3.5, row,(*it).rotation().getPhi() - val[3][irow]);
            align->Fill(4.5, row,(*it).rotation().getTheta() - val[3][irow]);
            align->Fill(5.5, row,(*it).rotation().getPsi() - val[5][irow]);
            row = row - 1.;
          }
          irow++;
        } // loop over alignment rows

      }  //  if payload.get()
      else
        return false;
      irun++;
    }      // loop over IOVs

    gStyle->SetPalette(1);
    gStyle->SetOptStat(0);
    TCanvas canvas("CC map", "CC map", 1000, 1000);
    TLatex t1;
    t1.SetNDC();
    t1.SetTextAlign(26);
    t1.SetTextSize(0.05);
    t1.SetTextColor(2);
    t1.DrawLatex(0.5, 0.96,Form("Ecal %s Alignment, IOV %i - %i", subdet.c_str(), run[1],run[0]));

    TPad* pad = new TPad("pad", "pad", 0.0, 0.0, 1.0, 0.94);
    pad->Draw();
    pad->cd();
    align->Draw("TEXT");
    TLine* l = new TLine;
    l->SetLineWidth(1);

    for (int i = 1; i < NbRows; i++) {
      double y = (double) i;
      l = new TLine(0., y, 6., y);
      l->Draw();
    }

    for (int i = 1; i < 6; i++) {
      double x = (double) i;
      double y = (double) NbRows;
      l = new TLine(x, 0., x, y);
      l->Draw();
    }
    
    align->GetXaxis()->SetTickLength(0.);
    align->GetXaxis()->SetLabelSize(0.);
    align->GetYaxis()->SetTickLength(0.);
    align->GetYaxis()->SetLabelSize(0.);

    std::string ImageName(m_imageFileName);
    canvas.SaveAs(ImageName.c_str());
    return true;
  }      // fill method
};

} // close namespace

// Register the classes as boost python plugin
PAYLOAD_INSPECTOR_MODULE(EcalAlignment) {
  PAYLOAD_INSPECTOR_CLASS(EcalAlignmentPlot);
  PAYLOAD_INSPECTOR_CLASS (EcalAlignmentDiff);
}