rgb l2

geosentinel/rasterWkt.py

@@ -35,26 +35,19 @@ def rasterWkt(wkt, inputfile, outputfile):
 	intput = gdal.Open(inputfile)
 	rows, cols, geotransform = intput.RasterYSize, intput.RasterXSize, intput.GetGeoTransform()
 	# Read the input bands as numpy arrays.
-	np_intput_1 = intput.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
-	np_intput_2 = intput.GetRasterBand(2).ReadAsArray(0,0,cols,rows)
-	np_intput_3 = intput.GetRasterBand(3).ReadAsArray(0,0,cols,rows)
+
 
 	driverTiff = gdal.GetDriverByName('GTiff')
 	output = driverTiff.Create(outputfile, cols, rows, 3, gdal.GDT_Byte)
 	print ("BANDS")
 	print (output.RasterCount)
 
-	output.GetRasterBand(1).SetNoDataValue(-99)
-	output.GetRasterBand(1).WriteArray(np_intput_1)
-	output.GetRasterBand(1).FlushCache()
-
-	output.GetRasterBand(2).SetNoDataValue(-99)
-	output.GetRasterBand(2).WriteArray(np_intput_2)
-	output.GetRasterBand(2).FlushCache()
+	for b in range(output.RasterCount):
+		np_input = intput.GetRasterBand(b).ReadAsArray(0,0,cols,rows)
+		output.GetRasterBand(b).SetNoDataValue(-99)
+		output.GetRasterBand(b).WriteArray(np_input)
+		output.GetRasterBand(b).FlushCache()
 
-	output.GetRasterBand(3).SetNoDataValue(-99)
-	output.GetRasterBand(3).WriteArray(np_intput_3)
-	output.GetRasterBand(3).FlushCache()
 
 	output.SetGeoTransform(geotransform)