CSGUniverseList.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* 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

#include "CSGUniverseList.h"

namespace CSG
{

CSGUniverseList::CSGUniverseList()
{
  // create root universe by default when CSG object is initialized
  std::string root_name = "ROOT_UNIVERSE";
  auto root_universe = std::make_unique<CSGUniverse>(root_name, /* is_root = */ true);
  _root_universe = root_universe.get();
  _universes.insert(std::make_pair(root_name, std::move(root_universe)));
}

CSGUniverse &
CSGUniverseList::getUniverse(const std::string & name) const
{
  auto univ = _universes.find(name);
  if (univ == _universes.end())
    mooseError("No universe by name " + name + " exists in the geometry.");
  else
    return *(univ->second);
}

std::vector<std::reference_wrapper<const CSGUniverse>>
CSGUniverseList::getAllUniverses() const
{
  std::vector<std::reference_wrapper<const CSGUniverse>> univs;
  for (auto it = _universes.begin(); it != _universes.end(); ++it)
    univs.push_back(*(it->second.get()));
  return univs;
}

CSGUniverse &
CSGUniverseList::addUniverse(const std::string & name)
{
  return addUniverse(std::make_unique<CSGUniverse>(name));
}

CSGUniverse &
CSGUniverseList::addUniverse(std::unique_ptr<CSGUniverse> universe)
{
  auto name = universe->getName();
  auto [it, inserted] = _universes.emplace(name, std::move(universe));
  if (!inserted)
    mooseError("Universe with name " + name + " already exists in geometry.");
  return *it->second;
}

void
CSGUniverseList::renameUniverse(const CSGUniverse & universe, const std::string & name)
{
  // check that this universe passed in is actually in the same universe that is in the universe
  // list
  auto prev_name = universe.getName();
  auto it = _universes.find(prev_name);
  if (it == _universes.end() || it->second.get() != &universe)
    mooseError("Universe " + prev_name + " cannot be renamed to " + name +
               " as it does not exist in this CSGBase instance.");

  auto existing_univ = std::move(it->second);
  existing_univ->setName(name);
  _universes.erase(prev_name);
  addUniverse(std::move(existing_univ));
}

} // namespace CSG