/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (C) 2005-2006 Refractions Research Inc.
 * Copyright (C) 2001-2002 Vivid Solutions Inc.
 *
 * This is free software; you can redistribute and/or modify it under
 * the terms of the GNU Lesser General Public Licence as published
 * by the Free Software Foundation.
 * See the COPYING file for more information.
 *
 **********************************************************************
 *
 * Last port: geomgraph/NodeMap.java rev. 1.3 (JTS-1.10)
 *
 **********************************************************************/


#pragma once

#include <geos/export.h>
#include <map>
#include <memory>
#include <vector>
#include <string>

#include <geos/geom/Coordinate.h> // for CoordinateLessThan
#include <geos/geomgraph/Node.h> // for testInvariant


#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable: 4251) // warning C4251: needs to have dll-interface to be used by clients of class
#endif

// Forward declarations
namespace geos {
namespace geomgraph {
class Node;
class EdgeEnd;
class NodeFactory;
}
}

namespace geos {
namespace geomgraph { // geos.geomgraph

class GEOS_DLL NodeMap final {
public:

    typedef std::map<geom::Coordinate*, std::unique_ptr<Node>, geom::CoordinateLessThan> container;

    typedef container::iterator iterator;

    typedef container::const_iterator const_iterator;

    container nodeMap;

    const NodeFactory& nodeFact;

    /// \brief
    /// NodeMap will keep a reference to the NodeFactory,
    /// keep it alive for the whole NodeMap lifetime
    NodeMap(const NodeFactory& newNodeFact);

    Node* addNode(const geom::Coordinate& coord);

    Node* addNode(Node* n);

    /// \brief
    /// Adds a node for the start point of this EdgeEnd
    /// (if one does not already exist in this map).
    /// Adds the EdgeEnd to the (possibly new) node.
    ///
    /// If ownership of the EdgeEnd should be transferred
    /// to the Node, use the unique_ptr overload instead.
    void add(EdgeEnd* e);

    void add(std::unique_ptr<EdgeEnd>&& e);

    Node* find(const geom::Coordinate& coord) const;

    const_iterator
    begin() const
    {
        return nodeMap.begin();
    }

    const_iterator
    end() const
    {
        return nodeMap.end();
    }

    iterator
    begin()
    {
        return nodeMap.begin();
    }

    iterator
    end()
    {
        return nodeMap.end();
    }

    void getBoundaryNodes(uint8_t geomIndex,
                          std::vector<Node*>& bdyNodes) const;

    std::string print() const;

    void
    testInvariant()
    {
#ifndef NDEBUG
        // Each Coordinate key is a pointer inside the Node value
        for(const auto& nodeIt: nodeMap) {
            const auto* n = nodeIt.second.get();
            geom::Coordinate* c = const_cast<geom::Coordinate*>(
                                      &(n->getCoordinate())
                                  );
            assert(nodeIt.first == c);
            (void)c;
        }
#endif
    }

private:

    // Declare type as noncopyable
    NodeMap(const NodeMap& other) = delete;
    NodeMap& operator=(const NodeMap& rhs) = delete;
};

} // namespace geos.geomgraph
} // namespace geos

#ifdef _MSC_VER
#pragma warning(pop)
#endif