AdventureSample - support DIM of 2 arrays

brian-dot-net · Apr 22, 2018 · d31c296 · d31c296
1 parent c61c6bf
commit d31c296
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/projects/AdventureSample/src/GWBas2CS/BasicVisitor.cs b/projects/AdventureSample/src/GWBas2CS/BasicVisitor.cs
@@ -98,7 +98,7 @@ public void Many(string name, BasicExpression[] list)
                     this.AddPrint(list[0]);
                     break;
                 case "Dim":
-                    this.AddDim(list[0]);
+                    this.AddDim(list);
                     break;
                 default:
                     throw new NotImplementedException("Many:" + name);
@@ -233,6 +233,14 @@ private void AddCls()
             this.methods.Add(name, clsMethod);
         }
 
+        private void AddDim(BasicExpression[] exprs)
+        {
+            foreach (BasicExpression expr in exprs)
+            {
+                this.AddDim(expr);
+            }
+        }
+
         private void AddDim(BasicExpression expr)
         {
             ExpressionNode node = new ExpressionNode(this.generator, this.vars, this.methods);

diff --git a/projects/AdventureSample/test/GWBas2CS.Test/DimStatements.cs b/projects/AdventureSample/test/GWBas2CS.Test/DimStatements.cs
@@ -44,5 +44,24 @@ private bool Main()
 
             actual.Should().Match(Expected);
         }
+
+        [Fact]
+        public void TwoArrays()
+        {
+            const string Input = @"10 DIM A(1),B$(2)";
+            const string Expected = @"*
+    private bool Main()
+    {
+        this.Init();
+        DIM1_na(out A_na, 1);
+        DIM1_sa(out B_sa, 2);
+        return false;
+    }
+*";
+
+            string actual = Test.Translate("MyProg", Input);
+
+            actual.Should().Match(Expected);
+        }
     }
 }