ecg gen: added comments in the header file

alphanumericslab · May 23, 2024 · cab8f88 · cab8f88
1 parent be364e1
commit cab8f88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/codes/ecg-image-generator/helper_functions.py b/codes/ecg-image-generator/helper_functions.py
@@ -340,4 +340,8 @@ def write_wfdb_file(ecg_frame, filename, rate, header_file, write_dir, full_mode
                 adc_gain = header.adc_gain, baseline = header.baseline, 
                 base_time = header.base_time, base_date = header.base_date, 
                 write_dir = write_dir)
-
+
+    with open(os.path.join(write_dir, tail + '.hea'), "a") as f:
+        for line in header.comments:
+            f.write("#" + line)
+            f.write("\n")