drawing polygon

969b925e · Mario Chirinos Colunga · 31889f26 · 969b925e
Commit 969b925e authored Aug 20, 2018 by Mario Chirinos Colunga 💬
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

rasterWkt.py geosentinel/rasterWkt.py +3 -1

No files found.
--- a/geosentinel/rasterWkt.py
+++ b/geosentinel/rasterWkt.py
@@ -49,7 +49,8 @@ def rasterWkt(wkt, inputfile, outputfile):
 	polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
 	polygonRaster.GetRasterBand(1).FlushCache()
 	polygonRaster.SetGeoTransform(geotransform)
-
+	polygonRaster.SetGeoTransform(geotransform)
+	polygonRaster.SetProjection(sr.ExportToWkt())



@@ -70,6 +71,7 @@ def rasterWkt(wkt, inputfile, outputfile):
 	layer.CreateFeature(feature)

 	gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
+	polygonRaster.GetRasterBand(1).FlushCache()

 	#Draw output image
 	np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)