AlignmentMonitorBase.h

Go to the documentation of this file.

#ifndef CommonAlignmentMonitor_AlignmentMonitorBase_h
#define CommonAlignmentMonitor_AlignmentMonitorBase_h
// -*- C++ -*-
//
// Package:     CommonAlignmentMonitor
// Class  :     AlignmentMonitorBase
// 
//
// Original Author:  Jim Pivarski
//         Created:  Fri Mar 30 12:21:02 CDT 2007
// $Id: AlignmentMonitorBase.h,v 1.11 2010/01/06 15:23:09 mussgill Exp $
//

// system include files

#include "TrackingTools/PatternTools/interface/Trajectory.h"
#include "Alignment/CommonAlignment/interface/AlignableNavigator.h"
#include "Alignment/TrackerAlignment/interface/AlignableTracker.h"
#include "Alignment/MuonAlignment/interface/AlignableMuon.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentParameterStore.h"
#include "DataFormats/TrackReco/interface/Track.h"

// user include files
#include "CommonTools/Utils/interface/TFileDirectory.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TProfile.h"
#include "TTree.h"

// forward declarations

class AlignmentMonitorBase
{
   public:
      typedef std::pair<const Trajectory*, const reco::Track*> ConstTrajTrackPair; 
      typedef std::vector<ConstTrajTrackPair>  ConstTrajTrackPairCollection;

      AlignmentMonitorBase(const edm::ParameterSet &cfg, std::string name);
      
      virtual ~AlignmentMonitorBase() {}

      void beginOfJob(AlignableTracker *pTracker, AlignableMuon *pMuon, AlignmentParameterStore *pStore);

      void startingNewLoop();

      void duringLoop(const edm::Event &iEvent, const edm::EventSetup &iSetup, const ConstTrajTrackPairCollection &iTrajTracks);

      void endOfLoop();

      void endOfJob() {}


      virtual void book() = 0;

      virtual void event(const edm::Event &iEvent, const edm::EventSetup &iSetup, const ConstTrajTrackPairCollection &iTrajTracks) { }

      virtual void afterAlignment() { }

   protected:
      TH1F *book1D(std::string dir, std::string name, std::string title, int nchX, double lowX, double highX);
      TProfile *bookProfile(std::string dir, std::string name, std::string title, int nchX, double lowX, double highX, int nchY=1, double lowY=0., double highY=0., const char *option="s");
      TH2F *book2D(std::string dir, std::string name, std::string title, int nchX, double lowX, double highX, int nchY, double lowY, double highY);
      TFileDirectory *directory(std::string dir);
      
      int                     iteration()    { return m_iteration; }
      AlignableTracker        *pTracker()    { return mp_tracker; }
      AlignableMuon           *pMuon()       { return mp_muon; }
      AlignmentParameterStore *pStore()      { return mp_store; }
      AlignableNavigator      *pNavigator()  { return mp_navigator; }

      const edm::InputTag m_beamSpotTag;

   private:
      AlignmentMonitorBase(const AlignmentMonitorBase&) = delete; // stop default
      const AlignmentMonitorBase& operator=(const AlignmentMonitorBase&) = delete; // stop default

      // ---------- member data --------------------------------

      int m_iteration;
      AlignableTracker         *mp_tracker;
      AlignableMuon            *mp_muon;
      AlignmentParameterStore  *mp_store;
      AlignableNavigator       *mp_navigator;

      std::map<std::vector<std::string>, TFileDirectory*> m_baseDirMap, m_iterDirMap;
};

/*** Global typedefs ***/
using AlignmentMonitors = std::vector<std::unique_ptr<AlignmentMonitorBase> >;

#endif