RootDataConnection.h

Go to the documentation of this file.

// $Id: RootDataConnection.h,v 1.9 2010-09-17 09:00:12 frankb Exp $
#ifndef GAUDIROOT_ROOTDATACONNECTION_H
#define GAUDIROOT_ROOTDATACONNECTION_H

// Framework include files
#include "GaudiKernel/ClassID.h"
#include "GaudiUtils/IIODataManager.h"
#include <string>
#include <vector>
#include <map>
#include <set>

#include "TFile.h"
#include "RootCnv/RootRefs.h"

// Forward declarations
class TTree;
class TClass;
class TBranch;
class TTreePerfStats;

class MsgStream;
class IRegistry;
class DataObject;

/*
 *  Gaudi namespace declaration
 */
namespace Gaudi  {

  class GAUDI_API RootConnectionSetup {
  public:
    typedef std::vector<std::string> StringVec;
  protected:
    virtual ~RootConnectionSetup();
    int refCount;
    MsgStream* m_msgSvc;

  public:
    StringVec    cacheBranches;
    StringVec    vetoBranches;
    std::string  loadSection;
    int          cacheSize;
    int          learnEntries;
    int          autoFlush;
    int          basketSize;

    RootConnectionSetup();
    void addRef();
    void release();

    static long setCompression(const std::string& compression);
    static int compression();

    void setMessageSvc(MsgStream* m);
    MsgStream& msgSvc() const {  return *m_msgSvc; }
  };

  class GAUDI_API RootDataConnection : virtual public Gaudi::IDataConnection  {
  public:

    struct ContainerSection {
      ContainerSection() : start(-1), length(0) {}
      ContainerSection(int s, int l) : start(s), length(l) {}
      ContainerSection(const ContainerSection& s) : start(s.start), length(s.length) {}
      ContainerSection& operator=(const ContainerSection& s) { start=s.start;length=s.length; return *this;}
      int start;
      int length;
    };

    typedef std::vector<std::string>                                 StringVec;
    typedef std::vector<std::pair<std::string,std::string> >         ParamMap;
    typedef std::map<std::string,TTree*>                             Sections;
    typedef std::vector<ContainerSection>                            ContainerSections;
    typedef std::map<std::string,ContainerSections>                  MergeSections;
    typedef std::vector<RootRef>                                     LinkSections;
    typedef std::set<const IInterface*>                              Clients;

    MsgStream& msgSvc() const {  return m_setup->msgSvc(); }


  protected:
    RootConnectionSetup* m_setup;
    TTreePerfStats*      m_statistics;
    TFile*               m_file;
    TTree               *m_refs;
    Sections             m_sections;
    StringVec            m_dbs;
    StringVec            m_conts;
    StringVec            m_links;
    StringVec            m_mergeFIDs;
    ParamMap             m_params;
    MergeSections        m_mergeSects;
    LinkSections         m_linkSects;
    Clients              m_clients;
    std::string          m_empty;

    const std::string& empty() const;

    StatusCode saveRefs();

  public:
    class Tool {
    protected:
      typedef RootDataConnection::StringVec          StringVec;
      typedef RootDataConnection::ParamMap           ParamMap;
      typedef RootDataConnection::Sections           Sections;
      typedef RootDataConnection::MergeSections      MergeSections;
      typedef RootDataConnection::LinkSections       LinkSections;
      typedef RootDataConnection::ContainerSection   ContainerSection;
      typedef RootDataConnection::ContainerSections  ContainerSections;

      RootDataConnection* c;
    public:
      TTree*             refs()               const { return c->m_refs;       }
      StringVec&         dbs()                const { return c->m_dbs;        }
      StringVec&         conts()              const { return c->m_conts;      }
      StringVec&         links()              const { return c->m_links;      }
      ParamMap&          params()             const { return c->m_params;     }
      MsgStream&         msgSvc()             const { return c->msgSvc();     }
      const std::string& name()               const { return c->m_name;       }
      Sections&          sections()           const { return c->m_sections;   }
      LinkSections&      linkSections()       const { return c->m_linkSects;  }
      MergeSections&     mergeSections()      const { return c->m_mergeSects; }

      virtual ~Tool() {}
      virtual void release() { delete this; }
      virtual TBranch* getBranch(const std::string&  section, const std::string& n) = 0;
      virtual RootRef poolRef(size_t /* which */) const { return RootRef(); }
        
      virtual StatusCode readRefs() = 0;
      virtual StatusCode saveRefs() = 0;
      virtual int loadRefs(const std::string& section, const std::string& cnt, unsigned long entry, RootObjectRefs& refs) = 0;
    } *m_tool;
    friend class Tool;

    Tool* makeTool();

  public:

    RootDataConnection(const IInterface* own, const std::string& nam, RootConnectionSetup* setup);
    virtual ~RootDataConnection();

    TFile* file() const                         {  return m_file;                              }
    virtual bool isConnected() const            {  return m_file != 0;                         }
    bool isWritable() const                     {  return m_file != 0 && m_file->IsWritable(); }
    Tool* tool() const                          {  return m_tool;                              }
    const MergeSections& mergeSections() const  {  return m_mergeSects;                        }
    const StringVec& mergeFIDs() const          {  return m_mergeFIDs;                         }


    void addClient(const IInterface* client);
    size_t removeClient(const IInterface* client);
    bool lookupClient(const IInterface* client) const;

    std::pair<const RootRef*,const ContainerSection*>  getMergeSection(const std::string& container, int entry) const;

    void enableStatistics(const std::string& section);
    void saveStatistics(const std::string& statisticsFile);

    int loadObj(const std::string& section, const std::string& cnt, unsigned long entry, DataObject*& pObj);

    int loadRefs(const std::string& section, const std::string& cnt, unsigned long entry, RootObjectRefs& refs)
    { return m_tool->loadRefs(section,cnt,entry,refs); }

    std::pair<int,unsigned long> saveObj(const std::string& section,const std::string& cnt, TClass* cl, DataObject* pObj,bool fill_missing=false);
    std::pair<int,unsigned long> save(const std::string& section,const std::string& cnt, TClass* cl, void* pObj,bool fill_missing=false);


    virtual StatusCode connectRead();
    virtual StatusCode connectWrite(IoType typ);
    virtual StatusCode disconnect();
    virtual StatusCode read(void* const, size_t)   { return StatusCode::FAILURE; }
    virtual StatusCode write(const void*, int)     { return StatusCode::FAILURE; }
    virtual long long int seek(long long int, int) { return -1; }

    TTree* getSection(const std::string& sect, bool create=false);

    TBranch* getBranch(const std::string& section,const std::string& branch_name) 
    { return m_tool->getBranch(section,branch_name); }
    TBranch* getBranch(const std::string& section, const std::string& branch_name, TClass* cl, void* ptr);

    void makeRef(IRegistry* pA, RootRef& ref);
    void makeRef(const std::string& name, long clid, int tech, const std::string& db, const std::string& cnt, int entry, RootRef& ref);

    int makeLink(const std::string& p);

    const std::string& getDb(int which) const;
    
    const std::string& getCont(int which) const 
      { return (which>=0)&&(size_t(which)<m_conts.size()) ? *(m_conts.begin()+which) : empty();    }
    
    const std::string& getLink(int which) const 
      { return (which>=0)&&(size_t(which)<m_links.size()) ? *(m_links.begin()+which) : empty();    }
  };
}         // End namespace Gaudi
#endif    // GAUDIROOT_ROOTDATACONNECTION_H