HYTOS

using System;

using System.Collections.Generic;

using System.Collections;

using System.Configuration;

using System.Data;

using System.Data.Common;

using System.Data.SqlClient;

using System.IO;

using Autodesk.AutoCAD.ApplicationServices;

using Autodesk.AutoCAD.ApplicationServices.Core;

using Autodesk.AutoCAD.DatabaseServices;

using Autodesk.AutoCAD.EditorInput;

using Autodesk.AutoCAD.Geometry;

using Autodesk.AutoCAD.Interop;

using Autodesk.AutoCAD.Interop.Common;

using Autodesk.AutoCAD.Runtime;

using Autodesk.AutoCAD.Windows;

using AVEVA.PID.Components;

using AVEVA.PID.GUI;

using AVEVA.PID.Common;

using AVEVA.PID.DWGSelector;

using AVEVA.PID.Utilities;

using AVEVA.PID.CustomizationUtility.DB;

using AVEVA.PID.CustomizationUtility.Model;

using AVEVA.PID.CustomizationUtility.Properties;

using DevExpress.XtraSplashScreen;

namespace AVEVA.PID.CustomizationUtility

{

    public class AutoModeling

    {

        public static AutoModeling Running { get; set; }

        Model.Document document;

        System.Data.DataTable AvevaSymbolTable = null;

        public AutoModeling(Model.Document document, System.Data.DataTable AvevaSymbolTable)

        {

            this.document = document;

            this.AvevaSymbolTable = AvevaSymbolTable;

        }

        public void CreateDrawing()

        {

            string outPath = Project_DB.GetDirectiveValue("DRGPTH");

            string tempPath = Path.GetTempPath();

            string selectedFullPath = Path.Combine(tempPath, document.AvevaTemplateName + ".DWT");

            DWTSelector.strDrawingNumber = document.AvevaDrawingNumber;

            DWTSelector.strSheetNumber = document.AvevaSheetNumber;

            DWTSelector.strCadFileName = document.AvevaDrawingNumber + document.AvevaSheetNumber;

            DWTSelector.strCadFilePath = outPath;

            try

            {

                string text = ReadWriteDrawingData.ReadTemplateData(document.AvevaTemplateName, document.AvevaTemplateID, selectedFullPath, false);

                Autodesk.AutoCAD.Interop.AcadApplication acadApplication = Autodesk.AutoCAD.ApplicationServices.Application.AcadApplication as Autodesk.AutoCAD.Interop.AcadApplication;

                Autodesk.AutoCAD.Interop.AcadDocument acadDocument = acadApplication.Documents.Add(text);

                Commands.addLatestPidvesrionToDictionary();

                AVVPropCmd.ActivateAcad();

                if (acadDocument != null)

                {

                    Running = this;

                    acadApplication.ActiveDocument = acadDocument;

                    acadDocument.SendCommand("QSAVE ");

                    acadDocument.SendCommand("APPIDRun ");

                    acadDocument.SendCommand("QSAVE ");

                    Running = null;

                }

            }

            catch (System.Exception ex)

            {

            }

            GC.Collect();

        }

        public void Run()

        {

            try

            {

                SplashScreenManager.ShowForm(typeof(APIDSplashScreen), true, true);

                SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetAllStepCount, 2);

                SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetDocumentName, document.AvevaDrawingNumber + document.AvevaSheetNumber);

                RunLineModeling();

                RunSymbolModeling();

            }

            catch (System.Exception ex)

            {

                System.Windows.Forms.MessageBox.Show(ex.Message + @"\n" + ex.StackTrace);

            }

            finally

