RootD3PD.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef D3PDMAKERROOT_ROOTD3PD_H
#define D3PDMAKERROOT_ROOTD3PD_H
 
 
// STL include(s):
#include <vector>
#include <string>
#include <map>
 
 
#include "D3PDMakerInterfaces/ID3PD.h"
class TTree;
class TClass;
 
 
namespace D3PD {
 
 
// Helper for clearing tuple variables.
namespace Root {
class Cleartable;
}
class FakeProxy;
 
 
class RootD3PD
  : public ID3PD
{
public:
  RootD3PD (TTree* tree,
            const std::string& master,
            const std::vector< std::string >& allowedNames =
            std::vector< std::string >(),
            const std::vector< std::string >& vetoedNames =
            std::vector< std::string >(),
            int basketSize = -1,
            int entryOffsetLen = -1);
 
 
  ~RootD3PD();
 
 
  virtual StatusCode addVariable (const std::string& name,
                                  const std::type_info& ti,
                                  void* & ptr,
                                  const std::string& docstring = "",
                                  const void* defval = 0) override;
 
 
  virtual StatusCode
  addDimensionedVariable (const std::string& name,
                          const std::type_info& ti,
                          void* & ptr,
                          const std::string& dim,
                          const std::string& docstring = "",
                          const void* defval = 0) override;
 
 
  virtual StatusCode capture () override;
 
 
  virtual StatusCode clear () override;
 
 
  virtual StatusCode redim (const Dim_t* ptr) override;
 
 
  const TTree* tree() const;
  TTree* tree();
 
 
  const std::string& master() const;
 
 
  virtual StatusCode addMetadata (const std::string& key,
                                  const void* obj,
                                  const std::type_info& ti) override;
 
 
  void setPoolFile (const std::string& poolFile);
 
  
  StatusCode attachPoolFile();
 
 
private:
  TClass* getClass (const std::type_info& ti);
 
  bool isAllowed (const std::string& name);
 
  StatusCode addFakeVariable (const std::string& name,
                              const std::type_info& ti,
                              void*& ptr);
 
  TTree* m_tree;
 
  std::string m_master;
 
  int m_basketSize;
 
  int m_entryOffsetLen;
 
  Root::Cleartable* m_cleartable;
 
  std::vector< std::string > m_allowedNames;
 
  std::vector< std::string > m_vetoedNames;
 
  std::map< std::string, FakeProxy* > m_fakeVars;
 
  std::string m_poolFile;
 
  // Disallow copying.
  RootD3PD (const RootD3PD&) = delete;
  RootD3PD& operator= (const RootD3PD&) = delete;
};
 
 
} // namespace D3PD
 
 
#endif // not D3PDMAKERROOT_ROOTD3PD_H