pyadps 0.1.0b0__py3-none-any.whl → 0.1.2__py3-none-any.whl

pyadps/pages/03_Download_Raw_File.py

@@ -18,16 +18,33 @@ if "fname" not in st.session_state:
     st.session_state.fname = "No file selected"
 
 if "rawfilename" not in st.session_state:
-    st.session_state.rawfilename = "rawfile.nc"
+    st.session_state.rawfilename = "RAW_DAT.nc"
+
+if "fleadfilename" not in st.session_state:
+    st.session_state.fleadfilename = "RAW_FIX.nc"
 
 if "vleadfilename" not in st.session_state:
-    st.session_state.vleadfilename = "vlead.nc"
+    st.session_state.vleadfilename = "RAW_VAR.nc"
 
 if "attributes" not in st.session_state:
     st.session_state.attributes = {}
-    
-if "add_attributes" not in st.session_state:
-    st.session_state.add_attributes = "No"  # Default value
+
+if "add_attributes_DRW" not in st.session_state:
+    st.session_state.add_attributes_DRW = "No"  # Default value
+
+if "file_prefix" not in st.session_state:
+    raw_basename = os.path.basename(st.session_state.fname)
+    st.session_state.filename = os.path.splitext(raw_basename)[0] 
+    st.session_state.file_prefix = st.session_state.filename
+
+
+if "prefix_saved" not in st.session_state:
+    st.session_state.prefix_saved = False
+
+if "filename" not in st.session_state:
+    st.session_state.filename = ""  # <-- Default file name if not passed
+
+
 
 ################ Functions #######################
 @st.cache_data()
@@ -38,6 +55,7 @@ def file_access(uploaded_file):
         f.write(uploaded_file.getvalue())
     return path
 
+
 @st.cache_data
 def read_file(filepath):
     ds = rd.ReadFile(st.session_state.fpath)
@@ -46,82 +64,200 @@ def read_file(filepath):
     st.session_state.ds = ds
 
 @st.cache_data
-def file_write(path, add_attributes=True):
+def get_prefixed_filename(base_name):
+    """Generates the file name with the optional prefix."""
+    if st.session_state.file_prefix:
+        return f"{st.session_state.file_prefix}_{base_name}"
+    return base_name
+
+
+@st.cache_data
+def file_write(path, axis_option, add_attributes=True):
     tempdirname = tempfile.TemporaryDirectory(delete=False)
-    st.session_state.rawfilename = tempdirname.name + "/rawfile.nc"
-    
+    st.session_state.rawfilename = os.path.join(tempdirname.name, get_prefixed_filename("RAW_DAT.nc"))
+
+    if add_attributes:
+        wr.rawnc(
+            path,
+            st.session_state.rawfilename,
+            st.session_state.date1,
+            axis_option,
+            attributes=st.session_state.attributes,
+        )
+    else:
+        wr.rawnc(
+            path, st.session_state.rawfilename, st.session_state.date1, axis_option
+        )
+
+@st.cache_data
+def file_write_flead(path, axis_option, add_attributes=True):
+    tempvardirname = tempfile.TemporaryDirectory(delete=False)
+    st.session_state.fleadfilename = os.path.join(tempvardirname.name, get_prefixed_filename("RAW_FIX.nc"))
+
     if add_attributes:
-        wr.main(path, st.session_state.rawfilename, attributes=st.session_state.attributes)
+        wr.flead_nc(
+            path,
+            st.session_state.fleadfilename,
+            st.session_state.date2,
+            axis_option,
+            attributes=st.session_state.attributes,
+        )
     else:
-        wr.main(path, st.session_state.rawfilename)
+        wr.flead_nc(
+            path, st.session_state.fleadfilename, st.session_state.date2, axis_option
+        )
 
 @st.cache_data
-def file_write_vlead(path, add_attributes=True):
+def file_write_vlead(path, axis_option, add_attributes=True):
     tempvardirname = tempfile.TemporaryDirectory(delete=False)
