THistSvc Class Reference

#include <src/THistSvc/THistSvc.h>

Inheritance diagram for THistSvc:

Collaboration diagram for THistSvc:

Classes
class	GlobalDirectoryRestore
struct	THistID

Public Member Functions
StatusCode	initialize () override
StatusCode	reinitialize () override
StatusCode	finalize () override
StatusCode	regHist (const std::string &name) override
StatusCode	regHist (const std::string &name, TH1 *) override
StatusCode	regHist (const std::string &name, TH2 *) override
StatusCode	regHist (const std::string &name, TH3 *) override
StatusCode	getHist (const std::string &name, TH1 *&) const override
StatusCode	getHist (const std::string &name, TH2 *&) const override
StatusCode	getHist (const std::string &name, TH3 *&) const override
StatusCode	regTree (const std::string &name) override
StatusCode	regTree (const std::string &name, TTree *) override
StatusCode	getTree (const std::string &name, TTree *&) const override
StatusCode	regGraph (const std::string &name) override
StatusCode	regGraph (const std::string &name, TGraph *) override
StatusCode	getGraph (const std::string &name, TGraph *&) const override
StatusCode	deReg (TObject *obj) override
StatusCode	deReg (const std::string &name) override
std::vector< std::string >	getHists () const override
std::vector< std::string >	getTrees () const override
std::vector< std::string >	getGraphs () const override
StatusCode	getTHists (TDirectory *td, TList &, bool recurse=false) const override
StatusCode	getTHists (const std::string &name, TList &, bool recurse=false) const override
StatusCode	getTHists (TDirectory *td, TList &tl, bool recurse=false, bool reg=false) override
StatusCode	getTHists (const std::string &name, TList &tl, bool recurse=false, bool reg=false) override
StatusCode	getTTrees (TDirectory *td, TList &, bool recurse=false) const override
StatusCode	getTTrees (const std::string &name, TList &, bool recurse=false) const override
StatusCode	getTTrees (TDirectory *td, TList &tl, bool recurse=false, bool reg=false) override
StatusCode	getTTrees (const std::string &name, TList &tl, bool recurse=false, bool reg=false) override
bool	exists (const std::string &name) const override
	THistSvc (const std::string &name, ISvcLocator *svc)
void	handle (const Incident &) override
StatusCode	io_reinit () override
	callback method to reinitialize the internal state of the component for I/O purposes (e.g. More...
Public Member Functions inherited from extends< Service, ITHistSvc, IIncidentListener, IIoComponent >
void *	i_cast (const InterfaceID &tid) const override
	Implementation of IInterface::i_cast. More...
StatusCode	queryInterface (const InterfaceID &ti, void **pp) override
	Implementation of IInterface::queryInterface. More...
std::vector< std::string >	getInterfaceNames () const override
	Implementation of IInterface::getInterfaceNames. More...
	~extends () override=default
	Virtual destructor. More...
Public Member Functions inherited from Service
const std::string &	name () const override
	Retrieve name of the service. More...
StatusCode	configure () override
StatusCode	initialize () override
StatusCode	start () override
StatusCode	stop () override
StatusCode	finalize () override
StatusCode	terminate () override
Gaudi::StateMachine::State	FSMState () const override
Gaudi::StateMachine::State	targetFSMState () const override
StatusCode	reinitialize () override
StatusCode	restart () override
StatusCode	sysInitialize () override
	Initialize Service. More...
StatusCode	sysStart () override
	Initialize Service. More...
StatusCode	sysStop () override
	Initialize Service. More...
StatusCode	sysFinalize () override
	Finalize Service. More...
StatusCode	sysReinitialize () override
	Re-initialize the Service. More...
StatusCode	sysRestart () override
	Re-initialize the Service. More...
	Service (std::string name, ISvcLocator *svcloc)
	Standard Constructor. More...
SmartIF< ISvcLocator > &	serviceLocator () const override
	Retrieve pointer to service locator. More...
StatusCode	setProperties ()
	Method for setting declared properties to the values specified for the job. More...
template<class T >
StatusCode	service (const std::string &name, const T *&psvc, bool createIf=true) const
	Access a service by name, creating it if it doesn't already exist. More...
template<class T >
StatusCode	service (const std::string &name, T *&psvc, bool createIf=true) const
template<typename IFace = IService>
SmartIF< IFace >	service (const std::string &name, bool createIf=true) const
template<class T >
StatusCode	service (const std::string &svcType, const std::string &svcName, T *&psvc) const
	Access a service by name and type, creating it if it doesn't already exist. More...
template<class T >
StatusCode	declareTool (ToolHandle< T > &handle, std::string toolTypeAndName, bool createIf=true)
	Declare used tool. More...
SmartIF< IAuditorSvc > &	auditorSvc () const
	The standard auditor service.May not be invoked before sysInitialize() has been invoked. More...
Public Member Functions inherited from PropertyHolder< CommonMessaging< implements< IService, IProperty, IStateful > > >
	PropertyHolder ()=default
	~PropertyHolder () override=default
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Details::PropertyBase &prop)
	Declare a property. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, TYPE &value, const std::string &doc="none")
	Helper to wrap a regular data member and use it as a regular property. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, TYPE &value, const std::string &doc="none") const
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, Gaudi::Property< TYPE, VERIFIER, HANDLERS > &prop, const std::string &doc="none")
	Declare a PropertyBase instance setting name and documentation. More...
Gaudi::Details::PropertyBase *	declareRemoteProperty (const std::string &name, IProperty *rsvc, const std::string &rname="")
	Declare a remote property. More...
StatusCode	setProperty (const Gaudi::Details::PropertyBase &p) override
	set the property form another property More...
StatusCode	setProperty (const std::string &s) override
	set the property from the formatted string More...
StatusCode	setProperty (const std::string &n, const std::string &v) override
	set the property from name and the value More...
StatusCode	setProperty (const std::string &name, const TYPE &value)
	set the property form the value More...
StatusCode	getProperty (Gaudi::Details::PropertyBase *p) const override
	get the property More...
const Gaudi::Details::PropertyBase &	getProperty (const std::string &name) const override
	get the property by name More...
StatusCode	getProperty (const std::string &n, std::string &v) const override
	convert the property to the string More...
const std::vector< Gaudi::Details::PropertyBase * > &	getProperties () const override
	get all properties More...
bool	hasProperty (const std::string &name) const override
	Return true if we have a property with the given name. More...
	PropertyHolder (const PropertyHolder &)=delete
PropertyHolder &	operator= (const PropertyHolder &)=delete
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, GaudiHandleBase &ref, const std::string &doc="none")
	Specializations for various GaudiHandles. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, GaudiHandleArrayBase &ref, const std::string &doc="none")
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, DataObjectHandleBase &ref, const std::string &doc="none")
Public Member Functions inherited from CommonMessagingBase
virtual	~CommonMessagingBase ()=default
	Virtual destructor. More...
const SmartIF< IMessageSvc > &	msgSvc () const
	The standard message service. More...
MsgStream &	msgStream () const
	Return an uninitialized MsgStream. More...
MsgStream &	msgStream (const MSG::Level level) const
	Predefined configurable message stream for the efficient printouts. More...
MsgStream &	always () const
	shortcut for the method msgStream(MSG::ALWAYS) More...
MsgStream &	fatal () const
	shortcut for the method msgStream(MSG::FATAL) More...
MsgStream &	err () const
	shortcut for the method msgStream(MSG::ERROR) More...
MsgStream &	error () const
	shortcut for the method msgStream(MSG::ERROR) More...
MsgStream &	warning () const
	shortcut for the method msgStream(MSG::WARNING) More...
MsgStream &	info () const
	shortcut for the method msgStream(MSG::INFO) More...
MsgStream &	debug () const
	shortcut for the method msgStream(MSG::DEBUG) More...
MsgStream &	verbose () const
	shortcut for the method msgStream(MSG::VERBOSE) More...
MsgStream &	msg () const
	shortcut for the method msgStream(MSG::INFO) More...
MSG::Level	msgLevel () const
	get the cached level (originally extracted from the embedded MsgStream) More...
MSG::Level	outputLevel () const __attribute__((deprecated))
	Backward compatibility function for getting the output level. More...
bool	msgLevel (MSG::Level lvl) const
	get the output level from the embedded MsgStream More...
Public Member Functions inherited from extend_interfaces< Interfaces... >
	~extend_interfaces () override=default
	Virtual destructor. More...

Protected Member Functions
	~THistSvc () override=default
Protected Member Functions inherited from Service
	~Service () override
	Standard Destructor. More...
int	outputLevel () const
	get the Service's output level More...
Protected Member Functions inherited from PropertyHolder< CommonMessaging< implements< IService, IProperty, IStateful > > >
Gaudi::Details::PropertyBase *	property (const std::string &name) const
Protected Member Functions inherited from CommonMessaging< implements< IService, IProperty, IStateful > >
MSG::Level	setUpMessaging ()
	Set up local caches. More...
MSG::Level	resetMessaging ()
	Reinitialize internal states. More...
void	updateMsgStreamOutputLevel (int level)
	Update the output level of the cached MsgStream. More...

Private Types
enum	Mode {   READ, WRITE, UPDATE, APPEND,   SHARE, INVALID }
typedef std::map< std::string, THistID >	uidMap
typedef std::multimap< std::string, THistID >	idMap
typedef std::map< TObject *, THistID >	objMap
typedef std::multimap< std::string, std::string >	streamMap

Private Member Functions
template<typename T >
StatusCode	regHist_i (T *hist, const std::string &name)
template<typename T >
StatusCode	getHist_i (const std::string &name, T *&hist, bool quiet=false) const
template<typename T >
StatusCode	readHist_i (const std::string &name, T *&hist) const
StatusCode	readHist (const std::string &name, TH1 *&) const
StatusCode	readHist (const std::string &name, TH2 *&) const
StatusCode	readHist (const std::string &name, TH3 *&) const
StatusCode	readTree (const std::string &name, TTree *&) const
void	updateFiles ()
StatusCode	write ()
StatusCode	connect (const std::string &)
TDirectory *	changeDir (const THistSvc::THistID &hid) const
std::string	dirname (std::string &dir) const
void	removeDoubleSlash (std::string &) const
bool	browseTDir (TDirectory *dir) const
bool	findStream (const std::string &name, std::string &root, std::string &rem, TFile *&file) const
void	parseString (const std::string &id, std::string &root, std::string &rem) const
void	setupInputFile (Gaudi::Details::PropertyBase &inputfile)
	call-back method to handle input stream property More...
void	setupOutputFile (Gaudi::Details::PropertyBase &outputfile)
	call-back method to handle output stream property More...
void	setupCompressionLevel (Gaudi::Details::PropertyBase &cmp)
void	copyFileLayout (TDirectory *, TDirectory *)
	helper function to recursively copy the layout of a TFile into a new TFile More...
