PyStore.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// Py::Store.h 
// Header file for class Py::Store
// Author: S.Binet<binet@cern.ch>
#ifndef PERFMONEVENT_PY_STORE_H 
#define PERFMONEVENT_PY_STORE_H 
 
// Python includes
// Work around a compilation warning.
#ifdef _POSIX_C_SOURCE
# undef _POSIX_C_SOURCE
#endif
#ifdef _XOPEN_SOURCE
# undef _XOPEN_SOURCE
#endif
#include "Python.h"
 
// STL includes
#include <string>
#include <vector>
 
// PerfMonEvent includes
#include "PerfMonEvent/PyStoreUtils.h"
#include "PerfMonEvent/PyStoreCnvUtils.h"
 
// Forward declaration
 
namespace Py {
 
class Store
{ 
 
  // Public methods: 
 public: 
 
  Store();
 
  Store( const std::size_t bufSize );
 
  ~Store(); 
 
  // Const methods: 
 
  const PyObject* pydict() const { Py_INCREF(m_store); return m_store; }
 
  std::size_t bufferSize() const { return m_bufSize; }
 
  std::string repr();
 
  // Non-const methods: 
 
  void setBufferSize( std::size_t bufferSize );
 
  //PyObject* pydict() { Py_INCREF(m_store); return m_store; }
 
  template< typename T >
  void book( const std::string& key,
         const std::string& dataName )
  {
    //PyObject* buf = Py::cnv<T>::toPy( T() );
    //PyObject* buf = PyList_New(0);
    //PyObject* buf = Py::cnv<T>::to( T() );
    this->book_impl( key, dataName, 'd' );
    //Py_XDECREF(buf);
  }
  
  template< typename T >
  void fill( const std::string& key,
         const std::string& dataName,
         const T& data )
  {
//     std::cerr << "PyStore::fill( " << key << ", " << dataName
//        << ", " << typeid(data).name() << " => "
//        << ClassName<T>::name() << "\n";
    PyObject* pyData = Py::cnv<T>::toPy( data );
    this->fill_impl( key, dataName, pyData );
    // fill_impl steals pyData reference...
    return;
  }
 
  void clear_store();
 
  void py_book( const std::string& key,
        const std::string& dataName,
        char dataType = 'd' )
  { return this->book_impl( key, dataName, dataType ); }
  
  void py_fill( const std::string& key,
        const std::string& dataName,
        PyObject* data )
  { return this->fill_impl( key, dataName, data ); }
 
  // Private methods: 
 private: 
 
  void book_impl( const std::string& key,
          const std::string& dataName,
          char dataType = 'd' );
 
  void fill_impl( const std::string& key,
          const std::string& dataName,
          PyObject* data );
 
  // Private data: 
 private: 
 
  std::size_t m_bufSize;
 
  PyObject* m_store;
  //arrayobject* m_aobj;
 
}; 
 
} // end namespace Py
 
#endif //> PERFMONEVENT_PY_DATASTORE_H