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.10 2009/09/15 17:09:49 pivarski 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(const edm::EventSetup &iSetup);

      void endOfJob();


      virtual void book() = 0;

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

      virtual void afterAlignment(const edm::EventSetup &iSetup) { };

   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; };

   private:
      AlignmentMonitorBase(const AlignmentMonitorBase&); // stop default
      const AlignmentMonitorBase& operator=(const AlignmentMonitorBase&); // 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;
};


#endif