THistSvc Class Reference

#include <THistSvc.h>

Inheritance diagram for THistSvc:

[legend]

Collaboration diagram for THistSvc:

[legend]

List of all members.

---

Detailed Description

Definition at line 24 of file THistSvc.h.

Public Member Functions
virtual StatusCode	initialize ()
	Initialization (from CONFIGURED to INITIALIZED).
virtual StatusCode	reinitialize ()
	Initialization (from INITIALIZED or RUNNING to INITIALIZED, via CONFIGURED).
virtual StatusCode	finalize ()
	Finalize (from INITIALIZED to CONFIGURED).
virtual StatusCode	queryInterface (const InterfaceID &riid, void **ppvInterface)
	Query interfaces of Interface.
virtual StatusCode	regHist (const std::string &name)
virtual StatusCode	regHist (const std::string &name, TH1 *)
virtual StatusCode	regHist (const std::string &name, TH2 *)
virtual StatusCode	regHist (const std::string &name, TH3 *)
virtual StatusCode	getHist (const std::string &name, TH1 *&) const
virtual StatusCode	getHist (const std::string &name, TH2 *&) const
virtual StatusCode	getHist (const std::string &name, TH3 *&) const
virtual StatusCode	regTree (const std::string &name)
virtual StatusCode	regTree (const std::string &name, TTree *)
virtual StatusCode	getTree (const std::string &name, TTree *&) const
virtual StatusCode	regGraph (const std::string &name)
virtual StatusCode	regGraph (const std::string &name, TGraph *)
virtual StatusCode	getGraph (const std::string &name, TGraph *&) const
virtual StatusCode	deReg (TObject *obj)
virtual StatusCode	deReg (const std::string &name)
virtual std::vector< std::string >	getHists () const
virtual std::vector< std::string >	getTrees () const
virtual std::vector< std::string >	getGraphs () const
virtual StatusCode	getTHists (TDirectory *td, TList &) const
virtual StatusCode	getTHists (const std::string &name, TList &) const
virtual StatusCode	getTTrees (TDirectory *td, TList &) const
virtual StatusCode	getTTrees (const std::string &name, TList &) const
	THistSvc (const std::string &name, ISvcLocator *svc)
Protected Member Functions
virtual	~THistSvc ()
Private Types
enum	Mode {   READ, WRITE, UPDATE, APPEND,   SHARE }
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) 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 (Property &inputfile)
	call-back method to handle input stream property
void	setupOutputFile (Property &outputfile)
	call-back method to handle input stream property
void	MergeRootFile (TDirectory *target, TDirectory *source)
Private Attributes
StringArrayProperty	m_inputfile
StringArrayProperty	m_outputfile
std::vector< std::string >	m_Rstream
std::vector< std::string >	m_Wstream
IntegerProperty	m_autoSave
BooleanProperty	m_print
std::set< std::string >	m_alreadyConnectedInFiles
	list of already connected files.
std::set< std::string >	m_alreadyConnectedOutFiles
	list of already connected files.
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
Classes
class	GlobalDirectoryRestore
struct	THistID

---

Member Typedef Documentation

typedef std::map<std::string, THistID> THistSvc::uidMap [private]

Definition at line 113 of file THistSvc.h.

typedef std::multimap<std::string, THistID> THistSvc::idMap [private]

Definition at line 114 of file THistSvc.h.

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

Definition at line 115 of file THistSvc.h.

typedef std::multimap<std::string, std::string> THistSvc::streamMap [private]

Definition at line 116 of file THistSvc.h.

---

Member Enumeration Documentation

enum THistSvc::Mode [private]

Enumerator:

READ
WRITE
UPDATE
APPEND
SHARE

Definition at line 105 of file THistSvc.h.

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

---

Constructor & Destructor Documentation

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

Definition at line 39 of file THistSvc.cpp.

  : Service(name, svc) {

  declareProperty ("Output", m_outputfile );
  declareProperty ("Input", m_inputfile );
  declareProperty ("AutoSave", m_autoSave=0 );
  declareProperty ("PrintAll", m_print=false);

  m_inputfile.declareUpdateHandler ( &THistSvc::setupInputFile,  this );
  m_outputfile.declareUpdateHandler( &THistSvc::setupOutputFile, this );
}

THistSvc::~THistSvc

(

)

[protected, virtual]

Definition at line 53 of file THistSvc.cpp.

                    {

}

---

Member Function Documentation

StatusCode THistSvc::initialize

(

)

[virtual]

Initialization (from CONFIGURED to INITIALIZED).

Reimplemented from Service.

