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() override;

    virtual void initialize() override;

    virtual void event() 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);
    static float CalculateChi2();
 
    //calibration objects
    SVDPulseShapeCalibrations m_PulseShapeCal; 
    SVDNoiseCalibrations m_NoiseCal; 
    SVDCoGTimeCalibrations m_TimeCal; 

    int m_NumberOfAPVSamples = 6;
 
  };

}