RawTrackCandCleaner.dcl.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 <tracking/trackFindingCDC/findlets/base/Findlet.h>
#include <tracking/vxdHoughTracking/findlets/RelationCreator.h>
#include <tracking/vxdHoughTracking/findlets/SVDHoughTrackingTreeSearcher.dcl.h>
#include <tracking/vxdHoughTracking/findlets/TrackCandidateResultRefiner.h>
#include <tracking/vxdHoughTracking/filters/relations/ChooseableRelationFilter.h>
#include <tracking/vxdHoughTracking/filters/pathFilters/ChooseablePathFilter.h>
 
#include <string>
#include <vector>
 
namespace Belle2 {
  class ModuleParamList;
  class SpacePointTrackCand;
  class VxdID;
 
  namespace vxdHoughTracking {
 
    template<class AHit>
    class RawTrackCandCleaner : public TrackFindingCDC::Findlet<std::vector<AHit*>, SpacePointTrackCand> {
      using Super =  TrackFindingCDC::Findlet<std::vector<AHit*>, SpacePointTrackCand>;
 
      using Result = std::vector<TrackFindingCDC::WithWeight<const AHit*>>;
 
    public:
      RawTrackCandCleaner();
 
      ~RawTrackCandCleaner();
 
      void exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix) override;
 
      void initialize() override;
 
      void apply(std::vector<std::vector<AHit*>>& rawTrackCandidates, std::vector<SpacePointTrackCand>& trackCandidates) override;
 
    private:
 
      RelationCreator<AHit, ChooseableRelationFilter> m_relationCreator;
 
      SVDHoughTrackingTreeSearcher<AHit, ChooseablePathFilter, Result> m_treeSearcher;
 
      TrackCandidateResultRefiner m_resultRefiner;
 
      std::vector<TrackFindingCDC::WeightedRelation<AHit>> m_relations;
 
      std::vector<Result> m_results;
      std::vector<SpacePointTrackCand> m_unfilteredResults;
      std::vector<SpacePointTrackCand> m_filteredResults;
 
      uint m_maxRelations = 100;
    };
 
  }
}