-    st.session_state.vleadfilename = tempvardirname.name + "/vlead.nc"
-    
+    st.session_state.vleadfilename = os.path.join(tempvardirname.name, get_prefixed_filename("RAW_VAR.nc"))
+
     if add_attributes:
-        wr.vlead_nc(path, st.session_state.vleadfilename, attributes=st.session_state.attributes)
+        wr.vlead_nc(
+            path,
+            st.session_state.vleadfilename,
+            st.session_state.date3,
+            axis_option,
+            attributes=st.session_state.attributes,
+        )
     else:
-        wr.vlead_nc(path, st.session_state.vleadfilename)
+        wr.vlead_nc(
+            path, st.session_state.vleadfilename, st.session_state.date3, axis_option
+        )
+
+
+
+if "axis_option" not in st.session_state:
+    st.session_state.axis_option = "ensemble"  # Default value
 
 # UI for attribute selection
 st.header("NetCDF File", divider="blue")
 
 # Option to add attributes
-add_attributes = st.radio("Do you want to add attributes to the NetCDF file?", ["No", "Yes"], horizontal=True)
+st.session_state.add_attributes_DRW = st.radio(
+    "Do you want to add attributes to the NetCDF file?", ["No", "Yes"], horizontal=True
+)
 
-if add_attributes == "Yes":
+if st.session_state.add_attributes_DRW == "Yes":
     st.write("### Please fill in the attributes:")
-    
+
     # Two-column layout
     col1, col2 = st.columns(2)
-    
-    with col1:
-        st.session_state.attributes['Cruise_No.'] = st.text_input("Cruise No.")
-        st.session_state.attributes['Ship_Name'] = st.text_input("Ship Name")
-        st.session_state.attributes['Project_No.'] = st.text_input("Project No.")
-        st.session_state.attributes['Water_Depth_m'] = st.text_input("Water Depth (m)")
-        st.session_state.attributes['Deployment_Depth_m'] = st.text_input("Deployment Depth (m)")
-        st.session_state.attributes['Deployment_Date'] = st.date_input("Deployment Date")
-        st.session_state.attributes['Recovery_Date'] = st.date_input("Recovery Date")
 
+    with col1:
+        st.session_state.attributes["Cruise_No."] = st.text_input("Cruise No.")
+        st.session_state.attributes["Ship_Name"] = st.text_input("Ship Name")
+        st.session_state.attributes["Project_No."] = st.text_input("Project No.")
+        st.session_state.attributes["Water_Depth_m"] = st.text_input("Water Depth (m)")
+        st.session_state.attributes["Deployment_Depth_m"] = st.text_input(
+            "Deployment Depth (m)"
+        )
+        st.session_state.attributes["Deployment_Date"] = st.date_input(
+            "Deployment Date"
+        )
+        st.session_state.attributes["Recovery_Date"] = st.date_input("Recovery Date")
 
     with col2:
-        st.session_state.attributes['Latitude'] = st.text_input("Latitude")
-        st.session_state.attributes['Longitude'] = st.text_input("Longitude")
-        st.session_state.attributes['Platform_Type'] = st.text_input("Platform Type")
-        st.session_state.attributes['Participants'] = st.text_area("Participants")
-        st.session_state.attributes['File_created_by'] = st.text_input("File created by")
-        st.session_state.attributes['Contact'] = st.text_input("Contact")
-        st.session_state.attributes['Comments'] = st.text_area("Comments")
+        st.session_state.attributes["Latitude"] = st.text_input("Latitude")
+        st.session_state.attributes["Longitude"] = st.text_input("Longitude")
+        st.session_state.attributes["Platform_Type"] = st.text_input("Platform Type")
+        st.session_state.attributes["Participants"] = st.text_area("Participants")
+        st.session_state.attributes["File_created_by"] = st.text_input(
+            "File created by"
+        )
+        st.session_state.attributes["Contact"] = st.text_input("Contact")
+        st.session_state.attributes["Comments"] = st.text_area("Comments")
 
     st.write("Attributes will be added to the NetCDF file once you submit.")
