wkt raster

geosentinel/rasterWkt.py

@@ -31,21 +31,11 @@ def getPolygonArray(image, wkt):
 	sr = osr.SpatialReference()
 	sr.ImportFromWkt(image.GetProjection())
 
-	driverTiff = gdal.GetDriverByName('GTiff')
-	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.SetProjection(sr.ExportToWkt())
-
 	# shapefile
 	driverShp = ogr.GetDriverByName("ESRI Shapefile")
 	data_source = driverShp.CreateDataSource("myShape.shp")
 
-	#Burn Polygon
+
 	srs = osr.SpatialReference()
 	srs.ImportFromEPSG(4326)
 	layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
@@ -56,6 +46,18 @@ def getPolygonArray(image, wkt):
 	feature.SetGeometry(geometry)
 	layer.CreateFeature(feature)
 
+
+	
+
+	#Set up polygon raster
+	driverTiff = gdal.GetDriverByName('GTiff')
+	polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte)
+	polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
+	polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
+	polygonRaster.GetRasterBand(1).FlushCache()
+	polygonRaster.SetGeoTransform(geotransform)
+	polygonRaster.SetProjection(sr.ExportToWkt())
+	#Burn Polygon
 	gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
 	polygonRaster.GetRasterBand(1).FlushCache()