VectorFillerToolBase.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-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
 
 
#ifndef D3PDMAKERUTILS_VECTORFILLERTOOLBASE_H
#define D3PDMAKERUTILS_VECTORFILLERTOOLBASE_H
 
 
#include "ObjectMetadata.h"
#include "D3PDMakerInterfaces/IAddVariable.h"
#include "D3PDMakerInterfaces/IBlockFillerTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include <vector>
#include <list>
#include <string>
#include <typeinfo>
class TVirtualCollectionProxy;
 
 
namespace D3PD {
 
 
class IIteration;
 

class VectorFillerToolBase
  : public AthAlgTool,
    public IAddVariable
{
public:
  VectorFillerToolBase (const std::string& type,
                        const std::string& name,
                        const IInterface* parent);
 

  virtual StatusCode initialize();
 

  virtual StatusCode finalize();
 

  virtual StatusCode addVariable (const std::string& name,
                                  const std::type_info& ti,
                                  void* & ptr,
                                  const std::string& docstring = "",
                                  const void* defval = 0);
 

  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);
 
 
protected:
  StatusCode doConfigure (IAddVariable* tree, IIteration& it);
 

  StatusCode doBook();
 
 
  StatusCode doFill (IIteration& it, size_t sizeHint);
 

  ObjectMetadata m_metadata;
 

  std::string m_prefix;
 
 
private:
  ToolHandleArray<IBlockFillerTool> m_blockFillers;

  std::string m_nrowName;

  std::string m_nobjName;

  std::string m_objIndexName;

  IAddVariable* m_tree;

  int* m_nrow;

  int* m_nobj;

  int* m_objIndex;
 

  class Var
  {
  public:
    Var (const std::string& name,
         const std::type_info& ti,
         void* & ptr,
         const std::string& dim,
         char* defval,
         size_t defsize);
 

    StatusCode init (IAddVariable* tree,
                     const std::string& prefix);
 

    void resize (size_t sz, size_t pos);
 

    void next();
 

    void free();
 
 
  private:
    std::string m_name;

    const std::type_info* m_ti;

    char** m_addr;

    std::string m_docstring;

    TVirtualCollectionProxy* m_proxy;

    size_t m_valdiff;

    void* m_contptr;

    char* m_default;

    size_t m_defsize;
  };
 

  struct Vars
    : public std::list<Var>
  {
    ~Vars();
 

    void resize_all (size_t sz, size_t pos = 0);
 

    void next_all ();
  };

  Vars m_vars;
 

  void maybeExtend (size_t& n, size_t& sz);
 

  StatusCode fillOnce (const void* obj,
                       size_t objndx,
                       size_t& n,
                       size_t& sz,
                       const std::vector<unsigned long>& codes_in,
                       std::vector<unsigned long>& codes_out);
};
 
 
} // namespace D3PD
 
 
#endif // not D3PDMAKERUTILS_VECTORFILLERTOOLBASE_H