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

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

GIS编程

开发平台：

Visual C++

DlgCreateContour.cpp：源码内容
  1. // DlgCreateContour.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "3D0214.h"
  6. #include "DlgCreateContour.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CDlgCreateContour dialog
  18. CDlgCreateContour::CDlgCreateContour(CWnd* pParent /*=NULL*/)
  19. : CDialog(CDlgCreateContour::IDD, pParent)
  20. {
  21. //{{AFX_DATA_INIT(CDlgCreateContour)
  22. m_edit_Contour_SurfaceInput = _T("");
  23. m_edit_ContourFeatureOutput = _T("");
  24. m_edit_Contour_MaxContour = 0.0;
  25. m_edit_Contour_MinContour = 0.0;
  26. m_edit_ContourZMax_in = 0.0;
  27. m_edit_ContourZMin_in = 0.0;
  28. m_edit_ContourTotalNumContour = 0;
  29. //}}AFX_DATA_INIT
  30. }
  33. void CDlgCreateContour::DoDataExchange(CDataExchange* pDX)
  34. {
  35. CDialog::DoDataExchange(pDX);
  36. //{{AFX_DATA_MAP(CDlgCreateContour)
  37. DDX_Control(pDX, IDC_EDIT_CONTOUR_CONTOURINTERVAL, m_edit_Contour_ContourInterval);
  38. DDX_Control(pDX, IDC_EDIT_CONTOUR_BASECONTOUR, m_edit_ContourBaseContour);
  39. DDX_Text(pDX, IDC_EDIT_CONTOUR_SURFACE_INPUT, m_edit_Contour_SurfaceInput);
  40. DDX_Text(pDX, IDC_EDIT_CONTOUR_FEATURE_OUTPUT, m_edit_ContourFeatureOutput);
  41. DDX_Text(pDX, IDC_EDIT_CONTOUR_MAXCONTOUR, m_edit_Contour_MaxContour);
  42. DDX_Text(pDX, IDC_EDIT_CONTOUR_MINCONTOUR, m_edit_Contour_MinContour);
  43. DDX_Text(pDX, IDC_EDIT_CONTOUR_ZMAX, m_edit_ContourZMax_in);
  44. DDX_Text(pDX, IDC_EDIT_CONTOUR_ZMIN, m_edit_ContourZMin_in);
  45. DDX_Text(pDX, IDC_EDIT_CONTOUR_TOTALNUMCONTOUR, m_edit_ContourTotalNumContour);
  46. //}}AFX_DATA_MAP
  47. }
  50. BEGIN_MESSAGE_MAP(CDlgCreateContour, CDialog)
  51. //{{AFX_MSG_MAP(CDlgCreateContour)
  52. ON_BN_CLICKED(IDC_BUTTON_CONTOUR_FEATURE_OUTPUT, OnButtonContourFeatureOutput)
  53. ON_BN_CLICKED(IDC_BUTTON_CONTOUR_SURFACE_INPUT, OnButtonContourSurfaceInput)
  54. ON_EN_CHANGE(IDC_EDIT_CONTOUR_BASECONTOUR, OnChangeEditContourBasecontour)
  55. ON_EN_CHANGE(IDC_EDIT_CONTOUR_CONTOURINTERVAL, OnChangeEditContourContourinterval)
  56. //}}AFX_MSG_MAP
  57. END_MESSAGE_MAP()
  59. /////////////////////////////////////////////////////////////////////////////
  60. // CDlgCreateContour message handlers
  62. void CDlgCreateContour::OnButtonContourFeatureOutput() 
  63. {
  64. // TODO: Add your control notification handler code here
  65. IGxDialogPtr ipGxDialog( CLSID_GxDialog);
  66. //FeatureClasses
  67. IGxObjectFilterPtr ipGxObjectFilter(CLSID_GxFilterFeatureClasses);
  68. HRESULT hr=ipGxDialog->putref_ObjectFilter( ipGxObjectFilter);
  69. if(FAILED(hr)) return;
  70. hr = ipGxDialog->put_Title(CComBSTR("保存等高线"));
  71. if(FAILED(hr)) return;
  72. VARIANT_BOOL bResult;
  73.     IEnumGxObjectPtr ipEnumGxObject;
  74. OLE_HANDLE hwnd=(OLE_HANDLE)this->GetSafeHwnd();
  75. hr=ipGxDialog->DoModalOpen(0,&ipEnumGxObject,&bResult);
  76. if(FAILED(hr)) return;
  77. if(bResult==VARIANT_FALSE)return;
  78. IGxObjectPtr ipGxObject;
  79. ipGxDialog->get_FinalLocation(&ipGxObject); 
  80. CComBSTR path;
  81. ipGxObject->get_FullName(&path);
  82. m_OutputFeatureDir=CString(path);
  83. CComBSTR name;
  84. ipGxDialog->get_Name(&name);
  85. m_OutputFeatureName=CString(name);
  86. //    m_edit_ContourFeatureOutput;
  87. // m_OutputFeatureDir+"\"+m_OutputFeatureName;
  88. UpdateData(FALSE);
  89. }
  91. void CDlgCreateContour::OnButtonContourSurfaceInput() 
  92. {
  93. // TODO: Add your control notification handler code here
  95. IGxDialogPtr ipGxDialog( CLSID_GxDialog);
  96. IGxObjectFilterPtr ipGxObjectFilter( CLSID_GxFilterSurfaceDatasets);
  97. HRESULT hr=ipGxDialog->putref_ObjectFilter(ipGxObjectFilter);
  98. if(FAILED(hr)) return;
  99. hr=ipGxDialog->put_Title(CComBSTR ("选择一个DEM表面数据"));
  100. if(FAILED( hr)) return;
  101. IEnumGxObjectPtr ipEnumGxObject;
  102. VARIANT_BOOL bResult;
  103. hr=ipGxDialog->DoModalOpen(0,&ipEnumGxObject,&bResult);
  104. if(FAILED(hr)) return;
  105. if(bResult==VARIANT_FALSE)return;
  106. hr=ipEnumGxObject->Reset();
  107. if(FAILED(hr)) return;
  108. IGxObjectPtr ipGxObject;
  109. ipGxDialog->get_FinalLocation(&ipGxObject);
  110. CComBSTR path;
  111. ipGxObject->get_FullName(&path);
  112. m_edit_Contour_SurfaceInput=CString(path);
  113. while(!ipEnumGxObject->Next(&ipGxObject))
  114. {
  115. IGxDatasetPtr ipGxDataset(ipGxObject);
  116. if(ipGxDataset!=0)
  117. {    
  118. IDatasetPtr ipDataset;    
  119. hr=ipGxDataset->get_Dataset(&ipDataset);    
  120. if(FAILED(hr)) return;
  121. IRasterDatasetPtr ipRasterDataset(ipDataset);
  122. if(ipRasterDataset!=NULL)
  123. {
  124. m_ipR=ipRasterDataset;
  125. CComBSTR bstrName;
  126. hr=ipRasterDataset->get_CompleteName(&bstrName);
  127. if(FAILED(hr)) return;
  128. m_edit_Contour_SurfaceInput= bstrName;
  129. }
  130.      ITinPtr ipTin(ipDataset);
  131.     ITinSurfacePtr ipTS(ipDataset);
  133.     if ( ipTin!=NULL)
  134. {
  135.     
  136. m_ipT=ipTin;    
  137. m_ipTS=ipTS;      
  138. CComBSTR bstrName;    
  139. hr=ipDataset->get_Name(&bstrName);    
  140. if(FAILED(hr )) return;    
  141. m_edit_Contour_SurfaceInput=CString(bstrName);
  142. }
  143.    UpdateData(FALSE);
  144.    if(ipRasterDataset!=NULL)  
  145.    {
  146.          ComputerHeightRange(ipRasterDataset);  
  147.    }
  148.   
  149.    if(ipTin!=NULL)  
  150.    {
  151.        ComputerHeightRange(ipTin);
  152.    }
  153. }
  154. }
  155. }
  156. void CDlgCreateContour::OnOK() 
  157. {
  158. // TODO: Add extra validation here
  159. UpdateData(TRUE);
  161. if(m_ipR!=NULL)
  162. {
  163. CreateContour(m_ipR);
  165. }
  166. if(m_ipT!=NULL)
  167. {
  168. CreateContour();
  169. }
  170. CDialog::OnOK();
  171. }
  173. void CDlgCreateContour::OnChangeEditContourBasecontour() 
  174. {
  175. // TODO: If this is a RICHEDIT control, the control will not
  176. // send this notification unless you override the CDialog::OnInitDialog()
  177. // function and call CRichEditCtrl().SetEventMask()
  178. // with the ENM_CHANGE flag ORed into the mask.
  180. // TODO: Add your control notification handler code here
  181. if(m_ipT!=NULL || m_ipR!=NULL)
  182.  
  183. {    
  184. CString str;    
  185. m_edit_Contour_ContourInterval.GetWindowText(str);    
  186. double interval=atof(str);
  187. m_edit_ContourBaseContour.GetWindowText(str);
  188. double base=atof ( str );
  189. ComputerOutputInfo(m_edit_ContourZMax_in,
  190.                           m_edit_ContourZMin_in,
  191.                           interval,
  192.                            base);
  193. }
  194. }
  196. void CDlgCreateContour::OnChangeEditContourContourinterval() 
  197. {
  198. // TODO: If this is a RICHEDIT control, the control will not
  199. // send this notification unless you override the CDialog::OnInitDialog()
  200. // function and call CRichEditCtrl().SetEventMask()
  201. // with the ENM_CHANGE flag ORed into the mask.
  203. // TODO: Add your control notification handler code here
  204. if(m_ipT!=NULL || m_ipR!=NULL)
  205. {
  206.         CString str;
  207.         m_edit_Contour_ContourInterval.GetWindowText(str);
  208.         double interval=atof (str);
  209.         m_edit_ContourBaseContour.GetWindowText(str);
  210.         double base=atof ( str);
  211.         ComputerOutputInfo(m_edit_ContourZMax_in,
  212.                             m_edit_ContourZMin_in,
  213.                            interval,
  214.                            base);
  215. }
  216. }
  217. void CDlgCreateContour::ComputerHeightRange(IRasterDataset * pRasterDataset)
  218. {   //如何计算RasterDataset 的统计信息
  219. HRESULT hr;
  220. IRasterBandCollectionPtr pRasterBC(pRasterDataset);
  221. IRasterBandPtr pRasterB;
  222. hr=pRasterBC->Item(0,&pRasterB);
  223. IRasterStatisticsPtr pRasterS;
  224. hr=pRasterB->get_Statistics(&pRasterS);
  225. hr=pRasterS->get_Maximum(&m_edit_ContourZMax_in);
  226. if(FAILED(hr )) return;
  227. hr=pRasterS->get_Minimum(&m_edit_ContourZMin_in);
  228.     UpdateData(FALSE);
  229. }
  230. void CDlgCreateContour::ComputerHeightRange(ITin *ipT)
  231. {   //如何计算Tin的空间范围统计信息
  232.    HRESULT hr;
  233.     IEnvelopePtr ipEnv;
  234.     m_ipT->get_Extent(&ipEnv); 
  235. hr= ipEnv->get_ZMin(&m_edit_ContourZMax_in);
  236. if(FAILED(hr ))return; 
  237. hr=ipEnv->get_ZMax(&m_edit_ContourZMax_in); 
  238. UpdateData(FALSE);
  239. }
  240. void CDlgCreateContour::ComputerOutputInfo(double Zmax,double Zmin,double interval,double base)
  241. {
  242.    if(interval==0)
  243.     {
  244.         AfxMessageBox("请不要把间隔设置为0");
  245.         return;
  246.     }
  247.     if(base>Zmin)
  248.     {
  249.           int  N=1;
  250.         for( int i=0;i<N;i++)
  251.         {
  252.             double temp= base-i *interval-Zmin;
  253.             if(temp<=interval&&temp>=0)
  254.               {
  255.                   N=1;
  256.                   m_edit_Contour_MinContour=base-i*interval;
  257.               }
  258. else{
  259. N=i+2;}
  260. }
  261. }
  262.     else
  263. {
  264. int N = 1;
  266. for( int i=0;i<N;i++)
  267. {
  269. double temp =base+i*interval-Zmin;
  270. if(temp<=interval&&temp>=0)
  271. {
  272. N=i;
  273. m_edit_Contour_MinContour=base+i*interval;
  274. }
  275.     else
  276. {
  278. N=i+2;
  279. }
  280. }
  281. }
  284. if(base>Zmax)
  285. {
  286. int N=1;
  287. for( int i=0;i<N;i++)
  288. {
  289. double temp=Zmax-(base-i*interval);
  290. if(temp<=interval&&temp>=0)
  291. {
  292. N=i;
  293. m_edit_Contour_MaxContour=base-i*interval;
  294. }        
  295. else
  296. {              
  297. N=i+2;
  298. }
  299. }
  300. }
  301. else
  302. {
  303. int N=1;
  304. for( int i=0;i<N;i++)
  305. {
  306.     
  307. double temp= Zmax-(base+i*interval);
  308.     
  309. if(temp<= interval&&temp>=0)
  310.     
  311. {
  312.           
  313. N=3;
  314.           
  315. m_edit_Contour_MaxContour = base+i*interval;
  316.       
  317. }
  318.       
  319. else
  320. {
  322. N=i+2; 
  323. }
  324.           
  325. m_edit_ContourTotalNumContour=
  326. (m_edit_Contour_MaxContour-m_edit_Contour_MinContour)/interval+1;
  327. UpdateData(FALSE);
  328. }
  329. }
  330. }
  331. void CDlgCreateContour::CreateContour()
  332. {
  333. //先创建featureclass对象，用于等高线的输出
  334. HRESULT hr;
  335. CString strShapeFieldName="Shape";
  336. //打开一个存在的文件夹作为工作区Workspace
  337. IWorkspaceFactoryPtr ipWorkspaceFactory(CLSID_ShapefileWorkspaceFactory);
  338. IWorkspacePtr ipWorkspace;
  339. hr=ipWorkspaceFactory->OpenFromFile((CComBSTR)m_OutputFeatureDir,0,&ipWorkspace);
  340. if( FAILED(hr)) return;
  341. IFeatureWorkspacePtr ipFeatWorkspace(ipWorkspace);
  342. //创建Fields对象集
  343. IFieldsPtr ipFields(CLSID_Fields);
  344. IFieldsEditPtr ipFieldsEdit(ipFields);
  345. //创建Shape字段
  346. //it will need a geometry definition，with a spatial reference
  347. IFieldPtr ipField(CLSID_Field);
  348. IFieldEditPtr ipFieldEdit( ipField);
  349. hr=ipFieldEdit->put_Name((CComBSTR)strShapeFieldName);
  350. if ( FAILED(hr)) return;
  351. hr=ipFieldEdit->put_Type(esriFieldTypeGeometry);
  352. if( FAILED(hr)) return;
  353. IGeometryDefPtr ipGeomDef ( CLSID_GeometryDef );
  354. IGeometryDefEditPtr ipGeomDefEdit(ipGeomDef );
  355. ISpatialReferencePtr ipSpatialRef( CLSID_UnknownCoordinateSystem);
  356. hr=ipGeomDefEdit->put_GeometryType(esriGeometryPolyline);
  357. if ( FAILED(hr)) return;
  358. hr=ipGeomDefEdit->putref_SpatialReference( ipSpatialRef );
  359. if ( FAILED(hr)) return;
  360. hr=ipFieldEdit->putref_GeometryDef( ipGeomDef );
  361. if ( FAILED(hr )) return;
  362. hr=ipFieldsEdit->AddField(ipField);
  363. if ( FAILED(hr)) return;
  364. //创建Shapefile(一些参数默认为NULL)
  365. IFeatureClassPtr   ipFeatureClass;
  366. hr=ipFeatWorkspace->CreateFeatureClass((CComBSTR)m_OutputFeatureName,
  367.                                     ipFields,
  368.                                                     NULL,
  369.                                                     NULL,
  370.                                                     esriFTSimple,
  371.                                                     (CComBSTR)strShapeFieldName,
  372.                                                     (CComBSTR)"",
  373.                                                     &ipFeatureClass);
  375. if ( FAILED(hr )) return;
  376. if ( ipFeatureClass==NULL)return;
  377. CString str;
  378. m_edit_Contour_ContourInterval.GetWindowText(str);
  379. double interval=atof ( str );
  380. m_edit_ContourBaseContour. GetWindowText(str);
  381. double base=atof(str);
  382. hr=m_ipTS->Contour(base,interval,ipFeatureClass,CComBSTR("Contour"),0);
  383. if(FAILED(hr))
  384. {
  385. if(hr==-2147467259)
  386. {
  387. AfxMessageBox("等高线恢复无法进行，请更换表面数据!");
  388. }
  389. return;
  390. }
  391. AfxMessageBox ("成功!");
  392. }
  393. void CDlgCreateContour::CreateContour( IRasterDataset*ipR)
  394. {
  395. HRESULT hr;
  396. ISurfaceOpPtr pSurfaceOp(CLSID_RasterSurface);
  397. IGeoDatasetPtr plnputDataset(ipR);
  398. //声明输出对象
  399. IGeoDatasetPtr pOutput;
  400. //设置输出工作区
  401. IWorkspacePtr pWo;
  402. IWorkspaceFactoryPtr pWSF(CLSID_ShapefileWorkspaceFactory);
  403. hr=pWSF->OpenFromFile((CComBSTR)m_OutputFeatureDir,NULL,&pWo);
  404. if(FAILED(hr))
  405. {
  406. AfxMessageBox( "输出路径不存在!");
  407. return;
  408. }
  409. IRasterAnalysisEnvironmentPtr pEnv(pSurfaceOp);
  410. hr = pEnv->putref_OutWorkspace(pWo);
  411. if(FAILED(hr))
  413. {
  414. AfxMessageBox("输出路径设置失败!");
  416. return;
  418. }
  419. //Contour
  420. CString str;
  421. m_edit_Contour_ContourInterval.GetWindowText(str);
  422. double interval=atof ( str );
  423. m_edit_ContourBaseContour. GetWindowText(str );
  424. double base= atof(str);
  425. VARIANT BASE;
  426. BASE.vt =VT_R8;
  427. BASE.dblVal=base;
  428. hr=pSurfaceOp->Contour(plnputDataset,interval,&BASE,&pOutput);
  429. if(FAILED(hr ))
  430. if(hr==-2147467259)
  431. {
  433. AfxMessageBox("等高线恢复无法进行，请更换表面数据!");
  434. return;
  435. }
  436. //重命名:因为Contour后的文件名为默认的，不符合要求
  437. IDatasetPtr pda(pOutput);
  438. VARIANT_BOOL CanRename;
  439. hr = pda->CanRename(&CanRename);
  440. if(FAILED(hr ))
  441. {
  442.         return;
  443. }
  444. if( CanRename == VARIANT_TRUE)
  445. {
  446. hr=pda->Rename((CComBSTR)m_OutputFeatureName);
  448. return;
  450. AfxMessageBox("成功!");
  451. }
  452. }