CDCDedxPIDModule.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/gearbox/Const.h>
 
#include <cdc/dataobjects/CDCDedxTrack.h>
#include <cdc/dataobjects/CDCDedxLikelihood.h>
 
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/MCParticle.h>
 
#include <tracking/dataobjects/RecoTrack.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
 
#include <cdc/dbobjects/CDCDedxScaleFactor.h>
#include <cdc/dbobjects/CDCDedxWireGain.h>
#include <cdc/dbobjects/CDCDedxRunGain.h>
#include <cdc/dbobjects/CDCDedxCosineCor.h>
#include <cdc/dbobjects/CDCDedx2DCell.h>
#include <cdc/dbobjects/CDCDedx1DCell.h>
#include <cdc/dbobjects/CDCDedxADCNonLinearity.h>
#include <cdc/dbobjects/CDCDedxCosineEdge.h>
#include <cdc/dbobjects/CDCDedxMeanPars.h>
#include <cdc/dbobjects/CDCDedxSigmaPars.h>
#include <cdc/dbobjects/CDCDedxHadronCor.h>
#include <cdc/dbobjects/CDCDedxInjectionTime.h>
#include <cdc/dbobjects/CDCdEdxPDFs.h>
 
#include <vector>
 
 
namespace Belle2 {

  class CDCDedxPIDModule : public Module {
 
  public:

    CDCDedxPIDModule();

    virtual ~CDCDedxPIDModule();

    virtual void initialize() override;

    virtual void event() override;

    virtual void terminate() override;
 
  private:
 
    // outputs
    StoreArray<CDCDedxTrack> m_dedxTracks; 
    StoreArray<CDCDedxLikelihood> m_dedxLikelihoods; 
 
    // required inputs
    StoreArray<Track> m_tracks; 
    StoreArray<RecoTrack> m_recoTracks; 
 
    // optional inputs
    StoreArray<MCParticle> m_mcparticles; 

    double meanCurve(double* x, double* par, int version) const;

    double getMean(double bg) const;

    double sigmaCurve(double* x, const double* par, int version) const;

    double getSigma(double dedx, double nhit, double cos, double timereso) const;

    double I2D(double cosTheta, double I) const;

    double D2I(double cosTheta, double D) const;

    void calculateMeans(double* mean, double* truncatedMean, double* truncatedMeanErr, const std::vector<double>& dedx) const;

    void saveChiValue(double(&chi)[Const::ChargedStable::c_SetSize], double(&predmean)[Const::ChargedStable::c_SetSize],
                      double(&predres)[Const::ChargedStable::c_SetSize], double p, double dedx, double cos, int nhit, double timereso) const;
 

    void saveLookupLogl(double(&logl)[Const::ChargedStable::c_SetSize], double p, double dedx);

    void checkPDFs();
 
    // parameters to determine the predicted means and resolutions
    std::vector<double> m_meanpars; 
    std::vector<double> m_sigmapars; 
 
    // pdfs for PID
    DBObjPtr<CDCdEdxPDFs> m_DBDedxPDFs; 
 
    bool m_trackLevel; 
    bool m_usePrediction; 
    double m_removeLowest; 
    double m_removeHighest; 
    bool m_backHalfCurlers; 
    bool m_enableDebugOutput; 
 
    bool m_useIndividualHits; 
 
    bool m_onlyPrimaryParticles; 
    bool m_ignoreMissingParticles; 
 
    // parameters: calibration constants
    DBObjPtr<CDCDedxScaleFactor> m_DBScaleFactor; 
    DBObjPtr<CDCDedxRunGain> m_DBRunGain; 
    DBObjPtr<CDCDedxInjectionTime> m_DBInjectTime; 
    DBObjPtr<CDCDedxCosineCor> m_DBCosineCor; 
    DBObjPtr<CDCDedxCosineEdge> m_DBCosEdgeCor; 
    DBObjPtr<CDCDedxWireGain> m_DBWireGains; 
    DBObjPtr<CDCDedx2DCell> m_DB2DCell; 
    DBObjPtr<CDCDedx1DCell> m_DB1DCell; 
    DBObjPtr<CDCDedxADCNonLinearity> m_DBNonlADC; 
    DBObjPtr<CDCDedxHadronCor> m_DBHadronCor; 
 
    std::vector<double> m_hadronpars; 
 
    int m_nLayerWires[9] = {}; 
 
    // parameters to determine the predicted means and resolutions and hadron correction
    DBObjPtr<CDCDedxMeanPars> m_DBMeanPars; 
    DBObjPtr<CDCDedxSigmaPars> m_DBSigmaPars; 
 
  };

} // Belle2 namespace