NTupleSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef EVENT_LOOP_ALGS__NTUPLE_SVC_H
#define EVENT_LOOP_ALGS__NTUPLE_SVC_H

 
 
 
#include <EventLoopAlgs/Global.h>
 
#include <set>
#include <EventLoop/Algorithm.h>
 
class TBranch;
class TFile;
class TTree;
 
namespace EL
{
  NTupleSvc *getNTupleSvc (IWorker *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:
    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