From b2684c23f67328970d93d9f52270a1024600ae20 Mon Sep 17 00:00:00 2001
From: nuluh <dam.ar@outlook.com>
Date: Tue, 27 Aug 2024 10:11:39 +0700
Subject: [PATCH] feat(script): Add zero-padding to CSV filenames to include
 sensors number

---
 generate_dummy_data.py | 39 +++++++++++++++++++++------------------
 1 file changed, 21 insertions(+), 18 deletions(-)

diff --git a/generate_dummy_data.py b/generate_dummy_data.py
index e764ed8..4283904 100644
--- a/generate_dummy_data.py
+++ b/generate_dummy_data.py
@@ -16,8 +16,10 @@ os.makedirs(processed_path, exist_ok=True)
 # Define the number of zeros to pad
 num_damages = 5
 num_tests = 10
+num_sensors = 2
 damage_pad = len(str(num_damages))
 test_pad = len(str(num_tests))
+sensor_pad = len(str(num_sensors))
 
 for damage in range(1, num_damages + 1):  # 5 Damage levels starts from 1
     damage_folder = f"DAMAGE_{damage:0{damage_pad}}"
@@ -25,23 +27,24 @@ for damage in range(1, num_damages + 1):  # 5 Damage levels starts from 1
     os.makedirs(damage_path, exist_ok=True)
 
     for test in range(1, 11):  # 10 Tests per damage level
-        # Filename for the CSV
-        csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}.csv"
-        csv_path = os.path.join(damage_path, csv_filename)
+        for sensor in range(1, 3):  # 2 Sensors per test
+            # Filename for the CSV
+            csv_filename = f"D{damage:0{damage_pad}}_TEST{test:0{test_pad}}_{sensor:0{sensor_pad}}.csv"
+            csv_path = os.path.join(damage_path, csv_filename)
+            
+            # Generate dummy data
+            num_rows = 10
+            start_time = datetime.now()
+            timestamps = [start_time + timedelta(seconds=i*0.0078125) for i in range(num_rows)]
+            values = np.random.randn(num_rows)  # Random float values
 
-        # Generate dummy data
-        num_rows = 10
-        start_time = datetime.now()
-        timestamps = [start_time + timedelta(seconds=i*0.0078125) for i in range(num_rows)]
-        values = np.random.randn(num_rows)  # Random float values
+            # Create DataFrame
+            df = pd.DataFrame({
+                "Time": timestamps,
+                "Value": values
+            })
 
-        # Create DataFrame
-        df = pd.DataFrame({
-            "Time": timestamps,
-            "Value": values
-        })
-
-        # Save the CSV file with a custom header
-        with open(csv_path, 'w') as file:
-            file.write('sep=,\n')  # Writing the separator hint
-            df.to_csv(file, index=False)
+            # Save the CSV file with a custom header
+            with open(csv_path, 'w') as file:
+                file.write('sep=,\n')  # Writing the separator hint
+                df.to_csv(file, index=False)