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

namespace YunyouXueYuan
{
    public class Tools
    {
        #region "添加实体"
        /// <summary>
        /// 添加实体。
        /// 将实体添加到当前模型空间上。
        /// </summary>
        /// <param name="entityID">实体ID</param>
        /// <returns>true：成功 false：失败</returns>
        public bool AddEntity(Entity entity, out ObjectId objId)
        {
            objId = ObjectId.Null;
            try
            {
                using (Database db = HostApplicationServices.WorkingDatabase)
                {
                    using (Transaction trans = db.TransactionManager.StartTransaction())
                    {
                        using (BlockTableRecord btr = (BlockTableRecord)trans.GetObject(db.CurrentSpaceId, OpenMode.ForWrite))
                        {
                            objId = btr.AppendEntity(entity);
                            trans.AddNewlyCreatedDBObject(entity, true);
                        }
                        trans.Commit();
                    }
                }
            }
            catch
            {
                return false;
            }
            return true;
        }
        /// <summary>
        /// 添加实体。
        /// </summary>
        /// <param name="entity"></param>
        /// <returns></returns>
        public bool AddEntity(Entity entity)
        {
            //返回结果id
            ObjectId objId = ObjectId.Null;
            try
            {
                using (DocumentLock docLock = Application.DocumentManager.MdiActiveDocument.LockDocument())
                {

                    using (Database db = HostApplicationServices.WorkingDatabase)
                    {

                        using (Transaction trans = db.TransactionManager.StartTransaction())
                        {

                            using (BlockTableRecord btr = (BlockTableRecord)trans.GetObject(db.CurrentSpaceId, OpenMode.ForWrite))
                            {

                                objId = btr.AppendEntity(entity);
                                trans.AddNewlyCreatedDBObject(entity, true);
                            }

                            trans.Commit();
                        }
                    }
                }
            }
            catch
            {
                //添加失败
                return false;
            }
            return true;
        }
        #endregion

        #region "添加多个实体"
        /// <summary>
        /// 添加实体。
        /// 将实体添加到当前模型空间上。
        /// </summary>
        /// <param name="entityID">实体ID</param>
        /// <returns>true：成功 false：失败</returns>
        public bool AddEntity(Entity[] ents, out ObjectIdCollection objId)
        {

            objId = new ObjectIdCollection();
            try
            {

                using (Database db = HostApplicationServices.WorkingDatabase)
                {

                    using (Transaction trans = db.TransactionManager.StartTransaction())
                    {

                        using (BlockTableRecord btr = (BlockTableRecord)trans.GetObject(db.CurrentSpaceId, OpenMode.ForWrite))
                        {
                            foreach (Entity ent in ents)
                            {

                                objId.Add(btr.AppendEntity(ent));
                                trans.AddNewlyCreatedDBObject(ent, true);
                            }
                        }

                        trans.Commit();
                    }
                }
            }
            catch
            {

                return false;
            }
            return true;
        }
        #endregion

        #region "删除实体"
        /// <summary>
        /// 删除实体。
        /// 删除当前模型空间上的实体。
        /// </summary>
        /// <param name="id">实体ID</param>
        /// <returns>true：成功 false：失败</returns>
        public bool DelEntity(ObjectId id)
        {
            try
            {

                if (!id.IsNull)
                {

                    using (Database db = HostApplicationServices.WorkingDatabase)
                    {

                        using (Transaction trans = db.TransactionManager.StartTransaction())
                        {

                            Entity entity = (Entity)trans.GetObject(id, OpenMode.ForWrite, true);

                            entity.Erase(true);

                            trans.Commit();
                        }
                    }
                }
                else
                {
                    return false;
                }
            }
            catch
            {

                return false;
            }
            return true;
        }
        #endregion

        #region "删除多个实体"
        /// <summary>
        /// 删除实体。
        /// 删除当前模型空间上的实体。
        /// </summary>
        /// <param name="entityID">实体ID</param>
        /// <returns>true：成功 false：失败</returns>
        public bool DelEntity(ObjectIdCollection ids)
        {
            try
            {

                using (Database db = HostApplicationServices.WorkingDatabase)
                {

                    using (Transaction trans = db.TransactionManager.StartTransaction())
                    {

                        using (BlockTableRecord btr = (BlockTableRecord)trans.GetObject(db.CurrentSpaceId, OpenMode.ForWrite))
                        {
                            foreach (ObjectId id in ids)
                            {

                                Entity entity = (Entity)trans.GetObject(id, OpenMode.ForWrite, true);
                                if (entity == null || entity.IsErased == true || entity is ProxyEntity)
                                {
                                    continue;
                                }

                                entity.Erase(true);
                            }
                        }
                        trans.Commit();
                    }
                }
            }
            catch
            {
                return false;
            }
            return true;
        }
        #endregion
    }
}