ParticleVertexFitterModule.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 <string>
#include <analysis/DecayDescriptor/DecayDescriptor.h>
 
// framework - DataStore
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreObjPtr.h>
 
#include <framework/geometry/B2Vector3.h>
 
// DataObjects
#include <analysis/dataobjects/ParticleList.h>
#include <mdst/dbobjects/BeamSpot.h>
 
// KFit
#include <analysis/VertexFitting/KFit/MassFitKFit.h>
#include <analysis/VertexFitting/KFit/FourCFitKFit.h>
#include <analysis/VertexFitting/KFit/MassFourCFitKFit.h>
#include <analysis/VertexFitting/KFit/MassPointingVertexFitKFit.h>
#include <analysis/VertexFitting/KFit/MassVertexFitKFit.h>
#include <analysis/VertexFitting/KFit/VertexFitKFit.h>
#include <analysis/VertexFitting/KFit/MakeMotherKFit.h>
#include <analysis/VertexFitting/KFit/RecoilMassKFit.h>
 
// Rave
#include <analysis/VertexFitting/RaveInterface/RaveSetup.h>
#include <analysis/VertexFitting/RaveInterface/RaveVertexFitter.h>
#include <analysis/VertexFitting/RaveInterface/RaveKinematicVertexFitter.h>
 
namespace Belle2 {
  class Particle;
 
  class ParticleVertexFitterModule : public Module {
 
  public:
 
    ParticleVertexFitterModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
  private:
 
    StoreObjPtr<ParticleList> m_plist; 
    std::string m_listName;       
    double m_confidenceLevel;     
    double m_Bfield;              
    std::string m_vertexFitter;   
    std::string m_fitType;        
    std::string m_withConstraint; 
    std::string m_decayString;    
    bool m_updateDaughters;       
    DecayDescriptor m_decaydescriptor; 
    bool m_hasCovMatrix = false;      
    B2Vector3D m_BeamSpotCenter;    
    TMatrixDSym m_beamSpotCov;    
    DBObjPtr<BeamSpot> m_beamSpotDB;
    double m_smearing;            
    double m_recoilMass;            
    std::vector<int> m_massConstraintList; 
    std::vector<std::string> m_massConstraintListParticlename; 
    bool doVertexFit(Particle* p);
 
    bool doKVertexFit(Particle* p, bool ipProfileConstraint, bool ipTubeConstraint);
 
    bool doKMassVertexFit(Particle* p);
 
    bool doKMassPointingVertexFit(Particle* p);
 
    bool doKMassFit(Particle* p);
 
    bool doKFourCFit(Particle* p);
 
    bool doKMassFourCFit(Particle* p);
 
    bool doKRecoilMassFit(Particle* p);
 
    bool makeKVertexMother(analysis::VertexFitKFit& kv, Particle* p);
 
    bool makeKMassVertexMother(analysis::MassVertexFitKFit& kv, Particle* p);
 
    bool makeKMassPointingVertexMother(analysis::MassPointingVertexFitKFit& kv, Particle* p);
 
    bool makeKMassMother(analysis::MassFitKFit& kv, Particle* p);
 
    bool makeKFourCMother(analysis::FourCFitKFit& kv, Particle* p);
 
    bool makeMassKFourCMother(analysis::MassFourCFitKFit& kv, Particle* p);
 
    bool makeKRecoilMassMother(analysis::RecoilMassKFit& kf, Particle* p);
 
    void updateMapOfTrackAndDaughter(unsigned& l, std::vector<std::vector<unsigned>>& pars, std::vector<unsigned>& pard,
                                     std::vector<Particle*>&  allparticles, const Particle* daughter);
 
    bool addChildofParticletoKFit(analysis::FourCFitKFit& kv, const Particle* particle);
 
    bool addChildofParticletoMassKFit(analysis::MassFourCFitKFit& kf, const Particle* particle, std::vector<unsigned>& particleId);
 
    void addIPProfileToKFit(analysis::VertexFitKFit& kv);
 
    void addIPTubeToKFit(analysis::VertexFitKFit& kv);
 
    bool fillFitParticles(const Particle* mother, std::vector<const Particle*>& fitChildren,
                          std::vector<const Particle*>& twoPhotonChildren);
 
    bool fillNotFitParticles(const Particle* mother, std::vector<const Particle*>& notFitChildren,
                             const std::vector<const Particle*>& fitChildren);
 
    bool redoTwoPhotonDaughterMassFit(Particle* postFit, const Particle* preFit, const analysis::VertexFitKFit& kv) ;
 
    bool doRaveFit(Particle* mother);
 
    bool allSelectedDaughters(const Particle* mother, const std::vector<const Particle*>& tracksVertex);
 
    void findConstraintBoost(double cut);
 
    void smearBeamSpot(double width);
 
    double getChi2TracksLBoost(const analysis::VertexFitKFit& kv);
  };
 
} // Belle2 namespace