void	MergeRootFile (TDirectory *target, TDirectory *source)
StatusCode	rootOpenAction (FILEMGR_CALLBACK_ARGS)
StatusCode	rootOpenErrAction (FILEMGR_CALLBACK_ARGS)

Private Attributes
Gaudi::Property< int >	m_autoSave {this, "AutoSave", 0}
Gaudi::Property< int >	m_autoFlush {this, "AutoFlush", 0}
Gaudi::Property< bool >	m_print {this, "PrintAll", false}
Gaudi::Property< int >	m_maxFileSize
Gaudi::Property< int >	m_compressionLevel {this, "CompressionLevel", 1}
Gaudi::Property< std::vector< std::string > >	m_outputfile {this, "Output", {}}
Gaudi::Property< std::vector< std::string > >	m_inputfile {this, "Input", {}}
std::vector< std::string >	m_Rstream
std::vector< std::string >	m_Wstream
std::set< std::string >	m_alreadyConnectedInFiles
	list of already connected files. More...
std::set< std::string >	m_alreadyConnectedOutFiles
	list of already connected files. More...
uidMap	m_uids
idMap	m_ids
objMap	m_tobjs
std::map< std::string, std::pair< TFile *, Mode > >	m_files
streamMap	m_fileStreams
std::map< std::string, std::string >	m_sharedFiles
bool	signaledStop = false
bool	m_delayConnect = false
bool	m_okToConnect = false
std::string	m_curstream
IIncidentSvc *	p_incSvc = nullptr
IFileMgr *	p_fileMgr = nullptr

Additional Inherited Members
Public Types inherited from extends< Service, ITHistSvc, IIncidentListener, IIoComponent >
using	base_class = extends
	Typedef to this class. More...
using	extend_interfaces_base = extend_interfaces< Interfaces... >
	Typedef to the base of this class. More...
Public Types inherited from Service
typedef Gaudi::PluginService::Factory< IService *, const std::string &, ISvcLocator * >	Factory
Public Types inherited from PropertyHolder< CommonMessaging< implements< IService, IProperty, IStateful > > >
using	PropertyHolderImpl = PropertyHolder< CommonMessaging< implements< IService, IProperty, IStateful > > >
	Typedef used to refer to this class from derived classes, as in. More...
Public Types inherited from CommonMessaging< implements< IService, IProperty, IStateful > >
using	base_class = CommonMessaging
Public Types inherited from extend_interfaces< Interfaces... >
using	ext_iids = typename Gaudi::interface_list_cat< typename Interfaces::ext_iids... >::type
	take union of the ext_iids of all Interfaces... More...
Protected Attributes inherited from Service
Gaudi::StateMachine::State	m_state = Gaudi::StateMachine::OFFLINE
	Service state. More...
Gaudi::StateMachine::State	m_targetState = Gaudi::StateMachine::OFFLINE
	Service state. More...
Gaudi::Property< int >	m_outputLevel {this, "OutputLevel", MSG::NIL, "output level"}
Gaudi::Property< bool >	m_auditInit {this, "AuditServices", false, "[[deprecated]] unused"}
Gaudi::Property< bool >	m_auditorInitialize {this, "AuditInitialize", false, "trigger auditor on initialize()"}
Gaudi::Property< bool >	m_auditorStart {this, "AuditStart", false, "trigger auditor on start()"}
Gaudi::Property< bool >	m_auditorStop {this, "AuditStop", false, "trigger auditor on stop()"}
Gaudi::Property< bool >	m_auditorFinalize {this, "AuditFinalize", false, "trigger auditor on finalize()"}
Gaudi::Property< bool >	m_auditorReinitialize {this, "AuditReinitialize", false, "trigger auditor on reinitialize()"}
Gaudi::Property< bool >	m_auditorRestart {this, "AuditRestart", false, "trigger auditor on restart()"}
SmartIF< IAuditorSvc >	m_pAuditorSvc
	Auditor Service. More...

Detailed Description

Definition at line 25 of file THistSvc.h.

Member Typedef Documentation

typedef std::multimap<std::string, THistID> THistSvc::idMap

private

Definition at line 174 of file THistSvc.h.

typedef std::map<TObject*, THistID> THistSvc::objMap

private

Definition at line 175 of file THistSvc.h.

typedef std::multimap<std::string, std::string> THistSvc::streamMap

private

Definition at line 176 of file THistSvc.h.

typedef std::map<std::string, THistID> THistSvc::uidMap

private

Definition at line 173 of file THistSvc.h.

Member Enumeration Documentation

enum THistSvc::Mode

private

Enumerator
READ
WRITE
UPDATE
APPEND
SHARE
INVALID

Definition at line 94 of file THistSvc.h.

{ READ, WRITE, UPDATE, APPEND, SHARE, INVALID };

Constructor & Destructor Documentation

THistSvc::THistSvc	(	const std::string &	name,
		ISvcLocator *	svc
	)

Definition at line 61 of file THistSvc.cpp.

                                                            : base_class( name, svc )
{
  m_compressionLevel.declareUpdateHandler( &THistSvc::setupCompressionLevel, this );
  m_outputfile.declareUpdateHandler( &THistSvc::setupOutputFile, this );
  m_inputfile.declareUpdateHandler( &THistSvc::setupInputFile, this );
}

THistSvc::~THistSvc ( )

overrideprotecteddefault

Member Function Documentation

bool THistSvc::browseTDir ( TDirectory *  dir ) const

private

Definition at line 344 of file THistSvc.cpp.

{

  if ( !dir ) {
    std::cerr << "TDirectory == 0" << std::endl;
    return false;
  }

  GlobalDirectoryRestore restore;

  dir->cd();

  cout << "-> " << dir->GetPath() << "  " << dir->GetListOfKeys()->GetSize() << endl;

  //  TIter nextkey(dir->GetListOfKeys());
  TIter nextkey( dir->GetList() );
  while ( TKey* key = (TKey*)nextkey() ) {

    TObject* obj = key->ReadObj();
    if ( !obj ) {
      cout << key->GetName() << " obj==0" << endl;
      continue;
    }
    //    if (obj->IsA()->InheritsFrom("TDirectory")) {
    cout << "  Key: " << key->GetName() << "   "
         << " tit: " << obj->GetTitle() << "   "
         << " (" << key->GetClassName() << ")" << endl;
    //    }
  }

  nextkey = dir->GetListOfKeys();
  while ( TKey* key = (TKey*)nextkey() ) {

    TObject* obj = key->ReadObj();
    if ( !obj ) {
      cout << key->GetName() << " obj==0" << endl;
      continue;
    }
    if ( obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      TDirectory* tt = dynamic_cast<TDirectory*>( obj );
      browseTDir( tt );
    }
  }

  return true;
}

TDirectory * THistSvc::changeDir ( const THistSvc::THistID &  hid ) const

private

Definition at line 1411 of file THistSvc.cpp.

{

  string uid  = hid.id;
  TFile* file = hid.file;
  string stream, fdir, bdir, dir, id;

  if ( file ) {
    file->cd( "/" );
  } else {
    gROOT->cd();
  }

  fdir = uid;
  bdir = dirname( fdir );

  while ( ( dir = dirname( fdir ) ) != "" ) {
    if ( !gDirectory->GetKey( dir.c_str() ) ) {
      gDirectory->mkdir( dir.c_str() );
    }
    gDirectory->cd( dir.c_str() );
  }

  return gDirectory;
}

StatusCode THistSvc::connect ( const std::string &  ident )

private

Definition at line 1217 of file THistSvc.cpp.