Definition at line 78 of file THistSvc.cpp.

                     {
  m_alreadyConnectedOutFiles.clear();
  m_alreadyConnectedInFiles.clear();

  StatusCode status = Service::initialize();

  GlobalDirectoryRestore restore;

  if (status.isFailure()) {
    MsgStream log ( msgSvc(), name() );
    log << MSG::ERROR << "initializing service" << endreq;
    return status;
  }

  vector<string>::const_iterator itr;
  StatusCode st(StatusCode::SUCCESS);

  try {
    setupOutputFile( m_outputfile );
  } catch ( GaudiException& err ) {
    MsgStream msg( msgSvc(), name() );
    msg << MSG::ERROR
        << "Caught: " << err << endreq;
    st = StatusCode::FAILURE;
  }

  try {
    setupInputFile( m_inputfile );
  } catch ( GaudiException& err ) {
    MsgStream msg( msgSvc(), name() );
    msg << MSG::ERROR
        << "Caught: " << err << endreq;
    st = StatusCode::FAILURE;
  }

  // Protect against multiple instances of TROOT
  if ( 0 == gROOT )   {
    static TROOT root("root","ROOT I/O");
    //    gDebug = 99;
  } else {
    MsgStream log ( msgSvc(), name() );
    log << MSG::VERBOSE << "ROOT already initialized, debug = "
        << gDebug<< endreq;
  }


  if (st.isFailure()) {
    MsgStream log ( msgSvc(), name() );
    log << MSG::FATAL << "Unable to initialize THistSvc" << endreq;
  }

  return st;

}

StatusCode THistSvc::reinitialize

(

)

[virtual]

Initialization (from INITIALIZED or RUNNING to INITIALIZED, via CONFIGURED).

Reimplemented from Service.

Definition at line 136 of file THistSvc.cpp.

                       {

  GlobalDirectoryRestore restore;

  MsgStream log ( msgSvc(), name() );
  log << MSG::WARNING << "reinitialize not implemented" << endreq;


  return StatusCode::SUCCESS;

}

StatusCode THistSvc::finalize

(

)

[virtual]

Finalize (from INITIALIZED to CONFIGURED).

Reimplemented from Service.

