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
 
//          
// 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 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