/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.28/EventLoopAlgs/EventLoopAlgs/NTupleSvc.h

#ifndef EVENT_LOOP_ALGS__NTUPLE_SVC_H
#define EVENT_LOOP_ALGS__NTUPLE_SVC_H

//          Copyright Nils Krumnack 2012.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

// Please feel free to contact me (krumnack@iastate.edu) for bug
// reports, feature suggestions, praise and complaints.





#include <EventLoopAlgs/Global.h>

#include <set>
#include <EventLoop/Algorithm.h>

class TBranch;
class TFile;
class TTree;

namespace D3PDReader
{
  class Event;
}

namespace EL
{
  NTupleSvc *getNTupleSvc (Worker *worker,
               const std::string& outputStream,
               const std::string& treeName = "");



  class NTupleSvc : public EL::Algorithm
  {
    //
    // public interface
    //

  public:
    void testInvariant () const;


  public:
    NTupleSvc (const std::string& val_outputName = "output");


  public:
    ~NTupleSvc ();


  public:
    void copyBranch (const std::string& name);


  public:
    void copyBranchList (const std::string& fileName);


  public:
    void addWhiteFloat (const std::string& varName);


  public:
    void addWhiteArray (const std::string& varName);


  public:
    TTree *tree () const;


  public:
    bool getFilterPassed () const;


  public:
    void setFilterPassed (bool passed = true);


  public:
    const std::string& treeName () const;
    void treeName (const std::string& val_treeName);


    //
    // interface inherited from Algorithm
    //

  public:
    virtual const char *GetName () const;


  private:
    virtual StatusCode setupJob (Job& job);


  private:
    virtual StatusCode changeInput (bool firstFile);


  private:
    virtual StatusCode initialize ();


  private:
    virtual StatusCode execute ();


  private:
    virtual StatusCode postExecute ();


  private:
    virtual bool hasName (const std::string& name) const;



    //
    // private interface
    //

  private:
    std::string m_outputName;

  private:
    std::string m_treeName;

  private:
    std::string m_tree_title;

  private:
    std::set<std::string> m_copyBranches;
    typedef std::set<std::string>::iterator m_copyBranchesIter;

  private:
    std::set<std::string> m_whiteFloat;
    typedef std::set<std::string>::iterator m_whiteFloatIter;

  private:
    std::set<std::string> m_whiteArray;
    typedef std::set<std::string>::iterator m_whiteArrayIter;

  private:
    TFile *m_file; 

  private:
    TTree *m_tree; 

  private:
    bool m_initialized; 

  private:
    bool m_taken; 

  private:
    bool m_connected; 

  private:
    D3PDReader::Event *m_d3pdreader; 

  private:
    WhiteBoardSvc *m_whiteboard; 

  private:
    struct CopyInfo
    {
    public:
      std::string name;

    public:
      TBranch *source;

    public:
      TBranch *target;

    public:
      std::vector<char> buffer;
    };
    std::vector<CopyInfo> m_copyInfo; 
    typedef std::vector<CopyInfo>::iterator copyInfoMIter;

  private:
    struct WhiteInfo
    {
    public:
      std::string name;

    public:
      bool array;

    public:
      std::vector<float> buffer;

    public:
      std::vector<float> *pointer;
    };
    std::vector<WhiteInfo> m_whiteInfo; 
    typedef std::vector<WhiteInfo>::iterator whiteInfoMIter;


  private:
    void initBranches ();


  private:
    void findBranches (std::set<std::string>& branchList);


  private:
    void initOutput (const std::string& branchName);


  private:
    void initOutput (const std::set<std::string>& branchList);


  private:
    void copyInput ();


    ClassDef(NTupleSvc, 1);
  };
}

#endif

---