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"
#include "TCanvas.h"
#include "TLine.h"
#include "TStyle.h"
#include "TLatex.h"
#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 lower and upper bounds on eta and phi
  enum {IX_MIN = 1, IY_MIN = 1, IX_MAX = 100, IY_MAX = 100};         // endcaps 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();
      std::shared_ptr<Alignments> payload = fetchPayload( std::get<1>(iov) );
      unsigned int run = std::get<0>(iov);
      TH2F* align;
      std::string subdet;
      int NbRows;
      if(payload.get()) {
    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);
}