Definition at line 151 of file THistSvc.cpp.

                   {

  GlobalDirectoryRestore restore;

  MsgStream log ( msgSvc(), name() );
  log << MSG::DEBUG << "THistSvc::finalize" << endreq;

  uidMap::const_iterator uitr;
  for (uitr=m_uids.begin(); uitr != m_uids.end(); ++uitr) {

    TObject* to = uitr->second.obj;

    string dirname("none");
    if (to->IsA()->InheritsFrom("TTree")) {
      TTree* tr = dynamic_cast<TTree*>(to);
      if (tr->GetDirectory() != 0) {
        dirname = tr->GetDirectory()->GetPath();
      }
    } else if (to->IsA()->InheritsFrom("TGraph")) {
      if (!uitr->second.temp) {
        dirname = uitr->second.file->GetPath();
        string id2(uitr->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->IsA()->InheritsFrom("TH1")) {
      TH1* th = dynamic_cast<TH1*>(to);
      if (th == 0) {
        log << MSG::ERROR << "Couldn't dcast: " << uitr->first << endreq;
      } else {
        if (th->GetDirectory() != 0) {
          dirname = th->GetDirectory()->GetPath();
        }
      }
    }

    MsgStream log ( msgSvc(), name() );
    log << MSG::DEBUG << "uid: \"" << uitr->first << "\"  temp: "
        << uitr->second.temp << "  dir: " << dirname
        << endreq;


//     if (uitr->second.temp == true) {
//       MsgStream log ( msgSvc(), name() );
//       log << MSG::INFO << "Deleting \"" << uitr->first << "\"" << endreq;
//       delete uitr->second.obj;
//     }
  }

  StatusCode sc = write();
  if (sc.isFailure()) {
    MsgStream log ( msgSvc(), name() );
    log << MSG::ERROR << "problems writing histograms" << endreq;
  }

  if (m_print) {
    log << MSG::INFO << "Listing contents of ROOT files: " << endreq;
  }
  vector<TFile*> deleted_files;
  map<string, pair<TFile*,Mode> >::const_iterator itr;
  for (itr = m_files.begin(); itr != m_files.end(); ++itr) {

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

      log << MSG::DEBUG << "finalizing stream/file " << itr->first << ":"
          << itr->second.first->GetName()
          << endreq;
    } else {
      log << MSG::DEBUG << "already finalized stream " << itr->first << endreq;
      continue;
    }


    if (m_print && log.level() <= MSG::INFO) {

      log << MSG::INFO;
      log << "==> File: " << itr->second.first->GetName()
          << "  stream: " << itr->first << endreq;

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

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

    itr->second.first->Close();

    if (itr->second.second==SHARE) {
      TFile *outputfile;
      //Merge File
      try {
        log << MSG::DEBUG << "Openning Final Output File: " <<m_sharedFiles[itr->first].c_str()<<endreq;
        outputfile = new TFile(m_sharedFiles[itr->first].c_str(), "UPDATE");
      } catch (const std::exception& Exception) {
      log << MSG::ERROR << "exception caught while trying to open root"
          << " file for appending: " << Exception.what() << std::endl
          << "  -> file probably corrupt." << endreq;
      return StatusCode::FAILURE;
      } catch (...) {
        log << MSG::ERROR << "Problems opening output file  \"" << m_sharedFiles[itr->first]
            << "\" for append: probably corrupt" << endreq;
        return StatusCode::FAILURE;
      }

      log << MSG::DEBUG << "THistSvc::write()::Merging Rootfile "<<endreq;
      TFile *inputfile;
      try {
        log << MSG::DEBUG << "Openning again Temporary File: " <<tmpfn.c_str()<<endreq;
        inputfile=new TFile(tmpfn.c_str(),"READ");
      } catch (const std::exception& Exception) {
        log << MSG::ERROR << "exception caught while trying to open root"
            << " file for appending: " << Exception.what() << std::endl
            << "  -> file probably corrupt." << endreq;
        return StatusCode::FAILURE;
      } catch (...) {
        log << MSG::ERROR << "Problems opening output file  \"" << tmpfn.c_str()
            << "\" for append: probably corrupt" << endreq;
        return StatusCode::FAILURE;
      }

      MergeRootFile(outputfile, inputfile);

      outputfile->Write();
      outputfile->Close();
      inputfile->Close();

      log << MSG::DEBUG << "Trying to remove temporary file \"" << tmpfn
          << "\""<<endreq;

      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();
}

StatusCode THistSvc::queryInterface	(	const InterfaceID &	riid,
		void **	ppvUnknown
	)			[virtual]

Query interfaces of Interface.

Parameters:

	riid	ID of Interface to be retrieved
	ppvUnknown	Pointer to Location for interface pointer

Reimplemented from Service.

Definition at line 58 of file THistSvc.cpp.

                                                              {
  StatusCode sc = StatusCode::FAILURE;
  if ( ppvInterface ) {
    *ppvInterface = 0;

    if ( ITHistSvc::interfaceID().versionMatch(riid) )    {
      *ppvInterface = static_cast<ITHistSvc*>(this);
      sc = StatusCode::SUCCESS;
      addRef();
    }
    else
      sc = Service::queryInterface( riid, ppvInterface );
  }
  return sc;
}

StatusCode THistSvc::regHist

(

const std::string &

name

)

[virtual]

Implements ITHistSvc.

Definition at line 534 of file THistSvc.cpp.

                                     {

  TH1 *hist(0);

  return regHist_i(hist, id);
}

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

Implements ITHistSvc.

Definition at line 544 of file THistSvc.cpp.

                                                {
  return regHist_i(hist, id);
}

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

Implements ITHistSvc.

Definition at line 551 of file THistSvc.cpp.

                                                {
  return regHist_i(hist, id);
}

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

Implements ITHistSvc.

Definition at line 558 of file THistSvc.cpp.

                                                {
  return regHist_i(hist, id);
}

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

Implements ITHistSvc.

Definition at line 613 of file THistSvc.cpp.

                                                       {
  return getHist_i(id, hist);
}

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

Implements ITHistSvc.

Definition at line 620 of file THistSvc.cpp.

                                                       {
  return getHist_i(id, hist);
}

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

Implements ITHistSvc.

Definition at line 627 of file THistSvc.cpp.

                                                       {
  return getHist_i(id, hist);
}

StatusCode THistSvc::regTree

(

const std::string &

name

)

[virtual]

Implements ITHistSvc.

Definition at line 565 of file THistSvc.cpp.

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

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

Implements ITHistSvc.

Definition at line 573 of file THistSvc.cpp.

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

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

Implements ITHistSvc.

Definition at line 655 of file THistSvc.cpp.

                                                         {
  return getHist_i(id, hist);
}

StatusCode THistSvc::regGraph

(

const std::string &

name

)

[virtual]

Implements ITHistSvc.

Definition at line 584 of file THistSvc.cpp.

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

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

Implements ITHistSvc.

Definition at line 592 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);
    }

    MsgStream log ( msgSvc(), name() );
    log << MSG::INFO << "setting name of TGraph id: \"" << id << "\" to \""
        << id2 << "\" since it is unset" << endreq;
    hist->SetName(id2.c_str());
  }

  return regHist_i(hist, id);
}

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

