From 1876b346fea647e8284a66bb6d62c38801035cff Mon Sep 17 00:00:00 2001
From: gdk <gab.dark.100@gmail.com>
Date: Sun, 13 Oct 2019 03:02:07 -0300
Subject: Initial work

---
 .../StructuredIr/AstBlockVisitor.cs                | 68 ++++++++++++++++++++++
 1 file changed, 68 insertions(+)
 create mode 100644 Ryujinx.Graphics.Shader/StructuredIr/AstBlockVisitor.cs

(limited to 'Ryujinx.Graphics.Shader/StructuredIr/AstBlockVisitor.cs')

diff --git a/Ryujinx.Graphics.Shader/StructuredIr/AstBlockVisitor.cs b/Ryujinx.Graphics.Shader/StructuredIr/AstBlockVisitor.cs
new file mode 100644
index 00000000..10d5dce0
--- /dev/null
+++ b/Ryujinx.Graphics.Shader/StructuredIr/AstBlockVisitor.cs
@@ -0,0 +1,68 @@
+using System;
+using System.Collections.Generic;
+
+using static Ryujinx.Graphics.Shader.StructuredIr.AstHelper;
+
+namespace Ryujinx.Graphics.Shader.StructuredIr
+{
+    class AstBlockVisitor
+    {
+        public AstBlock Block { get; private set; }
+
+        public class BlockVisitationEventArgs : EventArgs
+        {
+            public AstBlock Block { get; }
+
+            public BlockVisitationEventArgs(AstBlock block)
+            {
+                Block = block;
+            }
+        }
+
+        public event EventHandler<BlockVisitationEventArgs> BlockEntered;
+        public event EventHandler<BlockVisitationEventArgs> BlockLeft;
+
+        public AstBlockVisitor(AstBlock mainBlock)
+        {
+            Block = mainBlock;
+        }
+
+        public IEnumerable<IAstNode> Visit()
+        {
+            IAstNode node = Block.First;
+
+            while (node != null)
+            {
+                // We reached a child block, visit the nodes inside.
+                while (node is AstBlock childBlock)
+                {
+                    Block = childBlock;
+
+                    node = childBlock.First;
+
+                    BlockEntered?.Invoke(this, new BlockVisitationEventArgs(Block));
+                }
+
+                // Node may be null, if the block is empty.
+                if (node != null)
+                {
+                    IAstNode next = Next(node);
+
+                    yield return node;
+
+                    node = next;
+                }
+
+                // We reached the end of the list, go up on tree to the parent blocks.
+                while (node == null && Block.Type != AstBlockType.Main)
+                {
+                    BlockLeft?.Invoke(this, new BlockVisitationEventArgs(Block));
+
+                    node = Next(Block);
+
+                    Block = Block.Parent;
+                }
+            }
+        }
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3