Geometry.hpp

/* Copyright (c) 2013       David Sichau <mail"at"sichau"dot"eu>
 *               2013-2015  Simon Tanaka <tanakas"at"gmx"dot"ch>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
#ifndef GEOMETRY_GEOMETRY_HPP
#define GEOMETRY_GEOMETRY_HPP

#include <LbmLib/include/nodes/GeometryNode.hpp>
#include <LbmLib/include/Constants.hpp>
#include <UtilLib/include/geometry/FastNeighborList.hpp>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include <boost/geometry/geometry.hpp>
#include <boost/geometry/geometries/register/point.hpp>
#include <boost/geometry/index/rtree.hpp>

BOOST_GEOMETRY_REGISTER_POINT_2D_GET_SET(LbmLib::nodes::GeometryNode,
                                         double,
                                         cs::cartesian,
                                         LbmLib::nodes::LagrangianPoint::getXPos,
                                         LbmLib::nodes::LagrangianPoint::getYPos,
                                         LbmLib::nodes::LagrangianPoint::setXPos,
                                         LbmLib::nodes::LagrangianPoint::setYPos)

namespace boost {
namespace geometry {
namespace index {

//specializing indexable to store shared_ptr in rtree:
template <typename Value>
struct indexable <std::shared_ptr<Value> >
{
    typedef Value const& result_type;
    result_type operator() (std::shared_ptr<Value> const& v) const { return *v; }
};

} // end namespace index
} // end namespace geometry
} // end namespace boost

namespace LbmLib {
namespace geometry {

class Connection;

class Geometry {
 public:
    explicit Geometry(const std::string& filename);

    ~Geometry();

    void writeGeometry(const std::string& fileName) const;

    const std::map<unsigned int,
            std::shared_ptr<nodes::GeometryNode> >& getGeometryNodes() const {
        return geometryNodes_;
    }

    const std::vector<std::shared_ptr<nodes::GeometryNode> > getGeometryNodesWithinRadius(const double x,
                                                                                          const double y,
                                                                                          const double radius) const;



    const std::vector<std::shared_ptr<nodes::GeometryNode> > getGeometryNodesWithinRadiusWithAvoidance(const double x,
                                                                                                       const double y,
                                                                                                       const double radius,
                                                                                                       const unsigned int avoidDomainID) const;
    std::shared_ptr<nodes::GeometryNode> getGeometryNodesWithinRadiusWithAvoidanceClosest(const double x,
                                                     const double y,
                                                     const double radius,
                                                     const unsigned int avoidDomainID) const;

    const std::vector<std::shared_ptr<Connection> >& getConnections() const;

    unsigned int addGeometryNode(const double x,const double y);

    unsigned int removeGeometryNode(const unsigned int nodeid);

    void eraseConnection(std::shared_ptr<Connection> toDelete);

    void addConnection(std::shared_ptr<nodes::GeometryNode> p1,
                       std::shared_ptr<nodes::GeometryNode> p2,
                       const std::map<std::string, std::vector<std::string> > boundaryConditionDescriptor,
                       const unsigned int domainIdentifier);

    void moveGeometryNodes();

    bool checkGeometryIntegrity() const;

    void invalidateRangeQueryDataStructure();

    void reconstructRangeQueryDataStructure() const;

    std::map<unsigned int, unsigned int> getCellTypeTrackerMap(void) const;

private:
    void loadGeometryTXT(const std::string& fileName);

    void loadGeometryVTK(const std::string& fileName);

    std::vector<std::shared_ptr<Connection> > connections_;

    std::map<unsigned int,
            std::shared_ptr<nodes::GeometryNode> > geometryNodes_;

    mutable UtilLib::geometry::fastneighborlist< std::shared_ptr<nodes::GeometryNode> > fastneighborlist_;

    mutable bool isValidRangeQueryDataStructure_;

    std::map<unsigned int,unsigned int> cellTypeTrackerMap_;
};

}   // end namespace
}  // end namespace
#endif  // GEOMETRY_GEOMETRY_HPP