Implements ITHistSvc.

Definition at line 683 of file THistSvc.cpp.

                                                           {
  return getHist_i(id, hist);
}

StatusCode THistSvc::deReg

(

TObject *

obj

)

[virtual]

Implements ITHistSvc.

Definition at line 454 of file THistSvc.cpp.

                            {

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

    uidMap::iterator itr2 = m_uids.find(hid.id);
    if (itr2 == m_uids.end()) {
      MsgStream log ( msgSvc(), name() );
      log << MSG::ERROR << "Problems deregistering TObject \""
          << obj->GetName()
          << "\" with id \"" << hid.id << "\"" << endreq;
      return StatusCode::FAILURE;
    }

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

    idMap::iterator itr3;
    bool found(false);

    std::pair<idMap::iterator, idMap::iterator> mitr = m_ids.equal_range(rem);
    if (mitr.first == mitr.second) {
      MsgStream log ( msgSvc(), name() );
      log << MSG::ERROR << "Problems deregistering TObject \""
          << obj->GetName()
          << "\" with id \"" << hid.id << "\"" << endreq;
      return StatusCode::FAILURE;
    } else {
      for (itr3 = mitr.first; itr3 != mitr.second; ++itr3) {
        if (itr3->second.obj == obj) {
          found = true;
          break;
        }
      }
      if (!found) {
        MsgStream log ( msgSvc(), name() );
        log << MSG::ERROR << "Problems deregistering TObject \""
            << obj->GetName()
            << "\" with id \"" << hid.id << "\"" << endreq;
      }
    }

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

    return StatusCode::SUCCESS;

  } else {
    MsgStream log ( msgSvc(), name() );
    log << MSG::ERROR << "Cannot unregister TObject \"" << obj->GetName()
        << "\": not known to THistSvc" << endreq;
    return StatusCode::FAILURE;
  }

}

StatusCode THistSvc::deReg

(

const std::string &

name

)

[virtual]

Implements ITHistSvc.

Definition at line 516 of file THistSvc.cpp.

                                   {

  uidMap::iterator itr = m_uids.find(id);
  if (itr == m_uids.end()) {
    MsgStream log ( msgSvc(), name() );
    log << MSG::ERROR << "Problems deregistering id \""
        << id << "\"" << endreq;
    return StatusCode::FAILURE;
  }

  TObject* obj = itr->second.obj;

  return deReg(obj);
}

std::vector< std::string > THistSvc::getHists

(

)

const [virtual]

Implements ITHistSvc.

Definition at line 634 of file THistSvc.cpp.

                         {

  std::vector<std::string> names;
  names.reserve(m_uids.size());

  uidMap::const_iterator itr;
  for (itr = m_uids.begin(); itr != m_uids.end(); ++itr) {
    THistID tid = itr->second;

    if (tid.obj->IsA()->InheritsFrom("TH1")) {
      names.push_back(itr->first);
    }

  }

  return names;

}

std::vector< std::string > THistSvc::getTrees

(

)

const [virtual]

Implements ITHistSvc.

Definition at line 662 of file THistSvc.cpp.

                         {

  std::vector<std::string> names;
  names.reserve(m_uids.size());

  uidMap::const_iterator itr;
  for (itr = m_uids.begin(); itr != m_uids.end(); ++itr) {
    THistID tid = itr->second;

    if (tid.obj->IsA()->InheritsFrom("TTree")) {
      names.push_back(itr->first);
    }

  }

  return names;

}

std::vector< std::string > THistSvc::getGraphs

(

)

const [virtual]

Implements ITHistSvc.

Definition at line 690 of file THistSvc.cpp.

                          {

  std::vector<std::string> names;
  names.reserve(m_uids.size());

  uidMap::const_iterator itr;
  for (itr = m_uids.begin(); itr != m_uids.end(); ++itr) {
    THistID tid = itr->second;

    if (tid.obj->IsA()->InheritsFrom("TGraph")) {
      names.push_back(itr->first);
    }

  }

  return names;

}

StatusCode THistSvc::getTHists	(	TDirectory *	td,
		TList &	tl
	)			const [virtual]

Implements ITHistSvc.

