TOPValidationCollectorModule.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 <calibration/CalibrationCollectorModule.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
#include <top/dataobjects/TOPDigit.h>
#include <top/dataobjects/TOPRecBunch.h>
 
#include <framework/database/DBObjPtr.h>
#include <top/dbobjects/TOPCalTimebase.h>
#include <top/dbobjects/TOPCalChannelT0.h>
#include <top/dbobjects/TOPCalChannelMask.h>
#include <top/dbobjects/TOPCalChannelThresholdEff.h>
#include <top/dbobjects/TOPCalAsicShift.h>
#include <top/dbobjects/TOPCalEventT0Offset.h>
#include <top/dbobjects/TOPCalFillPatternOffset.h>
 
#include <top/utilities/TrackSelector.h>
#include <top/utilities/Chi2MinimumFinder1D.h>
#include <top/reconstruction_cpp/PDFConstructor.h>
#include <top/calibration/ValidationTreeStruct.h>
 
#include <string>
#include <vector>
 
namespace Belle2 {
  class TOPValidationCollectorModule : public CalibrationCollectorModule {
 
  public:
    TOPValidationCollectorModule();
 
  private:
 
    virtual void prepare() final;
 
    virtual void collect() final;
 
    virtual void startRun() final;
 
    virtual void closeRun() final;
 
    enum {c_numModules = TOP::ValidationTreeStruct::c_numModules,  
          c_numChannels = 512, 
          c_numSets = 2,  
         };
 
    // steering parameters
    int m_numBins;      
    double m_timeRange; 
    double m_sigmaSmear;  
    std::string m_sample; 
    double m_deltaEcms; 
    double m_dr; 
    double m_dz; 
    double m_minZ; 
    double m_maxZ; 
    std::string m_pdfOption;   
    // procedure
    TOP::TrackSelector m_selector; 
    TOP::PDFConstructor::EPDFOption m_PDFOption = TOP::PDFConstructor::c_Rough; 
    std::vector<std::string> m_namesChi; 
    std::vector<std::string> m_namesHit; 
    std::vector<TOP::Chi2MinimumFinder1D> m_finders[c_numSets]; 
    // collections
    StoreArray<TOPDigit> m_digits; 
    StoreArray<Track> m_tracks;    
    StoreArray<ExtHit> m_extHits;  
    StoreObjPtr<TOPRecBunch> m_recBunch; 
    // payloads
    DBObjPtr<TOPCalTimebase> m_timebase;   
    DBObjPtr<TOPCalChannelT0> m_channelT0; 
    DBObjPtr<TOPCalChannelMask> m_channelMask; 
    DBObjPtr<TOPCalChannelThresholdEff> m_thresholdEff; 
    DBObjPtr<TOPCalAsicShift> m_asicShift; 
    DBObjPtr<TOPCalEventT0Offset> m_eventT0Offset; 
    DBObjPtr<TOPCalFillPatternOffset> m_fillPatternOffset; 
    // tree variables
    TOP::ValidationTreeStruct m_treeEntry; 
  };
 
} // end namespace Belle2