ECLFinalizerModule.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/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
#include <ecl/dbobjects/ECLClusteringParameters.h>
 
namespace Belle2 {
 
  class EventT0;
  class ECLShower;
  class ECLCluster;
  class ECLCalDigit;
  class EventLevelClusteringInfo;

  class ECLFinalizerModule : public Module {
 
  public:
    ECLFinalizerModule();

    ~ECLFinalizerModule() override;

    virtual void initialize() override;

    virtual void beginRun() override;

    virtual void event() override;

    virtual void endRun() override;

    virtual void terminate() override;
 
  private:
    double m_clusterEnergyCutMin; 
    bool m_useParametersFromDatabase; 
    double m_clusterTimeCutMaxEnergy; 
    double m_clusterLossyFraction; 
 
    StoreArray<ECLCalDigit> m_eclCalDigits; 
    StoreArray<ECLShower> m_eclShowers; 
    StoreArray<ECLCluster> m_eclClusters; 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo; 
    StoreObjPtr<EventT0> m_eventT0; 

    DBObjPtr<ECLClusteringParameters> m_eclClusteringParameters;
 
    int makeCluster(int, double); 
 
  public:

    virtual const char* eclShowerArrayName() const
    { return "ECLShowers" ; }

    virtual const char* eclClusterArrayName() const
    { return "ECLClusters"; }

    virtual const char* eclCalDigitArrayName() const
    {return "ECLCalDigits";}
 
  }; // end of ECLFinalizerModule
 

  class ECLFinalizerPureCsIModule : public ECLFinalizerModule {
  public:

    virtual const char* eclShowerArrayName() const override
    { return "ECLShowersPureCsI" ; }

    virtual const char* eclClusterArrayName() const override
    { return "ECLClustersPureCsI"; }

    virtual const char* eclCalDigitArrayName() const override
    {return "ECLCalDigitsPureCsI";}
  }; // end of ECLFinalizerPureCsIModule

} // end of Belle2 namespace