from convert import *
from joblib import dump, load

# a = generate_damage_files_index(
#     num_damage=6, file_index_start=1, col=5, base_path="D:/thesis/data/dataset_A"
# )

b = generate_damage_files_index(
    num_damage=6,
    file_index_start=1,
    col=5,
    base_path="D:/thesis/data/dataset_B",
    prefix="zzzBD",
)
# data_A = DataProcessor(file_index=a)
# # data.create_vector_column(overwrite=True)
# data_A.create_limited_sensor_vector_column(overwrite=True)
# data_A.export_to_csv("D:/thesis/data/converted/raw")

data_B = DataProcessor(file_index=b)
# data.create_vector_column(overwrite=True)
data_B.create_limited_sensor_vector_column(overwrite=True)
data_B.export_to_csv("D:/thesis/data/converted/raw_B")
# a = load("D:/cache.joblib")
# breakpoint()