blob: 7da638e4997503f5391dd6ece6fb1f8c870813d7 [file] [log] [blame]
// Boost.Polygon library voronoi_builder.hpp header file
// Copyright Andrii Sydorchuk 2010-2012.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
// See http://www.boost.org for updates, documentation, and revision history.
#ifndef BOOST_POLYGON_VORONOI_BUILDER
#define BOOST_POLYGON_VORONOI_BUILDER
#include <algorithm>
#include <map>
#include <queue>
#include <utility>
#include <vector>
#include "detail/voronoi_ctypes.hpp"
#include "detail/voronoi_predicates.hpp"
#include "detail/voronoi_structures.hpp"
#include "voronoi_geometry_type.hpp"
namespace boost {
namespace polygon {
// GENERAL INFO:
// The sweepline algorithm implementation to compute Voronoi diagram of
// points and non-intersecting segments (excluding endpoints).
// Complexity - O(N*logN), memory usage - O(N), where N is the total number
// of input geometries.
//
// CONTRACT:
// 1) Input geometries should have integral (e.g. int32, int64) coordinate type.
// 2) Input geometries should not intersect except their endpoints.
//
// IMPLEMENTATION DETAILS:
// Each input point creates one input site. Each input segment creates three
// input sites: two for its endpoints and one for the segment itself (this is
// made to simplify output construction). All the site objects are constructed
// and sorted at the algorithm initialization step. Priority queue is used to
// dynamically hold circle events. At each step of the algorithm execution the
// leftmost event is retrieved by comparing the current site event and the
// topmost element from the circle event queue. STL map (red-black tree)
// container was chosen to hold state of the beach line. The keys of the map
// correspond to the neighboring sites that form a bisector and values map to
// the corresponding Voronoi edges in the output data structure.
template <typename T,
typename CTT = detail::voronoi_ctype_traits<T>,
typename VP = detail::voronoi_predicates<CTT> >
class voronoi_builder {
public:
typedef typename CTT::int_type int_type;
typedef typename CTT::fpt_type fpt_type;
voronoi_builder() : index_(0) {}
// Each point creates a single site event.
std::size_t insert_point(const int_type& x, const int_type& y) {
site_events_.push_back(site_event_type(x, y));
site_events_.back().initial_index(index_);
site_events_.back().source_category(SOURCE_CATEGORY_SINGLE_POINT);
return index_++;
}
// Each segment creates three site events that correspond to:
// 1) the start point of the segment;
// 2) the end point of the segment;
// 3) the segment itself defined by its start point.
std::size_t insert_segment(
const int_type& x1, const int_type& y1,
const int_type& x2, const int_type& y2) {
// Set up start point site.
point_type p1(x1, y1);
site_events_.push_back(site_event_type(p1));
site_events_.back().initial_index(index_);
site_events_.back().source_category(SOURCE_CATEGORY_SEGMENT_START_POINT);
// Set up end point site.
point_type p2(x2, y2);
site_events_.push_back(site_event_type(p2));
site_events_.back().initial_index(index_);
site_events_.back().source_category(SOURCE_CATEGORY_SEGMENT_END_POINT);
// Set up segment site.
if (point_comparison_(p1, p2)) {
site_events_.push_back(site_event_type(p1, p2));
site_events_.back().source_category(SOURCE_CATEGORY_INITIAL_SEGMENT);
} else {
site_events_.push_back(site_event_type(p2, p1));
site_events_.back().source_category(SOURCE_CATEGORY_REVERSE_SEGMENT);
}
site_events_.back().initial_index(index_);
return index_++;
}
// Run sweepline algorithm and fill output data structure.
template <typename OUTPUT>
void construct(OUTPUT* output) {
// Init structures.
output->_reserve(site_events_.size());
init_sites_queue();
init_beach_line(output);
// The algorithm stops when there are no events to process.
event_comparison_predicate event_comparison;
while (!circle_events_.empty() ||
!(site_event_iterator_ == site_events_.end())) {
if (circle_events_.empty()) {
process_site_event(output);
} else if (site_event_iterator_ == site_events_.end()) {
process_circle_event(output);
} else {
if (event_comparison(*site_event_iterator_,
circle_events_.top().first)) {
process_site_event(output);
} else {
process_circle_event(output);
}
}
while (!circle_events_.empty() &&
!circle_events_.top().first.is_active()) {
circle_events_.pop();
}
}
beach_line_.clear();
// Finish construction.
output->_build();
}
void clear() {
index_ = 0;
site_events_.clear();
}
private:
typedef detail::point_2d<int_type> point_type;
typedef detail::site_event<int_type> site_event_type;
typedef typename std::vector<site_event_type>::const_iterator
site_event_iterator_type;
typedef detail::circle_event<fpt_type> circle_event_type;
typedef typename VP::template point_comparison_predicate<point_type>
point_comparison_predicate;
typedef typename VP::
template event_comparison_predicate<site_event_type, circle_event_type>
event_comparison_predicate;
typedef typename VP::
template circle_formation_predicate<site_event_type, circle_event_type>
circle_formation_predicate_type;
typedef void edge_type;
typedef detail::beach_line_node_key<site_event_type> key_type;
typedef detail::beach_line_node_data<edge_type, circle_event_type>
value_type;
typedef typename VP::template node_comparison_predicate<key_type>
node_comparer_type;
typedef std::map< key_type, value_type, node_comparer_type > beach_line_type;
typedef typename beach_line_type::iterator beach_line_iterator;
typedef std::pair<circle_event_type, beach_line_iterator> event_type;
typedef struct {
bool operator()(const event_type& lhs, const event_type& rhs) const {
return predicate(rhs.first, lhs.first);
}
event_comparison_predicate predicate;
} event_comparison_type;
typedef detail::ordered_queue<event_type, event_comparison_type>
circle_event_queue_type;
typedef std::pair<point_type, beach_line_iterator> end_point_type;
void init_sites_queue() {
// Sort site events.
std::sort(site_events_.begin(), site_events_.end(),
event_comparison_predicate());
// Remove duplicates.
site_events_.erase(std::unique(
site_events_.begin(), site_events_.end()), site_events_.end());
// Index sites.
for (std::size_t cur = 0; cur < site_events_.size(); ++cur) {
site_events_[cur].sorted_index(cur);
}
// Init site iterator.
site_event_iterator_ = site_events_.begin();
}
template <typename OUTPUT>
void init_beach_line(OUTPUT* output) {
if (site_events_.empty())
return;
if (site_events_.size() == 1) {
// Handle single site event case.
output->_process_single_site(site_events_[0]);
++site_event_iterator_;
} else {
int skip = 0;
while (site_event_iterator_ != site_events_.end() &&
VP::is_vertical(site_event_iterator_->point0(),
site_events_.begin()->point0()) &&
VP::is_vertical(*site_event_iterator_)) {
++site_event_iterator_;
++skip;
}
if (skip == 1) {
// Init beach line with the first two sites.
init_beach_line_default(output);
} else {
// Init beach line with collinear vertical sites.
init_beach_line_collinear_sites(output);
}
}
}
// Init beach line with the two first sites.
// The first site is always a point.
template <typename OUTPUT>
void init_beach_line_default(OUTPUT* output) {
// Get the first and the second site event.
site_event_iterator_type it_first = site_events_.begin();
site_event_iterator_type it_second = site_events_.begin();
++it_second;
insert_new_arc(
*it_first, *it_first, *it_second, beach_line_.end(), output);
// The second site was already processed. Move the iterator.
++site_event_iterator_;
}
// Init beach line with collinear sites.
template <typename OUTPUT>
void init_beach_line_collinear_sites(OUTPUT* output) {
site_event_iterator_type it_first = site_events_.begin();
site_event_iterator_type it_second = site_events_.begin();
++it_second;
while (it_second != site_event_iterator_) {
// Create a new beach line node.
key_type new_node(*it_first, *it_second);
// Update the output.
edge_type* edge = output->_insert_new_edge(*it_first, *it_second).first;
// Insert a new bisector into the beach line.
beach_line_.insert(beach_line_.end(),
std::pair<key_type, value_type>(new_node, value_type(edge)));
// Update iterators.
++it_first;
++it_second;
}
}
void deactivate_circle_event(value_type* value) {
if (value->circle_event()) {
value->circle_event()->deactivate();
value->circle_event(NULL);
}
}
template <typename OUTPUT>
void process_site_event(OUTPUT* output) {
// Get next site event to process.
site_event_type site_event = *site_event_iterator_;
// Move site iterator.
site_event_iterator_type last = site_event_iterator_ + 1;
// If a new site is an end point of some segment,
// remove temporary nodes from the beach line data structure.
if (!site_event.is_segment()) {
while (!end_points_.empty() &&
end_points_.top().first == site_event.point0()) {
beach_line_iterator b_it = end_points_.top().second;
end_points_.pop();
beach_line_.erase(b_it);
}
} else {
while (last != site_events_.end() &&
last->is_segment() && last->point0() == site_event.point0())
++last;
}
// Find the node in the binary search tree with left arc
// lying above the new site point.
key_type new_key(*site_event_iterator_);
beach_line_iterator right_it = beach_line_.lower_bound(new_key);
for (; site_event_iterator_ != last; ++site_event_iterator_) {
site_event = *site_event_iterator_;
beach_line_iterator left_it = right_it;
// Do further processing depending on the above node position.
// For any two neighboring nodes the second site of the first node
// is the same as the first site of the second node.
if (right_it == beach_line_.end()) {
// The above arc corresponds to the second arc of the last node.
// Move the iterator to the last node.
--left_it;
// Get the second site of the last node
const site_event_type& site_arc = left_it->first.right_site();
// Insert new nodes into the beach line. Update the output.
right_it = insert_new_arc(
site_arc, site_arc, site_event, right_it, output);
// Add a candidate circle to the circle event queue.
// There could be only one new circle event formed by
// a new bisector and the one on the left.
activate_circle_event(left_it->first.left_site(),
left_it->first.right_site(),
site_event, right_it);
} else if (right_it == beach_line_.begin()) {
// The above arc corresponds to the first site of the first node.
const site_event_type& site_arc = right_it->first.left_site();
// Insert new nodes into the beach line. Update the output.
left_it = insert_new_arc(
site_arc, site_arc, site_event, right_it, output);
// If the site event is a segment, update its direction.
if (site_event.is_segment()) {
site_event.inverse();
}
// Add a candidate circle to the circle event queue.
// There could be only one new circle event formed by
// a new bisector and the one on the right.
activate_circle_event(site_event, right_it->first.left_site(),
right_it->first.right_site(), right_it);
right_it = left_it;
} else {
// The above arc corresponds neither to the first,
// nor to the last site in the beach line.
const site_event_type& site_arc2 = right_it->first.left_site();
const site_event_type& site3 = right_it->first.right_site();
// Remove the candidate circle from the event queue.
deactivate_circle_event(&right_it->second);
--left_it;
const site_event_type& site_arc1 = left_it->first.right_site();
const site_event_type& site1 = left_it->first.left_site();
// Insert new nodes into the beach line. Update the output.
beach_line_iterator new_node_it =
insert_new_arc(site_arc1, site_arc2, site_event, right_it, output);
// Add candidate circles to the circle event queue.
// There could be up to two circle events formed by
// a new bisector and the one on the left or right.
activate_circle_event(site1, site_arc1, site_event, new_node_it);
// If the site event is a segment, update its direction.
if (site_event.is_segment()) {
site_event.inverse();
}
activate_circle_event(site_event, site_arc2, site3, right_it);
right_it = new_node_it;
}
}
}
// In general case circle event is made of the three consecutive sites
// that form two bisectors in the beach line data structure.
// Let circle event sites be A, B, C, two bisectors that define
// circle event are (A, B), (B, C). During circle event processing
// we remove (A, B), (B, C) and insert (A, C). As beach line comparison
// works correctly only if one of the nodes is a new one we remove
// (B, C) bisector and change (A, B) bisector to the (A, C). That's
// why we use const_cast there and take all the responsibility that
// map data structure keeps correct ordering.
template <typename OUTPUT>
void process_circle_event(OUTPUT* output) {
// Get the topmost circle event.
const event_type& e = circle_events_.top();
const circle_event_type& circle_event = e.first;
beach_line_iterator it_first = e.second;
beach_line_iterator it_last = it_first;
// Get the C site.
site_event_type site3 = it_first->first.right_site();
// Get the half-edge corresponding to the second bisector - (B, C).
edge_type* bisector2 = it_first->second.edge();
// Get the half-edge corresponding to the first bisector - (A, B).
--it_first;
edge_type* bisector1 = it_first->second.edge();
// Get the A site.
site_event_type site1 = it_first->first.left_site();
if (!site1.is_segment() && site3.is_segment() &&
site3.point1() == site1.point0()) {
site3.inverse();
}
// Change the (A, B) bisector node to the (A, C) bisector node.
const_cast<key_type&>(it_first->first).right_site(site3);
// Insert the new bisector into the beach line.
it_first->second.edge(output->_insert_new_edge(
site1, site3, circle_event, bisector1, bisector2).first);
// Remove the (B, C) bisector node from the beach line.
beach_line_.erase(it_last);
it_last = it_first;
// Pop the topmost circle event from the event queue.
circle_events_.pop();
// Check new triplets formed by the neighboring arcs
// to the left for potential circle events.
if (it_first != beach_line_.begin()) {
deactivate_circle_event(&it_first->second);
--it_first;
const site_event_type& site_l1 = it_first->first.left_site();
activate_circle_event(site_l1, site1, site3, it_last);
}
// Check the new triplet formed by the neighboring arcs
// to the right for potential circle events.
++it_last;
if (it_last != beach_line_.end()) {
deactivate_circle_event(&it_last->second);
const site_event_type& site_r1 = it_last->first.right_site();
activate_circle_event(site1, site3, site_r1, it_last);
}
}
// Insert new nodes into the beach line. Update the output.
template <typename OUTPUT>
beach_line_iterator insert_new_arc(
const site_event_type& site_arc1, const site_event_type &site_arc2,
const site_event_type& site_event, beach_line_iterator position,
OUTPUT* output) {
// Create two new bisectors with opposite directions.
key_type new_left_node(site_arc1, site_event);
key_type new_right_node(site_event, site_arc2);
// Set correct orientation for the first site of the second node.
if (site_event.is_segment()) {
new_right_node.left_site().inverse();
}
// Update the output.
std::pair<edge_type*, edge_type*> edges =
output->_insert_new_edge(site_arc2, site_event);
position = beach_line_.insert(position,
typename beach_line_type::value_type(
new_right_node, value_type(edges.second)));
if (site_event.is_segment()) {
// Update the beach line with temporary bisector, that will
// disappear after processing site event corresponding to the
// second endpoint of the segment site.
key_type new_node(site_event, site_event);
new_node.right_site().inverse();
position = beach_line_.insert(position,
typename beach_line_type::value_type(new_node, value_type(NULL)));
// Update the data structure that holds temporary bisectors.
end_points_.push(std::make_pair(site_event.point1(), position));
}
position = beach_line_.insert(position,
typename beach_line_type::value_type(
new_left_node, value_type(edges.first)));
return position;
}
// Add a new circle event to the event queue.
// bisector_node corresponds to the (site2, site3) bisector.
void activate_circle_event(const site_event_type& site1,
const site_event_type& site2,
const site_event_type& site3,
beach_line_iterator bisector_node) {
circle_event_type c_event;
// Check if the three input sites create a circle event.
if (circle_formation_predicate_(site1, site2, site3, c_event)) {
// Add the new circle event to the circle events queue.
// Update bisector's circle event iterator to point to the
// new circle event in the circle event queue.
event_type& e = circle_events_.push(
std::pair<circle_event_type, beach_line_iterator>(
c_event, bisector_node));
bisector_node->second.circle_event(&e.first);
}
}
private:
point_comparison_predicate point_comparison_;
struct end_point_comparison {
bool operator() (const end_point_type& end1,
const end_point_type& end2) const {
return point_comparison(end2.first, end1.first);
}
point_comparison_predicate point_comparison;
};
std::vector<site_event_type> site_events_;
site_event_iterator_type site_event_iterator_;
std::priority_queue< end_point_type, std::vector<end_point_type>,
end_point_comparison > end_points_;
circle_event_queue_type circle_events_;
beach_line_type beach_line_;
circle_formation_predicate_type circle_formation_predicate_;
std::size_t index_;
// Disallow copy constructor and operator=
voronoi_builder(const voronoi_builder&);
void operator=(const voronoi_builder&);
};
typedef voronoi_builder<detail::int32> default_voronoi_builder;
} // polygon
} // boost
#endif // BOOST_POLYGON_VORONOI_BUILDER