DataWrapper.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2022 Whitney Armstrong
 
#ifndef JUGBASE_DATAWRAPPER_H
#define JUGBASE_DATAWRAPPER_H
 
#include <type_traits>
 
// Include files
#include <GaudiKernel/DataObject.h>
#include <podio/CollectionBase.h>
 
/** Data wrapper.
 *
 * \ingroup base
 */
class GAUDI_API DataWrapperBase : public DataObject {
public:
  // ugly hack to circumvent the usage of boost::any yet
  // DataSvc would need a templated register method
  virtual podio::CollectionBase* collectionBase() = 0;
  virtual ~DataWrapperBase(){};
};
 
/** Data wrapper.
 *
 * \ingroup base
 */
template <class T>
class GAUDI_API DataWrapper : public DataWrapperBase {
public:
  DataWrapper() : DataWrapperBase(), m_data(nullptr){};
  virtual ~DataWrapper();
 
  const T* getData() { return m_data; }
  void setData(T* data) { m_data = data; }
  /// try to cast to collectionBase; may return nullptr;
  virtual podio::CollectionBase* collectionBase();
 
private:
  T* m_data;
};
 
template <class T>
DataWrapper<T>::~DataWrapper() {
  if (m_data != nullptr) delete m_data;
}
 
template <class T>
podio::CollectionBase* DataWrapper<T>::collectionBase() {
  if (std::is_base_of<podio::CollectionBase, T>::value) {
    return reinterpret_cast<podio::CollectionBase*>(m_data);
  }
  return nullptr;
}
 
#endif