main file generator

Author: EhsanKhodadad

examples/patmos/s4noc_fed_lf/build.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
+
 LF_MAIN=S4NoCFedLF
 BIN_DIR=bin
 CC=patmos-clang
@@ -32,6 +33,9 @@ for i in $(seq 1 $N); do
     A_FILES="$A_FILES ./$LF_MAIN/r$i/bin/$LF_MAIN.a"
 done
 
+chmod +x ./gen_main.sh
+./gen_main.sh N=$N
+
 $CC -O2 -Wall -Wextra main.c $A_FILES -o $BIN_DIR/$LF_MAIN
 
 read -n 1 -t 10 -p "Choose action: [e]mulate or [f]pga? (default: e) " action

examples/patmos/s4noc_fed_lf/gen_main.sh

@@ -0,0 +1,72 @@
+#!/bin/bash
+
+# Number of federates
+N=2
+rm -rf main.c
+
+cat >> main.c <<EOF
+#include <stdio.h>
+#include <pthread.h>
+
+EOF
+
+# Generate include headers for each federate
+for i in $(seq 1 $N); do
+cat >> main.c <<EOF
+#include "S4NoCFedLF/r$i/src-gen/S4NoCFedLF/r$i/lf_start.h"
+EOF
+done
+
+cat >> main.c <<EOF
+
+void* f1_thread(void* arg) {
+    printf("Starting federate 1 on core/thread 1\\n");
+    lf_start();
+    return NULL;
+}
+EOF
+
+# Generate thread functions for each federate
+for i in $(seq 2 $N); do
+cat >> main.c <<EOF
+
+void* f${i}_thread(void* arg) {
+    printf("Starting federate $i on core/thread $i\\n");
+    lf_start_$i();
+    return NULL;
+}
+EOF
+done
+
+# Generate main() function
+cat >> main.c <<EOF
+
+int main(void) {
+    pthread_t threads[$N];
+    printf("Starting S4NOC Federated LF Example\\n");
+EOF
+
+for i in $(seq 1 $N); do
+cat >> main.c <<EOF
+    pthread_create(&threads[$((i-1))], NULL, f${i}_thread, NULL);
+EOF
+done
+
+cat >> main.c <<EOF
+
+    printf("Threads created for federates.\\n");
+
+EOF
+
+for i in $(seq 1 $N); do
+cat >> main.c <<EOF
+    pthread_join(threads[$((i-1))], NULL);
+EOF
+done
+
+cat >> main.c <<EOF
+
+    printf("All federates finished.\\n");
+    return 0;
+}
+EOF

examples/patmos/s4noc_fed_lf/main.c

@@ -4,33 +4,29 @@
 #include "S4NoCFedLF/r1/src-gen/S4NoCFedLF/r1/lf_start.h"
 #include "S4NoCFedLF/r2/src-gen/S4NoCFedLF/r2/lf_start.h"
 
-
 void* f1_thread(void* arg) {
     printf("Starting federate 1 on core/thread 1\n");
-    lf_start(); 
+    lf_start();
     return NULL;
 }
 
 void* f2_thread(void* arg) {
     printf("Starting federate 2 on core/thread 2\n");
-    lf_start_2(); 
+    lf_start_2();
     return NULL;
 }
 
 int main(void) {
-    pthread_t thread1, thread2;
+    pthread_t threads[2];
     printf("Starting S4NOC Federated LF Example\n");
-    
-    // Create threads for each federate
-    pthread_create(&thread1, NULL, f1_thread, NULL);
-    pthread_create(&thread2, NULL, f2_thread, NULL);
+    pthread_create(&threads[0], NULL, f1_thread, NULL);
+    pthread_create(&threads[1], NULL, f2_thread, NULL);
 
     printf("Threads created for federates.\n");
 
-    // Wait for both federates to finish
-    pthread_join(thread1, NULL);
-    pthread_join(thread2, NULL);
+    pthread_join(threads[0], NULL);
+    pthread_join(threads[1], NULL);
 
     printf("All federates finished.\n");
-    return 0; 
-}
+    return 0;
+}