RestartableDataMap.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "RestartableData.h"
 
#include "UniqueStorage.h"
 
// C++ includes
#include <memory>
#include <vector>
#include <unordered_map>
 
class RestartableDataMap
{
public:
  class Data : public UniqueStorage<RestartableDataValue>
  {
  public:
    class WriteKey
    {
      friend class RestartableDataMap;
      WriteKey() {}
      WriteKey(const WriteKey &) {}
    };
 
    RestartableDataValue & addPointer(std::unique_ptr<RestartableDataValue> && ptr, const WriteKey)
    {
      return UniqueStorage<RestartableDataValue>::addPointer(std::move(ptr));
    }
  };
 
  RestartableDataMap();
 
  RestartableDataValue & addData(std::unique_ptr<RestartableDataValue> data);
 
  bool hasData(const std::string & name) const;
 
  const RestartableDataValue * findData(const std::string & name) const;
  RestartableDataValue * findData(const std::string & name);
 
  RestartableDataValue & data(const std::string & name);
 
  auto begin() { return _data.begin(); }
  auto end() { return _data.end(); }
  auto begin() const { return _data.begin(); }
  auto end() const { return _data.end(); }
 
  auto size() const { return _data.size(); }
  auto empty() const { return _data.empty(); }
 
private:
  RestartableDataMap::Data _data;
  std::unordered_map<std::string, std::size_t> _name_to_data_index;
};