Kink.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 <mdst/dataobjects/TrackFitResult.h>
#include <mdst/dataobjects/Track.h>
#include <framework/datastore/RelationsObject.h>
#include <framework/datastore/StoreArray.h>
#include <framework/gearbox/Const.h>
#include <utility>
 
namespace Belle2 {
  class Kink : public RelationsObject {
  public:
    Kink();
 
    Kink(const std::pair<const Belle2::Track*,
         std::pair<const Belle2::TrackFitResult*, const Belle2::TrackFitResult*> >& trackPairMother,
         const std::pair<const Belle2::Track*, const Belle2::TrackFitResult*>& trackPairDaughter,
         const Double32_t vertexX, const Double32_t vertexY, const Double32_t vertexZ, const short filterFlag);
 
    Track* getMotherTrack() const;
 
    Track* getDaughterTrack() const;
 
    short getMotherTrackIndex() const
    {
      return m_trackIndexMother;
    }
 
    short getDaughterTrackIndex() const
    {
      return m_trackIndexDaughter;
    }
 
    TrackFitResult* getMotherTrackFitResultStart() const;
 
    TrackFitResult* getMotherTrackFitResultEnd() const;
 
    TrackFitResult* getDaughterTrackFitResult() const;
 
    short getTrackFitResultIndexMotherStart() const
    {
      return m_trackFitResultIndexMotherStart;
    }
 
    short getTrackFitResultIndexMotherEnd() const
    {
      return m_trackFitResultIndexMotherEnd;
    }
 
    short getTrackFitResultIndexDaughter() const
    {
      return m_trackFitResultIndexDaughter;
    }
 
    Double32_t getFittedVertexX() const { return m_fittedVertexX; }
 
    Double32_t getFittedVertexY() const { return m_fittedVertexY; }
 
    Double32_t getFittedVertexZ() const { return m_fittedVertexZ; }
 
    ROOT::Math::XYZVector getFittedVertexPosition() const
    {
      return ROOT::Math::XYZVector(m_fittedVertexX, m_fittedVertexY, m_fittedVertexZ);
    }
 
    short getFilterFlag() const
    {
      return m_filterFlag;
    }
 
    short getPrefilterFlag() const
    {
      return abs(m_filterFlag) % 10;
    }
 
    short getCombinedFitResultFlag() const
    {
      return (abs(m_filterFlag) % 1000 - getPrefilterFlag()) / 10;
    }
 
    short getSplitTrackDistanceAtVertexFlag() const
    {
      return getCombinedFitResultFlag();
    }
 
    short getNumberOfReassignedHits() const
    {
      return m_filterFlag / 1000;
    }
 
  private:
    short m_trackIndexMother = -1;
 
    short m_trackIndexDaughter = -1;
 
    short m_trackFitResultIndexMotherStart = -1;
 
    short m_trackFitResultIndexMotherEnd = -1;
 
    short m_trackFitResultIndexDaughter = -1;
 
    Double32_t m_fittedVertexX = 0.0;
 
    Double32_t m_fittedVertexY = 0.0;
 
    Double32_t m_fittedVertexZ = 0.0;
 
    short m_filterFlag = 0;
 
    ClassDef(Kink, 1);
 
    friend class FixMergedObjectsModule;
  };
}