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;
};