wkt raster

geosentinel/rasterWkt.py

@@ -25,27 +25,40 @@ def createLayer(wkt, layerName="wkt"):
 	feature = None
 	return layer
 #-------------------------------------------------------------------------------
-def getPolygonArray(image, wkt):
+def getPolygonArray(image, wkt_geom):
 	rows, cols, geotransform = image.RasterYSize, image.RasterXSize, image.GetGeoTransform()
 	print(rows, cols, geotransform)
 	sr = osr.SpatialReference()
 	sr.ImportFromWkt(image.GetProjection())
 
-	# shapefile
-	driverShp = ogr.GetDriverByName("ESRI Shapefile")
-	data_source = driverShp.CreateDataSource("myShape.shp")
+#	# shapefile
+#	driverShp = ogr.GetDriverByName("ESRI Shapefile")
+#	data_source = driverShp.CreateDataSource("myShape.shp")
 
 
-	srs = osr.SpatialReference()
-	srs.ImportFromEPSG(4326)
-	layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
-	geometry = ogr.CreateGeometryFromWkt(wkt)
-	feature = ogr.Feature(layer.GetLayerDefn())
+#	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)
+#	
+#	data_source = None
+
+	rast_ogr_ds = ogr.GetDriverByName('Memory').CreateDataSource( 'wrk' )
+	rast_mem_lyr = rast_ogr_ds.CreateLayer( 'poly', srs=sr )
+
+
+	feat = ogr.Feature( rast_mem_lyr.GetLayerDefn() )
+	feat.SetGeometryDirectly( ogr.Geometry(wkt = wkt_geom) )
+
+	rast_mem_lyr.CreateFeature( feat )
 
+	# Run the algorithm.
+#	err = gdal.RasterizeLayer( target_ds, [3,2,1], rast_mem_lyr, burn_values = [200,220,240] )
 
 	#Set up polygon raster
 	driverTiff = gdal.GetDriverByName('GTiff')
@@ -53,18 +66,18 @@ def getPolygonArray(image, wkt):
 	polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
 	polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
 	polygonRaster.GetRasterBand(1).FlushCache()
-#	polygonRaster.SetGeoTransform(geotransform)
-#	polygonRaster.SetProjection(sr.ExportToWkt())
 	polygonRaster.SetProjection(image.GetProjectionRef())
 	polygonRaster.SetGeoTransform(image.GetGeoTransform()) 
+
+
 	#Burn Polygon
-	gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
+	gdal.RasterizeLayer(polygonRaster, [1], rast_mem_lyr, burn_values=[255])
 	polygonRaster.GetRasterBand(1).FlushCache()
 
 	np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
 
-	feature = None
-	data_source = None
+	rast_ogr_ds = None
+#	data_source = None
 	polygonRaster = None
 
 	return np_polygon