lsMaterialMap.hpp

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <set>
#include <vector>
 
namespace viennals {

class MaterialMap {
  std::vector<int> materialMap;
  std::set<int> materials;
 
public:
  MaterialMap() = default;
  MaterialMap(const MaterialMap &) = default;
  MaterialMap &operator=(const MaterialMap &) = default;
  MaterialMap &operator=(MaterialMap &&) = default;
 
  void insertNextMaterial(const int passedMaterialId) {
    materialMap.push_back(passedMaterialId);
    materials.insert(passedMaterialId);
  }
 
  void removeLastMaterial() {
    if (materialMap.empty()) {
      return;
    }
 
    int idToRemove = materialMap.back();
    materialMap.pop_back();
 
    bool isUnique = std::find(materialMap.begin(), materialMap.end(),
                              idToRemove) == materialMap.end();
 
    if (isUnique) {
      materials.erase(idToRemove);
    }
  }
 
  void setMaterialId(const std::size_t index, const int materialId) {
    if (index >= materialMap.size()) {
      materialMap.resize(index + 1, -1); // Initialize new elements with -1
    } else {
      auto oldMaterialId = materialMap[index];
      materials.erase(oldMaterialId); // Remove old material ID if it exists
    }
    materialMap[index] = materialId;
    materials.insert(materialId);
  }
 
  std::size_t getNumberOfLayers() const { return materialMap.size(); }
 
  std::size_t getNumberOfMaterials() const { return materials.size(); }
 
  int getMaterialId(const std::size_t index) const {
    if (index >= materialMap.size())
      return -1;
    return materialMap[index];
  }
 
  // Additional utility methods
  bool isValidIndex(const std::size_t index) const {
    return index < materialMap.size();
  }
 
  void clear() {
    materialMap.clear();
    materials.clear();
  }
 
  void reserve(const std::size_t size) { materialMap.reserve(size); }
 
  bool hasMaterial(const int materialId) const {
    return materials.count(materialId) > 0;
  }
 
  const std::set<int> &getMaterials() const { return materials; }
 
  const std::vector<int> &getMaterialMap() const { return materialMap; }
};
 
} // namespace viennals