76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
|
#!/usr/bin/python
|
||
|
|
import shapefile
|
||
|
|
import os, sys
|
||
|
|
from lxml import etree
|
||
|
|
|
||
|
|
from optparse import OptionParser
|
||
|
|
parser = OptionParser()
|
||
|
|
parser.add_option("-f", "--file", dest="inputfilename",
|
||
|
|
help="Shapefile filename to process", metavar="FILE")
|
||
|
|
parser.add_option("-o", "--output", dest="outputfilename",
|
||
|
|
help="Output filename", metavar="FILE")
|
||
|
|
parser.add_option("-p", "--pretty",
|
||
|
|
action="store_true", dest="pretty", default=False,
|
||
|
|
help="pretty-print output xml file")
|
||
|
|
|
||
|
|
|
||
|
|
(options, args) = parser.parse_args()
|
||
|
|
if not options.inputfilename:
|
||
|
|
print "You must provide an input filename, -h or --help for help"
|
||
|
|
exit(1)
|
||
|
|
if not options.inputfilename:
|
||
|
|
print "You must provide an output filename, -h or --help for help"
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
out=open(options.outputfilename,'w')
|
||
|
|
sf = shapefile.Reader(options.inputfilename)
|
||
|
|
|
||
|
|
shapeRecs = sf.shapeRecords()
|
||
|
|
print "shp loaded"
|
||
|
|
|
||
|
|
out.write("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")
|
||
|
|
out.write("<osm generator=\"srtmshp2osm\" version=\"0.6\">")
|
||
|
|
|
||
|
|
index= 3000000000
|
||
|
|
l=len(shapeRecs)
|
||
|
|
for shapeRec in shapeRecs:
|
||
|
|
height=int(shapeRec.record[0]) # there only one field in my shapefile, 'HEIGHT'
|
||
|
|
points=shapeRec.shape.points # [[x1,y1],[x2,y2]...]
|
||
|
|
|
||
|
|
# nodes:
|
||
|
|
firstindex = index
|
||
|
|
for p in points:
|
||
|
|
node = etree.Element("node",
|
||
|
|
id=str(index),
|
||
|
|
lat=str(p[1]),
|
||
|
|
lon=str(p[0]),
|
||
|
|
version="1")
|
||
|
|
out.write(etree.tostring(node, pretty_print=options.pretty))
|
||
|
|
index +=1
|
||
|
|
lastindex=index-1
|
||
|
|
|
||
|
|
# way:
|
||
|
|
way = etree.Element("way", id=str(index),version="1")
|
||
|
|
index +=1
|
||
|
|
for i in range(firstindex,lastindex):
|
||
|
|
way.append(etree.Element("nd", ref=str(i)))
|
||
|
|
|
||
|
|
# tags:
|
||
|
|
way.append(etree.Element("tag",k="elevation", v=str(height)))
|
||
|
|
way.append(etree.Element("tag",k="contour", v="elevation"))
|
||
|
|
if height % 100 ==0:
|
||
|
|
way.append(etree.Element("tag",k="name", v=str(height)))
|
||
|
|
|
||
|
|
out.write(etree.tostring(way, pretty_print=options.pretty))
|
||
|
|
|
||
|
|
#progress:
|
||
|
|
l-=1
|
||
|
|
sys.stdout.write("\r")
|
||
|
|
sys.stdout.write("% 12d" % l)
|
||
|
|
sys.stdout.flush()
|
||
|
|
|
||
|
|
|
||
|
|
print ""
|
||
|
|
out.write("</osm>")
|
||
|
|
out.close()
|