Djalim Simaila
1c1a9bac94
🐛 fix(data_processing.py): fix formula for R_V_scan to include parentheses for correct order of operations 🐛 fix(output.py): add delta_z parameter to generate_headers function to correctly display the delta_z value in the output file ✨ feat(output.py): add delta_z value to output file headers if add_headers setting is enabled ✨ feat(MainWindow.py): add delta_z value to output file headers if add_headers setting is enabled ✨ feat(DiscreteDataWorker.py): add delta_z value to output file headers if add_headers setting is enabled ✨ feat(RawDataWorker.py): add delta_z value to output file headers if add_headers setting is enabled The changes in data_processing.py fix a variable name and a formula to correctly calculate the R_max and R_V_scan values.
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
"""
|
|
Created on Mon Apr 17 2023
|
|
@name: output.py
|
|
@desc: Output file formatting and saving
|
|
@auth: Djalim Simaila
|
|
@e-mail: djalim.simaila@inrae.fr
|
|
"""
|
|
import datetime
|
|
from utils.settings.SettingManager import SettingManager
|
|
|
|
def generate_headers(filename:str,delta_z:float) -> str:
|
|
"""
|
|
"""
|
|
headers = f"""##############################
|
|
Analyse Morphologique
|
|
version :{SettingManager.get_instance().get_setting("version")}
|
|
|
|
filename : {filename}
|
|
date : {datetime.datetime.now().strftime("%d/%m/%Y %H:%M:%S")}
|
|
discretisation : {SettingManager.get_instance().get_setting("discretisation_method")}
|
|
delta_z : {delta_z}
|
|
had been veticalised : {"yes" if SettingManager.get_instance().get_setting("should_verticalise") else "no"}
|
|
##############################
|
|
"""
|
|
return headers
|
|
|
|
|
|
def format_data(data:dict, separator:str, selected_columns:list = None) -> str:
|
|
"""
|
|
Format the data to be saved in the output file.
|
|
|
|
:param data: Data to be formatted
|
|
:param selected_columns: Columns to be saved (in the specified order)
|
|
:param separator: Separator of the columns
|
|
:return: Formatted data
|
|
|
|
Example:
|
|
>>> data = {
|
|
... 'col1': [1, 2, 3],
|
|
... 'col2': [4, 5, 6],
|
|
... 'col3': [7, 8, 9]
|
|
... }
|
|
>>> format_data(data, separator=';')
|
|
'col1;col2;col3
|
|
1 ;4 ;7
|
|
2 ;5 ;8
|
|
3 ;6 ;9
|
|
'
|
|
"""
|
|
# Get the setting for the output file
|
|
is_prettier = SettingManager.get_instance().get_setting('pretiffy_output_file')
|
|
output = ''
|
|
# If the columns are not specified, we take all the columns from dict
|
|
if selected_columns is None:
|
|
selected_columns = list(data.keys())
|
|
|
|
# Write the columns headers
|
|
for column_name in selected_columns:
|
|
if is_prettier:
|
|
output += column_name.ljust(len(column_name) if len(column_name) > 8 else 9 ) + separator
|
|
else:
|
|
output += column_name + separator
|
|
output += '\n'
|
|
# Write the columns values
|
|
for i in range(len(data[selected_columns[0]])):
|
|
for column in selected_columns:
|
|
if is_prettier:
|
|
output += str(data[column][i]).ljust(len(column) if len(column) > 8 else 9 ) + separator
|
|
else:
|
|
output += str(data[column][i]) + separator
|
|
output += '\n'
|
|
return output
|
|
|
|
def save_output_file(output_file:str, content:str):
|
|
"""
|
|
Save the output file.
|
|
|
|
:param output_file: Path to the output file
|
|
:param content: Content of the output file
|
|
"""
|
|
with open(output_file, 'w',encoding='utf-8') as file:
|
|
file.write(content)
|