-    
+
+
+st.info(f"Current file name: **{st.session_state.filename}**")
+
+# Prefix editing option
+st.session_state.use_custom_filename = st.radio(
+    "Do you want to edit Output Filename?",
+    ["No", "Yes"],
+    horizontal=True,
+)
+
+if st.session_state.use_custom_filename == "Yes" and not st.session_state.prefix_saved:
+    st.session_state.file_prefix = st.text_input(
+        "Enter file name (e.g., GD10A000)",
+        value=st.session_state.file_prefix,
+    )
+
+    if st.button("Save Filename"):
+        if st.session_state.file_prefix.strip():
+            st.session_state.prefix_saved = True
+            st.rerun()
+        else:
+            st.warning("Please enter a valid filename before saving.")
+
+if st.session_state.prefix_saved:
+    st.success(f"Filename saved as: **{st.session_state.file_prefix}**")
+
+# Dropdown for axis_option
+st.session_state.axis_option_DRW = st.selectbox(
+    "Select x-axis option:",
+    options=["time", "ensemble"],
+    index=0,  # Default to "time"
+)
+
+# Ensure it is passed correctly
+# st.session_state.axis_option = axis_option
+
 # Buttons to generate files
-download_button = st.button("Generate Raw NetCDF File")
-download_var_button = st.button("Generate Raw Variable Leader NetCDF File")
+st.session_state.rawnc_download_DRW = st.button("Generate Raw NetCDF File")
+st.session_state.fleadnc_download_DRW = st.button(
+    "Generate Raw Fixed Leader NetCDF File"
+)
+st.session_state.vleadnc_download_DRW = st.button(
+    "Generate Raw Variable Leader NetCDF File"
+)
 
-if download_button:
-    file_write(st.session_state.fpath, add_attributes == "Yes")
+if st.session_state.rawnc_download_DRW:
+    file_write(
+        st.session_state.fpath,
+        st.session_state.axis_option_DRW,
+        st.session_state.add_attributes_DRW == "Yes",
+    )
     st.write(st.session_state.rawfilename)
     with open(st.session_state.rawfilename, "rb") as file:
         st.download_button(
             label="Download Raw File",
             data=file,
-            file_name="rawfile.nc",
+            file_name=get_prefixed_filename("RAW_DAT.nc"),
         )
 
-if download_var_button:
-    file_write_vlead(st.session_state.fpath, add_attributes == "Yes")
+if st.session_state.fleadnc_download_DRW:
+    file_write_flead(
+        st.session_state.fpath,
+        st.session_state.axis_option,
+        st.session_state.add_attributes_DRW == "Yes",
+    )
+    st.write(st.session_state.fleadfilename)
+    with open(st.session_state.fleadfilename, "rb") as file:
+        st.download_button(
+            label="Download Fixed Leader",
+            data=file,
+            file_name=get_prefixed_filename("RAW_FIX.nc"),
+        )
+
+if st.session_state.vleadnc_download_DRW:
+    file_write_vlead(
+        st.session_state.fpath,
+        st.session_state.axis_option,
+        st.session_state.add_attributes_DRW == "Yes",
+    )
     st.write(st.session_state.vleadfilename)
     with open(st.session_state.vleadfilename, "rb") as file:
         st.download_button(
             label="Download Variable Leader",
             data=file,
-            file_name="vlead.nc",
+            file_name=get_prefixed_filename("RAW_VAR.nc"),
         )
 
+
 def download_csv_with_ensemble(data, filename):
     # Create ensemble numbers from 1 to the number of rows in the data
     ensembles = np.arange(1, len(next(iter(data.values()))) + 1)
@@ -131,60 +267,62 @@ def download_csv_with_ensemble(data, filename):
     df.insert(0, "RDI_Ensemble", ensembles)  # Add ensemble numbers as the first column
 
     # Export the DataFrame as a CSV
-    csv = df.to_csv(index=False).encode('utf-8')
+    csv = df.to_csv(index=False).encode("utf-8")
     return st.download_button(
         label=f"Download {filename} as CSV",
         data=csv,
         file_name=f"{filename}.csv",
-        mime='text/csv',
+        mime="text/csv",
     )
