Skip to content

GeoSentinel
GeoSentinel
Commit dd90b14e authored by Mario Chirinos Colunga's avatar Mario Chirinos Colunga 💬
Browse files
Options

wkt raster

parent 525e9b2a
Hide whitespace changes
Inline Side-by-side
Showing with 37 additions and 36 deletions
geosentinel/rasterWkt.py
View file @ dd90b14e
...@@ -46,16 +46,17 @@ def getPolygonArray(image, wkt): ...@@ -46,16 +46,17 @@ def getPolygonArray(image, wkt):
data_source = driverShp.CreateDataSource("myShape.shp") data_source = driverShp.CreateDataSource("myShape.shp")
#Burn Polygon #Burn Polygon
srs = osr.SpatialReference() # srs = osr.SpatialReference()
srs.ImportFromEPSG(4326) # srs.ImportFromEPSG(4326)
layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon) # layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
geometry = ogr.CreateGeometryFromWkt(wkt) # geometry = ogr.CreateGeometryFromWkt(wkt)
feature = ogr.Feature(layer.GetLayerDefn()) # feature = ogr.Feature(layer.GetLayerDefn())
feature.SetStyleString("PEN(c:#FF0000,w:5px);") # feature.SetStyleString("PEN(c:#FF0000,w:5px);")
feature.SetGeometry(geometry) # feature.SetGeometry(geometry)
layer.CreateFeature(feature) # layer.CreateFeature(feature)
layer = data_source.getLayer()
gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255]) gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
polygonRaster.GetRasterBand(1).FlushCache() polygonRaster.GetRasterBand(1).FlushCache()
...@@ -82,43 +83,43 @@ def rasterWkt(wkt, inputfile, outputfile): ...@@ -82,43 +83,43 @@ def rasterWkt(wkt, inputfile, outputfile):
print("driver") print("driver")
driverTiff = gdal.GetDriverByName('GTiff') driverTiff = gdal.GetDriverByName('GTiff')
output = driverTiff.Create(outputfile, cols, rows, inputImage.RasterCount, gdal.GDT_Byte) 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 # #Set up polygon raster
polygonRaster.GetRasterBand(1).SetNoDataValue(-99) # polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols))) # polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
polygonRaster.GetRasterBand(1).FlushCache() # polygonRaster.GetRasterBand(1).FlushCache()
polygonRaster.SetGeoTransform(geotransform) # polygonRaster.SetGeoTransform(geotransform)
polygonRaster.SetGeoTransform(geotransform) # polygonRaster.SetGeoTransform(geotransform)
polygonRaster.SetProjection(sr.ExportToWkt()) # polygonRaster.SetProjection(sr.ExportToWkt())
print("Shapefile") # print("Shapefile")
# shapefile # # shapefile
driver = ogr.GetDriverByName("ESRI Shapefile") # driver = ogr.GetDriverByName("ESRI Shapefile")
data_source = driver.CreateDataSource("myShape.shp") # data_source = driver.CreateDataSource("myShape.shp")
print("Burn Polygon") # print("Burn Polygon")
#Burn Polygon # #Burn Polygon
srs = osr.SpatialReference() # srs = osr.SpatialReference()
srs.ImportFromEPSG(4326) # srs.ImportFromEPSG(4326)
layer = data_source.GetLayer() # layer = data_source.GetLayer()
# layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon) ## layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
# geometry = ogr.CreateGeometryFromWkt(wkt) ## geometry = ogr.CreateGeometryFromWkt(wkt)
# feature = ogr.Feature(layer.GetLayerDefn()) ## feature = ogr.Feature(layer.GetLayerDefn())
print("features") # print("features")
# feature.SetStyleString("PEN(c:#FF0000,w:5px);") ## feature.SetStyleString("PEN(c:#FF0000,w:5px);")
# feature.SetGeometry(geometry) ## feature.SetGeometry(geometry)
# layer.CreateFeature(feature) ## layer.CreateFeature(feature)
gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255]) # gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
polygonRaster.GetRasterBand(1).FlushCache() # polygonRaster.GetRasterBand(1).FlushCache()
# Draw output image ## Draw output image
np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows) # np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
np_polygon = getPolygonArray(inputImage, wkt) np_polygon = getPolygonArray(inputImage, wkt)
print("Setup") print("Setup")
for b in range(1, output.RasterCount+1): for b in range(1, output.RasterCount+1):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment