BoundaryChainNoder.h

/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (c) Martin Davis 2022
 *
 * 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.
 *
 **********************************************************************/
 
#pragma once
 
#include <geos/noding/Noder.h>     // for composition
#include <geos/noding/SegmentString.h>     // for composition
#include <geos/geom/LineSegment.h> // for composition
 
#include <unordered_set>
 
// Forward declarations
namespace geos {
namespace geom {
class CoordinateSequence;
class Coordinate;
}
namespace noding {
class NodedSegmentString;
}
}
 
namespace geos {   // geos
namespace noding { // geos::noding
 
class GEOS_DLL BoundaryChainNoder : public Noder {
    using Coordinate = geos::geom::Coordinate;
 
private:
 
    class BoundarySegmentMap {
 
        private:
 
            // Members
            SegmentString* segString;
            std::vector<bool> isBoundary;
 
            static SegmentString* createChain(
                const SegmentString* segString,
                std::size_t startIndex,
                std::size_t endIndex,
                bool constructZ,
                bool constructM);
 
            std::size_t findChainStart(std::size_t index) const;
            std::size_t findChainEnd(std::size_t index) const;
 
        public:
 
            BoundarySegmentMap(SegmentString* ss)
                : segString(ss) {
                    isBoundary.resize(ss->size()-1, false);
                };
 
            void setBoundarySegment(std::size_t index);
            void createChains(std::vector<SegmentString*>& chainList, bool constructZ, bool constructM);
    };
 
    class Segment {
    public:
        Segment(const geom::CoordinateSequence& seq,
            BoundarySegmentMap& segMap,
            std::size_t index)
            : m_seq(seq)
            , m_segMap(segMap)
            , m_index(index)
            , m_flip(seq.getAt<geom::CoordinateXY>(index).compareTo(seq.getAt<geom::CoordinateXY>(index + 1)) < 0)
        {}
 
        const geom::CoordinateXY& p0() const {
            return m_seq.getAt<geom::CoordinateXY>(m_flip ? m_index : m_index + 1);
        }
 
        const geom::CoordinateXY& p1() const {
            return m_seq.getAt<geom::CoordinateXY>(m_flip ? m_index + 1 : m_index);
        }
 
        void markInBoundary() const {
            m_segMap.setBoundarySegment(m_index);
        };
 
        bool operator==(const Segment& other) const {
            return p0().equals2D(other.p0()) && p1().equals2D(other.p1());
        }
 
        struct HashCode {
            std::size_t operator()(const Segment& s) const {
                std::size_t h = std::hash<double>{}(s.p0().x);
                h ^= (std::hash<double>{}(s.p0().y) << 1);
                h ^= (std::hash<double>{}(s.p1().x) << 1);
                h ^= (std::hash<double>{}(s.p1().y) << 1);
                return h;
            }
        };
 
    private:
        const geom::CoordinateSequence& m_seq;
        BoundarySegmentMap& m_segMap;
        std::size_t m_index;
        bool m_flip;
    };
 
public:
    using SegmentSet = std::unordered_set<Segment, Segment::HashCode>;
 
    BoundaryChainNoder() : chainList(nullptr), m_constructZ(false), m_constructM(false) {};
 
    // Noder virtual methods
    std::vector<SegmentString*>* getNodedSubstrings() const override;
    void computeNodes(std::vector<SegmentString*>* inputSegStrings) override;
 
 
private:
 
    // Members
    std::vector<SegmentString*>* chainList;
    bool m_constructZ;
    bool m_constructM;
 
    // Methods
    void addSegments(std::vector<SegmentString*>* segStrings,
        SegmentSet& segSet,
        std::vector<BoundarySegmentMap>& includedSegs);
 
    static void addSegments(SegmentString* segString,
        BoundarySegmentMap& segInclude,
        SegmentSet& segSet);
 
    static void markBoundarySegments(SegmentSet& segSet);
 
    std::vector<SegmentString*>* extractChains(std::vector<BoundarySegmentMap>& sections) const;
 
    static bool segSetContains(SegmentSet& segSet, Segment& seg);
 
};
 
} // namespace geos::noding
} // namespace geos