CS_GIS/Main/GNetworkProcessTool/DataOperateHelper.cs


using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using ESRI.ArcGIS.Carto;
using ESRI.ArcGIS.Geodatabase;

namespace GNetworkProcessTool
{
    public class DataOperateHelper
    {
        /// <summary>
        /// 获取GDB工作空间通过GDB文件路径
        /// </summary>
        /// <param name="strGdbFilePath">GDB文件路径</param>
        /// <param name="pFeatureWorkspace">输出的工作空间</param>
        public static void GetWorkspaceFromGDBFile(string strGdbFilePath, out IFeatureWorkspace pFeatureWorkspace)
        {
            IWorkspaceFactory pWorkspaceFactory = new ESRI.ArcGIS.DataSourcesGDB.FileGDBWorkspaceFactoryClass();
            IWorkspace pWorkspace = null;
            try
            {
                pWorkspace = pWorkspaceFactory.OpenFromFile(strGdbFilePath, 0);
            }
            catch
            {
                throw new Exception("打开GDB失败：无效的GDB文件！");
            }    
            pFeatureWorkspace = pWorkspace as IFeatureWorkspace;

            System.Runtime.InteropServices.Marshal.ReleaseComObject(pWorkspaceFactory);
            //System.Runtime.InteropServices.Marshal.ReleaseComObject(pWorkspace);
        }

        /// <summary>
        /// 从要素集获得要素图层（排除几何网络关系图层）
        /// </summary>
        /// <param name="pFeatureWorkspace">工作空间</param>
        /// <param name="pFeatureDataset">要素集</param>
        /// <returns>图层名（去除前缀英文）,要素类</图层名></returns>
        public static Dictionary<string, IFeatureClass> GetFeatureClassFromDataset(IFeatureWorkspace pFeatureWorkspace, IFeatureDataset pFeatureDataset)
        {
            Dictionary<string, IFeatureClass> DicFeatureClass = new Dictionary<string, IFeatureClass>();

            IEnumDataset pEnumDataset = pFeatureDataset.Subsets;
            pEnumDataset.Reset();
            IDataset pDataset = pEnumDataset.Next();
            IFeatureLayer pFeatureLayer = new FeatureLayerClass();
            while (pDataset != null)
            {
                if (pDataset.Type == esriDatasetType.esriDTFeatureClass && !pDataset.Name.Contains("_Net"))//排除几何网络关系图层
                {
                    pFeatureLayer.FeatureClass = pFeatureWorkspace.OpenFeatureClass(pDataset.Name);
                    if (pDataset.Name.Contains("."))
                    {
                        pFeatureLayer.Name = pDataset.Name.Substring(pDataset.Name.LastIndexOf(".") + 1);//截取去除前缀
                    }
                    else
                    {
                        pFeatureLayer.Name = pDataset.Name;
                    }
                    DicFeatureClass.Add(pFeatureLayer.Name, pFeatureLayer.FeatureClass);
                }
                pDataset = pEnumDataset.Next();
            }

            System.Runtime.InteropServices.Marshal.ReleaseComObject(pEnumDataset);
            return DicFeatureClass;
        }

        /// <summary>
        /// 从要素集获得要素图层（排除几何网络关系图层和不需要的图层）
        /// </summary>
        /// <param name="pFeatureWorkspace">工作空间</param>
        /// <param name="pFeatureDataset">要素集</param>
        /// <param name="lstRemoveClass">不需要图层名称集合</param>
        /// <returns>图层名（去除前缀英文）,要素类</returns>
        public static Dictionary<string, IFeatureClass> GetFeatureClassFromDataset(IFeatureWorkspace pFeatureWorkspace, IFeatureDataset pFeatureDataset, List<string> lstRemoveClass)
        {
            Dictionary<string, IFeatureClass> DicFeatureClass = new Dictionary<string, IFeatureClass>();

            IEnumDataset pEnumDataset = pFeatureDataset.Subsets;
            pEnumDataset.Reset();
            IDataset pDataset = pEnumDataset.Next();
            IFeatureLayer pFeatureLayer = new FeatureLayerClass();
            while (pDataset != null)
            {
                if ((pDataset.Type == esriDatasetType.esriDTFeatureClass && !pDataset.Name.Contains("_Net")) && !lstRemoveClass.Exists(p=>p == (pDataset.Name.Contains(".") ? pDataset.Name.Substring(pDataset.Name.LastIndexOf(".") + 1) : pDataset.Name)))//排除几何网络关系图层
                {
                    pFeatureLayer.FeatureClass = pFeatureWorkspace.OpenFeatureClass(pDataset.Name);
                    if (pDataset.Name.Contains("."))
                    {
                        pFeatureLayer.Name = pDataset.Name.Substring(pDataset.Name.LastIndexOf(".") + 1);//截取去除前缀
                    }
                    else
                    {
                        pFeatureLayer.Name = pDataset.Name;
                    }
                    DicFeatureClass.Add(pFeatureLayer.Name, pFeatureLayer.FeatureClass);
                }
                pDataset = pEnumDataset.Next();
            }

            System.Runtime.InteropServices.Marshal.ReleaseComObject(pEnumDataset);
            return DicFeatureClass;
        }

        /// <summary>
        /// 获取要素个数通过要素类（游标）
        /// </summary>
        /// <param name="pFeatureclass">要素类</param>
        /// <returns>要素个数</returns>
        public static int GetFeatureCount(IFeatureClass pFeatureclass)
        {
            IFeatureCursor pFeatureCursor = pFeatureclass.Search(null, false);
            IFeature pFeature = pFeatureCursor.NextFeature();
            int pFeatureCount = 0;
            while (pFeature != null)
            {
                pFeatureCount++;
                pFeature = pFeatureCursor.NextFeature();
            }

            System.Runtime.InteropServices.Marshal.ReleaseComObject(pFeatureCursor);
            return pFeatureCount;
        }

        /// <summary>
        /// 获取要素个数通过要素类（转表）
        /// </summary>
        /// <param name="pFeatureclass">要素类</param>
        /// <param name="pQueryFilter">过滤条件</param>
        /// <returns>要素个数</returns>
        public static int GetFeatureCount(IFeatureClass pFeatureclass, IQueryFilter pQueryFilter)
        {
            return (pFeatureclass as ITable).RowCount(pQueryFilter);
        }
    }
}