geometry_container.h

Go to the documentation of this file.

//    |  /           |
//    ' /   __| _` | __|  _ \   __|
//    . \  |   (   | |   (   |\__ `
//   _|\_\_|  \__,_|\__|\___/ ____/
//                   Multi-Physics
//
//  License:         BSD License
//                   Kratos default license: kratos/license.txt
//
//  Main authors:    Pooyan Dadvand
//
 
#pragma once
 
// System includes
 
// External includes
 
// Project includes
#include "includes/define.h"
#include "containers/pointer_hash_map_set.h"
 
 
namespace Kratos
{
 
 
template<class TGeometryType>
class GeometryContainer
{
    class GetGeometryId
    {
    public:
        std::size_t const& operator()(const TGeometryType& rGeometry) const
        {
            return rGeometry.Id();
        }
    };
 
public:
 
    KRATOS_CLASS_POINTER_DEFINITION(GeometryContainer);
 
    typedef std::size_t IndexType;
    typedef std::size_t SizeType;
 
    typedef typename TGeometryType::Pointer GeometryPointerType;
 
 
    // Stores with hash of Ids to corresponding geometries.
    typedef PointerHashMapSet<
        TGeometryType,
        std::hash<std::size_t>,
        GetGeometryId,
        GeometryPointerType
        > GeometriesMapType;
 
    typedef typename GeometriesMapType::iterator GeometryIterator;
 
    typedef typename GeometriesMapType::const_iterator GeometryConstantIterator;
 
 
    GeometryContainer()
        : mGeometries()
    {}
 
    GeometryContainer(GeometryContainer const& rOther)
        : mGeometries(rOther.mGeometries)
    {}
 
    GeometryContainer(
        GeometriesMapType& NewGeometries)
        : mGeometries(NewGeometries)
    {}
 
    ~GeometryContainer() = default;
 
 
    GeometryContainer Clone()
    {
        typename GeometriesMapType::Pointer p_geometries(new GeometriesMapType(*mGeometries));
 
        return GeometryContainer(p_geometries);
    }
 
    void Clear()
    {
        mGeometries.clear();
    }
 
 
    SizeType NumberOfGeometries() const
    {
        return mGeometries.size();
    }
 
 
    GeometryIterator AddGeometry(GeometryPointerType pNewGeometry)
    {
        auto i = mGeometries.find(pNewGeometry->Id());
        if (i == mGeometries.end()) {
            return mGeometries.insert(pNewGeometry);
        } else if (&(*i) == pNewGeometry.get()) { // check if the pointee coincides
            return i;
        } else {
            KRATOS_ERROR << "Attempting to add Geometry with Id: " << pNewGeometry->Id() << ", unfortunately a (different) geometry with the same Id already exists!" << std::endl;
        }
    }
 
 
    GeometryPointerType pGetGeometry(IndexType GeometryId)
    {
        auto i = mGeometries.find(GeometryId);
        KRATOS_ERROR_IF(i == mGeometries.end())
            << " geometry index not found: " << GeometryId << ".";
        return (i.base()->second);
    }
 
    const GeometryPointerType pGetGeometry(IndexType GeometryId) const
    {
        auto i = mGeometries.find(GeometryId);
        KRATOS_ERROR_IF(i == mGeometries.end())
            << " geometry index not found: " << GeometryId << ".";
        return (i.base()->second);
    }
 
    GeometryPointerType pGetGeometry(std::string GeometryName)
    {
        auto hash_index = TGeometryType::GenerateId(GeometryName);
        auto i = mGeometries.find(hash_index);
        KRATOS_ERROR_IF(i == mGeometries.end())
            << " geometry index not found: " << GeometryName << ".";
        return (i.base()->second);
    }
 
    const GeometryPointerType pGetGeometry(std::string GeometryName) const
    {
        auto hash_index = TGeometryType::GenerateId(GeometryName);
        auto i = mGeometries.find(hash_index);
        KRATOS_ERROR_IF(i == mGeometries.end())
            << " geometry index not found: " << GeometryName << ".";
        return (i.base()->second);
    }
 
    TGeometryType& GetGeometry(IndexType GeometryId)
    {
        return *pGetGeometry(GeometryId);
    }
 
    const TGeometryType& GetGeometry(IndexType GeometryId) const
    {
        return *pGetGeometry(GeometryId);
    }
 
    TGeometryType& GetGeometry(std::string GeometryName)
    {
        return *pGetGeometry(GeometryName);
    }
 
    const TGeometryType& GetGeometry(std::string GeometryName) const
    {
        return *pGetGeometry(GeometryName);
    }
 
 
    void RemoveGeometry(IndexType GeometryId)
    {
        mGeometries.erase(GeometryId);
    }
 
    void RemoveGeometry(std::string GeometryName)
    {
        auto index = TGeometryType::GenerateId(GeometryName);
 
        mGeometries.erase(index);
    }
 
 
    bool HasGeometry(IndexType GeometryId) const
    {
        return (mGeometries.find(GeometryId) != mGeometries.end());
    }
 
    bool HasGeometry(std::string GeometryName) const
    {
        auto hash_index = TGeometryType::GenerateId(GeometryName);
 
        return (mGeometries.find(hash_index) != mGeometries.end());
    }
 
 
    GeometryIterator GeometriesBegin()
    {
        return mGeometries.begin();
    }
 
    GeometryConstantIterator GeometriesBegin() const
    {
        return mGeometries.begin();
    }
 
    GeometryIterator GeometriesEnd()
    {
        return mGeometries.end();
    }
 
    GeometryConstantIterator GeometriesEnd() const
    {
        return mGeometries.end();
    }
 
 
    GeometriesMapType& Geometries()
    {
        return mGeometries;
    }
 
    const GeometriesMapType& Geometries() const
    {
        return mGeometries;
    }
 
 
    std::string Info() const
    {
        return "GeometryContainer";
    }
 
    void PrintInfo(std::ostream& rOStream) const
    {
        rOStream << Info();
    }
 
    void PrintData(std::ostream& rOStream) const
    {
        rOStream << "Number of Geometries: " << mGeometries.size() << std::endl;
    }
 
    virtual void PrintInfo(std::ostream& rOStream, std::string const& PrefixString) const
    {
        rOStream << PrefixString << Info();
    }
 
    virtual void PrintData(std::ostream& rOStream, std::string const& PrefixString ) const
    {
        rOStream << PrefixString << "Number of Geometries: " << mGeometries.size() << std::endl;
    }
 
 
private:
 
    GeometriesMapType mGeometries;
 
 
    friend class Serializer;
 
    void save(Serializer& rSerializer) const
    {
        rSerializer.save("Geometries", mGeometries);
    }
 
    void load(Serializer& rSerializer)
    {
        rSerializer.load("Geometries", mGeometries);
    }
 
 
    GeometryContainer& operator=(const GeometryContainer& rOther)
    {
        mGeometries = rOther.mGeometries;
        return *this;
    }
 
 
}; // Class GeometryContainer
 
 
template<class TGeometryType>
inline std::istream& operator >> (std::istream& rIStream,
                                  GeometryContainer<TGeometryType>& rThis);
 
template<class TGeometryType>
inline std::ostream& operator << (std::ostream& rOStream,
                                  const GeometryContainer<TGeometryType>& rThis)
{
    rThis.PrintInfo(rOStream);
    rOStream << std::endl;
    rThis.PrintData(rOStream);
 
    return rOStream;
}
 
}  // namespace Kratos.