Implement getInheritedClasses in NFApi. (#7905)

OpenModelica · Sep 16, 2021 · 9e0e8de · 9e0e8de
1 parent 1e6f45a
commit 9e0e8de
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 1 deletion.
diff --git a/OMCompiler/Compiler/NFFrontEnd/NFInstNode.mo b/OMCompiler/Compiler/NFFrontEnd/NFInstNode.mo
@@ -988,6 +988,7 @@ uniontype InstNode
   function scopePath
     input InstNode node;
     input Boolean includeRoot = false "Whether to include the root class name or not.";
+    input Boolean ignoreBaseClass = false "Ignore that a class is a base class if true.";
     output Absyn.Path path;
   algorithm
     path := match node
@@ -997,7 +998,7 @@ uniontype InstNode
       case CLASS_NODE(nodeType = it)
         then
           match it
-            case InstNodeType.BASE_CLASS() then scopePath(it.parent, includeRoot);
+            case InstNodeType.BASE_CLASS() guard not ignoreBaseClass then scopePath(it.parent, includeRoot);
             else scopePath2(node.parentScope, includeRoot, Absyn.IDENT(node.name));
           end match;
 

diff --git a/OMCompiler/Compiler/Script/Interactive.mo b/OMCompiler/Compiler/Script/Interactive.mo
@@ -9287,6 +9287,10 @@ public function getInheritedClasses
   input Absyn.Path inPath;
   output list<Absyn.Path> outPaths;
 algorithm
+  if Flags.isSet(Flags.NF_API) then
+    outPaths := NFApi.getInheritedClasses(inPath, SymbolTable.getAbsyn());
+    return;
+  end if;
 
 try
 

diff --git a/OMCompiler/Compiler/Script/NFApi.mo b/OMCompiler/Compiler/Script/NFApi.mo
@@ -805,5 +805,29 @@ algorithm
 
 end frontEndLookup_dispatch;
 
+public
+function getInheritedClasses
+  input Absyn.Path classPath;
+  input Absyn.Program program;
+  output list<Absyn.Path> extendsPaths;
+protected
+  InstNode cls_node;
+  Class cls;
+algorithm
+  (_, _, cls_node) := frontEndLookup(program, classPath);
+
+  if not InstNode.isClass(cls_node) then
+    extendsPaths := {};
+    return;
+  end if;
+
+  cls := InstNode.getClass(cls_node);
+
+  extendsPaths := match cls
+    case Class.EXPANDED_DERIVED() then {InstNode.scopePath(cls.baseClass, true, true)};
+    else list(InstNode.scopePath(e, true, true) for e in ClassTree.getExtends(Class.classTree(cls)));
+  end match;
+end getInheritedClasses;
+
   annotation(__OpenModelica_Interface="backend");
 end NFApi;
diff --git a/OMCompiler/Compiler/Stubs/NFApi.mo b/OMCompiler/Compiler/Stubs/NFApi.mo
@@ -60,5 +60,11 @@ function mkFullyQual
   output Absyn.Path qualPath = pathToQualify;
 end mkFullyQual;
 
+function getInheritedClasses
+  input Absyn.Path classPath;
+  input Absyn.Program program;
+  output list<Absyn.Path> extendsPaths;
+end getInheritedClasses;
+
   annotation(__OpenModelica_Interface="backend");
 end NFApi;