AANTupleStream.h

Go to the documentation of this file.

// dear emacs, this is -*- C++ -*-
 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ANALYSISTOOLS_AANTUPLESTREAM_H
#define ANALYSISTOOLS_AANTUPLESTREAM_H 1
 
#include "GaudiKernel/IAddressCreator.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/IIoComponent.h"
#include "AthenaBaseComps/AthLegacySequence.h"
#include "CxxUtils/checker_macros.h"
 
#include <map>
#include <vector>
 
namespace coral
{
  class AttributeList;
  class AttributeListSpecification;
}
 
class StoreGateSvc;
 
class ITHistSvc;
 
class TFile;
class TTree;
class TBranch;
 
class ATLAS_NOT_THREAD_SAFE AANTupleStream : virtual public IIoComponent,
                                             public AthLegacySequence
{
public:
    
  AANTupleStream(const std::string& name, ISvcLocator* pSvcLocator); 
 
  virtual ~AANTupleStream();
 
  virtual StatusCode initialize();
 
  virtual StatusCode finalize();
 
  virtual StatusCode execute();
 
private:
 
  StatusCode initCollection();
 
  StatusCode getRef(CLID id, const std::string& key,
            std::string& ref);
 
  StatusCode splitAddress( const std::string& address,
               const std::string& match,
               std::string&       address_header,
               std::string&       address_data ) const;
 
  StatusCode initialize_subAlgos ();
 
  StatusCode execute_subAlgos ();
 
  void writeAttributeListSpecification();
 
  void setupTree();
 
  bool writeTokenAttrList( const std::string& token, const coral::AttributeList& attributeList );
  
  StatusCode getFilters();
 
  bool isEventAccepted() const;
 
  StatusCode initSchema();
 
  StatusCode io_reinit ();
 
  ServiceHandle<IAddressCreator> m_persSvc;
 
  coral::AttributeListSpecification* m_attribSpec;
 
  std::string m_fileName;
  
  StringArrayProperty m_extraRefNames;
 
  BooleanProperty          m_writeInputDH;
 
  BooleanProperty          m_existDH;
 
  std::string m_streamName;
 
  BooleanProperty m_lateSchema;
 
  std::string m_macro;
 
  bool m_schemaDone;
 
  std::string m_treeName;
 
  std::vector<std::string> m_acceptNames;
 
  std::vector<Algorithm*> m_acceptAlgs;
 
  TTree * m_tree;
 
  TBranch * m_tokenBranch;
 
  ServiceHandle<ITHistSvc> m_tHistSvc;
 
  // Sub-algorithms  as given in jobOptions type/name
  std::vector<std::string> m_membersNames;
 
  static const unsigned int s_maxLengthOfStrings = 5000;
  char m_tokenCString[s_maxLengthOfStrings];
};
 
#endif