From 3be2d4a39d1e9573256d2367d59075573c907981 Mon Sep 17 00:00:00 2001
From: Armin Zavada <zavadaarmin@edu.bme.hu>
Date: Tue, 15 Oct 2024 22:09:12 +0200
Subject: [PATCH] Implemented workflow template

---
 .../similarity/SimilarityWorkflow.java        | 96 -------------------
 similarity/src/main/jinja/workflow.java.j2    | 56 ++++++++++-
 2 files changed, 55 insertions(+), 97 deletions(-)
 delete mode 100644 similarity/src/main/java/hu/bme/mit/ase/shingler/similarity/SimilarityWorkflow.java

diff --git a/similarity/src/main/java/hu/bme/mit/ase/shingler/similarity/SimilarityWorkflow.java b/similarity/src/main/java/hu/bme/mit/ase/shingler/similarity/SimilarityWorkflow.java
deleted file mode 100644
index 8ba58e5..0000000
--- a/similarity/src/main/java/hu/bme/mit/ase/shingler/similarity/SimilarityWorkflow.java
+++ /dev/null
@@ -1,96 +0,0 @@
-package hu.bme.mit.ase.shingler.similarity;
-
-import hu.bme.mit.ase.shingler.workflow.impl.*;
-import hu.bme.mit.ase.shingler.workflow.lib.*;
-
-public class SimilarityWorkflow extends Workflow<Double> {
-
-    // Input pin declarations
-
-    public final Pin<String> tokenizerAInput = new Pin<>();
-    public final Pin<String> tokenizerBInput = new Pin<>();
-
-    // Parameter declarations
-
-    private final boolean granularity;
-    private final int size;
-
-    public SimilarityWorkflow(boolean granularity, int size) {
-        this.granularity = granularity;
-        this.size = size;
-    }
-
-    @Override
-    protected void initialize() {
-        // Worker declarations
-
-        var tokenizerA = new TokenizerWorker(granularity);
-        var tokenizerB = new TokenizerWorker(granularity);
-
-        var shinglerA = new ShinglerWorker(size);
-        var shinglerB = new ShinglerWorker(size);
-
-        var vectorAA = new VectorMultiplierWorker();
-        var vectorAB = new VectorMultiplierWorker();
-        var vectorBB = new VectorMultiplierWorker();
-
-        var cosine = new CosineSimilarityWorker();
-
-        // Adding all workers
-
-        addWorker(tokenizerA);
-        addWorker(tokenizerB);
-        addWorker(shinglerA);
-        addWorker(shinglerB);
-        addWorker(vectorAA);
-        addWorker(vectorAB);
-        addWorker(vectorBB);
-        addWorker(cosine);
-
-        // Set output pin
-
-        setOutputPin(cosine.outputPin);
-
-        // Input pin channel declarations
-
-        var input_ta_input = new Channel<>(tokenizerAInput, tokenizerA.inputPin);
-        var input_tb_input = new Channel<>(tokenizerBInput, tokenizerB.inputPin);
-
-        // Channel declarations
-
-        var ta_sa = new Channel<>(tokenizerA.outputPin, shinglerA.inputPin);
-        var tb_sb = new Channel<>(tokenizerB.outputPin, shinglerB.inputPin);
-
-        var sa_vaa_a = new Channel<>(shinglerA.outputPin, vectorAA.aPin);
-        var sa_vaa_b = new Channel<>(shinglerA.outputPin, vectorAA.bPin);
-        var sa_vab_a = new Channel<>(shinglerA.outputPin, vectorAB.aPin);
-
-        var sb_vab_b = new Channel<>(shinglerB.outputPin, vectorAB.bPin);
-        var sb_vbb_a = new Channel<>(shinglerB.outputPin, vectorBB.aPin);
-        var sb_vbb_b = new Channel<>(shinglerB.outputPin, vectorBB.bPin);
-
-        var vaa_c_aa = new Channel<>(vectorAA.outputPin, cosine.aaPin);
-        var vab_c_ab = new Channel<>(vectorAB.outputPin, cosine.abPin);
-        var vbb_c_bb = new Channel<>(vectorBB.outputPin, cosine.aaPin);
-
-        // Add input pin channels
-
-        addChannel(input_ta_input);
-        addChannel(input_tb_input);
-
-        // Add channels
-
-        addChannel(ta_sa);
-        addChannel(tb_sb);
-        addChannel(sa_vaa_a);
-        addChannel(sa_vaa_b);
-        addChannel(sa_vab_a);
-        addChannel(sb_vab_b);
-        addChannel(sb_vbb_a);
-        addChannel(sb_vbb_b);
-        addChannel(vaa_c_aa);
-        addChannel(vab_c_ab);
-        addChannel(vbb_c_bb);
-    }
-
-}
diff --git a/similarity/src/main/jinja/workflow.java.j2 b/similarity/src/main/jinja/workflow.java.j2
index 14efe6e..d68d33a 100644
--- a/similarity/src/main/jinja/workflow.java.j2
+++ b/similarity/src/main/jinja/workflow.java.j2
@@ -1 +1,55 @@
-// TODO: create Workflow template!
+package hu.bme.mit.ase.shingler.similarity;
+
+import hu.bme.mit.ase.shingler.workflow.impl.*;
+import hu.bme.mit.ase.shingler.workflow.lib.*;
+
+public class {{ name }}Workflow extends Workflow<Double> {
+
+    // Input pin declarations
+    {%- for in_pin in inPins %}
+    public final Pin<String> {{ in_pin.name }} = new Pin<>();
+    {%- endfor %}
+
+    // Parameter declarations
+    {%- for param in parameters %}
+    private final {{ param.type }} {{ param.name }};
+    {%- endfor %}
+
+    public {{ name }}Workflow({% for param in parameters %}{{ param.type }} {{ param.name }}{% if not loop.last %}, {% endif %}{% endfor %}) {
+        {%- for param in parameters %}
+        this.{{ param.name }} = {{ param.name }};
+        {%- endfor %}
+    }
+
+    @Override
+    protected void initialize() {
+        {%- for worker in workers %}
+        var {{ worker.name }} = new {{ worker.type }}Worker(
+                {%- if worker.arguments -%}
+                {%- for argument in worker.arguments -%}
+                {{ argument }}
+                {%- if not loop.last -%}, {% endif -%}
+                {%- endfor -%}
+                {%- endif -%}
+        );
+        {%- endfor %}
+        {% for worker in workers %}
+        addWorker({{ worker.name }});
+        {%- endfor %}
+
+        setOutputPin({{ outPin.worker }}.outputPin);
+        {% for in_pin in inPins %}
+        var input{{ in_pin.name | capitalize }} = new Channel<>({{ in_pin.name }}, {{ in_pin.worker }}.{{ in_pin.pin }}Pin);
+        {%- endfor %}
+        {% for in_pin in inPins %}
+        addChannel(input{{ in_pin.name | capitalize }});
+        {%- endfor %}
+        {% for channel in channels %}
+        var {{ channel.name }} = new Channel<>({{ channel.fromWorker }}.outputPin, {{ channel.toWorker }}.{{ channel.toPin }}Pin);
+        {%- endfor %}
+        {% for channel in channels %}
+        addChannel({{ channel.name }});
+        {%- endfor %}
+    }
+
+}