开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > GIS编程 > 查看源码
DlgSlope.cpp
资源名称：GISCode.rar　[点击查看]
上传用户：juying163
上传日期：2014-09-24
资源大小：5867k
文件大小：10k
源码类别：

GIS编程

开发平台：

Visual C++

DlgSlope.cpp：源码内容
  1. // DlgSlope.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "3D0214.h"
  6. #include "DlgSlope.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CDlgSlope dialog
  18. CDlgSlope::CDlgSlope(CWnd* pParent /*=NULL*/)
  19. : CDialog(CDlgSlope::IDD, pParent)
  20. {
  21. //{{AFX_DATA_INIT(CDlgSlope)
  22. m_edit_inputSurf_slope = _T("");
  23. m_edit_OutputCellsizeSlope = 0.0;
  24. m_edit_outputRaster_slope = _T("");
  25. m_degree=-1;
  26. //}}AFX_DATA_INIT
  27. }
  30. void CDlgSlope::DoDataExchange(CDataExchange* pDX)
  31. {
  32. CDialog::DoDataExchange(pDX);
  33. //{{AFX_DATA_MAP(CDlgSlope)
  34. DDX_Text(pDX, IDC_EDIT_INPUT_SURF_SLOPE, m_edit_inputSurf_slope);
  35. DDX_Text(pDX, IDC_EDIT_OUTPUT_CELLSIZE_SLOPE, m_edit_OutputCellsizeSlope);
  36. DDX_Text(pDX, IDC_EDIT_OUTPUT_RASTER_SLOPE, m_edit_outputRaster_slope);
  37. DDX_Text(pDX, IDC_RADIO_DEGREE_SLOPE, m_degree);
  39. //}}AFX_DATA_MAP
  40. }
  43. BEGIN_MESSAGE_MAP(CDlgSlope, CDialog)
  44. //{{AFX_MSG_MAP(CDlgSlope)
  45. ON_BN_CLICKED(IDC_BUTTON_INPUTSURF_SLOPE, OnButtonInputsurfSlope)
  46. ON_BN_CLICKED(IDC_BUTTON_OUTPUTRASTER_SLOPE, OnButtonOutputrasterSlope)
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  50. /////////////////////////////////////////////////////////////////////////////
  51. // CDlgSlope message handlers
  53. void CDlgSlope::OnButtonInputsurfSlope() 
  54. {
  55. // TODO: Add your control notification handler code here
  56. IGxDialogPtr ipGxDialog(CLSID_GxDialog);
  57.     IGxObjectFilterPtr ipGxObjectFilter( CLSID_GxFilterSurfaceDatasets);
  58.     HRESULT hr=ipGxDialog->putref_ObjectFilter(ipGxObjectFilter);
  59.     if(FAILED(hr)) return;
  60.     hr = ipGxDialog->put_Title(CComBSTR("选择一个DEM表面数据"));
  61.     if(FAILED(hr)) return;
  63. IEnumGxObjectPtr ipEnumGxObject;
  65. VARIANT_BOOL bResult;
  67. hr=ipGxDialog->DoModalOpen(0,&ipEnumGxObject,&bResult);
  69. if(FAILED(hr)) return;
  70.   if(bResult==VARIANT_FALSE) return;
  71. hr=ipEnumGxObject->Reset();
  72. if(FAILED(hr)) return;
  73. IGxObjectPtr ipGxObject;
  74. ipGxDialog->get_FinalLocation( &ipGxObject);
  75.  
  76. while(!ipEnumGxObject->Next(&ipGxObject))
  77. {
  78.  
  79. IGxDatasetPtr ipGxDataset(ipGxObject);
  80. if(ipGxDataset!=0)
  81. {
  82. IDatasetPtr ipDataset;
  83. hr=ipGxDataset->get_Dataset(&ipDataset);
  84. if(FAILED(hr)) return;
  85. IRasterDatasetPtr ipRasterDataset(ipDataset);
  86. if(ipRasterDataset!=NULL)
  87. {
  88. m_ipR=ipRasterDataset;
  89. CComBSTR bstrName;
  90. hr=ipRasterDataset->get_CompleteName(&bstrName);
  91. if(FAILED(hr)) return;
  92. m_edit_inputSurf_slope=bstrName;
  93. IRasterPtr ipRas;
  94. //获得输人Raster表面的网格大小,
  95. //作为输出Raster网格大小的默认值
  96. hr=ipRasterDataset->CreateDefaultRaster(&ipRas);
  97. IRasterAnalysisPropsPtr ipRA(ipRas);
  98. double defaultcellsize;
  99. hr=ipRA->get_PixelHeight( &defaultcellsize);
  100. m_edit_OutputCellsizeSlope=defaultcellsize;
  101. UpdateData(FALSE);
  102. }
  104. ITinPtr ipTin(ipDataset);
  106. if(ipTin!=NULL)
  107. {
  108.   
  109. m_ipT=ipTin ;
  110. CComBSTR bstrName;  
  111. hr= ipDataset->get_Name(&bstrName);  
  112. if(FAILED(hr)) return;  
  113. m_edit_inputSurf_slope=CString(bstrName);    
  114. IGeoDatasetPtr plnTin(m_ipT);    
  115. //通过获得Tin的范围设置默认。ellsize大小    
  116. IEnvelopePtr pExtent;    
  117. hr=plnTin->get_Extent(&pExtent);    
  118. double width,height;  
  119. hr=pExtent->get_Width(&width);    
  120. hr=pExtent->get_Height(&height);    
  121. double defaultcellsize;    
  122. if ( width>height)        
  123. defaultcellsize=height/250;      
  124. else        
  125. defaultcellsize=width/250;    
  126. m_edit_OutputCellsizeSlope=defaultcellsize;    
  127. UpdateData(FALSE);
  128. }
  129. UpdateData(FALSE);
  131. }
  132. }
  133. }
  135. void CDlgSlope::OnButtonOutputrasterSlope() //输出栅格数据路径按钮按下消息响应函数
  136. {
  137. // TODO: Add your control notification handler code here
  138.  IGxDialogPtr ipGxDialog(CLSID_GxDialog);
  139.     //Set the Filter to FeatureDatasets and FeatureClasses
  140.     IGxObjectFilterPtr ipGxObjectFilter(CLSID_GxFilterRasterDatasets);
  141.     HRESULT hr = ipGxDialog->putref_ObjectFilter(ipGxObjectFilter);
  142.     if(FAILED(hr))return;
  143.     hr = ipGxDialog->put_Title(CComBSTR("保存坡度图"));
  144.     if(FAILED(hr)) return;
  145. VARIANT_BOOL bResult;
  146. OLE_HANDLE hwnd=(OLE_HANDLE)this->GetSafeHwnd();
  147. hr=ipGxDialog->DoModalSave(hwnd,&bResult);
  148. if(FAILED(hr))return;
  149. IGxObjectPtr ipGxObject;
  150. ipGxDialog->get_FinalLocation(&ipGxObject);
  151. CComBSTR path;
  152. ipGxObject->get_FullName(&path);
  153. m_OutputPath=CString(path);
  154. CComBSTR name;
  155. ipGxDialog->get_Name(&name);
  156. m_OutputRasterName_Slop=CString(name);
  157. UpdateData(TRUE);
  158. m_edit_outputRaster_slope=m_OutputPath+"\"+m_OutputRasterName_Slop;
  159. UpdateData(FALSE);
  160. }
  162. void CDlgSlope::OnOK() 
  163. {
  164. // TODO: Add extra validation here
  165. ::OleInitialize(NULL);//组件操作初始化 
  166. if(m_ipR!=NULL)
  167. {
  168. UpdateData(TRUE);
  169. CreateSlope(m_ipR);
  170. }
  171. if(m_ipT!=NULL)
  172. {
  173. UpdateData(TRUE);
  174. CreateSlope(m_ipT);
  175. }
  176. CDialog::OnOK();
  177. }
  179. void CDlgSlope::CreateSlope(ITin* ipT) //为TIN创建坡度过程函数。
  180. { 
  181.  ::OleInitialize(NULL);//组件操作初始化 
  183.   HRESULT hr;
  184.     IGeoDatasetPtr pInTin(ipT);
  185. IEnvelopePtr pExtent;
  186. hr=pInTin->get_Extent(&pExtent);
  187.     IPointPtr pOrigin;
  188. hr=pExtent->get_LowerLeft(&pOrigin);
  189. double x,y;
  190. hr=pOrigin->get_X(&x);
  191. hr=pOrigin->get_Y(&y); 
  192. hr=pOrigin->put_X(x-m_edit_OutputCellsizeSlope*0.5);
  193. hr=pOrigin->put_Y(y-m_edit_OutputCellsizeSlope*0.5);
  194. long nCol,nRow;
  195. double width,height;
  196. hr=pExtent->get_Width(&width);
  197. hr=pExtent->get_Height(&height);
  198. nCol=int(width/m_edit_OutputCellsizeSlope)+1;
  199. nRow=int(height/m_edit_OutputCellsizeSlope)+1;
  200. ISpatialReferencePtr pSR;
  201. hr=pInTin->get_SpatialReference(&pSR);
  202. IRasterDatasetPtr pRDS;
  203. IWorkspaceFactoryPtr rWksFac(CLSID_RasterWorkspaceFactory);
  204. IWorkspacePtr wks;
  205. hr = rWksFac->OpenFromFile(CComBSTR(m_OutputPath),0,&wks);
  206. IRasterWorkspace2Ptr rWks(wks);
  207. long numbands=1;
  208.   rWks->CreateRasterDataset((CComBSTR)m_OutputRasterName_Slop,
  209.                                CComBSTR("GRID"),
  210.                                pOrigin,
  211.                                nCol,
  212.                                nRow,
  213.                                m_edit_OutputCellsizeSlope,
  214.                                m_edit_OutputCellsizeSlope,
  215.                                numbands,
  216.                                PT_FLOAT,
  217.                                pSR,
  218.                                VARIANT_TRUE,     
  219.                                &pRDS);
  221. if(FAILED(hr))
  222. {
  223.       AfxMessageBox ( " Raster文件生成失败!");
  224.       return;
  225. }
  226.  AfxMessageBox ( " 1文件生成失败!");
  227. //得到RawPixels
  228. IRasterBandCollectionPtr pBandCollection(pRDS);
  229. long cont;
  230.     CString str;
  232.  hr=pBandCollection->get_Count(&cont);
  233.         str.Format("(%fn",(double) cont);
  234. AfxMessageBox(str); 
  236. IRasterBandPtr pRasterBand;
  237.   AfxMessageBox ( " 2r文件生成失败!");
  238. hr=pBandCollection->Item(0,&pRasterBand);
  239.  AfxMessageBox ( " 3文件生成失败!");
  240.  
  241. IRawPixelsPtr pRawPixels(pRasterBand);
  242.  
  243. IPntPtr pBlockSize(CLSID_DblPnt);
  244. hr=pBlockSize->put_X(nCol);
  245. hr=pBlockSize->put_Y(nRow);
  246. IPixelBlockPtr pPixelBlock;
  247. hr=pRawPixels->CreatePixelBlock(pBlockSize,&pPixelBlock);
  248. VARIANT val;
  249. val.vt = VT_SAFEARRAY ;//要指定类型
  250. hr=pPixelBlock->get_SafeArray(0,&val);
  251. ITinSurfacePtr pTinSurf(ipT); 
  252. IRasterPropsPtr pRasterProps(pRawPixels);
  253. VARIANT nodatafloat;
  254. nodatafloat.vt = VT_R4 ;//要指定类型
  255. // QueryPixelBlock使用左上角点的中心为原点
  256. //下面为计算中心坐标
  257.  
  258. hr=pOrigin->get_X(&x);
  259. hr=pOrigin->get_Y(&y);
  260. hr=pOrigin->put_X(x+m_edit_OutputCellsizeSlope*0.5);
  261. hr = pOrigin->put_Y(y+m_edit_OutputCellsizeSlope*nRow-m_edit_OutputCellsizeSlope*0.5);
  262. hr=pRasterProps->get_NoDataValue(&nodatafloat);
  263. hr = pOrigin->get_X(&x);
  264. hr=pOrigin->get_Y(&y);
  265. esriRasterizationType Type;
  266. UpdateData(TRUE);
  268. m_degree=0;
  269. if ( m_degree==0)
  270. Type=esriDegreeSlopeAsRaster;
  271. if(m_degree==1)
  272. Type=esriPercentageSlopeAsRaster;
  273. if(m_degree ==-1)
  274. {
  275. AfxMessageBox("请选择输出单位!");
  276. return;
  277. }
  278. // AfxMessageBox ( " 2....!");
  280. hr=pTinSurf->QueryPixelBlock(x,y,
  281.                                   m_edit_OutputCellsizeSlope,
  282.                                   m_edit_OutputCellsizeSlope,
  283.                                   Type,
  284.                                   nodatafloat,
  285.                                   val);
  287. IPntPtr pOffset(CLSID_DblPnt);
  288. hr=pOffset->put_X(0);
  289. hr = pOffset->put_Y(0);
  290. hr=pRawPixels->Write(pOffset,pPixelBlock);
  291.     //注意:只有指针pRDS被释放,Raster才会从内存写人硬盘。
  292. }
  294. void CDlgSlope::CreateSlope(IRasterDataset* ipR)  //为Raster创建坡度图过程函数。
  295. {
  296.     HRESULT hr;
  297.    ISurfaceOpPtr pSurfaceOp(CLSID_RasterSurfaceOp);
  298.    IGeoDatasetPtr pInputDataset(ipR);
  299.    //声明输出对象
  300.    IGeoDatasetPtr pOutputRaster;
  301.    //设置输出工作区
  302.    IRasterAnalysisEnvironmentPtr pEnv(pSurfaceOp);
  303.    hr=pEnv->put_VerifyType(esriRasterVerifyOn);
  304.    VARIANT vCellsize;
  305.    vCellsize.vt=VT_R8;
  306.    vCellsize.dblVal=m_edit_OutputCellsizeSlope;
  307.    hr=pEnv->SetCellSize(esriRasterEnvValue,&vCellsize);
  308.    if(FAILED(hr)) return;
  309.    IWorkspacePtr pWo;
  310.    IWorkspaceFactoryPtr pWSF(CLSID_RasterWorkspaceFactory);
  311.    hr=pWSF->OpenFromFile((CComBSTR)m_OutputPath,NULL,&pWo);
  312.    if(FAILED(hr))
  313.    {
  314.    AfxMessageBox("失败!");      
  315.    return;
  316.    }
  317.    //Slope开始
  318.    VARIANT vZfactor;
  319.    vZfactor.vt= VT_R8;
  320.    vZfactor.dblVal=1;
  321.    enum esriGeoAnalysisSlopeEnum slopeType;
  322. UpdateData(TRUE);
  323.    if(m_degree==0) slopeType=esriGeoAnalysisSlopeDegrees;
  324.    if(m_degree==1) slopeType=esriGeoAnalysisSlopePercentrise;
  325.    if(m_degree==-1)
  326.    {
  327.     AfxMessageBox("请选择输出单位!");
  328.         return;
  329.    }
  330.    hr=pSurfaceOp->Slope(pInputDataset,slopeType,&vZfactor,&pOutputRaster);
  331.    if(FAILED(hr))
  332.    {
  333.    AfxMessageBox ("坡度图绘制失败!");
  334.    return;
  335.    }
  336.    IRasterBandCollectionPtr pBands(pOutputRaster);
  337.    IDatasetPtr pRD;
  338.    hr= pBands->SaveAs(CComBSTR(m_OutputRasterName_Slop),
  339.                                    pWo,
  340.    CComBSTR("GRID"),
  341.                                    &pRD);
  342. }