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

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

GIS编程

开发平台：

Visual C++

DlgViewshed.cpp：源码内容
  1. // DlgViewshed.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "3D0214.h"
  6. #include "DlgViewshed.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CDlgViewshed dialog
  18. CDlgViewshed::CDlgViewshed(CWnd* pParent /*=NULL*/)
  19. : CDialog(CDlgViewshed::IDD, pParent)
  20. {
  21. //{{AFX_DATA_INIT(CDlgViewshed)
  22. m_edit_viewshedSurfaceinput = _T("");
  23. m_edit_viewshedRasteroutput = _T("");
  24. m_edit_viewshedCellsize = 0.0;
  25. m_edit_viewshedobserverpoints = _T("");
  26. //}}AFX_DATA_INIT
  27. }
  30. void CDlgViewshed::DoDataExchange(CDataExchange* pDX)
  31. {
  32. CDialog::DoDataExchange(pDX);
  33. //{{AFX_DATA_MAP(CDlgViewshed)
  34. DDX_Text(pDX, IDC_EDIT_VIEWSHED_SURFACEINPUT, m_edit_viewshedSurfaceinput);
  35. DDX_Text(pDX, IDC_EDIT_VIEWSHED_RASTEROUTPUT, m_edit_viewshedRasteroutput);
  36. DDX_Text(pDX, IDC_EDIT_VIEWSHED_CELLSIZE, m_edit_viewshedCellsize);
  37. DDX_Text(pDX, IDC_EDIT_VIEWSHED_OBSERVERPOINTS, m_edit_viewshedobserverpoints);
  38. //}}AFX_DATA_MAP
  39. }
  42. BEGIN_MESSAGE_MAP(CDlgViewshed, CDialog)
  43. //{{AFX_MSG_MAP(CDlgViewshed)
  44. ON_BN_CLICKED(IDC_BUTTON_VIEWSHED_OBSERVERPOINTS, OnButtonViewshedObserverpoints)
  45. ON_BN_CLICKED(IDC_BUTTON_VIEWSHED_RASTEROUTPUT, OnButtonViewshedRasteroutput)
  46. ON_BN_CLICKED(IDC_BUTTON_VIEWSHED_SURFACEINPUT, OnButtonViewshedSurfaceinput)
  47. //}}AFX_MSG_MAP
  48. END_MESSAGE_MAP()
  50. /////////////////////////////////////////////////////////////////////////////
  51. // CDlgViewshed message handlers
  53. void CDlgViewshed::OnButtonViewshedObserverpoints() 
  54. {
  55. // TODO: Add your control notification handler code here
  56. //打开对话框         
  57. IGxDialogPtr ipGxDialog(CLSID_GxDialog);         
  58. IGxObjectFilterPtr ipGxObjectFiiter( CLSID_GxFilterPointFeatureClasses);
  59.     //注意:这里使用了GxFilterPointFeatureClasses ,
  60.     //可以过滤点特征集，满足功能只用点内插的需要          
  61. HRESULT hr=ipGxDialog->putref_ObjectFilter(ipGxObjectFiiter);            
  62. if(FAILED(hr)) return;
  63.     hr=ipGxDialog->put_Title(CComBSTR("选择一个点数据集"));
  64.     if(FAILED(hr))return;
  65. IEnumGxObjectPtr ipEnumGxObject;
  66. VARIANT_BOOL bResult;
  67. hr=ipGxDialog->DoModalOpen(0,&ipEnumGxObject,&bResult);
  68. if(FAILED(hr)) return;
  69. if(bResult==VARIANT_FALSE)return;
  70. hr=ipEnumGxObject->Reset();
  71. if(FAILED(hr)) return;
  72. IGxObjectPtr ipGxObject;
  73. while(!ipEnumGxObject->Next(&ipGxObject))
  74. {
  75. IGxDatasetPtr ipGxDataset(ipGxObject);
  76. if(ipGxDataset!=0)
  77. {   
  78. IDatasetPtr ipDataset;    
  79. hr=ipGxDataset->get_Dataset(&ipDataset);    
  80. if(FAILED(hr))return;    
  81. IFeatureClassPtr ipFeatureClass(ipDataset);    
  82. IFeatureLayerPtr ipFeatureLayer( CLSID_FeatureLayer);    
  83. hr=ipFeatureLayer->putref_FeatureClass(ipFeatureClass);    
  84. if(FAILED(hr)) return;
  85. m_ipfeat=ipFeatureClass;
  86. CComBSTR bstrName; //*} book获得输入点层的全名
  87. ipGxObject->get_FullName(&bstrName);
  88. CString str( bstrName);
  89. m_edit_viewshedobserverpoints=str;
  90. UpdateData(FALSE);
  91. }
  92. }
  95. }
  97. void CDlgViewshed::OnButtonViewshedRasteroutput() 
  98. {
  99. // TODO: Add your control notification handler code here
  100. IGxDialogPtr ipGxDialog(CLSID_GxDialog);
  101.     IGxObjectFilterPtr ipGxObjectFilter(CLSID_GxFilterRasterDatasets);
  102. HRESULT hr=ipGxDialog->putref_ObjectFilter(ipGxObjectFilter);
  103. if(FAILED(hr))return;
  104. hr=ipGxDialog->put_Title( CComBSTR("保存"));
  105. if(FAILED(hr)) return;
  106.      VARIANT_BOOL bResult;
  107. OLE_HANDLE hwnd=(OLE_HANDLE)this->GetSafeHwnd();
  108. hr=ipGxDialog->DoModalSave(hwnd,&bResult);
  109. if(FAILED(hr)) return;
  110. if(bResult==VARIANT_FALSE) return;
  111. IGxObjectPtr ipGxObject;
  112. ipGxDialog->get_FinalLocation(&ipGxObject);
  113. CComBSTR path;
  114. ipGxObject->get_FullName(&path);
  115. m_OutputPath=CString(path);
  116. CComBSTR name;
  117. ipGxDialog->get_Name(&name);
  118. m_OutputName=CString(name);
  119. UpdateData(TRUE);
  120. m_edit_viewshedRasteroutput=m_OutputPath+"\"+m_OutputName;
  121. UpdateData(FALSE);
  123. }
  125. void CDlgViewshed::OnButtonViewshedSurfaceinput() 
  126. {
  127. // TODO: Add your control notification handler code here
  128. IGxDialogPtr ipGxDialog(CLSID_GxDialog);
  129. IGxObjectFilterPtr ipGxObjectFilter( CLSID_GxFilterSurfaceDatasets);
  130. HRESULT hr=ipGxDialog->putref_ObjectFilter( ipGxObjectFilter);
  131. if(FAILED(hr)) return;
  132. hr = ipGxDialog->put_Title(CComBSTR("选择一个DEM表面数据"));
  133. if(FAILED(hr)) return;
  134. IEnumGxObjectPtr ipEnumGxObject;
  135. VARIANT_BOOL bResult;
  136. hr=ipGxDialog->DoModalOpen(0,&ipEnumGxObject,&bResult);
  137. if(FAILED(hr)) return;
  138. if(bResult==VARIANT_FALSE)return;
  139. hr=ipEnumGxObject->Reset();
  140. if(FAILED(hr)) return;
  141. IGxObjectPtr ipGxObject;
  142. ipGxDialog->get_FinalLocation(&ipGxObject);
  143. while(!ipEnumGxObject->Next(&ipGxObject))
  144. {   
  145. IGxDatasetPtr ipGxDataset(ipGxObject);    
  146. if(ipGxDataset!=0)     
  147. {
  148. IDatasetPtr ipDataset;
  149. hr=ipGxDataset->get_Dataset(&ipDataset);
  150. if(FAILED(hr )) return;
  151. IRasterDatasetPtr ipRasterDataset(ipDataset);
  152. if(ipRasterDataset!=NULL)
  153. {  
  154. m_ipR=ipRasterDataset;   
  155. CComBSTR bstrName;   
  156. hr=ipRasterDataset->get_CompleteName(&bstrName);
  157. if(FAILED(hr)) return;
  158. m_edit_viewshedSurfaceinput=bstrName;
  159. IRasterPtr ipRas;
  160. //获得输人Raster表面的网格大小,作为viewshed的默认值
  161. hr=ipRasterDataset->CreateDefaultRaster(&ipRas);
  162. IRasterAnalysisPropsPtr ipRA(ipRas);
  163. double defaultcellsize;
  164. hr=ipRA->get_PixelHeight(&defaultcellsize);
  165. m_edit_viewshedCellsize=defaultcellsize;
  166. UpdateData(FALSE);
  167. }      
  168. ITinPtr ipTin(ipDataset);      
  169. ITinSurfacePtr ipTS(ipDataset);      
  170. if( ipTin!=NULL)  
  171. {   
  172. m_ipT=ipTin;   
  173. CComBSTR bstrName;   
  174. hr=ipDataset->get_Name(&bstrName);   
  175. if(FAILED(hr)) return;
  176. m_edit_viewshedSurfaceinput=bstrName;  
  177. IGeoDatasetPtr plnTin(m_ipT);  
  178. //通过获得Tin的范围设置默认。ellsize大小  
  179. IEnvelopePtr pExtent;  
  180. hr=plnTin->get_Extent(&pExtent);  
  181. double width,height;
  182. hr=pExtent->get_Width(&width);
  183. hr= pExtent->get_Height(&height);
  184. double defaultcellsize;
  185. if ( width>height)
  186. defaultcellsize=height/250;
  187. else
  188. defaultcellsize=width/250;
  189. m_edit_viewshedCellsize=defaultcellsize;
  190. UpdateData( FALSE);
  191. }
  192. }
  193. }
  196. }
  198. void CDlgViewshed::OnOK() 
  199. {
  200. // TODO: Add extra validation here
  201. if(m_ipT!=NULL)
  202. {
  203. TintoRaster( m_ipT);//将Tin转为Raster}
  204. }
  205. Viewshed();
  206. CDialog::OnOK();
  208. }
  209.  void CDlgViewshed::TintoRaster(ITin*ipT)
  210. {
  211. HRESULT hr;
  212.     IGeoDatasetPtr pInTin(ipT);
  213.     IEnvelopePtr pExtent;
  214.     hr=pInTin->get_Extent(&pExtent);
  215. IPointPtr pOrigin;
  216. hr=pExtent->get_LowerLeft(&pOrigin);
  217. double x,y;
  218. hr=pOrigin->get_X(&x);
  219. hr=pOrigin->get_Y(&y);
  220. hr = pOrigin->put_X(x-m_edit_viewshedCellsize*0.5);
  221. hr = pOrigin->put_Y(y-m_edit_viewshedCellsize*0.5);
  222. long nCol,nRow;
  223. double width,height;
  224. hr=pExtent->get_Width(&width);
  225. hr=pExtent->get_Height(&height);
  226. nCol=int(width/m_edit_viewshedCellsize)+1;
  227. nRow=int(height/m_edit_viewshedCellsize)+1;
  228. ISpatialReferencePtr pSR;
  229. hr=pInTin->get_SpatialReference(&pSR);
  230. IRasterDatasetPtr pRDS;
  231. IWorkspaceFactoryPtr rWksFac(CLSID_RasterWorkspaceFactory);
  232. IWorkspacePtr wks;
  233. hr=rWksFac->OpenFromFile(CComBSTR(m_OutputPath),0,&wks);
  234. IRasterWorkspace2Ptr rWks(wks);
  235. long numbands=1;
  236. hr=rWks->CreateRasterDataset(CComBSTR("tempToR"),
  237.                                  CComBSTR("GRID"),
  238.                                  pOrigin,
  239.                                  nCol,
  240.                                  nRow,
  241.  m_edit_viewshedCellsize,
  242.  m_edit_viewshedCellsize,
  243.  numbands,
  244.  PT_FLOAT,
  245.  pSR,
  246.  VARIANT_TRUE,
  247.  &pRDS);
  249. if(FAILED(hr))
  250. {
  251. if(hr=-2147467259)
  252. {
  253. AfxMessageBox( " 临时文件tempTtoR已经存在,请先删除");
  254. }
  255. AfxMessageBox ( " 临时文件Raster文件tempTtoR生成失败!");
  256. return;
  257. }
  258. //下面是调用GetRawPixel。函数 
  259. IRasterBandCollectionPtr pBandCollection(pRDS);
  260. IRasterBandPtr pRasterBand;
  261. hr=pBandCollection->Item(0,&pRasterBand);
  262. IRawPixelsPtr pRawPixels(pRasterBand);
  263. IPntPtr pBlockSize(CLSID_DblPnt);
  264. hr=pBlockSize->put_X(nCol);
  265. hr=pBlockSize->put_Y(nRow);
  266. IPixelBlockPtr pPixelBlock;
  267. hr=pRawPixels->CreatePixelBlock(pBlockSize,&pPixelBlock);
  268. VARIANT val;
  269. val.vt=VT_SAFEARRAY ;//要指定类型
  270. hr=pPixelBlock->get_SafeArray(0,&val);
  271. ITinSurfacePtr pTinSurf( ipT);
  272. IRasterPropsPtr pRasterProps(pRawPixels);
  273. VARIANT nodataFloat;
  274. nodataFloat.vt=VT_R4 ;//要指定类型
  275.     hr=pOrigin->get_X(&x);
  276.     hr=pOrigin->get_Y(&y);
  277.     hr=pOrigin->put_X(x+m_edit_viewshedCellsize*0.5);
  278.     hr=pOrigin->put_Y(y+m_edit_viewshedCellsize*nRow-m_edit_viewshedCellsize*0.5);
  279. hr=pRasterProps->get_NoDataValue(&nodataFloat);
  280. hr=pOrigin->get_X(&x);
  281. hr=pOrigin->get_Y(&y);
  282. hr= pTinSurf->QueryPixelBlock(x,y,
  283. m_edit_viewshedCellsize,
  284. m_edit_viewshedCellsize,
  285. esriElevationAsRaster,
  286. nodataFloat,
  287. val);
  288. IPntPtr pOffset(CLSID_DblPnt);
  289. hr=pOffset->put_X(0);
  290. hr=pOffset->put_Y(0);
  291. hr=pRawPixels->Write(pOffset,pPixelBlock);
  292.     //注意:只有指针pRDS被释放,Raster才会从内存写人硬盘。
  293. }
  294. void CDlgViewshed::Viewshed()
  295. {
  296. HRESULT hr;
  297. if( m_ipT!=NULL)
  298. {
  299. IWorkspacePtr ppRW ;//将生成的Raster赋给ipR   
  300. IWorkspaceFactoryPtr pWF(CLSID_RasterWorkspaceFactory);    
  301. hr=pWF->OpenFromFile(CComBSTR(m_OutputPath),0,&ppRW);    
  302. IRasterWorkspacePtr pRW(ppRW);   
  303. hr=pRW->OpenRasterDataset(CComBSTR("tempTtoR"),&m_ipR);
  304. }
  305. //创建RasterSurface0p对象
  306. ISurfaceOpPtr pSurfaceOp(CLSID_RasterSurfaceOp);
  307. //声明输人表面对象
  308. IGeoDatasetPtr pElevation(m_ipR);
  309. //声明观察点对象
  310. IGeoDatasetPtr pObservers(m_ipfeat);
  311. //声明输出对象
  312. IGeoDatasetPtr pOutputRaster;
  313. //通过设置环境接口IRasterAnalysisEnvironment的
  314. //函数OutV}Torkspace改变路径
  315. IRasterAnalysisEnvironmentPtr pEnv(pSurfaceOp);
  316. hr=pEnv->Reset();
  317. VARIANT cellsize;//设置、iewshed输出结果cellsize的大小
  318. cellsize.vt=VT_R8;
  319. cellsize.dblVal=m_edit_viewshedCellsize;
  320. hr=pEnv->SetCellSize(esriRasterEnvValue,&cellsize);
  321. IWorkspacePtr pWo;
  322. IWorkspaceFactoryPtr pWSF(CLSID_RasterWorkspaceFactory);
  323. hr=pWSF->OpenFromFile(CComBSTR(m_OutputPath),NULL,&pWo);
  324. if(FAILED(hr))
  325. {
  326.        AfxMessageBox ("打开表面数据失败!");
  327.        return; 
  328. }
  330. hr=pEnv->putref_OutWorkspace(pWo);
  332. if(FAILED(hr ))
  333. {
  334. AfxMessageBox("输出路径设置失败!");
  335. return;
  336. }
  338. hr=pSurfaceOp->Visibility(pElevation,
  339.                               pObservers,
  340.                               esriGeoAnalysisVisibilityFrequency,
  341.                               &pOutputRaster);
  343. if(FAILED(hr))
  344. {    
  345. AfxMessageBox("viewshed failed!");     
  346. return;
  347. }
  348. //永久保存生成的Raster
  349. IRasterPtr pOutRaster(pOutputRaster);
  350. IRasterBandCollectionPtr pRasBandC(pOutRaster);
  351. IDatasetPtr pOut;
  352. hr=pRasBandC->SaveAs(CComBSTR(m_OutputName),pWo,CComBSTR("GRID"),&pOut);
  354. if(FAILED(hr))
  355. {
  356.     
  357. AfxMessageBox("分析失败!");
  358.         return;
  359. }
  360. }