diff --git a/tools/model_generator/src/com/libiec61850/scl/SclParser.java b/tools/model_generator/src/com/libiec61850/scl/SclParser.java index 4ed22e51..4faa143b 100644 --- a/tools/model_generator/src/com/libiec61850/scl/SclParser.java +++ b/tools/model_generator/src/com/libiec61850/scl/SclParser.java @@ -30,6 +30,7 @@ import java.io.InputStream; import java.util.LinkedList; import java.util.List; import java.util.Stack; +import java.util.stream.Stream; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; @@ -118,6 +119,15 @@ public class SclParser { public Communication getCommunication() { return communication; } + + public Stream getIeds() { + return ieds.stream(); + } + + public int getIedsCount() { + return ieds.size(); + } + public static Document parseXmlWithLineNumberInformation(InputStream xmlInputStream) throws IOException, SAXException { final Document xmlDocument; @@ -314,4 +324,4 @@ public class SclParser { return null; } -} \ No newline at end of file +}