Definition at line 352 of file THistSvc.cpp.

                                                    {

  MsgStream log ( msgSvc(), name() );
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if (!td->cd()) {
    log << MSG::ERROR << "No such TDirectory \"" << td->GetPath() << "\""
        << endreq;
    return StatusCode::FAILURE;
  }

  TIter nextkey(td->GetList());
  while (TObject *key = (TObject*)nextkey()) {
    if (key != 0) {
      if (key->IsA()->InheritsFrom("TH1")) { tl.Add(key); }
    }
  }


  return StatusCode::SUCCESS;

}

StatusCode THistSvc::getTHists	(	const std::string &	name,
		TList &	tl
	)			const [virtual]

Implements ITHistSvc.

Definition at line 380 of file THistSvc.cpp.

                                                          {

  MsgStream log ( msgSvc(), name() );
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  if (!gDirectory->cd(dir.c_str())) {
    log << MSG::ERROR << "No such TDirectory \"" << dir << "\""
         << endreq;
    sc = StatusCode::FAILURE;
  } else {
    sc = getTHists(gDirectory,tl);
  }

  return sc;

}

StatusCode THistSvc::getTTrees	(	TDirectory *	td,
		TList &	tl
	)			const [virtual]

Implements ITHistSvc.

Definition at line 403 of file THistSvc.cpp.

                                                    {

  MsgStream log ( msgSvc(), name() );
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  if (!td->cd()) {
    log << MSG::ERROR << "No such TDirectory \"" << td->GetPath() << "\""
        << endreq;
    return StatusCode::FAILURE;
  }

  TIter nextkey(td->GetList());
  while (TObject *key = (TObject*)nextkey()) {
    if (key != 0) {
      if (key->IsA()->InheritsFrom("TTree")) { tl.Add(key); }
    }
  }

  return StatusCode::SUCCESS;

}

StatusCode THistSvc::getTTrees	(	const std::string &	name,
		TList &	tl
	)			const [virtual]

Implements ITHistSvc.

Definition at line 430 of file THistSvc.cpp.

                                                          {

  MsgStream log ( msgSvc(), name() );
  GlobalDirectoryRestore restore;

  gErrorIgnoreLevel = kBreak;

  StatusCode sc;

  if (!gDirectory->cd(dir.c_str())) {
    log << MSG::ERROR << "No such TDirectory \"" << dir << "\""
         << endreq;
    sc = StatusCode::FAILURE;
  } else {
    sc = getTTrees(gDirectory,tl);
  }

  return sc;

}

template<typename T>

StatusCode THistSvc::regHist_i	(	T *	hist,
		const std::string &	name
	)			[inline, private]

template<typename T>

StatusCode THistSvc::getHist_i	(	const std::string &	name,
		T *&	hist
	)			const [inline, private]

template<typename T>

StatusCode THistSvc::readHist_i	(	const std::string &	name,
		T *&	hist
	)			const [inline, private]

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

Definition at line 711 of file THistSvc.cpp.

                                                        {
  return readHist_i(id, hist);
}

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

Definition at line 718 of file THistSvc.cpp.

                                                        {
  return readHist_i(id, hist);
}

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

Definition at line 725 of file THistSvc.cpp.

                                                        {
  return readHist_i(id, hist);
}

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

Definition at line 732 of file THistSvc.cpp.

                                                          {
  return readHist_i(id, hist);
}

void THistSvc::updateFiles

(

)

[private]

Definition at line 869 of file THistSvc.cpp.

                      {

  // If TTrees grow beyond TTree::fgMaxFileSize, 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.

  uidMap::iterator uitr, uitr2;
  for (uitr=m_uids.begin(); uitr != m_uids.end(); ++uitr) {
    TObject* to = uitr->second.obj;
    TFile* oldFile = uitr->second.file;
    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;
        findStream(uitr->second.id, streamName, rem, dummy);

        MsgStream log( msgSvc(), name() );
        log << MSG::DEBUG << "migrating uid: " << uitr->second.id
            << "   stream: " << streamName << "   newFile: " << newFileName
            << endreq;

        map<string, pair<TFile*,Mode> >::iterator itr;
        for (itr=m_files.begin(); itr!= m_files.end(); ++itr) {
          if (itr->second.first == oldFile) {
            itr->second.first = newFile;

          }
        }

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

        streamMap::iterator sitr;
        for (sitr = m_fileStreams.begin(); sitr!=m_fileStreams.end(); ++sitr) {
          if (sitr->second == streamName) {
            oldFileName = sitr->first;
            break;
          }
        }

        if (oldFileName != "") {
          while ( (sitr=m_fileStreams.find(oldFileName)) != m_fileStreams.end() ) {
            log << MSG::DEBUG << "changing filename \"" << oldFileName
                << "\" to \"" << newFileName << "\" for stream \""
                << sitr->second << "\"" << endreq;
            m_fileStreams.erase(sitr);
            m_fileStreams.insert( make_pair<std::string,std::string>(newFileName,streamName) );
          }


        } else {
          log << MSG::ERROR
              << "Problems updating fileStreams with new file name" << endreq;
        }

      }

    }
  }

}

