GlobalCache.h

Go to the documentation of this file.

// Definition of objects placed in the edm::GlobalCache.
 
#ifndef RecoHGCal_TICL_GlobalCache_H__
#define RecoHGCal_TICL_GlobalCache_H__
 
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
 
namespace ticl {
  // base class across ticl for objects hold in the edm::GlobalCache by plugins
  class CacheBase {
  public:
    CacheBase(const edm::ParameterSet& params) {}
 
    virtual ~CacheBase() {}
  };
 
  // data structure hold by TrackstersProducer to store the TF graph for energy regression and ID
  class TrackstersCache : public CacheBase {
  public:
    TrackstersCache(const edm::ParameterSet& params) : CacheBase(params), eidGraphDef(nullptr) {}
 
    ~TrackstersCache() override {}
 
    std::atomic<tensorflow::GraphDef*> eidGraphDef;
  };
}  // namespace ticl
 
#endif  // RecoHGCal_TICL_GlobalCache_H__