RootOutputStream.h

Go to the documentation of this file.

#ifndef ROOTOUTPUTSTREAM_H
#define ROOTOUTPUTSTREAM_H

#include "RootIOStream.h"

#include "DybKernel/MsgStreamMember.h"

#include <string>

class TFile;
class TTree;
class TDirectory;

class RootOutputStream : public RootIOStream {

    TFile* m_file;
    TTree* m_tree;
    TDirectory* m_dir;

    std::string m_classname, m_treepath, m_branchname;

    int m_entries;

    Dyb::MsgStreamMember log;
public:
    RootOutputStream(void* addr, const std::string& classname,
                     const std::string& treepath = "",
                     const std::string& branchname = "");
    virtual ~RootOutputStream();

    void setIndexExpression(const std::string& major, const std::string& minor);

    std::string path();

    bool write();

    bool close();

    bool newFile(const std::string& filename,
                 const std::string& treepath = "",
                 const std::string& branchname = "");

    int entries();

    int fileEntries();

private:
    
    bool handle_user_data();
    std::string m_majorIndex, m_minorIndex;
};

#endif  // ROOTOUTPUTSTREAM_H