TrackBuilder.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 <string>
#include <framework/geometry/B2Vector3.h>
#include <genfit/AbsTrackRep.h>
 
namespace Belle2 {
 
  class RecoTrack;

  class TrackBuilder {
  public:
    TrackBuilder(
      const std::string& trackColName,
      const std::string& trackFitResultColName,
      const B2Vector3D& beamSpot = B2Vector3D(0., 0., 0.),
      const B2Vector3D& beamAxis = B2Vector3D(0., 0., 1.)
    ) :
      m_trackColName(trackColName),
      m_trackFitResultColName(trackFitResultColName),
      m_beamSpot(beamSpot),
      m_beamAxis(beamAxis)
    {};

    bool storeTrackFromRecoTrack(RecoTrack& recoTrack,
                                 const bool useClosestHitToIP = false, const bool useBFieldAtHit = false);

    static uint32_t getHitPatternVXDInitializer(const RecoTrack& recoTrack, const genfit::AbsTrackRep* representation = nullptr);
    static uint64_t getHitPatternCDCInitializer(const RecoTrack& recoTrack, const genfit::AbsTrackRep* representation = nullptr);
 
  private:
    std::string m_trackColName;
    std::string m_trackFitResultColName;
    B2Vector3D m_beamSpot;
    B2Vector3D m_beamAxis;
  };

}