wkt raster

geosentinel/rasterWkt.py

@@ -27,7 +27,7 @@ def createLayer(wkt, layerName="wkt"):
 #-------------------------------------------------------------------------------
 def getPolygonArray(image, wkt):
 	rows, cols, geotransform = image.RasterYSize, image.RasterXSize, image.GetGeoTransform()
-	print(rows, cols, geotransform)
+
 	sr = osr.SpatialReference()
 	sr.ImportFromWkt(image.GetProjection())
 
@@ -52,7 +52,7 @@ def getPolygonArray(image, wkt):
 	geometry = ogr.CreateGeometryFromWkt(wkt)
 	feature = ogr.Feature(layer.GetLayerDefn())
 
-#	feature.SetStyleString("PEN(c:#FF0000,w:5px);")
+	feature.SetStyleString("PEN(c:#FF0000,w:5px);")
 	feature.SetGeometry(geometry)
 	layer.CreateFeature(feature)
 
@@ -61,7 +61,6 @@ def getPolygonArray(image, wkt):
 
 	np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
 
-	feature = None
 	data_source = None
 	polygonRaster = None
 
@@ -105,7 +104,6 @@ def rasterWkt(wkt, inputfile, outputfile):
 	#Burn Polygon
 	srs = osr.SpatialReference()
 	srs.ImportFromEPSG(4326)
-
 	layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
 	geometry = ogr.CreateGeometryFromWkt(wkt)
 	feature = ogr.Feature(layer.GetLayerDefn())
@@ -120,7 +118,7 @@ def rasterWkt(wkt, inputfile, outputfile):
 
 #	Draw output image
 	np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
-#	np_polygon = getPolygonArray(inputImage, wkt)
+	np_polygon = getPolygonArray(inputImage, wkt)
 	print("Setup")
 	for b in range(1, output.RasterCount+1):
 		np_input = np_polygon & inputImage.GetRasterBand(b).ReadAsArray(0,0,cols,rows)
@@ -133,9 +131,9 @@ def rasterWkt(wkt, inputfile, outputfile):
 
 	print("end")
 	#Close files
-#	feature = None
-#	data_source = None
-#	polygonRaster = None
+	feature = None
+	data_source = None
+	polygonRaster = None
 	output = None
 	inputImage = None