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