StatusCode THistSvc::write

(

)

[private]

Definition at line 943 of file THistSvc.cpp.

                {

  updateFiles();

  MsgStream log(msgSvc(), name() );

  map<string, pair<TFile*,Mode> >::const_iterator itr;
  for (itr=m_files.begin(); itr!= m_files.end(); ++itr) {
    if (itr->second.second == WRITE || itr->second.second == UPDATE
        ||itr->second.second==SHARE) {
      itr->second.first->Write("",TObject::kOverwrite);
    } else if (itr->second.second == APPEND) {
      itr->second.first->Write("");
    }
  }


  log << MSG::DEBUG << "THistSvc::write()::List of Files connected in ROOT "
      <<endreq;

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

  return StatusCode::SUCCESS;

}

StatusCode THistSvc::connect

(

const std::string &

ident

)

[private]

Definition at line 977 of file THistSvc.cpp.

                                        {

  MsgStream log ( msgSvc(), name() );
  Tokenizer tok(true);

  string::size_type 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 val,VAL,TAG,filename,db_typ("ROOT");

  tok.analyse(ident.substr(loc+1,ident.length()), " ", "", "", "=", "'", "'");

  for ( Tokenizer::Items::iterator i = tok.items().begin(); i != tok.items().end(); i++)    {
    const std::string& tag = (*i).tag();
    TAG = tag;
    toupper(TAG);

    val = (*i).value();
    VAL = val;
    toupper(VAL);

    if (TAG == "FILE" || TAG == "DATAFILE") {
      filename = val;
      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 {
        log << MSG::ERROR << "Unknown OPT: \"" << (*i).value() << "\""
            << endreq;
        typ = 0;
      }
    } else if (TAG == "TYP") {
      db_typ = (*i).value();
    } else {
      props.push_back( Prop((*i).tag(), (*i).value()));
    }

  }

  if (stream == "temp") {
    log << MSG::ERROR << "in JobOption \"" << ident
        << "\": stream name \"temp\" reserved."
        << endreq;
    return StatusCode::FAILURE;
  }

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


  if (m_files.find(stream) != m_files.end()) {
    log << MSG::ERROR << "in JobOption \"" << ident
        << "\":\n stream \"" << stream << "\" already connected to file: \""
        << m_files[stream].first->GetName() << "\""
        << endreq;
    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?
    log << MSG::ERROR << "No OPT= specified or unknown access mode in: "
        << ident << endreq;
    return StatusCode::FAILURE;
  }

  // Is this file already connected to another stream?
  if (m_fileStreams.find(filename) != m_fileStreams.end()) {
    std::pair<streamMap::iterator, streamMap::iterator> fitr =
      m_fileStreams.equal_range(filename);

    std::string oldstream = (fitr.first)->second;

    std::pair<TFile*,Mode> f_info = m_files[oldstream];

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


  TFile *f(0) ;
  if (newMode == THistSvc::READ) {
    // old file

    try {
      f = TFile::Open(filename.c_str(),"READ");
    } catch (const std::exception& Exception) {
      log << MSG::ERROR << "exception caught while trying to open root"
          << " file for reading: " << Exception.what() << std::endl
          << "  -> file probably corrupt." << endreq;
      return StatusCode::FAILURE;
    } catch (...) {
      log << MSG::ERROR << "Problems opening input file  \"" << filename
          << "\": probably corrupt" << endreq;
      return StatusCode::FAILURE;
    }

    if (!f->IsOpen()) {
      log << MSG::ERROR << "Unable to open input file \"" << filename
          << "\": file does not exist" << endreq;
      return StatusCode::FAILURE;
    }


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

    f = TFile::Open(filename.c_str(),"NEW");
    if (!f->IsOpen()) {
      log << MSG::ERROR << "Unable to create new output file \"" << filename
          << "\" for writing: file already exists" << endreq;
      return StatusCode::FAILURE;
    }

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

    try {
      f =  TFile::Open(filename.c_str(),"UPDATE");
    } catch (const std::exception& Exception) {
      log << MSG::ERROR << "exception caught while trying to open root"
          << " file for appending: " << Exception.what() << std::endl
          << "  -> file probably corrupt." << endreq;
      return StatusCode::FAILURE;
    } catch (...) {
      log << MSG::ERROR << "Problems opening output file  \"" << filename
          << "\" for append: probably corrupt" << endreq;
      return StatusCode::FAILURE;
    }

    if (!f->IsOpen()) {
      log << MSG::ERROR << "Unable to open output file \"" << filename
          << "\" for appending" << endreq;
      return StatusCode::FAILURE;
    }

  } 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;
  stringstream out;
  string realfilename=filename;
  out << ishared;
  filename = string("tmp_THistSvc_")+out.str()+string(".root");

  log << MSG::DEBUG << "Creating temp file \"" << filename
      << "\" and realfilename="<<realfilename << endreq;
  m_sharedFiles[stream]=realfilename;

  try {
  f = TFile::Open(filename.c_str(),"NEW");
  } catch (const std::exception& Exception) {
    log << MSG::ERROR << "exception caught while trying to open root"
        << " file for appending: " << Exception.what() << std::endl
        << "  -> file probably corrupt." << endreq;
  return StatusCode::FAILURE;
  } catch (...) {
    log << MSG::ERROR << "Problems opening output file  \"" << filename
        << "\" for append: probably corrupt" << endreq;
  return StatusCode::FAILURE;
  }

  if (!f->IsOpen()) {
    log << MSG::ERROR << "Unable to open output file \"" << filename
        << "\" for appending" << endreq;
  return StatusCode::FAILURE;
  }

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

    try {
      f =  TFile::Open(filename.c_str(),"RECREATE");
    } catch (const std::exception& Exception) {
      log << MSG::ERROR << "exception caught while trying to open root"
          << " file for updating: " << Exception.what() << std::endl
          << "  -> file probably corrupt." << endreq;
      return StatusCode::FAILURE;
    } catch (...) {
      log << MSG::ERROR << "Problems opening output file  \"" << filename
          << "\" for update: probably corrupt" << endreq;
      return StatusCode::FAILURE;
    }

    if (!f->IsOpen()) {
      log << MSG::ERROR << "Unable to open output file \"" << filename
          << "\" for updating" << endreq;
      return StatusCode::FAILURE;
    }

  }

  m_files[stream] = make_pair<TFile*,Mode>(f,newMode);
  m_fileStreams.insert(make_pair<std::string,std::string>(filename,stream));

  log << MSG::DEBUG << "Opening TFile \"" << filename << "\"  stream: \""
      << stream << "\"  mode: \"" << typ << "\""
      << endreq;

  return StatusCode::SUCCESS;
}

