SVDCoGTimeEstimatorModule.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 <framework/core/Module.h>
#include <framework/datastore/RelationArray.h>
#include <framework/datastore/StoreArray.h>
#include <svd/dataobjects/SVDEventInfo.h>
#include <svd/dataobjects/SVDShaperDigit.h>
#include <svd/dataobjects/SVDRecoDigit.h>
#include <svd/calibration/SVDPulseShapeCalibrations.h>
#include <svd/calibration/SVDNoiseCalibrations.h>
#include <svd/calibration/SVDCoGTimeCalibrations.h>
 
#include <mdst/dataobjects/MCParticle.h>
#include <svd/dataobjects/SVDTrueHit.h>
 
#include <string>
 
namespace Belle2 {
  class SVDCoGTimeEstimatorModule : public Module {
  public:
 
    typedef std::vector<const RelationElement*> RelationLookup;
 
    SVDCoGTimeEstimatorModule();
 
    virtual ~SVDCoGTimeEstimatorModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
 
  private:
 
    StoreArray<SVDShaperDigit> m_storeShaper; 
    StoreArray<SVDRecoDigit> m_storeReco; 
    StoreArray<SVDTrueHit> m_storeTrueHits; 
    StoreArray<MCParticle> m_storeMCParticles; 
    StoreObjPtr<SVDEventInfo> m_storeSVDEvtInfo; 
    float m_weightedMeanTime = 0;
    float m_weightedMeanTimeError = 0;
 
    float m_amplitude = 0;
    float m_amplitudeError = 0;
 
    float m_chi2 = 0;
 
    float m_DeltaT = 31.44; //ns
 
    bool m_StopCreationReco = false;
 
  protected:
 
    void createRelationLookup(const RelationArray& relation, RelationLookup& lookup,
                              size_t digits);
 
    void fillRelationMap(const RelationLookup& lookup, std::map<unsigned int, float>&
                         relation, unsigned int index);
 
    // Relation lookups
    RelationLookup m_mcRelation;
    RelationLookup m_trueRelation;
 
    std::string m_svdEventInfoName = "SVDEventInfo";
    std::string m_storeMCParticlesName = "MCParticles";
    std::string m_storeTrueHitsName = "SVDTrueHits";
    std::string m_storeShaperDigitsName = "SVDShaperDigits";
    std::string m_storeRecoDigitsName = "SVDRecoDigits";
    std::string m_relRecoDigitShaperDigitName = "";
 
    bool m_calEventT0 = true; 
    bool m_corrPeakTime = true; 
    std::string m_relShaperDigitMCParticleName = "";
    std::string m_relShaperDigitTrueHitName = "";
    std::string m_relRecoDigitMCParticleName = "";
    std::string m_relRecoDigitTrueHitName = "";
 
    float m_AmplitudeArbitraryError = 10;
 
    float CalculateWeightedMeanPeakTime(Belle2::SVDShaperDigit::APVFloatSamples samples);
    float CalculateAmplitude(Belle2::SVDShaperDigit::APVFloatSamples samples);
    float CalculateWeightedMeanPeakTimeError(Belle2::SVDShaperDigit::APVFloatSamples samples);
    float CalculateAmplitudeError(VxdID ThisSensorID, bool ThisSide, int ThisCellID);
    float CalculateChi2();
 
    //calibration objects
    SVDPulseShapeCalibrations m_PulseShapeCal; 
    SVDNoiseCalibrations m_NoiseCal; 
    SVDCoGTimeCalibrations m_TimeCal; 
    int m_NumberOfAPVSamples = 6;
 
  };
}