BeamBkgTagSetterModule.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 <simulation/background/BeamBGTypes.h>
#include <framework/dataobjects/BackgroundMetaData.h>
#include <string>
 
// SimHits
#include <pxd/dataobjects/PXDSimHit.h>
#include <svd/dataobjects/SVDSimHit.h>
#include <cdc/dataobjects/CDCSimHit.h>
#include <top/dataobjects/TOPSimHit.h>
#include <arich/dataobjects/ARICHSimHit.h>
#include <ecl/dataobjects/ECLSimHit.h>
#include <ecl/dataobjects/ECLHit.h>
#include <klm/dataobjects/KLMSimHit.h>
 
// BEAST SimHits
#include <beast/beamabort/dataobjects/BeamabortSimHit.h>
#include <beast/claw/dataobjects/ClawSimHit.h>
#include <beast/claws/dataobjects/CLAWSSimHit.h>
#include <beast/fangs/dataobjects/FANGSSimHit.h>
#include <beast/plume/dataobjects/PlumeSimHit.h>
#include <beast/pindiode/dataobjects/PindiodeSimHit.h>
#include <beast/he3tube/dataobjects/He3tubeSimHit.h>
#include <beast/microtpc/dataobjects/MicrotpcSimHit.h>
#include <beast/qcsmonitor/dataobjects/QcsmonitorSimHit.h>
#include <beast/bgo/dataobjects/BgoSimHit.h>
#include <beast/csi/dataobjects/CsiSimHit.h>
 
 
namespace Belle2 {

  class BeamBkgTagSetterModule : public Module {
 
  public:

    BeamBkgTagSetterModule();

    virtual void initialize() override;

    virtual void event() override;
 
  private:
 
    std::string m_backgroundType; 
    double m_realTime; 
    int m_phase; 
    std::string m_specialFor; 
 
    background::BeamBGTypes m_bgTypes;  
    BackgroundMetaData::BG_TAG m_backgroundTag; 
    BackgroundMetaData::EFileType m_fileType; 
 
    StoreArray<PXDSimHit> m_pxdSimHits; 
    StoreArray<SVDSimHit> m_svdSimHits; 
    StoreArray<CDCSimHit> m_cdcSimHits; 
    StoreArray<TOPSimHit> m_topSimHits; 
    StoreArray<ARICHSimHit> m_arichSimHits; 
    StoreArray<ECLSimHit> m_eclSimHits; 
    StoreArray<ECLHit> m_eclHits; 
    StoreArray<KLMSimHit> m_klmSimHits; 
 
    StoreArray<BeamabortSimHit> m_diaSimHits; 
    StoreArray<CLAWSSimHit> m_clw2SimHits;  
    StoreArray<ClawSimHit> m_clw1SimHits;  
    StoreArray<FANGSSimHit> m_fngSimHits; 
    StoreArray<PlumeSimHit> m_plmSimHits; 
    StoreArray<PindiodeSimHit> m_pinSimHits; 
    StoreArray<He3tubeSimHit> m_he3SimHits; 
    StoreArray<MicrotpcSimHit> m_tpcSimHits; 
    StoreArray<QcsmonitorSimHit> m_sciSimHits; 
    StoreArray<BgoSimHit> m_bgoSimHits; 
    StoreArray<CsiSimHit> m_csiSimHits; 

    template<class SIMHIT>
    int setBackgroundTag(StoreArray<SIMHIT>& simHits)
    {
      if (!simHits.isValid()) return 0;
 
      int numEntries = simHits.getEntries();
      for (int i = 0; i < numEntries; i++) {
        SIMHIT* simHit = simHits[i];
        simHit->setBackgroundTag(m_backgroundTag);
      }
      return numEntries;
    }
 
 
  };

} // Belle2 namespace