RootInputFile.h

Go to the documentation of this file.

#ifndef ROOTINPUTFILE_H
#define ROOTINPUTFILE_H

#include "DybKernel/MsgStreamMember.h"

#include <string>
#include <vector>

class TFile;
class TTree;

class RootInputFile {
    std::string m_filename;
    std::string m_treepath;
    std::string m_branchname;
    int m_pruneSimHits;
    TFile* m_file;
    TTree* m_tree;
    int m_entry, m_entries;
    void* m_addr;

    Dyb::MsgStreamMember log;

public:
    // Return list of all treepaths found in given file.
    static std::vector<std::string> TreePaths(const std::string& filename);

    // Return non zero if a branch of the given name at the given
    // treepath in the given file is found and is a subclass of
    // RootIOBaseObject.  The non zero return value is the object's
    // class ID.
    static int TestForObject(const std::string& filename,
                             const std::string& treepath,
                             const std::string& branchname,
                             bool ignore_missing=false);
                       

    RootInputFile(const std::string& filename, 
                  const std::string& treepath,
                  const std::string& branchname);
    ~RootInputFile();

    const std::string& treepath() { return m_treepath; }
    const std::string& branchname() { return m_branchname; }
    const std::string& filename() { return m_filename; }

    bool open();

    bool leave();

    int entry();

    bool setEntry(int entry);

    bool read(void* addr);

    bool read(void* addr, int& nbytes);

    bool setAddr(void* addr);

    int entries();

    bool next(int steps = 1);

    bool beginning();
    bool ending();

    bool prev(int steps = 1);

};

class RootInputFileList : public std::vector<RootInputFile*> {
    int m_index;                // current
    Dyb::MsgStreamMember log;
public:
    RootInputFileList();
    ~RootInputFileList();

    int index();

    RootInputFile* current();

    bool next();

    bool prev();
    
    bool last();

    bool first();

    bool jump(int index);

    int entriesBefore(int index);

};

#endif  // ROOTINPUTFILE_H