TDirectory * THistSvc::changeDir

(

const THistSvc::THistID &

hid

)

const [private]

Definition at line 1221 of file THistSvc.cpp.

                                                    {

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

  if (file != 0) {
    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;

}

std::string THistSvc::dirname

(

std::string &

dir

)

const [private]

Definition at line 1250 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;

}

void THistSvc::removeDoubleSlash

(

std::string &

id

)

const [private]

Definition at line 1288 of file THistSvc.cpp.

                                               {

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

}

bool THistSvc::browseTDir

(

TDirectory *

dir

)

const [private]

Definition at line 305 of file THistSvc.cpp.

                                          {

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

  GlobalDirectoryRestore restore;

  MsgStream log(msgSvc(), name());

  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 == 0) { 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 == 0) { cout << key->GetName() << " obj==0"<< endl; continue; }
    if (obj->IsA()->InheritsFrom("TDirectory")) {
      TDirectory *tt = dynamic_cast<TDirectory*>(obj);
      browseTDir(tt);
    }
  }

  return true;
}

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

Definition at line 739 of file THistSvc.cpp.

                                       {

  string::size_type pos = id.find("/");

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

    string::size_type pos2 = id.find("/",pos+1);

    if (pos2 == string::npos) {
      MsgStream log( msgSvc(), name() );
      log << MSG::ERROR << "badly formed Hist/Tree id: \"" << id << "\""
          << endreq;
      return false;
    }

    parseString(id,stream,rem);

  }

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

  map< string,pair<TFile*,Mode> >::const_iterator itr = m_files.find(stream);
  if (itr != m_files.end()) {
    file = itr->second.first;
  } else {
    file = 0;
    MsgStream log( msgSvc(), name() );
    log << MSG::WARNING << "no stream \"" << stream
        << "\" associated with id: \"" << id << "\""
        << endreq;
  }

  return true;

}

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

