libraries/pyshapelib/pytest.py

import shapelib, dbflib, shptree

#
#       The the shapefile module
#

def make_shapefile(filename):
    # Create a shapefile with polygons
    outfile = shapelib.create(filename, shapelib.SHPT_POLYGON)

    # Create one very simple polygon and write it to the shapefile.  The
    # vertices should be given in clockwise order to comply with the
    # shapefile specification.
    obj = shapelib.SHPObject(shapelib.SHPT_POLYGON, 1,
                             [[(10, 10), (10, 20), (20, 20), (10, 10)]])
    print obj.extents()
    print obj.vertices()
    outfile.write_object(-1, obj)

    # Create a polygon with a hole.  Note that according to the
    # shapefile specification, the vertices of the outer ring have to be
    # in clockwise order and the inner rings have to be in counter
    # clockwise order.
    #
    # There's an optional fourth parameter which when given must be a
    # list of part types, one for each part of the shape.  For polygons,
    # the part type is always shapelib.SHPP_RING, though.  The part
    # types are only relevant for SHPT_MULTIPATCH shapefiles.
    obj = shapelib.SHPObject(shapelib.SHPT_POLYGON, 1,
                             [[(0, 0), (0, 40), (40, 40), (40, 0), (0, 0)],
                              [(10, 10), (20, 10), (20, 20), (10, 20),(10, 10)],
                              ])
    print obj.extents()
    print obj.vertices()
    outfile.write_object(-1, obj)

    # close the file.
    outfile.close()

def read_shapefile(filename):
    # open the shapefile
    shp = shapelib.ShapeFile(filename)

    # the info method returns a tuple (num_shapes, type, min, max) where
    # num_shapes is the number of shapes, type is the type code (one of
    # the SHPT* constants defined in the shapelib module) and min and
    # max are 4-element lists with the min. and max. values of the
    # vertices.
    print shp.info()

    # read_object reads a shape
    obj = shp.read_object(0)

    # The vertices method returns the shape as a list of lists of tuples.
    print obj.vertices()[0][:10]

    # The extents returns a tuple with two 4-element lists with the min.
    # and max. values of the vertices.
    print obj.extents()

    # The type attribute is the type code (one of the SHPT* constants
    # defined in the shapelib module)
    print obj.type

    # The id attribute is the shape id
    print obj.id

    # the cobject method returns a PyCObject containing the shapelib
    # SHPHandle. This is useful for passing shapefile objects to
    # C-Python extensions.
    print shp.cobject()

    # build a quad tree from the shapefile. The first argument must be
    # the return value of the shape file object's cobject method (this
    # is currently needed to access the shape file at the C-level). The
    # second argument is the dimension and the third the maximum depth.
    # 0 means to guess an appropriate depth
    tree = shptree.SHPTree(shp.cobject(), 2, 0)

    # Retrieve the ids for a region. Here we just use the extents of the
    # object previously read from the shapefile
    minima, maxima = obj.extents()
    print tree.find_shapes(minima[:2], maxima[:2])


make_shapefile("testfile")
read_shapefile("testfile")

#
#       Test the DBF file module.
#

def make_dbf(file):
    # create a new dbf file and add three fields.
    dbf = dbflib.create(file)
    dbf.add_field("NAME", dbflib.FTString, 20, 0)
    dbf.add_field("INT", dbflib.FTInteger, 10, 0)
    dbf.add_field("FLOAT", dbflib.FTDouble, 10, 4)

def add_dbf_records(file):
    # add some records to file
    dbf = dbflib.open(file, "r+b")
    # Records can be added as a dictionary...
    dbf.write_record(0, {'NAME': "Weatherwax", "INT":1, "FLOAT":3.1415926535})
    # ... or as a sequence
    dbf.write_record(1, ("Ogg", 2, -1000.1234))

def list_dbf(file):
    # print the contents of a dbf file to stdout
    dbf = dbflib.DBFFile(file)
    print "%d records, %d fields" % (dbf.record_count(), dbf.field_count())
    format = ""
    for i in range(dbf.field_count()):
        type, name, len, decc = dbf.field_info(i)
        if type == 0:
            format = format + " %%(%s)%ds" % (name, len)
        elif type == 1:
            format = format + " %%(%s)%dd" % (name, len)
        elif type == 2:
            format = format + " %%(%s)%dg" % (name, len)
    print format
    for i in range(dbf.record_count()):
        print format % dbf.read_record(i)


