KLMClustersReconstructorModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Kirill Chilikin                                          *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/dataobjects/bklm/BKLMHit2d.h>
#include <klm/dataobjects/eklm/EKLMHit2d.h>
 
/* Belle 2 headers. */
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/KLMCluster.h>
 
namespace Belle2 {
  class KLMClustersReconstructorModule : public Module {
 
    enum PositionMode {
      c_FullAverage, 
      c_FirstLayer,  
    };
 
    enum ClusterMode {
      c_AnyHit,   
      c_FirstHit, 
    };
 
  public:
 
    KLMClustersReconstructorModule();
 
    ~KLMClustersReconstructorModule();
 
    void initialize() override;
 
    void beginRun() override;
 
    void event() override;
 
    void endRun() override;
 
    void terminate() override;
 
  private:
 
    double m_ClusteringAngle;
 
    std::string m_PositionModeString;
 
    enum PositionMode m_PositionMode;
 
    std::string m_ClusterModeString;
 
    enum ClusterMode m_ClusterMode;
 
    StoreArray<KLMCluster> m_KLMClusters;
 
    StoreArray<BKLMHit2d> m_BKLMHit2ds;
 
    StoreArray<EKLMHit2d> m_EKLMHit2ds;
 
  };
 
}