TrackFinderVXDBasicPathFinderModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jakob Lettenbichler, Jonas Wagner, Felix Metzner         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/core/Module.h>
 
#include <mdst/dataobjects/EventLevelTrackingInfo.h>
 
#include <tracking/trackFindingVXD/algorithms/CellularAutomaton.h>
#include <tracking/trackFindingVXD/algorithms/PathCollectorRecursive.h>
#include <tracking/trackFindingVXD/algorithms/NodeFamilyDefiner.h>
#include <tracking/trackFindingVXD/algorithms/SPTCSelectorXBestPerFamily.h>
 
#include <tracking/trackFindingVXD/segmentNetwork/CACell.h>
#include <tracking/trackFindingVXD/segmentNetwork/DirectedNodeNetworkContainer.h>
 
#include <tracking/trackFindingVXD/tcTools/SpacePointTrackCandCreator.h>
#include <tracking/spacePointCreation/SpacePointTrackCand.h>
 
#include <tracking/trackFindingVXD/algorithms/CAValidator.h>
#include <tracking/trackFindingVXD/algorithms/NodeCompatibilityCheckerBase.h>
 
 
namespace Belle2 {
  class TrackFinderVXDBasicPathFinderModule final : public Module {
  private:
    using NodeType = Belle2::DirectedNode<Belle2::Segment<Belle2::TrackNode>, Belle2::CACell>;
    using NodeNetworkType = Belle2::DirectedNodeNetwork<Belle2::Segment<Belle2::TrackNode>, Belle2::CACell>;
    using Path = std::vector<NodeType*>;
 
  public:
 
    TrackFinderVXDBasicPathFinderModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
 
  protected:
 
 
    std::string m_PARAMSpacePointTrackCandArrayName;
 
    std::string m_PARAMNetworkName;
 
    std::string m_PARAMEventLevelTrackingInfoName;
 
    std::string m_PARAMsecMapName;
 
    std::vector<Path> m_collectedPaths;
 
    bool m_PARAMprintNetworks;
 
    bool m_PARAMstrictSeeding;
 
    bool m_PARAMstoreSubsets;
 
    bool m_PARAMsetFamilies;
 
    bool m_PARAMselectBestPerFamily;
 
    unsigned short m_PARAMxBestPerFamily = 5;
 
    unsigned short m_PARAMmaxFamilies = 10000;
 
    unsigned int m_PARAMmaxPaths = 400000;
 
 
    CellularAutomaton<NodeNetworkType, Belle2::CAValidator<Belle2::CACell>> m_cellularAutomaton;
 
    PathCollectorRecursive<NodeNetworkType, NodeType, Path, Belle2::NodeCompatibilityCheckerBase<NodeType>> m_pathCollector;
 
    SpacePointTrackCandCreator<StoreArray<Belle2::SpacePointTrackCand>> m_sptcCreator;
 
    NodeFamilyDefiner<NodeNetworkType, NodeType, Path> m_familyDefiner;
 
 
    StoreObjPtr<Belle2::DirectedNodeNetworkContainer> m_network;
 
 
    StoreArray<Belle2::SpacePointTrackCand> m_TCs;
 
    std::unique_ptr<SPTCSelectorXBestPerFamily> m_sptcSelector;
 
    StoreObjPtr<EventLevelTrackingInfo> m_eventLevelTrackingInfo;
 
    unsigned int m_eventCounter = 0;
  };
}