TOPPDFCheckerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2017 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/HistoModule.h>
#include <top/dataobjects/TOPDigit.h>
#include <mdst/dataobjects/MCParticle.h>
#include <set>
#include <TH2F.h>
#include <TVector3.h>
 
 
namespace Belle2 {
  class TOPPDFCheckerModule : public HistoModule {
 
  public:
 
    TOPPDFCheckerModule();
 
    virtual ~TOPPDFCheckerModule();
 
    virtual void defineHisto() override;
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    bool isFromThisParticle(const TOPDigit& digit, const MCParticle* particle);
 
    double m_minTime; 
    double m_maxTime; 
    int m_numBins;    
    TH2F* m_hits = 0; 
    TH2F* m_pdf = 0;  
    TH2F* m_hitsCol = 0; 
    TH2F* m_pdfCol = 0;  
    TVector3 m_avrgMomentum; 
    TVector3 m_avrgPosition; 
    int m_numTracks = 0;     
    std::set<int> m_slotIDs;      
    std::set<int> m_PDGCodes;     
  };
 
} // Belle2 namespace