SysWriteHandle.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/

 
 
#ifndef SYSTEMATICS_HANDLES__SYS_WRITE_HANDLE_H
#define SYSTEMATICS_HANDLES__SYS_WRITE_HANDLE_H
 
#include <AnaAlgorithm/AnaAlgorithm.h>
#include <AsgDataHandles/WriteHandleKey.h>
#include <AsgMessaging/AsgMessagingForward.h>
#include <AthContainers/CurrentContext.h>
#include <PATInterfaces/SystematicSet.h>
#include <SystematicsHandles/ISysHandleBase.h>
#include <memory>
#include <string>
#include <type_traits>
#include <unordered_map>
 
class StatusCode;
 
namespace CP
{
  class SysListHandle;
  class SystematicSet;
 

 
  template<typename T,typename Aux = void> class SysWriteHandle final
    : public ISysObjectHandleBase, public asg::AsgMessagingForward
  {
  public:

    template<typename T2>
    SysWriteHandle (T2 *owner, const std::string& propertyName,
                    const std::string& propertyValue,
                    const std::string& propertyDescription);

    template<typename T2>
    SysWriteHandle (const std::string &outputName, T2 *owner);

    virtual bool empty () const noexcept override;

    explicit operator bool () const noexcept;

    virtual std::string getNamePattern () const override;
 

    StatusCode initialize (SysListHandle& sysListHandle);
    StatusCode initialize (SysListHandle& sysListHandle, SG::AllowEmptyEnum);
 

    const std::string& getName (const CP::SystematicSet& sys) const;
 

    template<typename X = Aux,typename = std::enable_if_t<std::is_same_v<X,void>>>
    ::StatusCode record (std::unique_ptr<T> object,
                         const CP::SystematicSet& sys,
                         const EventContext& ctx = Gaudi::Hive::currentContext()) const;
 

    template<typename X = Aux,typename = std::enable_if_t<!std::is_same_v<X,void>>>
    ::StatusCode record (std::unique_ptr<T> object,
                         std::unique_ptr<Aux> aux,
                         const CP::SystematicSet& sys,
                         const EventContext& ctx = Gaudi::Hive::currentContext()) const;
 
 

  private:
 
    virtual CP::SystematicSet
    getInputAffecting (const ISystematicsSvc& svc) const override;
    virtual StatusCode
    fillSystematics (const ISystematicsSvc& svc,
                     const CP::SystematicSet& fullAffecting,
                     const std::vector<CP::SystematicSet>& sysList) override;
    virtual StatusCode
    addDecorationDependency (const ISystematicsSvc& svc, const std::string& decoName, bool decoWrite) override;
 
 

  private:

    std::string m_outputName;

    std::string m_outputType;

    struct SysData
    {
      SG::WriteHandleKey<T> writeHandle;
    };
    std::unordered_map<CP::SystematicSet,SysData> m_sysData;

    const SG::WriteHandleKey<T>& getWriteHandle (const CP::SystematicSet& sys) const;
 
 
#ifndef XAOD_STANDALONE
    std::function<void(const DataObjID&, Gaudi::DataHandle::Mode)> m_addAlgDependency;
#endif
  };
}
 
#include "SysWriteHandle.icc"
 
#endif