PyStoreObj.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/datastore/DataStore.h>
#include <framework/datastore/StoreAccessorBase.h>
 
#include <string>
#include <vector>
 
class TClass;
class TObject;
 
namespace Belle2 {
  class PyStoreObj {
  public:
    static std::vector<std::string> list(DataStore::EDurability durability = DataStore::EDurability::c_Event);
 
    static void printList(DataStore::EDurability durability = DataStore::EDurability::c_Event);
 
    explicit PyStoreObj(const std::string& name,
                        DataStore::EDurability durability =  DataStore::EDurability::c_Event);
 
    explicit PyStoreObj(TClass* objClass,
                        DataStore::EDurability durability =  DataStore::EDurability::c_Event);
 
    explicit PyStoreObj(TClass* objClass,
                        const std::string& name,
                        DataStore::EDurability durability =  DataStore::EDurability::c_Event);
 
    bool registerInDataStore(DataStore::EStoreFlags storeFlags);
 
    bool registerInDataStore(const std::string& name = "",
                             DataStore::EStoreFlags storeFlags = DataStore::EStoreFlags::c_WriteOut);
 
    bool isRequired(const std::string& name = "");
 
    bool isOptional(const std::string& name = "");
 
    bool hasValidClass() const;
 
    bool isValid() const;
 
    operator bool() const { return isValid(); }
 
    TObject* obj() const { ensureAttached(); return isValid() ? m_storeEntry->ptr : nullptr; }
 
    TObject* operator->() const { return obj(); }
 
    bool assign(TObject* object, bool replace = false);
 
  public:
    bool create(bool replace = false);
 
  private:
    void ensureAttached() const;
 
    void attach() const;
 
  private:
    StoreAccessorBase m_storeAccessor;
 
    mutable StoreEntry* m_storeEntry = nullptr;
  };
}