MillePedeMonitor.h

Go to the documentation of this file.

#ifndef MILLEPEDEMONITOR_H
#define MILLEPEDEMONITOR_H


#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"

#include "Alignment/ReferenceTrajectories/interface/ReferenceTrajectoryBase.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"

#include <vector>
#include <array>

#include <TString.h>
#include <TH1.h>
#include <TH2.h>

class TH1;
class TH2;
class TDirectory;
class Trajectory;

class TrackerTopology;

namespace reco {
  class Track;
}

class Alignable;
class AlignableDetOrUnitPtr;
class TrajectoryStateOnSurface;

/***************************************
****************************************/
class MillePedeMonitor
{
 public:
  // book histograms in constructor
  MillePedeMonitor(const TrackerTopology* tTopo,const char *rootFile = "trackMonitor.root");
  MillePedeMonitor(TDirectory *rootDir, const TrackerTopology* tTopo);
  // writes histograms in destructor
  ~MillePedeMonitor(); // non-virtual destructor: not intended to be parent class

  void fillTrack(const reco::Track *track);//, const Trajectory *traj);
  void fillUsedTrack(const reco::Track *track, unsigned int nHitX, unsigned int nHitY);//, const Trajectory *traj);
  void fillRefTrajectory(const ReferenceTrajectoryBase::ReferenceTrajectoryPtr &refTrajPtr);
  void fillDerivatives(const TransientTrackingRecHit::ConstRecHitPointer &recHit,
               const float *localDerivs, unsigned int nLocal,
               const float *globalDerivs, unsigned int nGlobal, const int *labels);
  void fillResiduals(const TransientTrackingRecHit::ConstRecHitPointer &recHit,
             const TrajectoryStateOnSurface &tsos, unsigned int nHit,
             float residuum, float sigma, bool isY);
  void fillFrameToFrame(const AlignableDetOrUnitPtr &aliDet, const Alignable *ali);

  void fillCorrelations2D(float corr, const TransientTrackingRecHit::ConstRecHitPointer &hit);
  
  void fillPxbSurveyHistsChi2(const float &chi2);
  void fillPxbSurveyHistsLocalPars(const float &a0, const float &a1, const float &S, const float &phi);

 private:
  bool init(TDirectory *directory);
  bool equidistLogBins(double* bins, int nBins, double first, double last) const;
  void fillResidualHists(const std::vector<TH1*> &hists, float phiSensToNorm,
             float residuum, float sigma);
  void fillResidualHitHists(const std::vector<TH1*> &hists, float angle,
                float residuum, float sigma, unsigned int nHit);
  void fillTrack(const reco::Track *track, std::vector<TH1*> &trackHists1D,
         std::vector<TH2*> &trackHists2D);

  template <class OBJECT_TYPE>  
    int GetIndex(const std::vector<OBJECT_TYPE*> &vec, const TString &name);
  template <class OBJECT_TYPE>  
  std::vector<OBJECT_TYPE*> cloneHists(const std::vector<OBJECT_TYPE*> &orgs,
                       const TString &namAd, const TString &titAd) const;
  template <class OBJECT_TYPE>  
  void addToDirectory(const std::vector<OBJECT_TYPE*> &objs, TDirectory *dir) const;
  template <typename T, size_t SIZE>
  std::array<int,SIZE> indexArray1D(const std::vector<T*>& hists, const char* title);
  template <typename T, size_t SIZE>
  std::array<std::array<int,SIZE>,SIZE> indexArray2D(const std::vector<T*>& hists,
                                                     const char* title);

  TDirectory *myRootDir;
  bool        myDeleteDir; 

  std::vector<TH1*> myTrackHists1D; // all input tracks 
  std::vector<TH2*> myTrackHists2D;
  std::vector<TH1*> myUsedTrackHists1D; // tracks used, i.e. tranferred to pede
  std::vector<TH2*> myUsedTrackHists2D;
  std::vector<TH1*> myTrajectoryHists1D;
  std::vector<TH2*> myTrajectoryHists2D;
  std::vector<TH2*> myDerivHists2D;
  std::vector<TH2*> myResidHists2D;
  std::vector<std::vector<TH1*> > myResidHistsVec1DX;
  std::vector<std::vector<TH1*> > myResidHistsVec1DY;
  std::vector<TH1*> myResidHitHists1DX;
  std::vector<TH1*> myResidHitHists1DY;
  std::vector<TH2*> myFrame2FrameHists2D;
  std::vector<TH1*> myCorrHists; // correlations
  std::vector<TH1*> myPxbSurveyHists; // correlations

  const TrackerTopology* trackerTopology;
};

template <class OBJECT_TYPE>  
int MillePedeMonitor::GetIndex(const std::vector<OBJECT_TYPE*> &vec, const TString &name)
{
  int result = 0;
  for (typename std::vector<OBJECT_TYPE*>::const_iterator iter = vec.begin(), iterEnd = vec.end();
       iter != iterEnd; ++iter, ++result) {
    if (*iter && (*iter)->GetName() == name) return result;
  }
  edm::LogError("Alignment") << "@SUB=MillePedeMonitor::GetIndex" << " could not find " << name;
  return -1;
}

template <class OBJECT_TYPE>  
std::vector<OBJECT_TYPE*> MillePedeMonitor::cloneHists(const std::vector<OBJECT_TYPE*> &orgs,
                               const TString &namAd,
                               const TString &titAd) const
{
  // OBJECT_TYPE required to have methods Clone(const char*), GetName(), SetTitle(const char*) and GetTitle()
  std::vector<OBJECT_TYPE*> result;
  for (typename std::vector<OBJECT_TYPE*>::const_iterator iter = orgs.begin(), iterEnd = orgs.end();
       iter != iterEnd; ++iter) {
    if (!(*iter)) continue;
    result.push_back(static_cast<OBJECT_TYPE*>((*iter)->Clone(namAd + (*iter)->GetName())));
    if (result.back()) result.back()->SetTitle((*iter)->GetTitle() + titAd);
    else edm::LogError("Alignment") <<"@SUB=MillePedeMonitor::cloneHists"
                    << "out of memory?";
  }
  
  return result;
}


template <class OBJECT_TYPE>  
void MillePedeMonitor::addToDirectory(const std::vector<OBJECT_TYPE*> &obs,
                       TDirectory *dir) const
{
  // OBJECT_TYPE is required to have method SetDirectory(TDirectory *dir)
  for (typename std::vector<OBJECT_TYPE*>::const_iterator iter = obs.begin(), iterEnd = obs.end();
       iter != iterEnd; ++iter) {
    if (*iter) (*iter)->SetDirectory(dir);
  }
}

template <typename T, size_t SIZE>
std::array<int,SIZE>
MillePedeMonitor::indexArray1D(const std::vector<T*>& hists, const char* title) {
  std::array<int,SIZE> result{};
  for (size_t i = 0; i < SIZE; ++i) {
    result[i] = this->GetIndex(hists, Form(title, i));
  }
  return result;
}

template <typename T, size_t SIZE>
std::array<std::array<int,SIZE>,SIZE>
MillePedeMonitor::indexArray2D(const std::vector<T*>& hists, const char* title) {
  std::array<std::array<int,SIZE>,SIZE> result{};
  for (size_t i = 0; i < SIZE; ++i) {
    for (size_t j = 0; j < SIZE; ++j) {
      result[i][j] = this->GetIndex(hists, Form(title, i, j));
    }
  }
  return result;
}

#endif