From b09fd62041cfd88306d1aeb26fdc0f1aec4f738f Mon Sep 17 00:00:00 2001
From: Yi Zhang <zhangyiss@icloud.com>
Date: Sat, 20 Nov 2021 16:23:07 +0800
Subject: [PATCH] update eemd_example

---
 .gitignore                  |  1 +
 src/examples/eemd_example.c | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/.gitignore b/.gitignore
index caf5815..69eefdb 100644
--- a/.gitignore
+++ b/.gitignore
@@ -51,6 +51,7 @@ Module.symvers
 Mkfile.old
 dkms.conf
 
+.vscode/
 build/
 .DS_Store
 *.sh
\ No newline at end of file
diff --git a/src/examples/eemd_example.c b/src/examples/eemd_example.c
index fd8ae87..0e882fe 100644
--- a/src/examples/eemd_example.c
+++ b/src/examples/eemd_example.c
@@ -56,6 +56,8 @@ int main(void) {
 	}
 	// Write output to file
 	FILE* fp = fopen(outfile, "w");
+	// Output in lines
+	/*
 	for (size_t j=0; j<N; j++) {
 		fprintf(fp, "%f ", inp[j]);
 	}
@@ -66,6 +68,15 @@ int main(void) {
 		}
 		fprintf(fp, "\n");
 	}
+	*/
+	// Output in columns
+	for (size_t j=0; j<N; j++) {
+		fprintf(fp, "%f ", inp[j]);
+		for (size_t i=0; i<M; i++) {
+			fprintf(fp, "%f ", outp[i*N+j]);
+		}
+		fprintf(fp, "\n");
+	}
 	printf("Done!\n");
 	// Cleanup
 	fclose(fp);