defineFunction support in Julia

hetalang · Jun 20, 2024 · 60aefbd · 60aefbd
1 parent e68fce7
commit 60aefbd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/julia-export/index.js b/src/julia-export/index.js
@@ -36,7 +36,7 @@ class JuliaExport extends AbstractExport {
   }
   // skipVersionCode means that the version will not be printed in output
   // this is required for autotests
-  makeText(skipVersionCode = false){
+  makeText(skipVersionCode = false) {
     //let logger = this._container.logger;
     // create image for multiple namespaces
     let nsImages = this.selectedNamespaces()
@@ -46,7 +46,8 @@ class JuliaExport extends AbstractExport {
     let image = {
       builderVersion: skipVersionCode ? '*' : pkg.version,
       options: this,
-      nsImages
+      nsImages,
+      functionDefArray: [...this._container.functionDefStorage.values()].filter((fd) => !fd.isCore)
     };
     let modelContent = this.getModelCode(image);
     let runContent = this.getRunCode(image);

diff --git a/src/templates/julia-model.jl.njk b/src/templates/julia-model.jl.njk
@@ -17,7 +17,13 @@
 =#
 
 __platform__ = (function()
-{% for img in nsImages %}
+{% for fd in functionDefArray %}
+function {{ fd.id }}({% for a in fd.arguments %}{{ a }}, {% endfor %}__args__...)
+  return {{ fd.math.toJuliaString() }}
+end
+{%- endfor %}
+
+{%- for img in nsImages %}
 {%- set timeScales = img.namespace.selectByClassName('TimeScale') | exclude2('id', 't') -%}
 {% set id = img.namespace.spaceName %}
 ### create default constants