TrackFinderSegmentTripleAutomaton.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/AxialSegmentPairCreator.h>
#include <tracking/trackFindingCDC/findlets/minimal/SegmentTripleCreator.h>
 
#include <tracking/trackFindingCDC/findlets/minimal/TrackCreatorSegmentTripleAutomaton.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/segmentTripleRelation/ChooseableSegmentTripleRelationFilter.h>
 
#include <tracking/trackFindingCDC/eventdata/tracks/CDCTrack.h>
#include <tracking/trackFindingCDC/eventdata/tracks/CDCSegmentTriple.h>
#include <tracking/trackFindingCDC/eventdata/tracks/CDCAxialSegmentPair.h>
 
#include <tracking/trackFindingCDC/findlets/minimal/WeightedRelationCreator.h>
#include <tracking/trackFindingCDC/findlets/base/StoreVectorSwapper.h>
 
#include <vector>
 
namespace Belle2 {
  namespace TrackFindingCDC {
 
    class TrackFinderSegmentTripleAutomaton : public Findlet<const CDCSegment2D, CDCTrack> {
 
    private:
      using Super = Findlet<const CDCSegment2D, CDCTrack>;
 
    public:
      TrackFinderSegmentTripleAutomaton();
 
      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
      AxialSegmentPairCreator m_axialSegmentPairCreator;
 
      SegmentTripleCreator m_segmentTripleCreator;
 
      WeightedRelationCreator<const CDCSegmentTriple, ChooseableSegmentTripleRelationFilter> m_segmentTripleRelationCreator;
 
      TrackCreatorSegmentTripleAutomaton m_trackCreatorSegmentTripleAutomaton;
 
      TrackCreatorSingleSegments m_trackCreatorSingleSegments;
 
      TrackLinker m_trackLinker;
 
      TrackOrienter m_trackOrienter;
 
      StoreVectorSwapper<CDCSegmentTriple> m_segmentTripleSwapper{"CDCSegmentTripleVector"};
 
      // object pools
      std::vector<CDCAxialSegmentPair> m_axialSegmentPairs;
 
      std::vector<CDCSegmentTriple> m_segmentTriples;
 
      std::vector<WeightedRelation<const CDCSegmentTriple> > m_segmentTripleRelations;
 
      std::vector<CDCTrack> m_preLinkingTracks;
 
      std::vector<CDCTrack> m_orientedTracks;
    };
  }
}