{

  auto loc      = ident.find( " " );
  string stream = ident.substr( 0, loc );
  char typ( 0 );
  typedef std::pair<std::string, std::string> Prop;
  std::vector<Prop> props;
  string filename, db_typ( "ROOT" );
  int cl( 1 );

  if ( loc != string::npos ) {
    using Parser = Gaudi::Utils::AttribStringParser;
    for ( auto attrib : Parser( ident.substr( loc + 1 ) ) ) {
      auto TAG = boost::algorithm::to_upper_copy( attrib.tag );
      auto VAL = boost::algorithm::to_upper_copy( attrib.value );

      if ( TAG == "FILE" || TAG == "DATAFILE" ) {
        filename = attrib.value;
        removeDoubleSlash( filename );
      } else if ( TAG == "OPT" ) {
        if ( VAL == "APPEND" || VAL == "UPDATE" ) {
          typ = 'A';
        } else if ( VAL == "CREATE" || VAL == "NEW" || VAL == "WRITE" ) {
          typ = 'N';
        } else if ( VAL == "RECREATE" ) {
          typ = 'R';
        } else if ( VAL == "SHARE" ) {
          typ = 'S';
        } else if ( VAL == "OLD" || VAL == "READ" ) {
          typ = 'O';
        } else {
          error() << "Unknown OPT: \"" << attrib.value << "\"" << endmsg;
          typ = 0;
        }
      } else if ( TAG == "TYP" ) {
        db_typ = std::move( attrib.value );
      } else if ( TAG == "CL" ) {
        cl = std::stoi( attrib.value );
      } else {
        props.emplace_back( attrib.tag, attrib.value );
      }
    }
  }

  if ( stream == "temp" ) {
    error() << "in JobOption \"" << ident << "\": stream name \"temp\" reserved." << endmsg;
    return StatusCode::FAILURE;
  }

  if ( db_typ != "ROOT" ) {
    error() << "in JobOption \"" << ident << "\": technology type \"" << db_typ << "\" not supported." << endmsg;
    return StatusCode::FAILURE;
  }

  if ( m_files.find( stream ) != m_files.end() ) {
    error() << "in JobOption \"" << ident << "\":\n stream \"" << stream << "\" already connected to file: \""
            << m_files[stream].first->GetName() << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  Mode newMode;
  if ( typ == 'O' ) {
    newMode = THistSvc::READ;
  } else if ( typ == 'N' ) {
    newMode = THistSvc::WRITE;
  } else if ( typ == 'A' ) {
    newMode = THistSvc::APPEND;
  } else if ( typ == 'R' ) {
    newMode = THistSvc::UPDATE;
  } else if ( typ == 'S' ) {
    newMode = THistSvc::SHARE;
  } else {
    // something else?
    error() << "No OPT= specified or unknown access mode in: " << ident << endmsg;
    return StatusCode::FAILURE;
  }

  // Is this file already connected to another stream?
  if ( m_fileStreams.find( filename ) != m_fileStreams.end() ) {
    auto fitr = m_fileStreams.equal_range( filename );

    const std::string& oldstream = fitr.first->second;

    const auto& f_info = m_files[oldstream];

    if ( newMode != f_info.second ) {
      error() << "in JobOption \"" << ident << "\":\n file \"" << filename << "\" already opened by stream: \""
              << oldstream << "\" with different access mode." << endmsg;
      return StatusCode::FAILURE;
    } else {
      TFile* f2       = f_info.first;
      m_files[stream] = make_pair( f2, newMode );
      if ( msgLevel( MSG::DEBUG ) )
        debug() << "Connecting stream: \"" << stream << "\" to previously opened TFile: \"" << filename << "\""
                << endmsg;
      return StatusCode::SUCCESS;
    }
  }

  IIncidentSvc* pi = nullptr;
  if ( service( "IncidentSvc", pi ).isFailure() ) {
    error() << "Unable to get the IncidentSvc" << endmsg;
    return StatusCode::FAILURE;
  }

  void* vf = nullptr;
  TFile* f = nullptr;

  if ( newMode == THistSvc::READ ) {
    // old file

    int r = p_fileMgr->open( Io::ROOT, name(), filename, Io::READ, vf, "HIST" );

    if ( r != 0 ) {
      error() << "Unable to open ROOT file " << filename << " for reading" << endmsg;
      return StatusCode::FAILURE;
    }

    f = (TFile*)vf;

    // FIX ME!
    pi->fireIncident( FileIncident( name(), "BeginHistFile", filename ) );

  } else if ( newMode == THistSvc::WRITE ) {
    // new file. error if file exists

    int r = p_fileMgr->open( Io::ROOT, name(), filename, ( Io::WRITE | Io::CREATE | Io::EXCL ), vf, "HIST" );

    if ( r != 0 ) {
      error() << "Unable to open ROOT file " << filename << " for writing" << endmsg;
      return StatusCode::FAILURE;
    }

    f = (TFile*)vf;

  } else if ( newMode == THistSvc::APPEND ) {
    // update file

    int r = p_fileMgr->open( Io::ROOT, name(), filename, ( Io::WRITE | Io::APPEND ), vf, "HIST" );
    if ( r != 0 ) {
      error() << "unable to open file \"" << filename << "\" for appending" << endmsg;
      return StatusCode::FAILURE;
    }

    f = (TFile*)vf;

  } else if ( newMode == THistSvc::SHARE ) {
    // SHARE file type
    // For SHARE files, all data will be stored in a temp file and will be merged into the target file
    // in write() when finalize(), this help to solve some confliction. e.g. with storegate

    static int ishared  = 0;
    string realfilename = filename;
    filename            = "tmp_THistSvc_" + std::to_string( ishared++ ) + ".root";

    if ( msgLevel( MSG::DEBUG ) )
      debug() << "Creating temp file \"" << filename << "\" and realfilename=" << realfilename << endmsg;
    m_sharedFiles[stream] = realfilename;

    int r = p_fileMgr->open( Io::ROOT, name(), filename, ( Io::WRITE | Io::CREATE | Io::EXCL ), vf, "HIST" );

    if ( r != 0 ) {
      error() << "Unable to open ROOT file " << filename << " for writing" << endmsg;
      return StatusCode::FAILURE;
    }

    f = (TFile*)vf;

  } else if ( newMode == THistSvc::UPDATE ) {
    // update file

    int r = p_fileMgr->open( Io::ROOT, name(), filename, ( Io::WRITE | Io::CREATE ), vf, "HIST" );

    if ( r != 0 ) {
      error() << "Unable to open ROOT file " << filename << " for appending" << endmsg;
      return StatusCode::FAILURE;
    }

    f = (TFile*)vf;
  }

  m_files[stream] = make_pair( f, newMode );
  m_fileStreams.insert( make_pair( filename, stream ) );

  if ( msgLevel( MSG::DEBUG ) )
    debug() << "Opening TFile \"" << filename << "\"  stream: \"" << stream << "\"  mode: \"" << typ << "\""
            << " comp level: " << cl << endmsg;

  return StatusCode::SUCCESS;
}

void THistSvc::copyFileLayout ( TDirectory *  dst, TDirectory *  src  )

private

helper function to recursively copy the layout of a TFile into a new TFile

Definition at line 1609 of file THistSvc.cpp.

{

  if ( msgLevel( MSG::DEBUG ) ) debug() << "copyFileLayout() to dst path: " << dst->GetPath() << endmsg;

  // strip out URLs
  TString path( (char*)strstr( dst->GetPath(), ":" ) );
  path.Remove( 0, 2 );

  src->cd( path );
  TDirectory* cur_src_dir = gDirectory;

  // loop over all keys in this directory
  TList* key_list = cur_src_dir->GetListOfKeys();
  int n           = key_list->GetEntries();
  for ( int j = 0; j < n; ++j ) {
    TKey* k                        = (TKey*)key_list->At( j );
    const std::string src_pathname = cur_src_dir->GetPath() + std::string( "/" ) + k->GetName();
    TObject* o                     = src->Get( src_pathname.c_str() );

    if ( o && o->IsA()->InheritsFrom( "TDirectory" ) ) {
      if ( msgLevel( MSG::VERBOSE ) ) verbose() << " subdir [" << o->GetName() << "]..." << endmsg;
      dst->cd();
      TDirectory* dst_dir = dst->mkdir( o->GetName(), o->GetTitle() );
      copyFileLayout( dst_dir, src );
    }
  } // loop over keys
  return;
}

StatusCode THistSvc::deReg ( TObject *  obj )

override

Definition at line 795 of file THistSvc.cpp.

{

  auto itr = m_tobjs.find( obj );
  if ( itr != m_tobjs.end() ) {
    THistID hid = itr->second;

    auto itr2 = m_uids.find( hid.id );
    if ( itr2 == m_uids.end() ) {
      error() << "Problems deregistering TObject \"" << obj->GetName() << "\" with id \"" << hid.id << "\"" << endmsg;
      return StatusCode::FAILURE;
    }

    std::string id, root, rem;
    parseString( hid.id, root, rem );

    auto mitr = m_ids.equal_range( rem );
    auto itr3 =
        std::find_if( mitr.first, mitr.second, [&]( idMap::const_reference i ) { return i.second.obj == obj; } );
    if ( itr3 == mitr.second ) {
      error() << "Problems deregistering TObject \"" << obj->GetName() << "\" with id \"" << hid.id << "\"" << endmsg;
      return StatusCode::FAILURE;
    }

    m_tobjs.erase( itr );
    m_uids.erase( itr2 );
    m_ids.erase( itr3 );

    return StatusCode::SUCCESS;

  } else {
    error() << "Cannot unregister TObject \"" << obj->GetName() << "\": not known to THistSvc" << endmsg;
    return StatusCode::FAILURE;
  }
}

StatusCode THistSvc::deReg ( const std::string &  name )

override

Definition at line 833 of file THistSvc.cpp.

{

  auto itr = m_uids.find( id );
  if ( itr == m_uids.end() ) {
    error() << "Problems deregistering id \"" << id << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  TObject* obj = itr->second.obj;
  return deReg( obj );
}

std::string THistSvc::dirname ( std::string &  dir ) const

private

Definition at line 1439 of file THistSvc.cpp.

{

  string::size_type i = dir.find( "/" );

  if ( i == string::npos ) return {};

  if ( i == 0 ) {
    dir.erase( 0, 1 );
    return dirname( dir );
  }

  string root = dir.substr( 0, i );
  dir.erase( 0, i );

  return root;
}

bool THistSvc::exists ( const std::string &  name ) const

override

Definition at line 1552 of file THistSvc.cpp.

{

  TH1* h;
  return getHist_i( name, h, true ).isSuccess();
}

StatusCode THistSvc::finalize ( )

override

Definition at line 199 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  if ( msgLevel( MSG::DEBUG ) ) debug() << "THistSvc::finalize" << endmsg;

#ifndef NDEBUG
  if ( msgLevel( MSG::DEBUG ) ) {
    for ( const auto& uid : m_uids ) {

      TObject* to = uid.second.obj;

      string dirname( "none" );
      if ( to && to->IsA()->InheritsFrom( "TTree" ) ) {
        TTree* tr = dynamic_cast<TTree*>( to );
        if ( tr->GetDirectory() != 0 ) {
          dirname = tr->GetDirectory()->GetPath();
        }
      } else if ( to && to->IsA()->InheritsFrom( "TGraph" ) ) {
        if ( !uid.second.temp ) {
          dirname = uid.second.file->GetPath();
          string id2( uid.second.id );
          id2.erase( 0, id2.find( "/", 1 ) );
          id2.erase( id2.rfind( "/" ), id2.length() );
          if ( id2.find( "/" ) == 0 ) {
            id2.erase( 0, 1 );
          }
          dirname += id2;
        } else {
          dirname = "/tmp";
        }
      } else if ( to && to->IsA()->InheritsFrom( "TH1" ) ) {
        TH1* th = dynamic_cast<TH1*>( to );
        if ( th == 0 ) {
          error() << "Couldn't dcast: " << uid.first << endmsg;
        } else {
          if ( th->GetDirectory() != 0 ) {
            dirname = th->GetDirectory()->GetPath();
          }
        }
      } else if ( !to ) {
        warning() << uid.first << " has NULL TObject ptr" << endmsg;
      }

      debug() << "uid: \"" << uid.first << "\"  temp: " << uid.second.temp << "  dir: " << dirname << endmsg;
    }
  }
#endif

  StatusCode sc = write();
  if ( sc.isFailure() ) {
    error() << "problems writing histograms" << endmsg;
  }

  if ( m_print ) {
    info() << "Listing contents of ROOT files: " << endmsg;
  }
  vector<TFile*> deleted_files;
  for ( auto& itr : m_files ) {

    if ( find( deleted_files.begin(), deleted_files.end(), itr.second.first ) == deleted_files.end() ) {
      deleted_files.push_back( itr.second.first );

#ifndef NDEBUG
      if ( msgLevel( MSG::DEBUG ) )
        debug() << "finalizing stream/file " << itr.first << ":" << itr.second.first->GetName() << endmsg;
#endif
    } else {
#ifndef NDEBUG
      if ( msgLevel( MSG::DEBUG ) ) debug() << "already finalized stream " << itr.first << endmsg;
#endif
      continue;
    }

    if ( m_print && msgLevel( MSG::INFO ) ) {

      info() << "==> File: " << itr.second.first->GetName() << "  stream: " << itr.first << endmsg;

      itr.second.first->Print( "base" );
    }

    string tmpfn = itr.second.first->GetName();

    p_fileMgr->close( itr.second.first, name() );

    IIncidentSvc* pi = nullptr;
    if ( service( "IncidentSvc", pi ).isFailure() ) {
      error() << "Unable to get the IncidentSvc" << endmsg;
      return StatusCode::FAILURE;
    }

    if ( itr.second.second == SHARE ) {

      // Merge File
      void* vf = nullptr;
      int r    = p_fileMgr->open( Io::ROOT, name(), m_sharedFiles[itr.first], Io::WRITE | Io::APPEND, vf, "HIST" );

      if ( r ) {
        error() << "unable to open Final Output File: \"" << m_sharedFiles[itr.first] << "\" for merging" << endmsg;
        return StatusCode::FAILURE;
      }

      TFile* outputfile = (TFile*)vf;
      pi->fireIncident( FileIncident( name(), IncidentType::WroteToOutputFile, m_sharedFiles[itr.first] ) );

      if ( msgLevel( MSG::DEBUG ) ) debug() << "THistSvc::write()::Merging Rootfile " << endmsg;

      vf = nullptr;
      r  = p_fileMgr->open( Io::ROOT, name(), tmpfn, Io::READ, vf, "HIST" );

      if ( r ) {
        error() << "unable to open temporary file: \"" << tmpfn << endmsg;
        return StatusCode::FAILURE;
      }

      TFile* inputfile = (TFile*)vf;

      outputfile->SetCompressionLevel( inputfile->GetCompressionLevel() );

      MergeRootFile( outputfile, inputfile );

      outputfile->Write();
      p_fileMgr->close( outputfile, name() );
      p_fileMgr->close( inputfile, name() );

      if ( msgLevel( MSG::DEBUG ) ) debug() << "Trying to remove temporary file \"" << tmpfn << "\"" << endmsg;

      std::remove( tmpfn.c_str() );
    }
    delete itr.second.first;
  }

  m_sharedFiles.clear();
  m_fileStreams.clear();
  m_files.clear();
  m_uids.clear();
  m_ids.clear();
  m_tobjs.clear();

  return Service::finalize();
}

bool THistSvc::findStream ( const std::string &  name, std::string &  root, std::string &  rem, TFile *&  file  ) const

private

Definition at line 985 of file THistSvc.cpp.

{

  auto pos = id.find( "/" );

  if ( pos == string::npos ) {
    stream = "temp";
    rem    = id;
  } else if ( pos != 0 ) {
    stream = "temp";
    rem    = id;
  } else {

    auto pos2 = id.find( "/", pos + 1 );

    if ( pos2 == string::npos ) {
      error() << "badly formed Hist/Tree id: \"" << id << "\"" << endmsg;
      return false;
    }

    parseString( id, stream, rem );
  }

  if ( stream == "temp" ) {
    file = nullptr;
    return true;
  }

  auto itr = m_files.find( stream );
  file     = ( itr != m_files.end() ? itr->second.first : nullptr );
  if ( !file ) {
    warning() << "no stream \"" << stream << "\" associated with id: \"" << id << "\"" << endmsg;
  }

  return true;
}

StatusCode THistSvc::getGraph ( const std::string &  name, TGraph *&  hist  ) const

override

Definition at line 954 of file THistSvc.cpp.

{ return getHist_i( id, hist ); }

std::vector< std::string > THistSvc::getGraphs ( ) const

override

Definition at line 958 of file THistSvc.cpp.

{

  std::vector<std::string> names;
  names.reserve( m_uids.size() );
  transform_if( std::begin( m_uids ), std::end( m_uids ), std::back_inserter( names ), select1st,
                []( uidMap::const_reference i ) { return i.second.obj->IsA()->InheritsFrom( "TTree" ); } );
  return names;
}

StatusCode THistSvc::getHist ( const std::string &  name, TH1 *&  hist  ) const

override

Definition at line 916 of file THistSvc.cpp.

{ return getHist_i( id, hist ); }

StatusCode THistSvc::getHist ( const std::string &  name, TH2 *&  hist  ) const

override

Definition at line 920 of file THistSvc.cpp.

{ return getHist_i( id, hist ); }

StatusCode THistSvc::getHist ( const std::string &  name, TH3 *&  hist  ) const

override

Definition at line 924 of file THistSvc.cpp.

{ return getHist_i( id, hist ); }

template<typename T >

StatusCode THistSvc::getHist_i ( const std::string &  name, T *&  hist, bool  quiet = false  ) const

private

Definition at line 142 of file THistSvc.icc.

                                                                   {
  // id starts with "/": unique

  GlobalDirectoryRestore restore;

  std::string idr(id);
  removeDoubleSlash( idr );

  if (idr.find("/") == 0) {
    auto itr = m_uids.find(id);
    if (itr == m_uids.end()) {
      if (!quiet) {
    error() << "Could not locate Hist with id \"" << idr << "\""
          << endmsg;
      }
      hist = nullptr;
      return StatusCode::FAILURE;
    }

    THistID hid = itr->second;
    if (!quiet) {
      hist = dynamic_cast<T*>(hid.obj);
      if ( !hist ) {
    error() << "dcast failed, Hist id: \"" << idr << "\""
          << endmsg;
    return StatusCode::FAILURE;
      }
      if (msgLevel(MSG::VERBOSE)) {
    verbose() << "found unique Hist title: \""
          << hist->GetTitle()
          << "\"  id: \"" << idr << "\"" << endmsg;
      }
    } else {
      if (msgLevel(MSG::VERBOSE)) {
    verbose() << "found unique Hist id: \"" << idr
          << "\" type: \"" << hid.obj->IsA()->GetName() << "\""
          << endmsg;
      }
    }

    return StatusCode::SUCCESS;


    // not necessarily unique
  } else {

      auto mitr = m_ids.equal_range(idr);


    if (mitr.first == mitr.second) {
      error() << "Could not locate Hist with id \"" << idr << "\""
        << endmsg;
      hist = nullptr;
      return StatusCode::FAILURE;
    } else {

      if (distance(mitr.first,mitr.second) == 1) {
        THistID hid = mitr.first->second;
    if (!quiet) {
      hist = dynamic_cast<T*>(hid.obj);
      if (hist == 0) {
        error() << "dcast failed" << endmsg;
        return StatusCode::FAILURE;
      }
      if (msgLevel(MSG::VERBOSE)) {
        verbose() << "found Hist title: \"" << hist->GetTitle()
          << "\"  id: \"" << idr << "\"" << endmsg;
      }
    } else {
      if (msgLevel(MSG::VERBOSE)) {
        verbose() << "found Hist id: \"" << idr << "\" type: \""
          << hid.obj->IsA()->GetName() << "\""
          << endmsg;
      }
    }
        return StatusCode::SUCCESS;
      } else {
    if (!quiet) {
      // return failure if trying to GET a single hist
      error() << "Multiple matches with id \"" << idr << "\"."
        << " Further specifications required."
        << endmsg;
      hist = nullptr;
      return StatusCode::FAILURE;
    } else {
      // return a SUCCESS if just INQUIRING
      info() << "Found multiple matches with id \"" << idr
          << "\"" << endmsg;
      hist = nullptr;
      return StatusCode::SUCCESS;
    }
      }
    }
  }
}

std::vector< std::string > THistSvc::getHists ( ) const

override

Definition at line 928 of file THistSvc.cpp.

{

  std::vector<std::string> names;
  names.reserve( m_uids.size() );
  transform_if( std::begin( m_uids ), std::end( m_uids ), std::back_inserter( names ), select1st,
                []( uidMap::const_reference i ) { return i.second.obj->IsA()->InheritsFrom( "TH1" ); } );
  return names;
}

StatusCode THistSvc::getTHists ( TDirectory *  td, TList &  tl, bool  recurse = false  ) const

override

Definition at line 393 of file THistSvc.cpp.

{
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if ( !td->cd() ) {
    error() << "getTHists: No such TDirectory \"" << td->GetPath() << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  if ( msgLevel( MSG::DEBUG ) )
    debug() << "getTHists: \"" << td->GetPath() << "\": found " << td->GetListOfKeys()->GetSize() << " keys" << endmsg;

  TIter nextkey( td->GetListOfKeys() );
  while ( TKey* key = (TKey*)nextkey() ) {
    auto& log = debug();
    if ( msgLevel( MSG::DEBUG ) ) log << "  key: " << key->GetName();
    TObject* obj = key->ReadObj();
    if ( obj != 0 && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
    } else if ( obj != 0 && obj->IsA()->InheritsFrom( "TH1" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
      tl.Add( obj );
    } else if ( obj != 0 ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " [" << obj->IsA()->GetName() << "]";
    }
    if ( msgLevel( MSG::DEBUG ) ) log << endmsg;
  }

  // operate recursively
  if ( rcs ) {
    nextkey = td->GetListOfKeys();
    while ( TKey* key = (TKey*)nextkey() ) {
      TObject* obj = key->ReadObj();
      if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
        TDirectory* tt = dynamic_cast<TDirectory*>( obj );
        getTHists( tt, tl, rcs );
      }
    }
  }

  return StatusCode::SUCCESS;
}

StatusCode THistSvc::getTHists ( const std::string &  name, TList &  tl, bool  recurse = false  ) const

override

Definition at line 440 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  std::string stream, rem, r2;
  parseString( dir, stream, rem );

  auto itr = m_files.find( stream );
  if ( itr != m_files.end() ) {
    r2 = itr->second.first->GetName();
    r2 += ":/";
    r2 += rem;

    if ( msgLevel( MSG::DEBUG ) )
      debug() << "getTHists: \"" << dir << "\" looks like a stream name."
              << " associated TFile: \"" << itr->second.first->GetName() << "\"" << endmsg;

    if ( gDirectory->cd( r2.c_str() ) ) {
      m_curstream = stream;
      sc          = getTHists( gDirectory, tl, rcs );
      m_curstream = "";
      return sc;
    } else {
      if ( msgLevel( MSG::DEBUG ) ) debug() << "getTHists: no such TDirectory \"" << r2 << "\"" << endmsg;
    }

  } else {
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTHists: stream \"" << stream << "\" not found" << endmsg;
  }

  if ( !gDirectory->cd( dir.c_str() ) ) {
    error() << "getTHists: No such TDirectory/stream \"" << dir << "\"" << endmsg;
    sc = StatusCode::FAILURE;
  } else {
    sc = getTHists( gDirectory, tl, rcs );
  }

  return sc;
}

StatusCode THistSvc::getTHists ( TDirectory *  td, TList &  tl, bool  recurse = false, bool  reg = false  )

override

Definition at line 573 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if ( !td->cd() ) {
    error() << "getTHists: No such TDirectory \"" << td->GetPath() << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  if ( msgLevel( MSG::DEBUG ) )
    debug() << "getTHists: \"" << td->GetPath() << "\": found " << td->GetListOfKeys()->GetSize() << " keys" << endmsg;

  TIter nextkey( td->GetListOfKeys() );
  while ( TKey* key = (TKey*)nextkey() ) {
    auto& log = debug();
    if ( msgLevel( MSG::DEBUG ) ) log << "  key: " << key->GetName();
    TObject* obj = key->ReadObj();
    if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
    } else if ( obj && obj->IsA()->InheritsFrom( "TH1" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
      tl.Add( obj );
      if ( reg && m_curstream != "" ) {
        string dir = td->GetPath();
        string fil = td->GetFile()->GetName();
        dir.erase( 0, fil.length() + 1 );
        string id = "/" + m_curstream;
        if ( dir == "/" ) {
          id = id + "/" + key->GetName();
        } else {
          id = id + dir + "/" + key->GetName();
        }
        if ( !exists( id ) ) {
          if ( msgLevel( MSG::DEBUG ) ) log << "  reg as \"" << id << "\"";
          regHist( id ).ignore();
        } else {
          if ( msgLevel( MSG::DEBUG ) ) log << "  already registered";
        }
      }
    } else if ( obj ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " [" << obj->IsA()->GetName() << "]";
    }
    if ( msgLevel( MSG::DEBUG ) ) log << endmsg;
  }

  // operate recursively
  if ( rcs ) {
    nextkey = td->GetListOfKeys();
    while ( TKey* key = (TKey*)nextkey() ) {
      TObject* obj = key->ReadObj();
      if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
        TDirectory* tt = dynamic_cast<TDirectory*>( obj );
        getTHists( tt, tl, rcs, reg );
      }
    }
  }

  return StatusCode::SUCCESS;
}

