From 8cc4d675f0722758fc40d68e2ab42754d8219d90 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?DALL=27AGNOLA=20C=C3=A9dric?= <ced.dall@gmail.com>
Date: Tue, 29 May 2018 16:14:01 +0200
Subject: [PATCH] Update SclParser.java

Add method to retrieve all IEDs
---
 .../src/com/libiec61850/scl/SclParser.java           | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

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<IED> 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
+}