            {

                SplashScreenManager.CloseForm(false);

            }

        }

        #region Run Method

        private void RunLineModeling()

        {

            SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetAllProgress, document.LINES.Count);

            SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetStep, "Line Modeling");

            foreach (var item in document.LINES)

            {

                if (item.Aveva.Handle == 0)

                    LineModeling(item);

                SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.UpProgress, null);

            }

        }

        private void RunSymbolModeling()

        {

            SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetAllProgress, document.SYMBOLS.Count);

            SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.SetStep, "Symbol Modeling");

            foreach (var item in document.SYMBOLS)

            {

                if (item.Aveva.Handle == 0)

                    SymbolModeling(item);

                SplashScreenManager.Default.SendCommand(APIDSplashScreen.SplashScreenCommand.UpProgress, null);

            }

        }

        #endregion

        #region Modeling Method

        private void LineModeling(Model.Line line)

        {

            List<Model.Line> groupLine = new List<Model.Line>();

            List<string> points = new List<string>();

            GetConnectedGroupLine(line, groupLine, false);

            GetGroupLinePoints(groupLine, points);

            long handle = 0;

            if (line.Aveva.Type == Model.Type.Pipe)

                handle = DrawPipe(line.Aveva.Name, points);

            else if (line.Aveva.Type == Model.Type.Signal)

                handle = DrawSignal(line.Aveva.Name, points);

            foreach (var item in groupLine)

                item.Aveva.Handle = handle;

        }

        private void SymbolModeling(Symbol symbol)

        {

            DataRow[] allRows = AvevaSymbolTable.Select(string.Format("Symbol_Name = '{0}'", symbol.Aveva.Name));

            if (allRows.Length == 1)

            {

                DataRow symbolRow = allRows[0];

                bool bAngle = CheckAngle(symbolRow);

                bool bBalloon = CheckBalloon(symbol);

                long handle = InsertSymbol(symbol, symbolRow, bAngle, bBalloon);

                if (handle != 0)

                {

                    symbol.Aveva.Handle = handle;

                    if (!bAngle)

                    {

                        ObjectId objectId = GetObjectIdByHandle(handle);

                        if (objectId != ObjectId.Null)

                        {

                            SetBlockReferenceRotation(objectId, symbol.ANGLE);

                        }

                    }

                }

            }

        }

        #endregion

        #region Drawing Method

        private long DrawPipe(string style, List<string> coordinates)

        {

            List<long> prevHandles = GetAllGroupHandles();

            GUIUtils.SetPipeStyle(style);

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Editor editor = acDoc.Editor;

            List<object> commandParam = new List<object>();

            //command name

            commandParam.Add("DRPIPE");

            //coordinate

            foreach (var item in coordinates)

                commandParam.Add(item);

            //enter

            commandParam.Add(null);

            //enter

            commandParam.Add(null);

            editor.Command(commandParam.ToArray());

            List<long> newHandles = GetAllGroupHandles();

            List<long> otherHandles = new List<long>();

            foreach (var item in newHandles)

                if (!prevHandles.Contains(item))

                    otherHandles.Add(item);

            if (otherHandles.Count == 1)

                return otherHandles[0];

            else

                return 0;

        }

        private long DrawPipe(string style, List<string> coordinates, ObjectId objectId)

        {

            List<long> prevHandles = GetAllGroupHandles();

            GUIUtils.SetPipeStyle(style);

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Editor editor = acDoc.Editor;

            List<object> commandParam = new List<object>();

            //command name

            commandParam.Add("DRPIPE");

            //coordinate

            foreach (var item in coordinates)

                commandParam.Add(item);

            //enter

            commandParam.Add(null);

            //Input Object ID

            commandParam.Add(objectId);

            //enter

            commandParam.Add(null);

            editor.Command(commandParam.ToArray());

            List<long> newHandles = GetAllGroupHandles();

            List<long> otherHandles = new List<long>();

            foreach (var item in newHandles)

                if (!prevHandles.Contains(item))

                    otherHandles.Add(item);

            if (otherHandles.Count == 1)

                return otherHandles[0];

            else

                return 0;

        }

        private long DrawSignal(string style, List<string> coordinates)

        {

            List<long> prevHandles = GetAllGroupHandles();

            GUIUtils.SetPipeStyle(style);

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Editor editor = acDoc.Editor;

            List<object> commandParam = new List<object>();

            //command name

            commandParam.Add("VPESIGNAL");

            //coordinate

            foreach (var item in coordinates)

                commandParam.Add(item);

            //enter

            commandParam.Add(null);

            editor.Command(commandParam.ToArray());

            List<long> newHandles = GetAllGroupHandles();

            List<long> otherHandles = new List<long>();

            foreach (var item in newHandles)

                if (!prevHandles.Contains(item))

                    otherHandles.Add(item);

            if (otherHandles.Count == 1)

                return otherHandles[0];

            else

                return 0;

        }

        private long InsertSymbol(Symbol symbol, DataRow symbolRow, bool needAngle, bool needBalloon)

        {

            long handle = 0;

            try

            {

                string insertSymbolName = symbol.Aveva.Name;

                double x = symbol.Aveva.X;

                double y = symbol.Aveva.Y;

                List<long> prevHandles = GetAllBlockHandlesByName(insertSymbolName);

                AVEVA.PID.Utilities.DrawingData drawingData = new AVEVA.PID.Utilities.DrawingData();

                drawingData.InsertSymbolName = insertSymbolName;

                Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

                Editor editor = acDoc.Editor;

                List<object> commandParam = new List<object>();

                commandParam.Add("INSSYM");

                // Insert Point

                Point3d point = new Point3d(x, y, 0);

                commandParam.Add(point);

                // Check Insert Point

                if (needAngle)

                    commandParam.Add(symbol.ANGLE);

                // Check Balloon

                if (needBalloon)

                {

                    point = new Point3d(x + 20, y + 20, 0);

                    commandParam.Add(point);

                    // Check LeaderLine

                    if (GraphicalUtility.IsLeaderReqOnSymbolBalloons())

                        commandParam.Add(x + "," + y);

                }

                for (int i = 0; i < 8; i++)

                    commandParam.Add(null);

                editor.Command(commandParam.ToArray());

                List<long> newHandles = GetAllBlockHandlesByName(insertSymbolName);

                List<long> otherHandles = new List<long>();

                foreach (var item in newHandles)

                    if (!prevHandles.Contains(item))

                        otherHandles.Add(item);

                if (otherHandles.Count == 1)

                    return otherHandles[0];

                else

                    return 0;

            }

            catch (System.Exception ex)

            {

            }

            return handle;

        }

        #endregion

        #region Autocad Utils

        private List<long> GetAllBlockHandlesByName(string name)

        {

            List<long> handles = new List<long>();

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Database acCurDb = acDoc.Database;

            using (Transaction acTrans = acCurDb.TransactionManager.StartTransaction())

            {

                BlockTable gd = (BlockTable)acTrans.GetObject(acCurDb.BlockTableId, OpenMode.ForRead);

                foreach (var entry in gd)

                {

                    BlockTableRecord blockTableRecord = acTrans.GetObject(entry, OpenMode.ForRead, true) as BlockTableRecord;

                    if (blockTableRecord != null)

                    {

                        IEnumerable<BlockReference> records = AcDbLinqExtensionMethods.GetBlockReferences(blockTableRecord);

                        foreach (var item in records)

                        {

                            if (item.Name == name)

                                handles.Add(item.Handle.Value);

                        }

                    }

                }

                acTrans.Commit();

            }

            return handles;

        }

        private List<long> GetAllGroupHandles()

        {

            List<long> handles = new List<long>();

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Database acCurDb = acDoc.Database;

            using (Transaction acTrans = acCurDb.TransactionManager.StartTransaction())

            {

                DBDictionary gd = (DBDictionary)acTrans.GetObject(acCurDb.GroupDictionaryId, OpenMode.ForRead);

                foreach (var entry in gd)

                    handles.Add(entry.Value.Handle.Value);

                acTrans.Commit();

            }

            return handles;

        }

        private ObjectId GetFirstPolyLine(long groupHandle)

        {

            ObjectId result = ObjectId.Null;

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Database acCurDb = acDoc.Database;

            using (Transaction acTrans = acCurDb.TransactionManager.StartTransaction())

            {

                DBDictionary gd = (DBDictionary)acTrans.GetObject(acCurDb.GroupDictionaryId, OpenMode.ForRead);

                foreach (var entry in gd)

                {

                    if (entry.Value.Handle.Value == groupHandle)

                    {

                        Autodesk.AutoCAD.DatabaseServices.Group group = acTrans.GetObject(entry.Value, OpenMode.ForRead) as Autodesk.AutoCAD.DatabaseServices.Group;

                        foreach (var item in group.GetAllEntityIds())

                        {

                            if (acTrans.GetObject(item, OpenMode.ForRead).GetType() == typeof(Autodesk.AutoCAD.DatabaseServices.Polyline))

                            {

                                result = item;

                                break;

                            }

                        }

                        break;

                    }

                }

                acTrans.Commit();

            }

            return result;

        }

        private ObjectId GetObjectIdByHandle(long lHandle)

        {

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Database acCurDb = acDoc.Database;

            Handle handle = new Handle(lHandle);

            return acCurDb.GetObjectId(false, handle, 0);

        }

        private void SetBlockReferenceRotation(ObjectId objectId, double rotation)

        {

            Autodesk.AutoCAD.ApplicationServices.Document acDoc = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument;

            Database acCurDb = acDoc.Database;

            Editor acDocEd = acDoc.Editor;

            using (Transaction acTrans = acCurDb.TransactionManager.StartTransaction())

            {

                DBObject objDB = acTrans.GetObject(objectId, OpenMode.ForWrite);

                if (objDB != null)

                {

                    if (objDB.GetType() == typeof(BlockReference))

                    {

                        BlockReference block = objDB as BlockReference;

                        block.Rotation = rotation;

                    }

                }

                acTrans.Commit();

            }

        }

        #endregion

        #region Modeling Utils

        private void GetConnectedGroupLine(Model.Line line, List<Model.Line> group, bool isInsert)

        {

            if (!group.Contains(line))

            {

                if (isInsert)

                    group.Insert(0, line);

                else

                    group.Add(line);

                if (line.CONNECTORS[0].ConnectedObject != null && !IsExistEndBreak(line, line.CONNECTORS[0].ConnectedObject))

                {

                    object connObj = line.CONNECTORS[0].ConnectedObject;

                    if (connObj.GetType() == typeof(Model.Line) &&

                        APIDUtils.IsConnectedLine(connObj as Model.Line, line))

                        GetConnectedGroupLine(connObj as Model.Line, group, true);

                    else if (connObj.GetType() == typeof(Model.Symbol))

                    {

                        Model.Line connLine = FindOtherLineByRun(line, connObj as Symbol);

                        if (connLine != null)

                            GetConnectedGroupLine(connLine as Model.Line, group, true);

                    }

                }

                if (line.CONNECTORS[1].ConnectedObject != null && !IsExistEndBreak(line, line.CONNECTORS[1].ConnectedObject))

                {

                    object connObj = line.CONNECTORS[1].ConnectedObject;

                    if (connObj.GetType() == typeof(Model.Line) &&

                        APIDUtils.IsConnectedLine(connObj as Model.Line, line))

                        GetConnectedGroupLine(connObj as Model.Line, group, false);

                    else if (connObj.GetType() == typeof(Model.Symbol))

                    {

                        Model.Line connLine = FindOtherLineByRun(line, connObj as Symbol);

                        if (connLine != null)

                            GetConnectedGroupLine(connLine as Model.Line, group, false);

                    }

                }

            }

        }

        private void GetGroupLinePoints(List<Model.Line> groupLine, List<string> points)

        {

            for (int i = 0; i < groupLine.Count; i++)

            {

                Model.Line line = groupLine[i];

                if (i == 0)

                {

                    object connObj = line.CONNECTORS[0].ConnectedObject;

                    if (connObj != null && connObj.GetType() == typeof(Symbol))

                    {

                        Symbol connSymbol = connObj as Symbol;

                        Connector connector = connSymbol.CONNECTORS.Find(x => x.ConnectedObject == line);

                        SlopeType slopeType = APIDUtils.CalcSlope(connSymbol.X, connSymbol.Y, connector.X, connector.Y);

                        if (slopeType == SlopeType.HORIZONTAL)

                            points.Add(connSymbol.X + "," + line.Aveva.Start_Y);

                        else if (slopeType == SlopeType.VERTICAL)

                            points.Add(line.Aveva.Start_X + "," + connSymbol.Y);

                        else

                            points.Add(line.Aveva.Start_X + "," + line.Aveva.Start_Y);

                    }

                    else

                        points.Add(line.Aveva.Start_X + "," + line.Aveva.Start_Y);

                }

                else

                {

                    Model.Line prevLine = groupLine[i - 1];

                    if (line.SlopeType == SlopeType.HORIZONTAL &&

                        prevLine.SlopeType == SlopeType.VERTICAL)

                        points.Add(prevLine.Aveva.End_X + "," + line.Aveva.Start_Y);

                    else if (line.SlopeType == SlopeType.VERTICAL &&

                        prevLine.SlopeType == SlopeType.HORIZONTAL)

                        points.Add(line.Aveva.Start_X + "," + prevLine.Aveva.End_Y);

                    else

                        points.Add(line.Aveva.Start_X + "," + line.Aveva.Start_Y);

                }

                if (i == groupLine.Count - 1)

                {

                    object connObj = line.CONNECTORS[1].ConnectedObject;

                    if (connObj != null && connObj.GetType() == typeof(Symbol))

                    {

                        Symbol connSymbol = connObj as Symbol;

                        Connector connector = connSymbol.CONNECTORS.Find(x => x.ConnectedObject == line);

                        SlopeType slopeType = APIDUtils.CalcSlope(connSymbol.X, connSymbol.Y, connector.X, connector.Y);

                        if (slopeType == SlopeType.HORIZONTAL)

                            points.Add(connSymbol.X + "," + line.Aveva.End_Y);

                        else if (slopeType == SlopeType.VERTICAL)

                            points.Add(line.Aveva.End_X + "," + connSymbol.Y);

                        else

                            points.Add(line.Aveva.End_X + "," + line.Aveva.End_Y);

                    }

                    else

                        points.Add(line.Aveva.End_X + "," + line.Aveva.End_Y);

                }

            }

        }

        private bool IsExistEndBreak(object item1, object item2)

        {

            bool result = false;

            if (item1 != null && item2 != null)

            {

                EndBreak endBreak = document.EndBreaks.Find(x =>

            (APIDUtils.FindObjectByUID(document, x.OWNER) == item1 &&

            APIDUtils.FindObjectByUID(document, x.PROPERTIES.Find(y => y.ATTRIBUTE == "Connected Item").VALUE) == item2) ||

            (APIDUtils.FindObjectByUID(document, x.OWNER) == item2 &&

            APIDUtils.FindObjectByUID(document, x.PROPERTIES.Find(y => y.ATTRIBUTE == "Connected Item").VALUE) == item1));

                if (endBreak != null)

                    result = true;

            }

            return result;

        }

        private LineRun FindLineRun(Model.Line line)

        {

            List<LineRun> allLineRun = new List<LineRun>();

            foreach (var item in document.LINENUMBERS)

                allLineRun.AddRange(item.RUNS);

            foreach (var item in document.TRIMLINES)

                allLineRun.AddRange(item.RUNS);

            return allLineRun.Find(x => x.RUNITEMS.Contains(line));

        }

        private Model.Line FindOtherLineByRun(Model.Line line, Symbol symbol)

        {

            Model.Line result = null;

            LineRun run = FindLineRun(line);

            Connector connector = symbol.CONNECTORS.Find(x =>

            x.ConnectedObject != null &&

            x.ConnectedObject != line &&

            run.RUNITEMS.Contains(x.ConnectedObject) &&

            x.ConnectedObject.GetType() == typeof(Model.Line));

            if (connector != null)

                result = connector.ConnectedObject as Model.Line;

            return result;

        }

        private bool CheckAngle(DataRow symbolRow)

        {

            return int.Parse(symbolRow["Insert_Points"].ToString()).Equals(2);

        }

        private bool CheckBalloon(Symbol symbol)

        {

            bool result = false;

            string text = string.Empty;

            Hashtable symbolDefinitions = GraphicsUtility.GetSymbolDefinitions(symbol.Aveva.Name);

            if (symbolDefinitions.Count != 0)

            {

                text = symbolDefinitions[SymbolDefinition.symAflowType].ToString().ToUpper();

                if (text.Equals("VLP"))

                    text = "VPO";

                else if (text.Equals("IVP"))

                    text = "IPO";

                if (SymbolCategory.astrLabelBalloonBlockNames.Contains(text))

                    result = true;//dumbDwgUpgrade.addLabelBalloon(symbolForInsertion, objectId);

            }

            return result;

        }

        #endregion

        #region Test Source

        public void test()

        {

            //InitGUI();

            //DrawSignal("SONIC", new List<string>() { "2,100", "100,100" });

            DrawPipe("Main Pipe", new List<string>() { "2,100", "100,100" });

            Symbol symbol = new Symbol();

            symbol.Aveva = new AvevaSymbolInfo();

            symbol.Aveva.Name = "INVB";

            symbol.Aveva.X = 50;

            symbol.Aveva.Y = 100;

            DataRow[] allRows = AvevaSymbolTable.Select(string.Format("Symbol_Name = '{0}'", symbol.Aveva.Name));

            if (allRows.Length == 1)

            {

                DataRow symbolRow = allRows[0];

                bool bAngle = CheckAngle(symbolRow);

                bool bBalloon = CheckBalloon(symbol);

                bBalloon = false;

                long handle = InsertSymbol(symbol, symbolRow, bAngle, bBalloon);

            }

            Hashtable symbolDefinitions = GraphicsUtility.GetSymbolDefinitions(symbol.Aveva.Name);

            string text = symbolDefinitions[SymbolDefinition.symAflowType].ToString().ToUpper();

            if (int.Parse(symbolDefinitions[SymbolDefinition.symNoOfInsertPoints].ToString()).Equals(2))

            {

                bool bIsPromptRotate = true;

            }

            int num = int.Parse(symbolDefinitions[SymbolDefinition.symFixedOrUserDefined].ToString());

            AssociateSymbolType eAssosiateSym = AssociateSymbolType.None;

            bool bIsSymbolHasAssociation = GraphicalUtility.FindSymbolHasAssociation(text, ref eAssosiateSym);

            bool bPlaceSimpleBlocks = SymbolCategory.astrSimplePlacedBlocks.Contains(text);

            //DrawPipe("Main Pipe", new List<string>() { "2,200", "100,200" });

            //DrawPipe("Sub Pipe", new List<string>() { "50,100", "50,200" });

        }

        public static void TESTStatic()

        {

            System.Data.DataTable avevaSymbolTable = Project_DB.SelectSymbolTable();

            AutoModeling auto = new AutoModeling(null, avevaSymbolTable);

            auto.test();

        }

        #endregion

    }

    public static class AcDbLinqExtensionMethods

    {

        /// <summary>

        /// Get all references to the given BlockTableRecord, including 

        /// references to anonymous dynamic BlockTableRecords.

        /// </summary>

        public static IEnumerable<BlockReference> GetBlockReferences(

           this BlockTableRecord btr,

           OpenMode mode = OpenMode.ForRead,

           bool directOnly = true)

        {

            if (btr == null)

                throw new ArgumentNullException("btr");

            var tr = btr.Database.TransactionManager.TopTransaction;

            if (tr == null)

                throw new InvalidOperationException("No transaction");

            var ids = btr.GetBlockReferenceIds(directOnly, true);

            int cnt = ids.Count;

            for (int i = 0; i < cnt; i++)

            {

                yield return (BlockReference)

                   tr.GetObject(ids[i], mode, false, false);

            }

            if (btr.IsDynamicBlock)

            {

                BlockTableRecord btr2 = null;

                var blockIds = btr.GetAnonymousBlockIds();

                cnt = blockIds.Count;

                for (int i = 0; i < cnt; i++)

                {

                    btr2 = (BlockTableRecord)tr.GetObject(blockIds[i],

                       OpenMode.ForRead, false, false);

                    ids = btr2.GetBlockReferenceIds(directOnly, true);

                    int cnt2 = ids.Count;

                    for (int j = 0; j < cnt2; j++)

                    {

                        yield return (BlockReference)

                           tr.GetObject(ids[j], mode, false, false);

                    }

                }

            }

        }

    }

}