StatusCode THistSvc::getTHists ( const std::string &  name, TList &  tl, bool  recurse = false, bool  reg = false  )

override

Definition at line 638 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  std::string stream, rem, r2;
  parseString( dir, stream, rem );

  auto itr = m_files.find( stream );
  if ( itr != m_files.end() ) {
    r2 = itr->second.first->GetName();
    r2 += ":/";
    r2 += rem;

    if ( msgLevel( MSG::DEBUG ) )
      debug() << "getTHists: \"" << dir << "\" looks like a stream name."
              << " associated TFile: \"" << itr->second.first->GetName() << "\"" << endmsg;

    if ( gDirectory->cd( r2.c_str() ) ) {
      m_curstream = stream;
      sc          = getTHists( gDirectory, tl, rcs, reg );
      m_curstream.clear();
      return sc;
    }
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTHists: no such TDirectory \"" << r2 << "\"" << endmsg;

  } else {
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTHists: stream \"" << stream << "\" not found" << endmsg;
  }

  if ( !gDirectory->cd( dir.c_str() ) ) {
    error() << "getTHists: No such TDirectory/stream \"" << dir << "\"" << endmsg;
    sc = StatusCode::FAILURE;
  } else {
    if ( reg ) {
      warning() << "Unable to register histograms automatically "
                << "without a valid stream name" << endmsg;
      reg = false;
    }
    sc = getTHists( gDirectory, tl, rcs, reg );
  }

  return sc;
}

