AthenaOutputStreamTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ATHENAOUTPUTSTREAMTOOL_H
#define ATHENAOUTPUTSTREAMTOOL_H
 
#include "GaudiKernel/ServiceHandle.h"
 
#include "AthenaKernel/IAthenaOutputStreamTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include <string>
 
class IClassIDSvc;
class IDecisionSvc;
class IConversionSvc;
class StoreGateSvc;
class DataHeader;
namespace SG {
   class IFolder;
}
 
class AthenaOutputStreamTool : public extends<::AthAlgTool, IAthenaOutputStreamTool> {
public:
   AthenaOutputStreamTool(const std::string& type,
       const std::string& name,
       const IInterface* parent);
   virtual ~AthenaOutputStreamTool();
 
   StatusCode initialize();
   StatusCode finalize();
 
   StatusCode connectServices(const std::string& dataStore,
       const std::string& cnvSvc,
       bool extendProvenenceRecord);
 
   StatusCode connectOutput(const std::string& outputName = "");
 
   StatusCode commitOutput(bool doCommit = false);
 
   StatusCode finalizeOutput();
 
   typedef std::pair<std::string, std::string> TypeKeyPair;
   typedef std::vector<TypeKeyPair>            TypeKeyPairs;
   virtual StatusCode streamObjects(const TypeKeyPairs& typeKeys, const std::string& outputName = "");
 
   typedef std::vector<DataObject*> DataObjectVec;
   virtual StatusCode streamObjects(const DataObjectVec& dataObjects, const std::string& outputName = "");
 
   virtual StatusCode getInputItemList(SG::IFolder* m_p2BWrittenFromTool);
 
private:
   virtual StatusCode connectServices();
 
private:
   StringProperty  m_outputName{ this, "OutputFile", "", "name of the output db name"};
   StringProperty  m_dataHeaderKey{ this, "DataHeaderKey", "", "name of the data header key: defaults to tool name"};
   StringProperty  m_processTag{ this, "ProcessingTag", "", "tag of processing stage: defaults to SG key of DataHeader (Stream name)"};
   StringProperty  m_outputCollection{ this, "OutputCollection", "", "custom container name prefix for DataHeader: default = "" (will result in \"POOLContainer_\")"};
   StringProperty  m_containerPrefix{ this, "PoolContainerPrefix", "", "prefix for top level POOL container: default = \"CollectionTree\""};
   StringProperty  m_containerNameHint{ this, "TopLevelContainerName", "0", "naming hint policy for top level POOL container: default = \"0\""};
   StringProperty  m_metaDataOutputCollection{ this, "MetaDataOutputCollection", "", "custom container name prefix for MetaDataHeader: default = "" (will result in \"MetaDataHdr\")"};
   StringProperty  m_metaDataContainerPrefix{ this, "MetaDataPoolContainerPrefix", "", "prefix for top level MetaData container: default = "" (will result in \"MetaData\")"};
   StringProperty  m_branchNameHint{ this, "SubLevelBranchName", "0", "naming hint policy for POOL branching: default = \"0\"" };
   std::string  m_outputAttributes{""};
   std::string  m_metaDataOutputAttributes{""};
   SG::ReadHandleKey<AthenaAttributeList>  m_attrListKey{this, "AttributeListKey", "", "optional key for AttributeList to be written as part of the DataHeader: default = \"\""};
   //SG::WriteHandleKey<AthenaAttributeList>  m_attrListWrite{this, "AttributeListWrite", "", "optional key for AttributeList to be written as part of the DataHeader: default = <AttributeListKey>+\"Decisions\""};
   std::string  m_attrListWrite{""};
 
   ServiceHandle<StoreGateSvc>   m_store{ this, "Store", "StoreGateSvc/DetectorStore", "Pointer to the data store"};
   ServiceHandle<IConversionSvc> m_conversionSvc;
   ServiceHandle<IClassIDSvc>    m_clidSvc;
   ServiceHandle<IDecisionSvc>   m_decSvc;
   DataHeader*     m_dataHeader;
   bool            m_connectionOpen;
   bool            m_extendProvenanceRecord;
   bool m_extend;
   
   std::set<std::string> m_skippedItems;
};
 
#endif