CDCDedxCorrectionModule.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 <cmath>
#include <algorithm>
#include <TMath.h>
#include <vector>
 
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
 
#include <reconstruction/dataobjects/CDCDedxTrack.h>
#include <reconstruction/dataobjects/DedxConstants.h>
#include <reconstruction/dbobjects/CDCDedxScaleFactor.h>
#include <reconstruction/dbobjects/CDCDedxMomentumCor.h>
#include <reconstruction/dbobjects/CDCDedxWireGain.h>
#include <reconstruction/dbobjects/CDCDedxRunGain.h>
#include <reconstruction/dbobjects/CDCDedxInjectionTime.h>
#include <reconstruction/dbobjects/CDCDedxCosineCor.h>
#include <reconstruction/dbobjects/CDCDedx2DCell.h>
#include <reconstruction/dbobjects/CDCDedx1DCell.h>
#include <reconstruction/dbobjects/CDCDedxHadronCor.h>
#include <reconstruction/dbobjects/CDCDedxADCNonLinearity.h>
#include <reconstruction/dbobjects/CDCDedxCosineEdge.h>
 
#include <mdst/dataobjects/EventLevelTriggerTimeInfo.h>
 
#include <cdc/geometry/CDCGeometryParConstants.h>
#include <cdc/geometry/CDCGeometryPar.h>
 
namespace Belle2 {
  class CDCDedxTrack;
 
  class CDCDedxCorrectionModule : public Module {
 
  public:
 
    CDCDedxCorrectionModule();
    virtual ~CDCDedxCorrectionModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
    void RunGainCorrection(double& dedx) const;
 
    void TimeGainCorrection(double& dedx, double ring, double time) const;
 
    void WireGainCorrection(int wireID, double& dedx, int layer) const;
 
    void TwoDCorrection(int layer, double doca, double enta, double& dedx) const;
 
    void OneDCorrection(int layer, double enta, double& dedx) const;
 
    void CosineCorrection(double costheta, double& dedx) const;
 
    void CosineEdgeCorrection(double costh, double& dedx) const;
 
    void HadronCorrection(double costheta,  double& dedx) const;
 
    void StandardCorrection(int adc, int layer, int wireID, double doca, double enta, double length,
                            double costheta, double ring, double time, double& dedx) const;
 
    double GetCorrection(int& adc, int layer, int wireID, double doca, double enta, double costheta, double ring, double time) const;
 
    double D2I(const double cosTheta, const double D) const;
 
    double I2D(const double cosTheta, const double I) const;
 
  private:
 
    bool m_relative; 
    bool m_momCor; 
    bool m_useDBMomCor; 
    bool m_scaleCor; 
    bool m_cosineCor; 
    bool m_wireGain; 
    bool m_runGain; 
    bool m_timeGain; 
    bool m_twoDCell; 
    bool m_oneDCell; 
    bool m_cosineEdge; 
    bool m_nonlADC; 
    StoreArray<CDCDedxTrack> m_cdcDedxTracks; 
    StoreObjPtr<EventLevelTriggerTimeInfo> m_TTDInfo; 
    //parameters: calibration constants
    DBObjPtr<CDCDedxScaleFactor> m_DBScaleFactor; 
    DBObjPtr<CDCDedxMomentumCor> m_DBMomentumCor; 
    DBObjPtr<CDCDedxWireGain> m_DBWireGains; 
    DBObjPtr<CDCDedxRunGain> m_DBRunGain; 
    DBObjPtr<CDCDedxInjectionTime> m_DBInjectTime; 
    DBObjPtr<CDCDedxCosineCor> m_DBCosineCor; 
    DBObjPtr<CDCDedx2DCell> m_DB2DCell; 
    DBObjPtr<CDCDedx1DCell> m_DB1DCell; 
    DBObjPtr<CDCDedxHadronCor> m_DBHadronCor; 
    DBObjPtr<CDCDedxADCNonLinearity> m_DBNonlADC; 
    DBObjPtr<CDCDedxCosineEdge> m_DBCosEdgeCor; 
    std::vector<double> m_hadronpars; 
    std::array<double, 56> m_lgainavg; 
    void calculateMeans(double* mean, double* truncatedMean, double* truncatedMeanErr, const std::vector<double>& dedx) const;
 
    double m_removeLowest;
    double m_removeHighest;
 
  };
} // Belle2 namespace