StatusCode THistSvc::getTree ( const std::string &  name, TTree *&  hist  ) const

override

Definition at line 939 of file THistSvc.cpp.

{ return getHist_i( id, hist ); }

std::vector< std::string > THistSvc::getTrees ( ) const

override

Definition at line 943 of file THistSvc.cpp.

{

  std::vector<std::string> names;
  names.reserve( m_uids.size() );
  transform_if( std::begin( m_uids ), std::end( m_uids ), std::back_inserter( names ), select1st,
                []( uidMap::const_reference i ) { return i.second.obj->IsA()->InheritsFrom( "TTree" ); } );
  return names;
}

StatusCode THistSvc::getTTrees ( TDirectory *  td, TList &  tl, bool  recurse = false  ) const

override

Definition at line 486 of file THistSvc.cpp.

{
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if ( !td->cd() ) {
    error() << "getTTrees: No such TDirectory \"" << td->GetPath() << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  if ( msgLevel( MSG::DEBUG ) )
    debug() << "getTHists: \"" << td->GetPath() << "\": found " << td->GetListOfKeys()->GetSize() << " keys" << endmsg;

  TIter nextkey( td->GetListOfKeys() );
  while ( TKey* key = (TKey*)nextkey() ) {
    auto& log = debug();
    if ( msgLevel( MSG::DEBUG ) ) log << "  key: " << key->GetName();
    TObject* obj = key->ReadObj();
    if ( obj != 0 && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
    } else if ( obj != 0 && obj->IsA()->InheritsFrom( "TTree" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
      tl.Add( obj );
    } else if ( obj != 0 ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " [" << obj->IsA()->GetName() << "]";
    }
    log << endmsg;
  }

  // operate recursively
  if ( rcs ) {
    nextkey = td->GetListOfKeys();
    while ( TKey* key = (TKey*)nextkey() ) {
      TObject* obj = key->ReadObj();
      if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
        TDirectory* tt = dynamic_cast<TDirectory*>( obj );
        getTTrees( tt, tl, rcs );
      }
    }
  }

  return StatusCode::SUCCESS;
}

StatusCode THistSvc::getTTrees ( const std::string &  name, TList &  tl, bool  recurse = false  ) const

override

Definition at line 533 of file THistSvc.cpp.

{
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  std::string stream, rem, r2;
  parseString( dir, stream, rem );

  auto itr = m_files.find( stream );
  if ( itr != m_files.end() ) {
    r2 = itr->second.first->GetName();
    r2 += ":/";
    r2 += rem;

    if ( msgLevel( MSG::DEBUG ) )
      debug() << "getTTrees: \"" << dir << "\" looks like a stream name."
              << " associated TFile: \"" << itr->second.first->GetName() << "\"" << endmsg;

    if ( gDirectory->cd( r2.c_str() ) ) {
      return getTTrees( gDirectory, tl, rcs );
    }
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTTrees: no such TDirectory \"" << r2 << "\"" << endmsg;
  } else {
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTTrees: stream \"" << stream << "\" not found" << endmsg;
  }

  if ( !gDirectory->cd( dir.c_str() ) ) {
    error() << "getTTrees: No such TDirectory/stream \"" << dir << "\"" << endmsg;
    sc = StatusCode::FAILURE;
  } else {
    sc = getTTrees( gDirectory, tl, rcs );
  }
  return sc;
}

StatusCode THistSvc::getTTrees ( TDirectory *  td, TList &  tl, bool  recurse = false, bool  reg = false  )

override

Definition at line 688 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if ( !td->cd() ) {
    error() << "getTTrees: No such TDirectory \"" << td->GetPath() << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  if ( msgLevel( MSG::DEBUG ) )
    debug() << "getTHists: \"" << td->GetPath() << "\": found " << td->GetListOfKeys()->GetSize() << " keys" << endmsg;

  TIter nextkey( td->GetListOfKeys() );
  while ( TKey* key = (TKey*)nextkey() ) {
    auto& log = debug();
    if ( msgLevel( MSG::DEBUG ) ) log << "  key: " << key->GetName();
    TObject* obj = key->ReadObj();
    if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
    } else if ( obj && obj->IsA()->InheritsFrom( "TTree" ) ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " (" << obj->IsA()->GetName() << ")";
      tl.Add( obj );
      if ( reg && m_curstream != "" ) {
        string dir = td->GetPath();
        string fil = td->GetFile()->GetName();
        dir.erase( 0, fil.length() + 1 );
        string id = "/" + m_curstream;
        if ( dir == "/" ) {
          id = id + "/" + key->GetName();
        } else {
          id = id + dir + "/" + key->GetName();
        }
        if ( !exists( id ) ) {
          if ( msgLevel( MSG::DEBUG ) ) log << "  reg as \"" << id << "\"";
          regHist( id ).ignore();
        } else {
          if ( msgLevel( MSG::DEBUG ) ) log << "  already registered";
        }
      }
    } else if ( obj != 0 ) {
      if ( msgLevel( MSG::DEBUG ) ) log << " [" << obj->IsA()->GetName() << "]";
    }
    if ( msgLevel( MSG::DEBUG ) ) log << endmsg;
  }

  // operate recursively
  if ( rcs ) {
    nextkey = td->GetListOfKeys();
    while ( TKey* key = (TKey*)nextkey() ) {
      TObject* obj = key->ReadObj();
      if ( obj && obj->IsA()->InheritsFrom( "TDirectory" ) ) {
        TDirectory* tt = dynamic_cast<TDirectory*>( obj );
        getTTrees( tt, tl, rcs, reg );
      }
    }
  }

  return StatusCode::SUCCESS;
}

StatusCode THistSvc::getTTrees ( const std::string &  name, TList &  tl, bool  recurse = false, bool  reg = false  )

override

Definition at line 753 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  std::string stream, rem, r2;
  parseString( dir, stream, rem );

  auto itr = m_files.find( stream );
  if ( itr != m_files.end() ) {
    r2 = itr->second.first->GetName();
    r2 += ":/";
    r2 += rem;

    if ( msgLevel( MSG::DEBUG ) )
      debug() << "getTTrees: \"" << dir << "\" looks like a stream name."
              << " associated TFile: \"" << itr->second.first->GetName() << "\"" << endmsg;

    if ( gDirectory->cd( r2.c_str() ) ) {
      return getTTrees( gDirectory, tl, rcs, reg );
    } else {
      if ( msgLevel( MSG::DEBUG ) ) debug() << "getTTrees: no such TDirectory \"" << r2 << "\"" << endmsg;
    }

  } else {
    if ( msgLevel( MSG::DEBUG ) ) debug() << "getTTrees: stream \"" << stream << "\" not found" << endmsg;
  }

  if ( !gDirectory->cd( dir.c_str() ) ) {
    error() << "getTTrees: No such TDirectory/stream \"" << dir << "\"" << endmsg;
    return StatusCode::FAILURE;
  }

  return getTTrees( gDirectory, tl, rcs, reg );
}

void THistSvc::handle ( const Incident &  )

override

Definition at line 1561 of file THistSvc.cpp.

{

  if ( signaledStop ) return;

  if ( m_maxFileSize.value() == -1 ) return;

  // convert to bytes.
  Long64_t mfs      = (Long64_t)m_maxFileSize.value() * (Long64_t)1048576;
  Long64_t mfs_warn = mfs * 95 / 100;

  updateFiles();

  map<string, pair<TFile*, Mode>>::const_iterator itr;
  for ( const auto& f : m_files ) {
    TFile* tf = f.second.first;

#ifndef NDEBUG
    if ( msgLevel( MSG::DEBUG ) )
      debug() << "stream: " << f.first << "  name: " << tf->GetName() << "  size: " << tf->GetSize() << endmsg;
#endif

    // Signal job to terminate if output file is too large
    if ( tf->GetSize() > mfs ) {

      signaledStop = true;

      fatal() << "file \"" << tf->GetName() << "\" associated with stream \"" << f.first
              << "\" has exceeded the max file size of " << m_maxFileSize.value() << "MB. Terminating Job." << endmsg;

      IEventProcessor* evt = nullptr;
      if ( service( "ApplicationMgr", evt, true ).isSuccess() ) {
        evt->stopRun();
        evt->release();
      } else {
        abort();
      }
    } else if ( tf->GetSize() > mfs_warn ) {
      warning() << "file \"" << tf->GetName() << "\" associated with stream \"" << f.first
                << "\" is at 95% of its maximum allowable file size of " << m_maxFileSize.value() << "MB" << endmsg;
    }
  }
}

