/**********************************************************************
 *
 * GEOS - Geometry Engine Open Source
 * http://geos.osgeo.org
 *
 * Copyright (C) 2016 Daniel Baston
 *
 * 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: operation/distance/FacetSequenceTreeBuilder.java (f6187ee2 JTS-1.14)
 *
 **********************************************************************/

#include <geos/operation/distance/FacetSequenceTreeBuilder.h>
#include <geos/geom/LineString.h>
#include <geos/geom/Point.h>

using namespace geos::geom;
using namespace geos::index::strtree;

namespace geos {
namespace operation {
namespace distance {

STRtree*
FacetSequenceTreeBuilder::build(const Geometry* g)
{
    std::unique_ptr<STRtree> tree(new STRtree(STR_TREE_NODE_CAPACITY));
    std::unique_ptr<std::vector<FacetSequence*> > sections(computeFacetSequences(g));
    for(std::vector<FacetSequence*>::iterator it = sections->begin(); it != sections->end(); ++it) {
        FacetSequence* section = *it;
        tree->insert(section->getEnvelope(), section);
    }

    tree->build();
    return tree.release();
}

std::vector<FacetSequence*>*
FacetSequenceTreeBuilder::computeFacetSequences(const Geometry* g)
{
    std::unique_ptr<std::vector<FacetSequence*> > sections(new std::vector<FacetSequence*>());

    class FacetSequenceAdder;
    class FacetSequenceAdder : public geom::GeometryComponentFilter {
        std::vector<FacetSequence*>*  m_sections;

    public :
        FacetSequenceAdder(std::vector<FacetSequence*>* p_sections) :
            m_sections(p_sections) {}
        void
        filter_ro(const Geometry* geom) override
        {
            if(const LineString* ls = dynamic_cast<const LineString*>(geom)) {
                const CoordinateSequence* seq = ls->getCoordinatesRO();
                addFacetSequences(geom, seq, *m_sections);
            }
            else if(const Point* pt = dynamic_cast<const Point*>(geom)) {
                const CoordinateSequence* seq = pt->getCoordinatesRO();
                addFacetSequences(geom, seq, *m_sections);
            }
        }
    };

    FacetSequenceAdder facetSequenceAdder(sections.get());
    g->apply_ro(&facetSequenceAdder);

    return sections.release();
}

void
FacetSequenceTreeBuilder::addFacetSequences(const Geometry* geom, const CoordinateSequence* pts,
        std::vector<FacetSequence*>& sections)
{
    size_t i = 0;
    size_t size = pts->size();

    while(i <= size - 1) {
        size_t end = i + FACET_SEQUENCE_SIZE + 1;
        // if only one point remains after this section, include it in this
        // section
        if(end >= size - 1) {
            end = size;
        }
        FacetSequence* sect = new FacetSequence(geom, pts, i, end);
        sections.push_back(sect);
        i += FACET_SEQUENCE_SIZE;
    }
}

}
}
}