#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
# Implementation of Ramer-Douglas-Peucker algorithm
__all__ = ['ramerdouglas']
def ramerdouglas(line, dist):
"""Does Ramer-Douglas-Peucker simplification of a curve with `dist`
threshold.
`line` is a list-of-tuples, where each tuple is a 2D coordinate
Usage is like so:
>>> myline = [(0.0, 0.0), (1.0, 2.0), (2.0, 1.0)]
>>> simplified = ramerdouglas(myline, dist = 1.0)
"""
if len(line) < 3:
return line
(begin, end) = (line[0], line[-1]) if line[0] != line[-1] else (line[0], line[-2])
begin_end_x = end[0] - begin[0]
begin_end_y = end[1] - begin[1]
dist_sq = []
dist_append = dist_sq.append
for curr in line[1:-1]:
begin_curr_x = curr[0] - begin[0]
begin_curr_y = curr[1] - begin[1]
dist_append((begin_curr_y * begin_end_x - begin_curr_x * begin_end_y)**2 /
(begin_end_x**2 + begin_end_y**2))
maxdist = max(dist_sq)
if maxdist < dist ** 2:
return [begin, end]
if len(dist_sq) > 500:
pos = len(dist_sq) // 2
else:
pos = dist_sq.index(maxdist)
return (ramerdouglas(line[:pos + 2], dist) +
ramerdouglas(line[pos + 1:], dist)[1:])
|