StatusCode THistSvc::initialize ( )

override

Definition at line 70 of file THistSvc.cpp.

{
  GlobalDirectoryRestore restore;

  StatusCode status = Service::initialize();

  if ( status.isFailure() ) {
    error() << "initializing service" << endmsg;
    return status;
  }

  StatusCode st( StatusCode::SUCCESS );

  try {
    setupOutputFile( m_outputfile );
  } catch ( GaudiException& err ) {
    error() << "Caught: " << err << endmsg;
    st = StatusCode::FAILURE;
  }

  try {
    setupInputFile( m_inputfile );
  } catch ( GaudiException& err ) {
    error() << "Caught: " << err << endmsg;
    st = StatusCode::FAILURE;
  }

  // Protect against multiple instances of TROOT
  if ( !gROOT ) {
    static TROOT root( "root", "ROOT I/O" );
    //    gDebug = 99;
  } else {
    if ( msgLevel( MSG::VERBOSE ) ) verbose() << "ROOT already initialized, debug = " << gDebug << endmsg;
  }

  if ( service( "IncidentSvc", p_incSvc, true ).isFailure() ) {
    error() << "unable to get the IncidentSvc" << endmsg;
    st = StatusCode::FAILURE;
  } else {
    p_incSvc->addListener( this, "EndEvent", 100, true );
  }

  if ( service( "FileMgr", p_fileMgr, true ).isFailure() ) {
    error() << "unable to get the FileMgr" << endmsg;
    st = StatusCode::FAILURE;
  } else {
    debug() << "got the FileMgr" << endmsg;
  }

  // Register open/close callback actions

  using namespace std::placeholders;
  auto boa = [this]( const Io::FileAttr* fa, const std::string& caller ) { return this->rootOpenAction( fa, caller ); };
  if ( p_fileMgr->regAction( boa, Io::OPEN, Io::ROOT ).isFailure() ) {
    error() << "unable to register ROOT file open action with FileMgr" << endmsg;
  }
  auto bea = [this]( const Io::FileAttr* fa, const std::string& caller ) {
    return this->rootOpenErrAction( fa, caller );
  };
  if ( p_fileMgr->regAction( bea, Io::OPEN_ERR, Io::ROOT ).isFailure() ) {
    error() << "unable to register ROOT file open Error action with FileMgr" << endmsg;
  }

  m_okToConnect = true;

  if ( m_delayConnect ) {
    if ( !m_inputfile.value().empty() ) {
      setupInputFile( m_inputfile );
    }
    if ( !m_outputfile.value().empty() ) {
      setupOutputFile( m_outputfile );
    }

    m_delayConnect = false;
  }
  m_alreadyConnectedOutFiles.clear();
  m_alreadyConnectedInFiles.clear();

  IIoComponentMgr* iomgr = nullptr;

  if ( service( "IoComponentMgr", iomgr, true ).isFailure() ) {
    error() << "unable to get the IoComponentMgr" << endmsg;
    st = StatusCode::FAILURE;
  } else {

    if ( !iomgr->io_register( this ).isSuccess() ) {
      error() << "could not register with the I/O component manager !" << endmsg;
      st = StatusCode::FAILURE;
    } else {
      bool all_good = true;
      // register input/output files...
      for ( const auto& reg : m_files ) {
        const std::string& fname = reg.second.first->GetName();
        const IIoComponentMgr::IoMode::Type iomode =
            ( reg.second.second == THistSvc::READ ? IIoComponentMgr::IoMode::READ : IIoComponentMgr::IoMode::WRITE );
        if ( !iomgr->io_register( this, iomode, fname ).isSuccess() ) {
          warning() << "could not register file [" << fname << "] with the I/O component manager..." << endmsg;
          all_good = false;
        } else {
          info() << "registered file [" << fname << "]... [ok]" << endmsg;
        }
      }
      if ( !all_good ) {
        error() << "problem while registering input/output files with "
                << "the I/O component manager !" << endmsg;
        st = StatusCode::FAILURE;
      }
    }
  }

  if ( st.isFailure() ) {
    fatal() << "Unable to initialize THistSvc" << endmsg;
  }

  return st;
}

StatusCode THistSvc::io_reinit ( )

override

callback method to reinitialize the internal state of the component for I/O purposes (e.g.

upon fork(2))

Definition at line 1643 of file THistSvc.cpp.

{
  bool all_good = true;
  if ( msgLevel( MSG::DEBUG ) ) debug() << "reinitializing I/O..." << endmsg;

  // retrieve the I/O component manager...

  IIoComponentMgr* iomgr = nullptr;

  if ( service( "IoComponentMgr", iomgr, true ).isFailure() ) {
    error() << "could not retrieve I/O component manager !" << endmsg;
    return StatusCode::FAILURE;
  }

  GlobalDirectoryRestore restore;
  // to hide the expected errors upon closing the files whose
  // file descriptors have been swept under the rug...
  gErrorIgnoreLevel = kFatal;

  for ( auto& ifile : m_files ) {
    TFile* f          = ifile.second.first;
    std::string fname = f->GetName();
    if ( msgLevel( MSG::DEBUG ) )
      debug() << "file [" << fname << "] mode: [" << f->GetOption() << "] r:" << f->GetFileBytesRead()
              << " w:" << f->GetFileBytesWritten() << " cnt:" << f->GetFileCounter() << endmsg;

    if ( ifile.second.second == READ ) {
      if ( msgLevel( MSG::DEBUG ) ) debug() << "  TFile opened in READ mode: not reassigning names" << endmsg;
      continue;
    }

    if ( !iomgr->io_retrieve( this, fname ).isSuccess() ) {
      error() << "could not retrieve new name for [" << fname << "] !!" << endmsg;
      all_good = false;
      continue;
    } else {
      if ( msgLevel( MSG::DEBUG ) ) debug() << "got a new name [" << fname << "]..." << endmsg;
    }
    // create a new TFile
    // TFile *newfile = TFile::Open (fname.c_str(), f->GetOption());

    void* vf;
    Option_t* opts = f->GetOption();
    int r          = p_fileMgr->open( Io::ROOT, name(), fname, Io::WRITE, vf, "HIST" );
    if ( r != 0 ) {
      error() << "unable to open file \"" << fname << "\" for writing" << endmsg;
      return StatusCode::FAILURE;
    }
    TFile* newfile = (TFile*)vf;
    newfile->SetOption( opts );

    if ( ifile.second.second != THistSvc::READ ) {
      copyFileLayout( newfile, f );
      ifile.second.first = newfile;
    }

    // loop over all uids and migrate them to the new file
    // XXX FIXME: this double loop sucks...
    for ( auto& uid : m_uids ) {
      THistID& hid = uid.second;
      if ( hid.file != f ) continue;
      TDirectory* olddir = this->changeDir( hid );
      hid.file           = newfile;
      // side-effect: create needed directories...
      TDirectory* newdir = this->changeDir( hid );
      TClass* cl         = hid.obj->IsA();

      // migrate the objects to the new file.
      // thanks to the object model of ROOT, it is super easy.
      if ( cl->InheritsFrom( "TTree" ) ) {
        dynamic_cast<TTree*>( hid.obj )->SetDirectory( newdir );
        dynamic_cast<TTree*>( hid.obj )->Reset();
      } else if ( cl->InheritsFrom( "TH1" ) ) {
        dynamic_cast<TH1*>( hid.obj )->SetDirectory( newdir );
        dynamic_cast<TH1*>( hid.obj )->Reset();
      } else if ( cl->InheritsFrom( "TGraph" ) ) {
        olddir->Remove( hid.obj );
        newdir->Append( hid.obj );
      } else {
        error() << "id: \"" << hid.id << "\" is not a inheriting from a class "
                << "we know how to handle (received [" << cl->GetName() << "], "
                << "expected [TTree, TH1 or TGraph]) !" << endmsg << "attaching to current dir [" << newdir->GetPath()
                << "] "
                << "nonetheless..." << endmsg;
        olddir->Remove( hid.obj );
        newdir->Append( hid.obj );
      }
    }
    f->ReOpen( "READ" );
    p_fileMgr->close( f, name() );
    f = newfile;
  }

  return all_good ? StatusCode::SUCCESS : StatusCode::FAILURE;
}

void THistSvc::MergeRootFile ( TDirectory *  target, TDirectory *  source  )

private

Definition at line 1485 of file THistSvc.cpp.

{

  if ( msgLevel( MSG::DEBUG ) ) debug() << "Target path: " << target->GetPath() << endmsg;
  TString path( (char*)strstr( target->GetPath(), ":" ) );
  path.Remove( 0, 2 );

  source->cd( path );
  TDirectory* current_sourcedir = gDirectory;

  // loop over all keys in this directory
  TList* lkeys = current_sourcedir->GetListOfKeys();
  int nkeys    = lkeys->GetEntries();
  TKey* key;
  for ( int jj = 0; jj < nkeys; jj++ ) {
    key                     = (TKey*)lkeys->At( jj );
    string pathnameinsource = current_sourcedir->GetPath() + string( "/" ) + key->GetName();
    if ( msgLevel( MSG::DEBUG ) ) debug() << "Reading Key:" << pathnameinsource << endmsg;
    // key->Dump();
    // TObject *obj=key->ReadObj();
    TObject* obj = source->Get( pathnameinsource.c_str() );

    if ( obj ) {
      if ( obj->IsA()->InheritsFrom( "TDirectory" ) ) {
        // it's a subdirectory

        if ( msgLevel( MSG::DEBUG ) ) debug() << "Found subdirectory " << obj->GetName() << endmsg;

        // create a new subdir of same name and title in the target file
        target->cd();
        TDirectory* newtargetdir = target->mkdir( obj->GetName(), obj->GetTitle() );

        MergeRootFile( newtargetdir, source );

      } else if ( obj->IsA()->InheritsFrom( "TTree" ) ) {
        if ( msgLevel( MSG::DEBUG ) ) debug() << "Found TTree " << obj->GetName() << endmsg;
        TTree* mytree = dynamic_cast<TTree*>( obj );
        int nentries  = (int)mytree->GetEntries();
        mytree->SetBranchStatus( "*", 1 );

        if ( msgLevel( MSG::DEBUG ) ) debug() << "Dumping TTree " << nentries << " entries" << endmsg;
        // mytree->Print();
        // for (int ij=0; ij<nentries; ij++) {
        // debug() << "Dumping TTree Show( " << ij <<" )"
        //<< endmsg;
        // mytree->Show(ij);
        //}
        target->cd();
        mytree->CloneTree();

        // debug() << "Writing TTree to target file: ( "
        //<< mycopiedtree->Write(key->GetName()) <<" ) bytes written"
        //<< endmsg;

      } else {
        target->cd();
        obj->Write( key->GetName() );
      }
    }

  } // while ( ( TKey *key = (TKey*)nextkey() ) )

  // save modifications to target file
}