make_dbf("testfile")
add_dbf_records("testfile")
list_dbf("testfile")
1	import shapelib, dbflib, shptree
2
3	#
4	# The the shapefile module
5	#
6
7	def make_shapefile(filename):
8	# Create a shapefile with polygons
9	outfile = shapelib.create(filename, shapelib.SHPT_POLYGON)
10
11	# Create one very simple polygon and write it to the shapefile. The
12	# vertices should be given in clockwise order to comply with the
13	# shapefile specification.
14	obj = shapelib.SHPObject(shapelib.SHPT_POLYGON, 1,
15	[[(10, 10), (10, 20), (20, 20), (10, 10)]])
16	print obj.extents()
17	print obj.vertices()
18	outfile.write_object(-1, obj)
19
20	# Create a polygon with a hole. Note that according to the
21	# shapefile specification, the vertices of the outer ring have to be
22	# in clockwise order and the inner rings have to be in counter
23	# clockwise order.
24	#
25	# There's an optional fourth parameter which when given must be a
26	# list of part types, one for each part of the shape. For polygons,
27	# the part type is always shapelib.SHPP_RING, though. The part
28	# types are only relevant for SHPT_MULTIPATCH shapefiles.
29	obj = shapelib.SHPObject(shapelib.SHPT_POLYGON, 1,
30	[[(0, 0), (0, 40), (40, 40), (40, 0), (0, 0)],
31	[(10, 10), (20, 10), (20, 20), (10, 20),(10, 10)],
32	])
33	print obj.extents()
34	print obj.vertices()
35	outfile.write_object(-1, obj)
36
37	# close the file.
38	outfile.close()
39
40	def read_shapefile(filename):
41	# open the shapefile
42	shp = shapelib.ShapeFile(filename)
43
44	# the info method returns a tuple (num_shapes, type, min, max) where
45	# num_shapes is the number of shapes, type is the type code (one of
46	# the SHPT* constants defined in the shapelib module) and min and
47	# max are 4-element lists with the min. and max. values of the
48	# vertices.
49	print shp.info()
50
51	# read_object reads a shape
52	obj = shp.read_object(0)
53
54	# The vertices method returns the shape as a list of lists of tuples.
55	print obj.vertices()[0][:10]
56
57	# The extents returns a tuple with two 4-element lists with the min.
58	# and max. values of the vertices.
59	print obj.extents()
60
61	# The type attribute is the type code (one of the SHPT* constants
62	# defined in the shapelib module)
63	print obj.type
64
65	# The id attribute is the shape id
66	print obj.id
67
68	# the cobject method returns a PyCObject containing the shapelib
69	# SHPHandle. This is useful for passing shapefile objects to
70	# C-Python extensions.
71	print shp.cobject()
72
73	# build a quad tree from the shapefile. The first argument must be
74	# the return value of the shape file object's cobject method (this
75	# is currently needed to access the shape file at the C-level). The
76	# second argument is the dimension and the third the maximum depth.
77	# 0 means to guess an appropriate depth
78	tree = shptree.SHPTree(shp.cobject(), 2, 0)
79
80	# Retrieve the ids for a region. Here we just use the extents of the
81	# object previously read from the shapefile
82	minima, maxima = obj.extents()
83	print tree.find_shapes(minima[:2], maxima[:2])
84
85
86	make_shapefile("testfile")
87	read_shapefile("testfile")
88
89	#
90	# Test the DBF file module.
91	#
92
93	def make_dbf(file):
94	# create a new dbf file and add three fields.
95	dbf = dbflib.create(file)
96	dbf.add_field("NAME", dbflib.FTString, 20, 0)
97	dbf.add_field("INT", dbflib.FTInteger, 10, 0)
98	dbf.add_field("FLOAT", dbflib.FTDouble, 10, 4)
99
100	def add_dbf_records(file):
101	# add some records to file
102	dbf = dbflib.open(file, "r+b")
103	# Records can be added as a dictionary...
104	dbf.write_record(0, {'NAME': "Weatherwax", "INT":1, "FLOAT":3.1415926535})
105	# ... or as a sequence
106	dbf.write_record(1, ("Ogg", 2, -1000.1234))
107
108	def list_dbf(file):
109	# print the contents of a dbf file to stdout
110	dbf = dbflib.DBFFile(file)
111	print "%d records, %d fields" % (dbf.record_count(), dbf.field_count())
112	format = ""
113	for i in range(dbf.field_count()):
114	type, name, len, decc = dbf.field_info(i)
115	if type == 0:
116	format = format + " %%(%s)%ds" % (name, len)
117	elif type == 1:
118	format = format + " %%(%s)%dd" % (name, len)
119	elif type == 2:
120	format = format + " %%(%s)%dg" % (name, len)
121	print format
122	for i in range(dbf.record_count()):
123	print format % dbf.read_record(i)
124
125
126	make_dbf("testfile")
127	add_dbf_records("testfile")
128	list_dbf("testfile")
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision