support distributor at function out

Author: jogibear9988

TiaCodeGen.Tests/SampleTests.cs

@@ -219,5 +219,33 @@ public void TestCallWithTOn()
             block.Interface = TestInterface;
             var xml = block.GetCode();
         }
+
+        [Test]
+        public void TestCallWithTPAndDistributor()
+        {
+            var codeblock = new CodeBlock() { Safety = false };
+
+            var nw = new Network("Test2", "Test2en");
+            var and =
+                new And(
+                new Signal("#aaa"),
+                new TPCall("PulseStartPrint",
+                    pt: new Signal("T#100ms", SignalType.ConstantTime),
+                    q: new Distributor(
+                        new Coil(new Signal("#bbbb")),
+                        new RCoil(new Signal("#cccc")),
+                        new SCoil(new Signal("#dddd"))
+                    )
+                )
+            );
+
+            nw.Add(and);
+
+            codeblock.Add(nw);
+
+            var block = new Block("Test", "blabla", codeblock);
+            block.Interface = TestInterface;
+            var xml = block.GetCode();
+        }
     }
 }

TiaCodegen/CodeGen/KopCodeHelper.cs

@@ -508,14 +508,33 @@ private void AddWires(IOperationOrSignal op)
                     {
                         _sb.AppendLine("<Wire UId=\"" + _currentId + "\">" + (debug ?? ("<!-- Wire 4 FunctionCall -->")));
                         _sb.AppendLine("<NameCon UId=\"" + op.OperationId + "\" Name=\"" + intf.Key + "\" />");
-                        if (sng is Signal)
+                        if (sng is Distributor db)
                         {
-                            _sb.AppendLine("<IdentCon UId=\"" + ((Signal)sng).SignalId + "\" />" + "  <!-- " + ((Signal)sng).Name + " -->");
+                            _sb.AppendLine("<!-- Distributor -->");
+                            foreach (var c in db.Children)
+                            {
+                                if (c is Signal)
+                                {
+                                    _sb.AppendLine("<IdentCon UId=\"" + ((Signal)c).SignalId + "\" />" + "  <!-- " + ((Signal)c).Name + " -->");
+                                }
+                                else
+                                {
+                                    var inName = c is FunctionCall ? ((c is InRangeCall || c is OutRangeCall) ? "pre" : "en") : "in";
+                                    _sb.AppendLine("<NameCon UId=\"" + c.OperationId + "\" Name=\"" + inName + "\" />");
+                                }
+                            }
                         }
                         else
-                        {
-                            var inName = sng is FunctionCall ? ((sng is InRangeCall || sng is OutRangeCall) ? "pre" : "en") : "in";
-                            _sb.AppendLine("<NameCon UId=\"" + sng.OperationId + "\" Name=\"" + inName + "\" />");
+                        {                            
+                            if (sng is Signal)
+                            {
+                                _sb.AppendLine("<IdentCon UId=\"" + ((Signal)sng).SignalId + "\" />" + "  <!-- " + ((Signal)sng).Name + " -->");
+                            }
+                            else
+                            {
+                                var inName = sng is FunctionCall ? ((sng is InRangeCall || sng is OutRangeCall) ? "pre" : "en") : "in";
+                                _sb.AppendLine("<NameCon UId=\"" + sng.OperationId + "\" Name=\"" + inName + "\" />");
+                            }
                         }
                         _sb.AppendLine("</Wire>");
                         _currentId++;