DynamicDataHelper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
/* Dear emacs, this is -*-c++-*- */
#ifndef _DynamicDataHelper_H_
#define _DynamicDataHelper_H_
 
#include "AthenaBaseComps/GaudiComponentVisitor.h"
#include "AthenaBaseComps/IDynamicDataConsumer.h"
#include "GaudiKernel/StatusCode.h"
 
// @TODO extend Gaudi's tool visitor to allow for an externally provided list of tools which have been visited already.
namespace Ath {
 
   struct DataObjID_PtrHasher {
      std::size_t operator()( const DataObjID* k ) const { return DataObjID_Hasher()(*k); }
   };
   struct DataObjID_PtrEqual {
      std::size_t operator()( const DataObjID* a, const DataObjID* b) const { return *a == *b; }
   };
   using DataObjIDPtrColl = std::unordered_set<const DataObjID *, DataObjID_PtrHasher, DataObjID_PtrEqual>;

   class DynamicDataHelper {
   public:
      void gatherDataHandlesAndDynamicConsumers(const std::string &parent_name, Gaudi::Algorithm *theAlgorithm);

      StatusCode addExtraDependencies(IClassIDSvc& clid_svc, std::vector<std::string> &undeclared_output_data, MsgStream &out);

      void updateDataNeeds(unsigned int max_pass,
                           MsgStream &out);
 
   protected:
      bool updateDataNeeds(const std::vector<const DataObjID *> &input_data_in,
                           const std::vector<const DataObjID *> &output_data_in,
                           std::vector<const DataObjID *> &input_data_out,
                           std::vector<const DataObjID *> &output_data_out,
                           MsgStream &out);
 
      std::vector<DataObjID> m_extraOutputIDs;
      DataObjIDPtrColl m_inputHandles;
      DataObjIDPtrColl m_outputHandles;
 
      std::unordered_map< IDynamicDataConsumer *,std::vector<Gaudi::Algorithm *> > m_dynDataConsumer;
 
   };
}
#endif