PIE SDK过滤控制

  1 方法（一）
  2     //前置过滤本示例以定标为例
  3     //风云4A数据为例
  4     IRasterLayer rasterLayer = mapControlMain.ActiveView.CurrentLayer as IRasterLayer;
  5     IRasterRender rasterRender = RenderFactory.ImportFromFile(@"C:\Users\zhangyiwei\Desktop\TestData\ICV-BD.xml");
  6 
  7     //实例化定标函数
  8     CaliCommon common = new CaliCommon();
  9     string hdfpath = (rasterLayer as ILayer).DataSourcePath;
 10     common.Initialize(hdfpath, "NOMChannel13");
 11 
 12     //添加前置过滤
 13     PIE.Carto.ICustomerPreRasterFilter preFilter = new PIE.Carto.CustomerPreRasterFilter();
 14     preFilter.SetPreFilterFun(common.CaliFunCallBack);
 15 
 16     IRasterClassifyColorRampRender classRender = rasterRender as IRasterClassifyColorRampRender;
 17     (classRender as IRasterFilterProps).AddPreRasterFilter(preFilter as IPreRasterFilter);
 18 
 19     //地图刷新
 20     rasterLayer.Render = rasterRender;
 21 mapControlMain.ActiveView.PartialRefresh(ViewDrawPhaseType.ViewAll);
 22 方法（二）
 23 /// <summary>
 24 /// 定标方法接口
 25 /// </summary>
 26 public interface ICaliFunction
 27 {
 28 /// <summary>
 29 /// 定标回调函数
 30 /// </summary>
 31 PreFilterFunCallback CaliFunCallBack
 32 {
 33     get;
 34 }
 35 
 36 /// <summary>
 37 /// 数据定标
 38 /// </summary>
 39 /// <param name="dataNeedCali">原始数据</param>
 40 /// <param name="width">宽度</param>
 41 /// <param name="height">高度</param>
 42 /// <param name="bandCount">波段</param>
 43 /// <returns>定标后数据</returns>
 44 float[] Cali<T>(T[] dataNeedCali, int width, int height, int bandCount);
 45 
 46 /// <summary>
 47 /// 定标算法函数
 48 /// </summary>
 49 unsafe bool CaliFilter(IntPtr valuesA, IntPtr valuesB);
 50 }
 51 
 52 /// <summary>
 53 /// 风云4A数据的定标方法
 54 /// </summary>
 55 public class CaliCommon : ICaliFunction
 56 {
 57 /// <summary>
 58 /// 定标数据
 59 /// </summary>
 60 private float[] m_DBDatas = null;
 61 
 62 /// <summary>
 63 /// 定标数据长度
 64 /// </summary>
 65 private int m_Count = 0;
 66 
 67 /// <summary>
 68 /// 定标回调函数
 69 /// </summary>
 70 private PreFilterFunCallback m_CaliFunCallBack = null;
 71 
 72 /// <summary>
 73 /// 构造函数
 74 /// </summary>
 75 public CaliCommon()
 76 {
 77 }
 78 
 79 /// <summary>
 80 /// 定标回调函数
 81 /// </summary>
 82 public PreFilterFunCallback CaliFunCallBack
 83 {
 84     get
 85     {
 86         return m_CaliFunCallBack;
 87     }
 88 }
 89 
 90 /// <summary>
 91 /// 初始化定标方法
 92 /// </summary>
 93 /// <param name="strFilePath_HDF">hdf文件路径</param>
 94 /// <param name="chanelName">通道名称</param>
 95 /// <returns></returns>
 96 public bool Initialize(string strFilePath_HDF, string chanelName)
 97 {
 98     string calName = chanelName.Replace("NOM", "CAL");
 99     string calFilePath = System.IO.Path.GetDirectoryName(strFilePath_HDF) + "\\" + calName+".tiff";
100     IRasterDataset rasterDataset = PIE.DataSource.DatasetFactory.OpenRasterDataset(calFilePath, OpenMode.ReadOnly);
101 
102     int height = rasterDataset.GetRasterYSize();
103 
104     m_DBDatas = new float[height];
105     int[] bandMap = { 1 };
106     rasterDataset.Read(0, 0, 1, height, m_DBDatas, 1, height, PixelDataType.Float32, 1, bandMap);
107 
108     (rasterDataset as IDisposable).Dispose();
109 
110     m_Count = m_DBDatas.Length;
111     m_CaliFunCallBack = new PreFilterFunCallback(CaliFilter);
112     GC.KeepAlive(m_CaliFunCallBack);
113     return true;
114 }
115 
116 /// <summary>
117 /// 数据定标
118 /// </summary>
119 /// <param name="dataNeedCali">原始数据</param>
120 /// <param name="width">宽度</param>
121 /// <param name="height">高度</param>
122 /// <param name="bandCount">波段</param>
123 /// <returns>定标后数据</returns>
124 public float[] Cali<T>(T[] dataNeedCali, int width, int height, int bandCount)
125 {
126     if (dataNeedCali == null || dataNeedCali.Length < 1)
127     {
128         return null;
129     }
130 
131     float[] dataCali = new float[width * height * bandCount];
132     for (int j = 0; j < height; j++)
133     {
134         for (int i = 0; i < width; i++)
135         {
136             for (int m = 0; m < bandCount; m++)
137             {
138                 int nIndex = (j * width + i) * bandCount + m;
139                 int value = Convert.ToInt32(dataNeedCali[nIndex]);
140                 if (value < 0 || value >= m_Count)
141                 {
142                     dataCali[nIndex] = 0;
143                 }
144                 else
145                 {
146                     dataCali[nIndex] = m_DBDatas[value];
147                 }
148             }
149         }
150     }
151     return dataCali;
152 }
153 
154 /// <summary>
155 /// 定标算法
156 /// </summary>
157 /// <param name="valuesA">定标前</param>
158 /// <param name="valuesB">定标后</param>
159 /// <returns></returns>
160 public unsafe bool CaliFilter(IntPtr valuesA, IntPtr valuesB)
161 {
162     if (m_DBDatas == null)
163     {
164         return false;
165     }
166     IPixelBuffer pixelBuffer = PIE.DataSource.DatasetFactory.ConstructCLRPixelBuffer(valuesA.ToPointer());
167     int width = pixelBuffer.Width;
168     int height = pixelBuffer.Height;
169     int bandCount = pixelBuffer.BandMap.Length;
170 
171     float[] valuesData = new float[width * height * bandCount];
172 
173     short* byteValues = (short*)(pixelBuffer.GetData_Ref().ToPointer());
174     for (int j = 0; j < height; j++)
175     {
176         for (int i = 0; i < width; i++)
177         {
178             for (int m = 0; m < bandCount; m++)
179             {
180                 int nIndex = (j * width + i) * bandCount + m;
181                 int value = *(byteValues + nIndex);
182                 if (value < 0 || value >= m_Count)
183                 {
184                     valuesData[nIndex] = 0;
185                 }
186                 else
187                 {
188                     valuesData[nIndex] = m_DBDatas[value];
189                 }
190             }
191         }
192     }
193     pixelBuffer.SetData(PixelDataType.Float32, valuesData);
194     return true;
195 }
196 }