Exemple #1
0
    def test__too_large_interp_dist__raises_runtimeerror(self):
        POLYGON = Polygon([[0, 0], [10, 0], [10, 10], [0, 10]])

        with self.assertRaises(RuntimeError):
            Centerline(POLYGON, 10)

        centerline = Centerline(POLYGON, 5)
        self.assertIsInstance(centerline, Centerline)
def generate_centerlines(polygon_shps):
    '''
    Create centerlines
    :param polygon_shps: input polygons
    :return: dictionary of linestrings
    '''
    dct_centerlines = {}
    for i, geom in enumerate(polygon_shps):
        print(" now running Centerline creation")
        center_obj = Centerline(geom, 5) # 5 is the guessed smallest nodes distance, to avoid QhullError
        center_line_shply_line = center_obj.create_centerline()
        dct_centerlines[i] = center_line_shply_line
    return dct_centerlines
def generate_centerlines(polygon_shps):
    '''
    Create centerlines
    :param polygon_shps: input polygons
    :return: dictionary of linestrings
    '''
    dct_centerlines = {}
    for i, geom in enumerate(polygon_shps):
        print(" now running Centerline creation ", i)
        center_obj = Centerline(geom, 0.00019) # works for image-extracted rivers (cell res=0.0002), if larger than this it will give me QhullError
        center_line_shply_line = center_obj.create_centerline()
        dct_centerlines[i] = center_line_shply_line
    return dct_centerlines
Exemple #4
0
    def run(self):
        """
        Starts processing the imported SHP file.
        It sedns the polygon's geometry allong with the interpolation distance
        to the Centerline class to create a Centerline object.
        The results (the polygon's ID and the geometry of the centerline) are
        added to the dictionary.
        """

        for key in self.dct_polygons.keys():
            poly_geom = self.dct_polygons[key]
            centerlineObj = Centerline(poly_geom, self.dist)

            self.dct_centerlines[key] = centerlineObj.createCenterline()
    def run(self):
        """
        Starts processing the imported SHP file.
        It sedns the polygon's geometry allong with the interpolation distance
        to the Centerline class to create a Centerline object.
        The results (the polygon's ID and the geometry of the centerline) are
        added to the dictionary.
        """

        for key in self.dct_polygons.keys():
            poly_geom = self.dct_polygons[key]
            centerlineObj = Centerline(poly_geom, self.dist)

            self.dct_centerlines[key] = centerlineObj.createCenterline()
Exemple #6
0
def generate_centerlines(polygon_shps):
    '''
    Create centerlines
    :param polygon_shps: input polygons
    :return: dictionary of linestrings
    '''
    dct_centerlines = {}

    for i, geom in enumerate(polygon_shps):
        print " now running Centerline creation"
        center_obj = Centerline(geom, 0.5)
        center_line_shply_line = center_obj.create_centerline()
        dct_centerlines[i] = center_line_shply_line

    return dct_centerlines
def generate_centerlines(polygon_shps):
    '''
    Create centerlines
    :param polygon_shps: input polygons
    :return: dictionary of linestrings
    '''
    dct_centerlines = {}

    for i, geom in enumerate(polygon_shps):
        print " now running Centerline creation"
        center_obj = Centerline(geom, 0.5)
        center_line_shply_line = center_obj.create_centerline()
        dct_centerlines[i] = center_line_shply_line

    return dct_centerlines
Exemple #8
0
    def test__polygon_with_interior_ring__returns_multilinestring(self):
        EXTERIOR = [(0, 0), (0, 2), (2, 2), (2, 0), (0, 0)]
        INTERIOR = [(1, 0), (0.5, 0.5), (1, 1), (1.5, 0.5), (1, 0)][::-1]
        POLYGON = Polygon(EXTERIOR, [INTERIOR])

        centerline = Centerline(POLYGON)

        self.assertIsInstance(centerline, MultiLineString)
Exemple #9
0
    def test__object_has_assigned_attributes(self):
        POLYGON = Polygon([[0, 0], [0, 4], [4, 4], [4, 0]])
        ATTRIBUTES = {'id': 1, 'name': 'polygon', 'valid': True}

        centerline = Centerline(POLYGON, **ATTRIBUTES)

        self.assertEqual(centerline.id, ATTRIBUTES.get('id'))
        self.assertEqual(centerline.name, ATTRIBUTES.get('name'))
        self.assertEqual(centerline.valid, ATTRIBUTES.get('valid'))
Exemple #10
0
    def test__geometry_collection__raises_valueerror(self):
        GEOMETRY_COLLECTION = GeometryCollection(
            (Point(0, 0),
             LineString([(0, 0), (0.8, 0.8), (1.8, 0.95),
                         (2.6, 0.5)]), Polygon([[0, 0], [0, 4], [4, 4], [4,
                                                                         0]])))

        with self.assertRaises(ValueError):
            Centerline(GEOMETRY_COLLECTION)
Exemple #11
0
    def test__multipolygon__raises_valueerror(self):
        POLYGONS = [Point(i, 0).buffer(0.1) for i in range(2)]
        MULTIPOLYGON = MultiPolygon(POLYGONS)

        with self.assertRaises(ValueError):
            Centerline(MULTIPOLYGON)
Exemple #12
0
 def setUpClass(cls):
     cls.POLYGON = Polygon([[0, 0], [0, 4], [4, 4], [4, 0]])
     cls.CENTERLINE = Centerline(cls.POLYGON)
Exemple #13
0
    def test__multilinestring__raises_valueerror(self):
        MULTILINESTRING = MultiLineString([((0, 0), (1, 1)),
                                           ((-1, 0), (1, 0))])

        with self.assertRaises(ValueError):
            Centerline(MULTILINESTRING)
Exemple #14
0
    def test__linestring__raises_valueerror(self):
        LINESTRING = LineString([(0, 0), (0.8, 0.8), (1.8, 0.95), (2.6, 0.5)])

        with self.assertRaises(ValueError):
            Centerline(LINESTRING)
Exemple #15
0
    def test__multipoint__raises_valueerror(self):
        MULTIPOINT = MultiPoint([Point(0, 0), Point(1, 1)])

        with self.assertRaises(ValueError):
            Centerline(MULTIPOINT)
Exemple #16
0
    def test__point__raises_valueerror(self):
        POINT = Point(0, 0)

        with self.assertRaises(ValueError):
            Centerline(POINT)
Exemple #17
0
 def setUpClass(cls):
     EXTERIOR = [(0, 0), (0, 2), (2, 2), (2, 0), (0, 0)]
     INTERIOR = [(1, 0), (0.5, 0.5), (1, 1), (1.5, 0.5), (1, 0)][::-1]
     cls.POLYGON = Polygon(EXTERIOR, [INTERIOR])
     cls.CENTERLINE = Centerline(cls.POLYGON)
Exemple #18
0
    def test__polygon__returns_multilinestring(self):
        POLYGON = Polygon([[0, 0], [0, 4], [4, 4], [4, 0]])

        centerline = Centerline(POLYGON)

        self.assertIsInstance(centerline, MultiLineString)