V0FinderModule.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
//Object with performing the actual algorithm:
#include <tracking/v0Finding/fitter/V0Fitter.h>
#include <tracking/v0Finding/fitter/NewV0Fitter.h>
 
#include <mdst/dataobjects/Track.h>
 
#include <framework/datastore/StoreArray.h>
#include <framework/core/Module.h>
 
#include <string>
#include <memory>
 
namespace Belle2 {
  class V0FinderModule : public Module {
 
  public:
 
    V0FinderModule();
 
    ~V0FinderModule() override = default;
 
    void initialize() override;
 
    void event() override;
 
    void terminate() override;
 
  private:
 
    std::string m_arrayNameTrack;     
    StoreArray<Track> m_tracks;       
 
    std::unique_ptr<V0Fitter> m_v0Fitter;  
    std::unique_ptr<NewV0Fitter> m_newV0Fitter;  
    std::string m_arrayNameRecoTrack;      
    std::string m_arrayNameCopiedRecoTrack;
    std::string m_arrayNameTFResult;       
    std::string m_arrayNameV0;             
 
    double m_beamPipeRadius;          
    double m_vertexChi2CutOutside;    
    int    m_v0FitterMode;            
    double m_precutRho;      
    double m_precutCosAlpha; 
    bool   m_useNewV0Fitter; 
    bool   m_useValidation;  
 
    std::tuple<double, double> m_invMassRangeKshort = { 0.425, 0.575 };
    std::tuple<double, double> m_invMassRangeLambda = { 1.09, 1.14 };
    std::tuple<double, double> m_invMassRangePhoton = { 0, 0.1 };
 
    std::string m_arrayNameV0ValidationVertex; 
 
    std::tuple<double, double> m_preFilterMassRangeKshort = { 0.45, Const::K0Mass + 0.015 };
    std::tuple<double, double> m_preFilterMassRangeLambda = { Const::Lambda.getMass() - 0.03, Const::Lambda.getMass() + 0.03 };
 
    bool isTrackPairSelected(const Track* track1, const Track* track2);
 
    bool preFilterTracks(const Track* trackPlus, const Track* trackMinus, const Const::ParticleType& v0Hypothesis);
 
    void fitAndStore(const Track* trackPlus, const Track* trackMinus, const Const::ParticleType& v0Hypothesis);
 
    // buffer some variables to speed up time, actual values will be calculated at initialization
    double m_mKshortMin2 = 0; 
    double m_mKshortMax2 = 0; 
    double m_mLambdaMin2 = 0; 
    double m_mLambdaMax2 = 0; 
 
    // counter for V0Finder statistics
    int m_allStored = 0;    
    int m_nForceStored = 0; 
    int m_nHitRemoved  = 0; 
  };
}