Merge branch 'unsupported_layer_check' into 'master'

added layer support checker

See merge request !15

src/main/java/de/monticore/lang/monticar/cnnarch/mxnetgenerator/CNNArch2MxNet.java

@@ -24,6 +24,9 @@ import de.monticore.io.paths.ModelPath;
 import de.monticore.lang.monticar.cnnarch.CNNArchGenerator;
 import de.monticore.lang.monticar.cnnarch._cocos.CNNArchCocos;
 import de.monticore.lang.monticar.cnnarch._symboltable.ArchitectureSymbol;
+import de.monticore.lang.monticar.cnnarch._symboltable.ArchitectureElementSymbol;
+import de.monticore.lang.monticar.cnnarch._symboltable.CompositeElementSymbol;
+import de.monticore.lang.monticar.cnnarch._symboltable.IOSymbol;
 import de.monticore.lang.monticar.cnnarch._symboltable.CNNArchCompilationUnitSymbol;
 import de.monticore.lang.monticar.cnnarch._symboltable.CNNArchLanguage;
 import de.monticore.lang.monticar.generator.FileContent;
@@ -39,11 +42,39 @@ import java.nio.file.Path;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Optional;
+import java.util.List;
 
 public class CNNArch2MxNet implements CNNArchGenerator {
 
     private String generationTargetPath;
 
+    private boolean isSupportedLayer(ArchitectureElementSymbol element, LayerSupportChecker layerChecker){
+        List<ArchitectureElementSymbol> constructLayerElemList;
+
+        if (!(element instanceof IOSymbol) && (element.getResolvedThis().get() instanceof CompositeElementSymbol))
+        {
+            constructLayerElemList = ((CompositeElementSymbol)element.getResolvedThis().get()).getElements();
+            for (ArchitectureElementSymbol constructedLayerElement : constructLayerElemList) {
+                if (!isSupportedLayer(constructedLayerElement, layerChecker)) return false;
+            }
+        }
+        if (!layerChecker.isSupported(element.toString())) {
+            Log.error("Unsupported layer " + "'" + element.getName() + "'" + " for the backend MXNET.");
+            return false;
+        }
+        else {
+            return true;
+        }
+    }
+
+    private boolean supportCheck(ArchitectureSymbol architecture){
+        LayerSupportChecker layerChecker = new LayerSupportChecker();
+        for (ArchitectureElementSymbol element : ((CompositeElementSymbol)architecture.getBody()).getElements()){
+            if(!isSupportedLayer(element, layerChecker)) return false;
+        }
+        return true;
+    }
+
     public CNNArch2MxNet() {
         setGenerationTargetPath("./target/generated-sources-cnnarch/");
     }
@@ -78,6 +109,10 @@ public class CNNArch2MxNet implements CNNArchGenerator {
         }
 
         CNNArchCocos.checkAll(compilationUnit.get());
+        if (!supportCheck(compilationUnit.get().getArchitecture())){
+            Log.error("Code generation aborted.");
+            System.exit(1);
+        }
 
         try{
             generateFiles(compilationUnit.get().getArchitecture());

src/main/java/de/monticore/lang/monticar/cnnarch/mxnetgenerator/LayerSupportChecker.java

+package de.monticore.lang.monticar.cnnarch.mxnetgenerator;
+
+import static de.monticore.lang.monticar.cnnarch.predefined.AllPredefinedLayers.*;
+import java.util.ArrayList;
+import java.util.List;
+
+
+public class LayerSupportChecker {
+
+    private List<String> unsupportedLayerList = new ArrayList();
+
+    public LayerSupportChecker() {
+        //Set the unsupported layers for the backend
+        //this.unsupportedLayerList.add(PREDEFINED_LAYER_NAME);
+    }
+
+    public boolean isSupported(String element) {
+        return !this.unsupportedLayerList.contains(element);
+    }
+}