TrackFinderSegmentPairAutomaton.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/findlets/minimal/SegmentPairCreator.h>
#include <tracking/trackFindingCDC/findlets/minimal/TrackCreatorSegmentPairAutomaton.h>
#include <tracking/trackFindingCDC/findlets/minimal/TrackCreatorSingleSegments.h>
#include <tracking/trackFindingCDC/findlets/minimal/TrackLinker.h>
#include <tracking/trackFindingCDC/findlets/minimal/TrackOrienter.h>
 
#include <tracking/trackFindingCDC/filters/segmentPairRelation/ChooseableSegmentPairRelationFilter.h>
 
#include <tracking/trackFindingCDC/eventdata/segments/CDCSegment2D.h>
#include <tracking/trackFindingCDC/eventdata/tracks/CDCTrack.h>
 
#include <tracking/trackFindingCDC/findlets/minimal/WeightedRelationCreator.h>
#include <tracking/trackFindingCDC/findlets/base/StoreVectorSwapper.h>
 
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    class TrackFinderSegmentPairAutomaton : public Findlet<const CDCSegment2D, CDCTrack> {
 
    private:
      using Super = Findlet<const CDCSegment2D, CDCTrack>;
 
    public:
      TrackFinderSegmentPairAutomaton();
 
      std::string getDescription() final;
 
      void exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix) final;
 
      void beginEvent() final;
 
      void apply(const std::vector<CDCSegment2D>& inputSegments, std::vector<CDCTrack>& tracks) final;
 
    private:
      // Findlets
      SegmentPairCreator m_segmentPairCreator;
 
      WeightedRelationCreator<const CDCSegmentPair, ChooseableSegmentPairRelationFilter> m_segmentPairRelationCreator;
 
      TrackCreatorSegmentPairAutomaton m_trackCreatorSegmentPairAutomaton;
 
      TrackCreatorSingleSegments m_trackCreatorSingleSegments;
 
      TrackLinker m_trackLinker;
 
      TrackOrienter m_trackOrienter;
 
      StoreVectorSwapper<CDCSegmentPair> m_segmentPairSwapper{"CDCSegmentPairVector"};
 
      // Object pools
      std::vector<CDCSegmentPair> m_segmentPairs;
 
      std::vector<WeightedRelation<const CDCSegmentPair> > m_segmentPairRelations;
 
      std::vector<CDCTrack> m_preLinkingTracks;
 
      std::vector<CDCTrack> m_orientedTracks;
    };
  }
}