-    
+
+
 def download_csv(data, filename):
     # Convert data to DataFrame if it's not already one
     if isinstance(data, dict):
         df = pd.DataFrame.from_dict(data, orient="index").T
     else:
         df = pd.DataFrame(data)
-    
+
     # Export the DataFrame as a CSV
-    csv = df.to_csv(index=False).encode('utf-8')
+    csv = df.to_csv(index=False).encode("utf-8")
     return st.download_button(
         label=f"Download {filename} as CSV",
         data=csv,
         file_name=f"{filename}.csv",
-        mime='text/csv',
+        mime="text/csv",
     )
-    
-    
+
+
 def download_csv1(data, filename):
     # Convert data to DataFrame if it's not already one
     if isinstance(data, dict):
         df = pd.DataFrame.from_dict(data, orient="index").T
     else:
         df = pd.DataFrame(data)
-    
+
     # Create ensemble and depth arrays
     ensembles = np.arange(1, df.shape[0] + 1)
     depths = np.arange(1, df.shape[1] + 1)
-    
+
     # Add ensemble numbers as the first column
-    df.insert(0, 'Ensemble', ensembles)
-    
+    df.insert(0, "Ensemble", ensembles)
+
     # Transpose the DataFrame to switch rows and columns
     df = df.T
-    
+
     # Add depth values as the first row
-    df.insert(0, 'Depth', [''] + list(depths))
-    
+    df.insert(0, "Depth", [""] + list(depths))
+
     # Export the DataFrame as a CSV
-    csv = df.to_csv(index=False, header=False).encode('utf-8')
+    csv = df.to_csv(index=False, header=False).encode("utf-8")
     return st.download_button(
         label=f"Download {filename} as CSV",
         data=csv,
         file_name=f"{filename}.csv",
-        mime='text/csv',
+        mime="text/csv",
     )
 
+
 # Load data
 fdata = st.session_state.flead.fleader
 vdata = st.session_state.vlead.vleader
@@ -202,25 +340,36 @@ X, Y = np.meshgrid(x, y)
 st.header("Download Raw Data CSV File", divider="blue")
 
 # Selection for the data category
-data_type = st.selectbox(
+st.session_state.rawcsv_option_DRW = st.selectbox(
     "Select data type to download:",
-    ["Fixed Leader", "Variable Leader", "Velocity", "Echo Intensity", "Correlation", "Percent Good"]
+    [
+        "Velocity",
+        "Echo Intensity",
+        "Correlation",
+        "Percent Good",
+        "Variable Leader",
+        "Fixed Leader",
+    ],
 )
 
 # Show corresponding variable options based on selection
-if data_type == "Fixed Leader":
+if st.session_state.rawcsv_option_DRW == "Fixed Leader":
     # Combine all variables of Fixed Leader into one DataFrame
     f_combined_data = {var: fdata[var] for var in fdata.keys()}
     download_csv_with_ensemble(f_combined_data, "Fixed_Leader_All_Variables")
 
-elif data_type == "Variable Leader":
+elif st.session_state.rawcsv_option_DRW == "Variable Leader":
     # Combine all variables of Variable Leader into one DataFrame
     v_combined_data = {var: vdata[var] for var in vdata.keys()}
     download_csv(v_combined_data, "Variable_Leader_All_Variables")
 
 else:
-    beam_download = st.radio("Select beam to download", (1, 2, 3, 4), horizontal=True)
+    st.session_state.rawcsv_beam_DRW = st.radio(
+        "Select beam to download", (1, 2, 3, 4), horizontal=True
+    )
 
+    data_type = st.session_state.rawcsv_option_DRW
+    beam_download = st.session_state.rawcsv_beam_DRW
     if data_type == "Velocity":
         download_data = velocity[beam_download - 1, :, :]
     elif data_type == "Echo Intensity":
@@ -231,4 +380,3 @@ else:
         download_data = pgood[beam_download - 1, :, :]
 
     download_csv1(download_data, f"{data_type}_Beam_{beam_download}")
-