ESCSP/es_analyze_script.py at main · ARISA-Lab-LLC/ESCSP · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#es_analyze.py
##########################################################
#Import Libaries
from escsp import *
import os
import datetime

##########################################################
#Set global variables
Volume="/media/tracy/ESCSPA00"
#Volume="/Volumes/ESCSPA00"
Annular=True
Total=False
verbose=1# False
Test= False

#Annular Eclipse
if Annular:
    top_folder="Annular_Analysis_Data"
    eclipse_type = "Annular"
    top_level_directory=os.path.join(Volume,top_folder)
    eclipse_data_csv=os.path.join(top_level_directory, "Spreadsheets", "2023_Annular_Eclipse_eclipse_data_all.csv")
    column_text="ESID,"
    column_text=column_text+"All Frequencies: 10/12/2023 V^2,"
    column_text=column_text+"All Frequencies: 10/13/2023 V^2,"
    column_text=column_text+"All Frequencies: Average V^2 of 10/12 & 10/13,"
    column_text=column_text+"All Frequencies: Standard Deviation 10/12 & 10/13,"
    column_text=column_text+"All Frequencies: Annular Eclipse 10/12/2023 V^2,"
    column_text=column_text+"All Frequencies: Difference /Standard Deviation (Annular Eclipse vs Average),"
    column_text=column_text+"Cricket Frequencies: 10/12/2023 PSD,"
    column_text=column_text+"Cricket Frequencies: 10/13/2023 PSD,"
    column_text=column_text+"Cricket Frequencies: Average V^2 of 10/12 & 10/13,"
    column_text=column_text+ "Cricket Frequencies: Standard Deviation 10/12 & 10/13,"
    column_text=column_text+"Cricket Frequencies: Eclipse 10/12/2023 V^2,"
    column_text=column_text+"Cricket Frequencies: Difference / Standard Deviation (Annular Eclipse vs Average) >2 Significant Increase <-2 Significant Increase"
    column_text=column_text+"\n"

#Total Eclipse
if Total:
    top_folder="Total_Analysis_Data"
    eclipse_type = "Total"
    top_level_directory=os.path.join(Volume,top_folder)
    eclipse_data_csv=os.path.join(top_level_directory, "Spreadsheets", "2024_Total_Eclipse_eclipse_data_all.csv")
    column_text="ESID,"
    column_text=column_text+"All Frequencies: 4/6/2024 V^2,"
    column_text=column_text+"All Frequencies: 4/7/2024 V^2,"
    column_text=column_text+"All Frequencies: Average V^2 of 4/6 & 4/7,"
    column_text=column_text+"All Frequencies: Standard Deviation 4/6 & 4/7,"
    column_text=column_text+"All Frequencies: Eclipse 4/8/2024 V^2,"
    column_text=column_text+"All Frequencies: Difference / Standard Deviation (Eclipse vs Average),"
    column_text=column_text+"Cricket Frequencies: 4/6/2024 PSD,"
    column_text=column_text+"Cricket Frequencies: 4/7/2024 PSD,"
    column_text=column_text+"Cricket Frequencies: Average V^2 of 4/6 & 4/7,"
    column_text=column_text+ "Cricket Frequencies: Standard Deviation 4/6 & 4/7,"
    column_text=column_text+"Cricket Frequencies: Eclipse 4/8/2024 V^2,"
    column_text=column_text+"Cricket Frequencies: Difference / Standard Deviation (Eclipse vs Average) >2 Significant Increase <-2 Significant Increase"
    column_text=column_text+"\n"


plots_folder=os.path.join(top_level_directory, "Plots")
outfile=os.path.join(top_level_directory,"Spreadsheets","ES_data_test.csv")
folders = get_es_folder_list(top_level_directory,  split = 1)
filelist=outfile

#folders=[os.path.join(top_level_directory,"ESID#201_TotalEclipse_AudioMothTimeChime_Split"),
#         os.path.join(top_level_directory,"ESID#232_TotalEclipse_AudioMothTimeChime_Split")
#         ]

##########################################################
if verbose : print(folders)


counter = 0
#csv files for (V**2 diff)/std
#Make a unique PSD analysis file
# Get the current time
current_time = datetime.datetime.now()
# Format the current time as YYYY_MM_DD_hh_mm
formatted_time = current_time.strftime("%Y_%m_%d_%H_%M")
psd_file=os.path.join(top_level_directory, "Spreadsheets", formatted_time+"_PSD_analysis.csv")
if os.path.exists(psd_file):
     os.system("rm "+psd_file)

f = open(psd_file, "w")
f.write(column_text)
#f.close()
##########################################################
if Test:
    folders=folders[0:10]

for folder in folders:
    print("Current Folder: ")
    print(folder)
    print(" ")
    #psd_file=os.path.join(top_level_directory, "Spreadsheets", "PSD_analysis.csv")
    psd_text=escsp_get_psd(folder, plots_folder, filelist=None, eclipse_type = eclipse_type, verbose=1, eclipse_data_csv=eclipse_data_csv)
    #f = open(psd_file, "a")
    f.write(psd_text)


f.close()