RootNtupleOutputStream.h

Go to the documentation of this file.

// Dear emacs, this is -*- C++ -*-
 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ATHENAROOTCOMPS_ROOTNTUPLEOUTPUTSTREAM_H
#define ATHENAROOTCOMPS_ROOTNTUPLEOUTPUTSTREAM_H
 
// STL include files
#include <memory>
#include <map>
#include <set>
#include <vector>
#include <string>
 
// fwk includes
#include "GaudiKernel/IDataSelector.h"
#include "GaudiKernel/ClassID.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/IIncidentListener.h"
#include "AthenaBaseComps/FilteredAlgorithm.h"
 
// forward declarations
template <class ConcreteAlgorithm> class AlgFactory;
class IClassIDSvc;
class StoreGateSvc;
class IAthenaOutputStreamTool;
class IAthenaOutputTool;
 
namespace SG {
   class DataProxy;
   class IFolder;
   class FolderItem;
}
 
namespace Athena {

class RootNtupleOutputStream 
  : virtual public IIncidentListener,
            public FilteredAlgorithm
{
  friend class AlgFactory<Athena::RootNtupleOutputStream>;
 
public:
  typedef std::vector<SG::DataProxy*>     Items;
 
protected:
  ServiceHandle<StoreGateSvc> m_dataStore;

  std::string              m_persName;
  StringProperty           m_writingTool;
  std::string              m_outputName;
  StringProperty           m_processTag;
   
  typedef ServiceHandle<IClassIDSvc> IClassIDSvc_t;
  IClassIDSvc_t m_pCLIDSvc;
   
  StringArrayProperty      m_itemList;
  IDataSelector            m_objects;
  int                      m_events;
  bool m_forceRead;
  bool m_persToPers;
  std::vector<unsigned int> m_exemptPersToPers;
  bool m_provideDef;
  bool m_writeOnExecute;
  bool m_writeOnFinalize;
  bool m_itemListFromTool;

  bool m_dynamicItemList;

  std::vector<SG::FolderItem> m_selection;

  ToolHandle<IAthenaOutputStreamTool> m_streamer;
  ToolHandleArray<IAthenaOutputTool> m_helperTools;
 
public:
  typedef std::vector<std::pair<std::string, std::string> > TypeKeyPairs;
  
  RootNtupleOutputStream(const std::string& name, ISvcLocator* pSvcLocator); 
  virtual ~RootNtupleOutputStream();


  virtual StatusCode initialize();
  virtual StatusCode finalize();
  virtual StatusCode execute();

  virtual StatusCode write();
 
private:
  void clearSelection();
  void collectAllObjects();
  IDataSelector* selectedObjects() {
    return &m_objects;
  }

  void handle(const Incident& incident);
};
 
} //> ns Athena
 
#endif // ATHENAROOTCOMPS_ROOTNTUPLEOUTPUTSTREAM_H