CDCTriggerNDFinderModule.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 "framework/core/Module.h"
#include "framework/datastore/StoreArray.h"
#include "trg/cdc/dataobjects/CDCTriggerTrack.h"
#include "trg/cdc/dataobjects/CDCTrigger3DFinderInfo.h"
#include "trg/cdc/dataobjects/CDCTriggerSegmentHit.h"
#include "trg/cdc/NDFinder.h"
#include "trg/cdc/Clusterizend.h"
 
namespace Belle2 {
  class CDCTriggerNDFinderModule : public Module {
  public:
    CDCTriggerNDFinderModule();
    virtual ~CDCTriggerNDFinderModule();
 
    virtual void initialize() override;
    virtual void beginRun() override;
    virtual void event() override;
    virtual void endRun() override;
    virtual void terminate() override;
 
  private:
 
    std::string m_TrackSegmentHitsName;
    std::string m_NDFinderTracksName;
    std::string m_NDFinderTrackToHitArrayName;
    std::string m_NDFinderInfosName;
 
    StoreArray<CDCTriggerSegmentHit> m_TrackSegmentHits;
    StoreArray<CDCTriggerTrack> m_NDFinderTracks;
    StoreArray<CDCTrigger3DFinderInfo> m_NDFinderInfos;
 
    NDFinder m_NDFinder;
    int m_minSuperAxial;
    int m_minSuperStereo;
    int m_minWeight;
    int m_minPts;
    double m_thresh;
    bool m_diagonal;
    int m_minCells;
    bool m_dbscanning;
    int m_minTotalWeight;
    int m_minPeakWeight;
    int m_iterations;
    int m_omegaTrim;
    int m_phiTrim;
    int m_thetaTrim;
    bool m_verbose;
    std::string m_axialFile;
    std::string m_stereoFile;
  };
}