wkt raster

geosentinel/rasterWkt.py

@@ -32,7 +32,7 @@ def getPolygonArray(image, wkt):
 	sr.ImportFromWkt(image.GetProjection())
 
 	driverTiff = gdal.GetDriverByName('GTiff')
-	polygonRaster = driverTiff.Create("polygonTmp.tiff", cols, rows, 1, gdal.GDT_Byte)
+	polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte)
 
 	#Set up polygon raster
 	polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
@@ -82,41 +82,41 @@ def rasterWkt(wkt, inputfile, outputfile):
 	print("driver")
 	driverTiff = gdal.GetDriverByName('GTiff')
 	output = driverTiff.Create(outputfile, cols, rows, inputImage.RasterCount, gdal.GDT_Byte)
-#	polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte)
+	polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte)
 
 
-#	#Set up polygon raster
-#	polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
-#	polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
-#	polygonRaster.GetRasterBand(1).FlushCache()
-#	polygonRaster.SetGeoTransform(geotransform)
-#	polygonRaster.SetGeoTransform(geotransform)
-#	polygonRaster.SetProjection(sr.ExportToWkt())
+	#Set up polygon raster
+	polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
+	polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
+	polygonRaster.GetRasterBand(1).FlushCache()
+	polygonRaster.SetGeoTransform(geotransform)
+	polygonRaster.SetGeoTransform(geotransform)
+	polygonRaster.SetProjection(sr.ExportToWkt())
 
 
 
-#	# shapefile
-#	driver = ogr.GetDriverByName("ESRI Shapefile")
-#	data_source = driver.CreateDataSource("myShape.shp")
+	# shapefile
+	driver = ogr.GetDriverByName("ESRI Shapefile")
+	data_source = driver.CreateDataSource("myShape.shp")
 
 
-#	#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())
+	#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())
 
-#	feature.SetStyleString("PEN(c:#FF0000,w:5px);")
-#	feature.SetGeometry(geometry)
-#	layer.CreateFeature(feature)
+	feature.SetStyleString("PEN(c:#FF0000,w:5px);")
+	feature.SetGeometry(geometry)
+	layer.CreateFeature(feature)
 
-#	gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
-#	polygonRaster.GetRasterBand(1).FlushCache()
+	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)
-	np_polygon = getPolygonArray(inputImage, wkt)
+	Draw output image
+	np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
+#	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)