GDAL의 pct2rgb.py는

Question

내가 (이 스크립트는 8 비트를 TIFF 이미지를 얻고 24-bit로의 비트 깊이를 변경) C#에 pct2rgb.py 스크립트를 재 작성하려고 단색 데이터 세트를 반환합니다. 출력이 HDD에 저장 될 때 작동했습니다. 스크립트를 조금 변경하고 싶습니다. 디스크에 저장하지 않고 즉시 Dataset을 반환해야합니다. 그것은 작동합니다 ... 거의 ... 문제는 반환 된 Dataset에 단색 데이터가 포함되어 있다는 것입니다 ... 그 이유는 모르겠습니다. 운전자가 GTiff에서 MEM (Driver gTiffDriver = Gdal.GetDriverByName("MEM");)으로 바뀌었기 때문일 수 있습니다. 하지만

다음은 내 수업의 ... 하드 디스크에 데이터를 저장하는 GTiff 드라이버를 사용하지 않는 방법을 모른다 :

using System; 
using System.Collections.Generic; 
using System.Drawing; 
using System.Drawing.Imaging; 
using System.IO; 
using System.Linq; 
using System.Windows.Media; 
using System.Windows.Media.Imaging; 
using OSGeo.GDAL; 
using OSGeo.OGR; 
using OSGeo.OSR; 
using Driver = OSGeo.GDAL.Driver; 
using Encoder = System.Drawing.Imaging.Encoder; 
using PixelFormat = System.Drawing.Imaging.PixelFormat; 

namespace RasterLoader 
{ 
    public class TiffConverter 
    { 
     private string _format = "GTiff"; 
     private string _srcFileName = null; 
     private int _outBands = 3; 
     private int _bandNumber = 1; 
     private string[] _args; 

     public TiffConverter(string[] args) 
     { 
      _args = args; 
     } 

     public Dataset Convert8To24Bit() 
     { 
      Gdal.AllRegister(); 

      string[] argv = Gdal.GeneralCmdLineProcessor(_args, 0); 

      int i = 1; 
      while (i < argv.Count()) 
      { 
       string arg = argv.ElementAt(i); 
       switch (arg) 
       { 
        case "-of": 
         i++; 
         _format = argv[i]; 
         break; 
        case "-b": 
         i++; 
         _bandNumber = int.Parse(argv[i]); 
         break; 
        case "-rgba": 
         _outBands = 4; 
         break; 
        default: 
         if (string.IsNullOrEmpty(_srcFileName)) 
         { 
          _srcFileName = argv[i]; 
         } 
         else 
         { 
          Usage(); 
         } 
         break; 
       } 
       i++; 
      } 

      string tmpFileName = _srcFileName + ".bak"; 

      // open source file 
      Dataset srcDS = Gdal.Open(_srcFileName, Access.GA_ReadOnly); 
      if (srcDS == null) 
      { 
       throw new Exception("Unable to open " + _srcFileName + "."); 
      } 

      Band srcBand = srcDS.GetRasterBand(_bandNumber); 

      // ensure we recognise the driver 
      Driver dstDriver = Gdal.GetDriverByName(_format); 
      if (dstDriver == null) 
      { 
       throw new Exception("\"" + _format + "\" not registered."); 
      } 

      // build color table 
      int[][] lookup = new int[4][]; 
      lookup[0] = Enumerable.Range(0, 256).ToArray(); 
      lookup[1] = Enumerable.Range(0, 256).ToArray(); 
      lookup[2] = Enumerable.Range(0, 256).ToArray(); 
      lookup[3] = new int[256]; 

      for (i = 0; i < 256; i++) 
      { 
       lookup[3][i] = 255; 
      } 

      ColorTable ct = srcBand.GetRasterColorTable(); 

      if (ct != null) 
      { 
       for (i = 0; i < Math.Min(256, ct.GetCount()); i++) 
       { 
        ColorEntry entry = ct.GetColorEntry(i); 
        for (int j = 0; j < 4; j++) 
        { 
         switch (j) 
         { 
          case 0: 
           lookup[j][i] = entry.c1; 
           break; 
          case 1: 
           lookup[j][i] = entry.c2; 
           break; 
          case 2: 
           lookup[j][i] = entry.c3; 
           break; 
          case 3: 
           lookup[j][i] = entry.c4; 
           break; 
         } 
        } 
       } 
      } 

      // create the working file 
      string tifFileName = string.Empty; 
      if (_format.Equals("GTiff", StringComparison.OrdinalIgnoreCase)) 
      { 
       tifFileName = tmpFileName; 
      } 
      else 
      { 
       tifFileName = Path.Combine(Directory.GetParent(tmpFileName).Name, "temp.gif"); 
      } 

//   Driver gTiffDriver = Gdal.GetDriverByName("GTiff"); 
      Driver gTiffDriver = Gdal.GetDriverByName("MEM"); 

      Dataset tifDS = gTiffDriver.Create(tifFileName, srcDS.RasterXSize, srcDS.RasterYSize, _outBands, DataType.GDT_Byte, 
               new string[] {}); 

      // we should copy projection information and so forth at this point 
      tifDS.SetProjection(srcDS.GetProjection()); 
      double[] geotransform = new double[6]; 
      srcDS.GetGeoTransform(geotransform); 
      tifDS.SetGeoTransform(geotransform); 

      if (srcDS.GetGCPCount() > 0) 
      { 
       tifDS.SetGCPs(srcDS.GetGCPs(), srcDS.GetGCPProjection()); 
      } 

      // do the processing one scanline at a time 
      for (int iY = 0; iY < srcDS.RasterYSize; iY++) 
      { 
       byte[] srcData = new byte[srcDS.RasterXSize*1]; 
       srcBand.ReadRaster(0, iY, srcDS.RasterXSize, 1, srcData, srcDS.RasterXSize, 1, 0, 0); 

       for (int iBand = 0; iBand < _outBands; iBand++) 
       { 
        int[] bandLookup = lookup[iBand]; 

        int[] dstData = new int[srcData.Count()]; 
        for (int index = 0; index < srcData.Count(); index++) 
        { 
         byte b = srcData[index]; 
         dstData[index] = bandLookup[b]; 
        } 

        tifDS.GetRasterBand(iBand + 1).WriteRaster(0, iY, srcDS.RasterXSize, 1, dstData, 
                   srcDS.RasterXSize, 1, 0, 0); 
       } 
      } 

      return tifDS; 
     } 

     private void Usage() 
     { 
      Console.WriteLine("Usage: pct2rgb.py [-of format] [-b <band>] [-rgba] source_file dest_file"); 
      throw new Exception("Bad arguments."); 
     } 
} 

Answer 1

좋아. 해결책을 찾았습니다. 내 문제는 모든 파일을 (심지어 좋은 형식으로) 변환하려고했습니다. 그런 다음 ct 변수 (ColorTable)는 null입니다.

솔루션 :

if (ct != null) 
{ 
// same code here 
} 
else 
{ 
    return srcDS; 
}