Definition at line 788 of file THistSvc.cpp.

                                                                       {
  string::size_type pos = id.find("/");

  if (pos == string::npos) {
    root = "";
    rem = id;
    return;
  }

  if (pos == 0) {
    parseString(id.substr(1,id.length()),root,rem);
  } else {
    root = id.substr(0,pos);
    rem = id.substr(pos+1,id.length());
  }

}

void THistSvc::setupInputFile

(

Property &

inputfile

)

[private]

call-back method to handle input stream property

Definition at line 809 of file THistSvc.cpp.

{
  StatusCode sc = StatusCode::SUCCESS;

  typedef std::vector<std::string> Strings_t;
  for ( Strings_t::const_iterator
          itr  = m_inputfile.value().begin(),
          iEnd = m_inputfile.value().end();
        itr != iEnd;
        ++itr ) {
    if ( m_alreadyConnectedInFiles.end() ==
         m_alreadyConnectedInFiles.find( *itr ) ) {
      if ( connect(*itr).isFailure() ) {
        sc = StatusCode::FAILURE;
      } else {
        m_alreadyConnectedInFiles.insert( *itr );
      }
    }
  }

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

void THistSvc::setupOutputFile

(

Property &

outputfile

)

[private]

call-back method to handle input stream property

Definition at line 839 of file THistSvc.cpp.

{
  StatusCode sc = StatusCode::SUCCESS;

  typedef std::vector<std::string> Strings_t;
  for ( Strings_t::const_iterator
          itr  = m_outputfile.value().begin(),
          iEnd = m_outputfile.value().end();
        itr != iEnd;
        ++itr ) {
    if ( m_alreadyConnectedOutFiles.end() ==
         m_alreadyConnectedOutFiles.find( *itr ) ) {
      if ( connect(*itr).isFailure() ) {
        sc = StatusCode::FAILURE;
      } else {
        m_alreadyConnectedOutFiles.insert( *itr );
      }
    }
  }

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

void THistSvc::MergeRootFile	(	TDirectory *	target,
		TDirectory *	source
	)			[private]

Definition at line 1298 of file THistSvc.cpp.

                                                                   {

  MsgStream log(msgSvc(), name() );

  log <<MSG::DEBUG << "Target path: " << target->GetPath() << endreq;
  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();
    log <<MSG::DEBUG << "Reading Key:" << pathnameinsource << endreq;
    //key->Dump();
    //TObject *obj=key->ReadObj();
    TObject *obj=source->Get(pathnameinsource.c_str());

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

      log <<MSG::DEBUG << "Found subdirectory " << obj->GetName()
          << endreq;

      // 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")) {
      log <<MSG::DEBUG << "Found TTree " << obj->GetName() << endreq;
      TTree *mytree=dynamic_cast<TTree*>(obj);
      int nentries=(int) mytree->GetEntries();
      mytree->SetBranchStatus("*",1);

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

      //log <<MSG::DEBUG << "Writing TTree to target file: ( "
      //<< mycopiedtree->Write(key->GetName()) <<" ) bytes written"
      //<< endreq;

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

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

  // save modifications to target file

}

---

Member Data Documentation

StringArrayProperty THistSvc::m_inputfile [private]

Definition at line 151 of file THistSvc.h.

StringArrayProperty THistSvc::m_outputfile [private]

Definition at line 151 of file THistSvc.h.

std::vector<std::string> THistSvc::m_Rstream [private]

Definition at line 152 of file THistSvc.h.

std::vector<std::string> THistSvc::m_Wstream [private]

Definition at line 152 of file THistSvc.h.

IntegerProperty THistSvc::m_autoSave [private]

Definition at line 153 of file THistSvc.h.

BooleanProperty THistSvc::m_print [private]

Definition at line 154 of file THistSvc.h.

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 159 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 163 of file THistSvc.h.

uidMap THistSvc::m_uids [private]

Definition at line 167 of file THistSvc.h.

idMap THistSvc::m_ids [private]

Definition at line 168 of file THistSvc.h.

objMap THistSvc::m_tobjs [private]

Definition at line 169 of file THistSvc.h.

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

Definition at line 171 of file THistSvc.h.

streamMap THistSvc::m_fileStreams [private]

Definition at line 172 of file THistSvc.h.

std::map<std::string, std::string > THistSvc::m_sharedFiles [private]

Definition at line 174 of file THistSvc.h.

---

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

• /afs/cern.ch/sw/Gaudi/releases/GAUDI/GAUDI_v20r4/GaudiSvc/src/THistSvc/THistSvc.h
• /afs/cern.ch/sw/Gaudi/releases/GAUDI/GAUDI_v20r4/GaudiSvc/src/THistSvc/THistSvc.cpp