MillepedeCollectorModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: tadeas                                                   *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/dataobjects/Particle.h>
#include <calibration/CalibrationCollectorModule.h>
#include <framework/core/ModuleParam.templateDetails.h>
#include <framework/geometry/B2Vector3.h>
#include <tracking/dataobjects/RecoTrack.h>
 
#include <genfit/GblTrajectory.h>
#include <genfit/Track.h>
#include <genfit/MeasuredStateOnPlane.h>
 
namespace Belle2 {
  class MillepedeCollectorModule : public CalibrationCollectorModule {
 
  public:
 
    MillepedeCollectorModule();
 
    virtual void prepare() override;
 
    virtual void collect() override;
 
    virtual void closeRun() override;
 
    virtual void finish() override;
 
    std::string getUniqueMilleName();
 
    std::vector<genfit::Track*> getParticlesTracks(std::vector<Particle*> particles, bool addVertexPoint = true);
 
    bool fitRecoTrack(RecoTrack& recoTrack, Particle* particle = nullptr);
 
    TMatrixD getGlobalToLocalTransform(const genfit::MeasuredStateOnPlane& msop);
 
    TMatrixD getLocalToGlobalTransform(const genfit::MeasuredStateOnPlane& msop);
 
    std::pair<TMatrixD, TMatrixD> getTwoBodyToLocalTransform(Particle& mother, double motherMass);
 
    void storeTrajectory(gbl::GblTrajectory& trajectory);
 
    std::tuple<B2Vector3D, TMatrixDSym> getPrimaryVertexAndCov() const;
 
 
  private:
    std::vector<std::string> m_tracks;
    std::vector<std::string> m_particles;
    std::vector<std::string> m_vertices;
    std::vector<std::string> m_primaryVertices;
    std::vector<std::string> m_twoBodyDecays;
    std::vector<std::string> m_primaryTwoBodyDecays;
    std::vector<std::string> m_primaryMassTwoBodyDecays;
    std::vector<std::string> m_primaryMassVertexTwoBodyDecays;
 
    double m_stableParticleWidth;
    bool m_doublePrecision;
    bool m_calibrateVertex;
    bool m_calibrateKinematics = true;
    double m_minPValue;
    bool m_useGblTree;
    bool m_absFilePaths;
    std::vector<std::string> m_components{};
    int m_externalIterations;
    std::string m_internalIterations;
    int m_recalcJacobians;
    bool m_fitTrackT0;
    bool m_updateCDCWeights;
    double m_minCDCHitWeight;
    double m_minUsedCDCHitFraction;
    int m_hierarchyType;
    bool m_enablePXDHierarchy;
    bool m_enableSVDHierarchy;
    bool m_enableWireByWireAlignment;
    bool m_enableWireSagging;
 
    std::vector<std::tuple<int, int, int>> m_eventNumbers{};
 
    std::vector< std::tuple< std::vector< int >, std::vector< std::tuple< int, int, int > > > > m_timedepConfig;
 
    std::map<std::string,  std::tuple<double, double>> m_customMassConfig;
 
    std::vector<gbl::GblData> m_currentGblData{};
 
    void updateMassWidthIfSet(std::string listName, double& mass, double& width);
  };
}