void THistSvc::parseString ( const std::string &  id, std::string &  root, std::string &  rem  ) const

private

Definition at line 1024 of file THistSvc.cpp.

{
  auto pos = id.find( "/" );

  if ( pos == string::npos ) {
    root.clear();
    rem = id;
  } else if ( pos == 0 ) {
    parseString( id.substr( 1 ), root, rem );
  } else {
    root = id.substr( 0, pos );
    rem  = id.substr( pos + 1 );
  }
}

StatusCode THistSvc::readHist ( const std::string &  name, TH1 *&  hist  ) const

private

Definition at line 969 of file THistSvc.cpp.

{ return readHist_i( id, hist ); }

StatusCode THistSvc::readHist ( const std::string &  name, TH2 *&  hist  ) const

private

Definition at line 973 of file THistSvc.cpp.

{ return readHist_i( id, hist ); }

StatusCode THistSvc::readHist ( const std::string &  name, TH3 *&  hist  ) const

private

Definition at line 977 of file THistSvc.cpp.

{ return readHist_i( id, hist ); }

template<typename T >

StatusCode THistSvc::readHist_i ( const std::string &  name, T *&  hist  ) const

private

Definition at line 242 of file THistSvc.icc.

                                                        {

  GlobalDirectoryRestore restore;

  std::string idr(id);
  removeDoubleSlash( idr );

  std::string stream, rem, dir, fdir, bdir, fdir2;
  TFile *file;

  if (!findStream(idr, stream, rem, file) ) {
    return StatusCode::FAILURE;
  }

  if ( !file ) {
    error() << "no associated file found" << endmsg;
    return StatusCode::FAILURE;
  }

  file->cd("/");

  fdir = idr;
  bdir = dirname(fdir);
  fdir2 = fdir;
  while ( (dir=dirname(fdir)) != "" ) {
    if (! gDirectory->GetKey(dir.c_str())) {
      error() << "Directory \"" << fdir2 << "\" doesnt exist in "
        << file->GetName() << endmsg;
      return StatusCode::FAILURE;
    }
    gDirectory->cd(dir.c_str());
  }

  TObject *to=nullptr;
  gDirectory->GetObject(fdir.c_str(), to);

  if ( !to ) {
    error() << "Could not get obj \"" << fdir << "\" in "
      << gDirectory->GetPath() << endmsg;
    return StatusCode::FAILURE;
  }



  hist = dynamic_cast<T*>(to);
  if ( !hist ) {
    error() << "Could not convert \"" << idr << "\" to a "
      << System::typeinfoName(typeid(*hist)) << " as is a "
      << to->IsA()->GetName()
      << endmsg;
    return StatusCode::FAILURE;
  }


  if (msgLevel(MSG::DEBUG)) {
    debug() << "Read in " << hist->IsA()->GetName() << "  \""
      << hist->GetName() << "\" from file "
      << file->GetName() << endmsg;
    hist->Print();
  }

  return StatusCode::SUCCESS;

}

StatusCode THistSvc::readTree ( const std::string &  name, TTree *&  hist  ) const

private

Definition at line 981 of file THistSvc.cpp.

{ return readHist_i( id, hist ); }

StatusCode THistSvc::regGraph ( const std::string &  name )

override

Definition at line 889 of file THistSvc.cpp.

{
  TGraph* hist = nullptr;
  return regHist_i( hist, id );
}

StatusCode THistSvc::regGraph ( const std::string &  name, TGraph *  hist  )

override

Definition at line 897 of file THistSvc.cpp.

{
  if ( strcmp( hist->GetName(), "Graph" ) == 0 ) {

    std::string id2( id );
    string::size_type i = id2.rfind( "/" );
    if ( i != string::npos ) {
      id2.erase( 0, i + 1 );
    }

    info() << "setting name of TGraph id: \"" << id << "\" to \"" << id2 << "\" since it is unset" << endmsg;
    hist->SetName( id2.c_str() );
  }

  return regHist_i( hist, id );
}

StatusCode THistSvc::regHist ( const std::string &  name )

override

Definition at line 848 of file THistSvc.cpp.

{

  TH1* hist = nullptr;
  return regHist_i( hist, id );
}

StatusCode THistSvc::regHist ( const std::string &  name, TH1 *  hist  )

override

Definition at line 857 of file THistSvc.cpp.

{ return regHist_i( hist, id ); }

StatusCode THistSvc::regHist ( const std::string &  name, TH2 *  hist  )

override

Definition at line 861 of file THistSvc.cpp.

{ return regHist_i( hist, id ); }

StatusCode THistSvc::regHist ( const std::string &  name, TH3 *  hist  )

override

Definition at line 865 of file THistSvc.cpp.

{ return regHist_i( hist, id ); }

template<typename T >

StatusCode THistSvc::regHist_i ( T *  hist, const std::string &  name  )

private

Definition at line 17 of file THistSvc.icc.

                                                           {

  GlobalDirectoryRestore restore;

  std::string idr(id);
  removeDoubleSlash( idr );

  if (idr.find("/") == idr.length()) {
    error() << "Badly formed identifier \"" << idr << "\": "
        << "Must not end with a /" << endmsg;
    return StatusCode::FAILURE;
  }


  TFile *f = nullptr;
  std::string stream,rem;
  if (!findStream(idr, stream, rem, f)) {
    error() << "Could not register id: \"" << idr << "\""
        << endmsg;
    return StatusCode::FAILURE;
  }

  std::string uid = "/" + stream + "/" + rem;
  auto itr = m_uids.find(uid);
  if (itr != m_uids.end()) {
    error() << "already registered an object with identifier \""
      << idr << "\"" << endmsg;
    return StatusCode::FAILURE;
  }


  bool temp = false;
  if ( !f ) {
    temp = true;
    if (msgLevel(MSG::DEBUG))
      debug() << "Historgram with id \"" << idr << "\" is temporary"
        << endmsg;
  }


  TObject *to = nullptr;
  THistID hid;

  // check to see if this hist is to be read in;
  if (!temp && m_files.find(stream)->second.second == READ) {

    if (hist != 0) {
      warning() <<  "Registering id: \"" << idr
        << "\" with non zero pointer!" << endmsg;
    }

    if (readHist_i(idr,hist).isFailure()) {
      error() <<  "Unable to read in hist" << endmsg;
      return StatusCode::FAILURE;
    }
    to = dynamic_cast<TObject*>(hist);
    hid = THistID(uid,temp,to,f,m_files.find(stream)->second.second);

  } else if (hist == 0) {
    error() << "Unable to read in hist with id: \""
      << idr << "\"" << endmsg;
    return StatusCode::FAILURE;

  } else {

    to = dynamic_cast<TObject*>(hist);
    if (to == 0) {
      error() << "Could not dcast to TObject. id: \"" << idr
        << "\"" << endmsg;
      return StatusCode::FAILURE;
    }

    auto oitr = m_tobjs.find(to);
    if (oitr != m_tobjs.end()) {
      error() << "already registered id: \"" << idr
        << "\" with identifier \"" << oitr->second.id << "\"" << endmsg;
      return StatusCode::FAILURE;
    }

    hid = THistID(uid,temp,to,f,m_files.find(stream)->second.second);
    TDirectory* dir = changeDir(hid);

    if ( dynamic_cast<TTree*>(hist) ) {
      dynamic_cast<TTree*>(hist)->SetDirectory(dir);
    } else if ( dynamic_cast<TH1*>(hist) ) {
      dynamic_cast<TH1*>(hist)->SetDirectory(dir);
    } else if ( dynamic_cast<TGraph*>(hist) ) {
      dir->Append(hist);
    } else {
      error() << "id: \"" << idr
        << "\" is not a TH, TTree, or TGraph. Attaching it to current dir."
        << endmsg;
      dir->Append(hist);
    }

  }

  std::string fname;
  if ( !f ) {
    fname = "none";
  } else {
    fname = f->GetName();
  }

  if (msgLevel(MSG::DEBUG))
    debug() << "Registering " << System::typeinfoName(typeid(*hist))
      << " title: \"" << hist->GetTitle()
      << "\"  id: \"" << uid << "\"  dir: "
      //      << hist->GetDirectory()->GetPath() << "  "
      << changeDir(hid)->GetPath()
      << "  file: " << fname
      << endmsg;

  m_ids.emplace(rem, hid);
  m_uids[uid] = hid;
  m_tobjs[to] = hid;

  return StatusCode::SUCCESS;

}

StatusCode THistSvc::regTree ( const std::string &  name )

override

Definition at line 869 of file THistSvc.cpp.

{
  TTree* hist = nullptr;
  return regHist_i( hist, id );
}

StatusCode THistSvc::regTree ( const std::string &  name, TTree *  hist  )

override

Definition at line 877 of file THistSvc.cpp.

{
  StatusCode sc = regHist_i( hist, id );
  if ( hist && sc.isSuccess() ) {
    if ( m_autoSave != 0 ) hist->SetAutoSave( m_autoSave );
    hist->SetAutoFlush( m_autoFlush );
  }
  return sc;
}

StatusCode THistSvc::reinitialize ( )

override

Definition at line 189 of file THistSvc.cpp.

{

  GlobalDirectoryRestore restore;
  warning() << "reinitialize not implemented" << endmsg;
  return StatusCode::SUCCESS;
}

void THistSvc::removeDoubleSlash ( std::string &  id ) const

private

Definition at line 1475 of file THistSvc.cpp.

{

  while ( id.find( "//" ) != std::string::npos ) {
    id.replace( id.find( "//" ), 2, "/" );
  }
}

StatusCode THistSvc::rootOpenAction ( FILEMGR_CALLBACK_ARGS  )

private

Definition at line 1741 of file THistSvc.cpp.

{

  if ( fa->tech() != Io::ROOT ) {
    // This should never happen
    return StatusCode::SUCCESS;
  }

  if ( fa->desc() != "HIST" ) {
    return StatusCode::SUCCESS;
  }

  p_incSvc->fireIncident( FileIncident( caller, "OpenHistFile", fa->name() ) );

  if ( fa->flags().isRead() ) {
    p_incSvc->fireIncident( FileIncident( caller, "BeginHistFile", fa->name() ) );
  } else if ( fa->flags().isWrite() ) {
    p_incSvc->fireIncident( FileIncident( caller, IncidentType::BeginOutputFile, fa->name() ) );
  } else {
    // for Io::RW
    p_incSvc->fireIncident( FileIncident( caller, IncidentType::BeginOutputFile, fa->name() ) );
  }

  return StatusCode::SUCCESS;
}

