CDCDedxCorrectionModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2012 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jake Bennett
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef CDCDEDXCORRECTIONMODULE_H
#define CDCDEDXCORRECTIONMODULE_H
 
#include <framework/core/Module.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
 
#include <reconstruction/dbobjects/CDCDedxScaleFactor.h>
#include <reconstruction/dbobjects/CDCDedxMomentumCor.h>
#include <reconstruction/dbobjects/CDCDedxWireGain.h>
#include <reconstruction/dbobjects/CDCDedxRunGain.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> //new in rel5
#include <reconstruction/dbobjects/CDCDedxCosineEdge.h> //new in rel5
 
#include <vector>
 
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 WireGainCorrection(int wireID, double& dedx) 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& dedx) const;
 
    double GetCorrection(int& adc, int layer, int wireID, double doca, double enta, double costheta) 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_twoDCell; 
    bool m_oneDCell; 
    bool m_cosineEdge; 
    bool m_nonlADC; 
    StoreArray<CDCDedxTrack> m_cdcDedxTracks; 
    //parameters: calibration constants
    DBObjPtr<CDCDedxScaleFactor> m_DBScaleFactor; 
    DBObjPtr<CDCDedxMomentumCor> m_DBMomentumCor; 
    DBObjPtr<CDCDedxWireGain> m_DBWireGains; 
    DBObjPtr<CDCDedxRunGain> m_DBRunGain; 
    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; 
    void calculateMeans(double* mean, double* truncatedMean, double* truncatedMeanErr, const std::vector<double>& dedx) const;
 
    double m_removeLowest;
    double m_removeHighest;
 
    double  m_alpha{ -1.};
    double  m_gamma{ -1.};
    double  m_delta{ -1.};
    double  m_power{ -1.};
    double  m_ratio{ -1.};
 
  };
} // Belle2 namespace
#endif