ctf: Make ctf2 trace have environment

Change-Id: If58b118283f9a5e3778e832edd94e358c23cded8
Signed-off-by: Matthew Khouzam <[email protected]>

Author: MatthewKhouzam

ctf/org.eclipse.tracecompass.ctf.core/src/org/eclipse/tracecompass/internal/ctf/core/event/metadata/IOStructGen.java

@@ -20,7 +20,10 @@
 
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.LinkedHashMap;
 import java.util.List;
+import java.util.Map;
+import java.util.Map.Entry;
 import java.util.stream.Collectors;
 
 import org.eclipse.jdt.annotation.NonNull;
@@ -46,6 +49,7 @@
 import org.eclipse.tracecompass.internal.ctf.core.utils.JsonMetadataStrings;
 
 import com.google.common.collect.Iterables;
+import com.google.gson.JsonElement;
 
 /**
  * IOStructGen
@@ -151,6 +155,15 @@ private void parseRoot(ICTFMetadataNode root) throws ParseException {
                     throw new ParseException("Only one trace block is allowed"); //$NON-NLS-1$
                 }
                 traceNode = child;
+                if (child instanceof JsonTraceMetadataNode) {
+                    JsonTraceMetadataNode node = (JsonTraceMetadataNode) child;
+                    Map<String, String> env = new LinkedHashMap<>();
+                    for( Entry<String, JsonElement> entry : node.getEnvironment().entrySet()) {
+                        env.put(entry.getKey(), entry.getValue().toString());
+                    }
+                    fTrace.setEnvironment(env);
+
+                }
                 parseTrace(traceNode);
             } else if (CTFParser.tokenNames[CTFParser.STREAM].equals(type) || JsonMetadataStrings.FRAGMENT_DATA_STREAM.equals(type)) {
                 StreamParser.INSTANCE.parse(child, new StreamParser.Param(fTrace, fRoot));