StatusCode THistSvc::rootOpenErrAction ( FILEMGR_CALLBACK_ARGS  )

private

Definition at line 1769 of file THistSvc.cpp.

{

  if ( fa->tech() != Io::ROOT ) {
    // This should never happen
    return StatusCode::SUCCESS;
  }

  if ( fa->desc() != "HIST" ) {
    return StatusCode::SUCCESS;
  }

  if ( fa->flags().isRead() ) {
    p_incSvc->fireIncident( FileIncident( caller, IncidentType::FailInputFile, fa->name() ) );
  } else if ( fa->flags().isWrite() ) {
    p_incSvc->fireIncident( FileIncident( caller, IncidentType::FailOutputFile, fa->name() ) );
  } else {
    // for Io::RW
    p_incSvc->fireIncident( FileIncident( caller, "FailRWFile", fa->name() ) );
  }

  return StatusCode::SUCCESS;
}

void THistSvc::setupCompressionLevel ( Gaudi::Details::PropertyBase &  cmp )

private

Definition at line 1041 of file THistSvc.cpp.

{

  warning() << "\"CompressionLevel\" Property has been deprecated. "
            << "Set it via the \"CL=\" parameter in the \"Output\" Property" << endmsg;
}

void THistSvc::setupInputFile ( Gaudi::Details::PropertyBase &  inputfile )

private

call-back method to handle input stream property

Definition at line 1050 of file THistSvc.cpp.

{

  if ( FSMState() < Gaudi::StateMachine::CONFIGURED || !m_okToConnect ) {

    debug() << "Delaying connection of Input Files until Initialize"
            << ". now in " << FSMState() << endmsg;

    m_delayConnect = true;
  } else {

    debug() << "Now connecting of Input Files" << endmsg;

    StatusCode sc = StatusCode::SUCCESS;

    for ( const auto& itr : m_inputfile.value() ) {
      if ( m_alreadyConnectedInFiles.end() != m_alreadyConnectedInFiles.find( itr ) ) continue;
      if ( connect( itr ).isFailure() ) {
        sc = StatusCode::FAILURE;
      } else {
        m_alreadyConnectedInFiles.insert( itr );
      }
    }

    if ( !sc.isSuccess() ) {
      throw GaudiException( "Problem connecting inputfile !!", name(), StatusCode::FAILURE );
    }
  }
}

void THistSvc::setupOutputFile ( Gaudi::Details::PropertyBase &  outputfile )

private

call-back method to handle output stream property

Definition at line 1082 of file THistSvc.cpp.

{
  if ( FSMState() < Gaudi::StateMachine::CONFIGURED || !m_okToConnect ) {
    debug() << "Delaying connection of Input Files until Initialize"
            << ". now in " << FSMState() << endmsg;
    m_delayConnect = true;
  } else {

    StatusCode sc = StatusCode::SUCCESS;
    for ( const auto& itr : m_outputfile.value() ) {
      if ( m_alreadyConnectedOutFiles.end() != m_alreadyConnectedOutFiles.find( itr ) ) continue;
      if ( connect( itr ).isFailure() ) {
        sc = StatusCode::FAILURE;
      } else {
        m_alreadyConnectedOutFiles.insert( itr );
      }
    }

    if ( !sc.isSuccess() ) {
      throw GaudiException( "Problem connecting outputfile !!", name(), StatusCode::FAILURE );
    }
  }
}

void THistSvc::updateFiles ( )

private

Definition at line 1108 of file THistSvc.cpp.

{

  // If TTrees grow beyond TTree::fgMaxTreeSize, a new file is
  // automatically created by root, and the old one closed. We
  // need to migrate all the UIDs over to show the correct file
  // pointer. This is ugly.

  if ( msgLevel( MSG::DEBUG ) ) debug() << "updateFiles()" << endmsg;

  for ( auto uitr = m_uids.begin(); uitr != m_uids.end(); ++uitr ) {
#ifndef NDEBUG
    if ( msgLevel( MSG::VERBOSE ) )
      verbose() << " update: " << uitr->first << " " << uitr->second.id << " " << uitr->second.mode << endmsg;
#endif
    TObject* to    = uitr->second.obj;
    TFile* oldFile = uitr->second.file;
    if ( !to ) {
      warning() << uitr->first << ": TObject == 0" << endmsg;
    } else if ( uitr->second.temp || uitr->second.mode == READ ) {
// do nothing - no need to check how big the file is since we
// are just reading it.
#ifndef NDEBUG
      if ( msgLevel( MSG::VERBOSE ) ) verbose() << "     skipping" << endmsg;
#endif

    } else if ( to->IsA()->InheritsFrom( "TTree" ) ) {
      TTree* tr      = dynamic_cast<TTree*>( to );
      TFile* newFile = tr->GetCurrentFile();

      if ( oldFile != newFile ) {
        std::string newFileName = newFile->GetName();
        std::string oldFileName, streamName, rem;
        TFile* dummy = nullptr;
        findStream( uitr->second.id, streamName, rem, dummy );

        for ( auto& itr : m_files ) {
          if ( itr.second.first == oldFile ) itr.second.first = newFile;
        }

        for ( auto uitr2 = uitr; uitr2 != m_uids.end(); ++uitr2 ) {
          if ( uitr2->second.file == oldFile ) {
            uitr2->second.file = newFile;
          }
        }

        auto sitr = std::find_if( std::begin( m_fileStreams ), std::end( m_fileStreams ),
                                  [&]( streamMap::const_reference s ) { return s.second == streamName; } );
        if ( sitr != std::end( m_fileStreams ) ) oldFileName = sitr->first;

#ifndef NDEBUG
        if ( msgLevel( MSG::DEBUG ) )
          debug() << "migrating uid: " << uitr->second.id << "   stream: " << streamName
                  << "   oldFile: " << oldFileName << "   newFile: " << newFileName << endmsg;
#endif

        if ( !oldFileName.empty() ) {
          auto i = m_fileStreams.lower_bound( oldFileName );
          while ( i != std::end( m_fileStreams ) && i->first == oldFileName ) {

#ifndef NDEBUG
            if ( msgLevel( MSG::DEBUG ) )
              debug() << "changing filename \"" << i->first << "\" to \"" << newFileName << "\" for stream \""
                      << i->second << "\"" << endmsg;
#endif
            std::string nm = std::move( i->second );
            i              = m_fileStreams.erase( i );
            m_fileStreams.emplace( newFileName, std::move( nm ) );
          }

        } else {
          error() << "Problems updating fileStreams with new file name" << endmsg;
        }
      }
    }
  }
}

StatusCode THistSvc::write ( )

private

Definition at line 1188 of file THistSvc.cpp.

{

  updateFiles();

  std::for_each( m_files.begin(), m_files.end(), []( std::pair<const std::string, std::pair<TFile*, Mode>>& i ) {
    auto mode = i.second.second;
    auto file = i.second.first;
    if ( mode == WRITE || mode == UPDATE || mode == SHARE ) {
      file->Write( "", TObject::kOverwrite );
    } else if ( mode == APPEND ) {
      file->Write( "" );
    }
  } );

  if ( msgLevel( MSG::DEBUG ) ) {
    debug() << "THistSvc::write()::List of Files connected in ROOT " << endmsg;
    TSeqCollection* filelist = gROOT->GetListOfFiles();
    for ( int ii = 0; ii < filelist->GetEntries(); ii++ ) {
      debug() << "THistSvc::write()::List of Files connected in ROOT: \"" << filelist->At( ii )->GetName() << "\""
              << endmsg;
    }
  }

  return StatusCode::SUCCESS;
}

Member Data Documentation

std::set<std::string> THistSvc::m_alreadyConnectedInFiles

private

list of already connected files.

This is to keep track of files registered by the setupInputFile callback method

Definition at line 167 of file THistSvc.h.

std::set<std::string> THistSvc::m_alreadyConnectedOutFiles

private

list of already connected files.

This is to keep track of files registered by the setupOutputFile callback method

Definition at line 171 of file THistSvc.h.

Gaudi::Property<int> THistSvc::m_autoFlush {this, "AutoFlush", 0}

private

Definition at line 155 of file THistSvc.h.

Gaudi::Property<int> THistSvc::m_autoSave {this, "AutoSave", 0}

private

Definition at line 154 of file THistSvc.h.

Gaudi::Property<int> THistSvc::m_compressionLevel {this, "CompressionLevel", 1}

private

Definition at line 159 of file THistSvc.h.

std::string THistSvc::m_curstream

mutableprivate

Definition at line 190 of file THistSvc.h.

bool THistSvc::m_delayConnect = false

private

Definition at line 188 of file THistSvc.h.

std::map<std::string, std::pair<TFile*, Mode> > THistSvc::m_files

private

Definition at line 182 of file THistSvc.h.

streamMap THistSvc::m_fileStreams

private

Definition at line 183 of file THistSvc.h.

idMap THistSvc::m_ids

private

Definition at line 179 of file THistSvc.h.

Gaudi::Property<std::vector<std::string> > THistSvc::m_inputfile {this, "Input", {}}

private

Definition at line 161 of file THistSvc.h.

Gaudi::Property<int> THistSvc::m_maxFileSize

private

Initial value:

{this, "MaxFileSize", 10240,
                                     "maximum file size in MB. if exceeded, will cause an abort. -1 to never check."}

Definition at line 157 of file THistSvc.h.

bool THistSvc::m_okToConnect = false

private

Definition at line 188 of file THistSvc.h.

Gaudi::Property<std::vector<std::string> > THistSvc::m_outputfile {this, "Output", {}}

private

Definition at line 160 of file THistSvc.h.

Gaudi::Property<bool> THistSvc::m_print {this, "PrintAll", false}

private

Definition at line 156 of file THistSvc.h.

std::vector<std::string> THistSvc::m_Rstream

private

Definition at line 163 of file THistSvc.h.

std::map<std::string, std::string> THistSvc::m_sharedFiles

private

Definition at line 185 of file THistSvc.h.

objMap THistSvc::m_tobjs

private

Definition at line 180 of file THistSvc.h.

uidMap THistSvc::m_uids

private

Definition at line 178 of file THistSvc.h.

std::vector<std::string> THistSvc::m_Wstream

private

Definition at line 163 of file THistSvc.h.

IFileMgr* THistSvc::p_fileMgr = nullptr

private

Definition at line 193 of file THistSvc.h.

IIncidentSvc* THistSvc::p_incSvc = nullptr

private

Definition at line 192 of file THistSvc.h.

bool THistSvc::signaledStop = false

private

Definition at line 187 of file THistSvc.h.

---

The documentation for this class was generated from the following files:

• GaudiSvc/src/THistSvc/THistSvc.h
• GaudiSvc/src/THistSvc/THistSvc.cpp
• GaudiSvc/src/THistSvc/THistSvc.icc