DQMEventProcessorBase.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <tracking/dqmUtils/DQMHistoModuleBase.h>
 
#include <tracking/dataobjects/RecoHitInformation.h>
 
namespace Belle2 {
  class DQMEventProcessorBase {
 
  public:
    DQMEventProcessorBase(DQMHistoModuleBase* histoModule, const std::string& recoTracksStoreArrayName,
                          const std::string& tracksStoreArrayName, bool runningOnHLT = false) :
      m_tracksStoreArrayName(tracksStoreArrayName),
      m_recoTracksStoreArrayName(recoTracksStoreArrayName),
      m_runningOnHLT(runningOnHLT)
    {
      m_histoModule = histoModule;
    }
 
    virtual void Run();
 
    void produce1Dres() {m_produce1Dres = true;};
    void produce2Dres() {m_produce2Dres = true;};
    void produceTRClusters() {m_produceTRClusters = true;};
 
  protected:
    virtual void ProcessTrack(const Track& track);
    virtual TString ConstructMessage(const TrackFitResult* trackFitResult, int nPXDClusters, int nSVDClusters, int nCDCHits);
    virtual void FillTrackFitResult(const TrackFitResult* trackFitResult);
    virtual void ProcessSuccessfulFit();
    virtual void ProcessRecoHit(RecoHitInformation* recoHitInfo);
    virtual void ProcessPXDRecoHit(RecoHitInformation* recoHitInfo);
    virtual void ProcessSVDRecoHit(RecoHitInformation* recoHitInfo);
    virtual void ComputeCommonVariables();
    virtual void FillCommonHistograms();
    virtual void SetCommonPrevVariables();
 
    static bool IsNotYang(int ladderNumber, int layerNumber);
 
    static bool IsNotMat(int ladderNumber, int layerNumber);
 
    DQMHistoModuleBase* m_histoModule = nullptr;
    std::string m_tracksStoreArrayName = "";
    std::string m_recoTracksStoreArrayName = "";
 
    bool m_runningOnHLT;
 
    bool m_produce1Dres = false;
 
    bool m_produce2Dres = false;
 
    bool m_produceTRClusters = false;
 
    int m_iTrack = 0;
    int m_iTrackVXD = 0;
    int m_iTrackCDC = 0;
    int m_iTrackVXDCDC = 0;
 
    RecoTrack* m_recoTrack = nullptr;
    bool m_isNotFirstHit = false;
 
    TVectorT<double>* m_rawSensorResidual = nullptr;
    VxdID m_sensorID = VxdID(0);
    VxdID m_sensorIDPrev = VxdID(0);
 
    ROOT::Math::XYZVector m_position = ROOT::Math::XYZVector();
    ROOT::Math::XYZVector m_residual_um = ROOT::Math::XYZVector();
    ROOT::Math::XYZVector m_globalResidual_um = ROOT::Math::XYZVector();
    float m_phi_deg = .0;
    float m_phiPrev_deg = .0;
    float m_theta_deg = .0;
    float m_thetaPrev_deg = .0;
    int m_layerNumber = 0;
    int m_layerNumberPrev = 0;
    int m_layerIndex = 0;
    int m_correlationIndex = 0;
    int m_sensorIndex = 0;
  };
}