AANTupleStream.h

Go to the documentation of this file.

// dear emacs, this is -*- C++ -*-
 
/*
  Copyright (C) 2002-2024 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<IAlgorithm*> 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