CSGEngUnitList.h

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

#pragma once

#include "CSGSurfaceEngUnit.h"
#include "CSGCellEngUnit.h"
#include "CSGUniverseEngUnit.h"

namespace CSG
{

class CSGEngUnitList
{
protected:
  CSGEngUnitList();

  virtual ~CSGEngUnitList() = default;

  bool hasEngUnit(const std::string & name) const
  {
    for (const auto * ptr : _eng_units)
      if (ptr->getName() == name)
        return true;
    return false;
  }

  std::vector<std::reference_wrapper<const CSGEngUnit>> getAllEngUnits() const;

  std::vector<std::reference_wrapper<const CSGSurfaceEngUnit>> getAllSurfaceEngUnits() const;

  std::vector<std::reference_wrapper<const CSGCellEngUnit>> getAllCellEngUnits() const;

  std::vector<std::reference_wrapper<const CSGUniverseEngUnit>> getAllUniverseEngUnits() const;

  CSGEngUnit & addEngUnit(CSGEngUnit & unit);

  CSGEngUnit & getEngUnit(const std::string & name) const;

  void removeEngUnit(const CSGEngUnit & unit);

  bool operator==(const CSGEngUnitList & other) const;

  bool operator!=(const CSGEngUnitList & other) const;

  std::vector<CSGEngUnit *> _eng_units;

  // Only CSGBase should be calling the methods in CSGEngUnitList
  friend class CSGBase;
};

} // namespace CSG