Restructuring

TESTING/testing.py

-import os
-import sys
-from unipath import Path
-import numpy as np
-import matplotlib.pyplot as plt
-from spacepy import pycdf
-from astropy.constants import R_sun
-
-# Append custom Python modules from parent directory
-sys.path.append(Path(sys.path[0]).parent)
-from MODULES.PSPops import data_quality as dq
-from MODULES.PSPops import data_transformation as dt
-from MODULES.Plotting import obs_plotset as ps
-from MODULES.PSPops import data_handling as dh
-from MODULES.Statistics import data_binning as db
-from MODULES.Statistics import stats as st
-
-
-# Plot set up
-ps.rc_setup()
-
-# CDF library (is needed to interface with the measurement data files)
-# see https://cdf.gsfc.nasa.gov/
-os.environ["CDF_LIB"] = "/usr/local/cdf/lib"
-
-# open CDF file and generate faulty index array
-cdf_data = pycdf.CDF("psp_swp_spc_l3i_20211117_v02.cdf")
-data = dh.data_generation(cdf_data)
-
-# Indices of non-usable data from general flag + reduction
-bad_ind = dq.general_flag(data["dqf"])
-data = dh.full_reduction(data, bad_ind)
-
-# Additional reduction from failed measurement indices
-mf_ind = dq.full_meas_eval(data)
-data = dh.full_reduction(data, mf_ind)
-
-# Add to data dictionary (spherical coordinates)
-data["r"], data["theta"], data["phi"] = dt.pos_cart_to_sph(data["pos"])
-
-# Add to data dictionary (temperature)
-data["T"] = dt.wp_to_temp(data["wp"])
-data["dThi"] = dt.wp_to_temp(data["dwphi"])
-data["dTlo"] = dt.wp_to_temp(data["dwplo"])
-
-# Testing binning into radial bins
-distance_bins = db.create_bins(0, 100, .5)
-bin_indices = db.sort_bins(distance_bins, data["r"] * 1e3 / R_sun.value)
-
-test_r = st.slice_index_list(data["r"], bin_indices["(46.5, 47.0)"]) * 1e3 / R_sun.value
-test_v = st.slice_index_list(data["vr"], bin_indices["(46.5, 47.0)"])
-
-plt.plot(test_r, test_v, zorder=2)
-plt.hlines(np.mean(test_v), np.min(test_r), np.max(test_r), color="black",
-           zorder=3, lw=2, ls="--")
-plt.fill_between(test_r, np.mean(test_v) + np.std(test_v),
-                 np.mean(test_v) - np.std(test_v),
-                 color="grey", alpha=0.35, zorder=1)
-plt.show()

TESTING/testing_plots.py

-import matplotlib.pyplot as plt
-import matplotlib as mpl
-import numpy as np
-
-"""Generalized plot attributes"""
-mpl.rcParams["xtick.direction"] = "in"
-mpl.rcParams["xtick.labelsize"] = "large"
-mpl.rcParams["xtick.major.width"] = 1.5
-mpl.rcParams["xtick.minor.width"] = 1.5
-mpl.rcParams["xtick.minor.visible"] = "True"
-mpl.rcParams["xtick.top"] = "True"
-
-mpl.rcParams["ytick.direction"] = "in"
-mpl.rcParams["ytick.labelsize"] = "large"
-mpl.rcParams["ytick.major.width"] = 1.5
-mpl.rcParams["ytick.minor.width"] = 1.5
-mpl.rcParams["ytick.minor.visible"] = "True"
-mpl.rcParams["ytick.right"] = "True"
-
-mpl.rcParams["axes.grid"] = "True"
-mpl.rcParams["axes.linewidth"] = 1.5
-mpl.rcParams["axes.labelsize"] = "large"
-
-x = np.arange(1, 10, 0.01)
-y1 = np.sin(x)
-y2 = np.sin(x+0.5)
-
-fig, ax = plt.subplots(1, 1, figsize=(15, 4.5))
-ax.plot(x, y1, lw=2.5, c="darkviolet")
-ax.plot(x, y2, lw=2.5, ls="--", c="fuchsia")
-
-plt.tight_layout()
-plt.show()

binned_stats.py

 import sys
 import os
 import numpy as np
-from MODULES.Plotting import general_plotset as gp
-from MODULES.Plotting import obs_plotset as op
-from MODULES.Statistics import stats as st
-from MODULES.Statistics import data_binning as db
+from MODULES.Plotting import plotset_general as gp
+from MODULES.Plotting import plotset_observations as op
+from MODULES.Statistics import stats_general as st
+from MODULES.Statistics import stats_databin as db
 
 # DESIGNATE BINNED DATA LOCATION
 BIN_DATA_LOCATION = f"{sys.path[0]}/STATISTICS/BINNED_DATA"

comp_data_plots.py → comparison_plots.py

 import sys
-from MODULES.Statistics import data_read
-from MODULES.Plotting import general_plotset as gp
-from MODULES.Plotting import comp_plotset as cp
+from MODULES.Statistics import stats_dataread
+from MODULES.Plotting import plotset_general as gp
+from MODULES.Plotting import plotset_comparison as cp
 
 
 # Necessary global variables

main_data.py → data_eval.py

@@ -2,14 +2,13 @@ import os
 import sys
 import numpy as np
 from spacepy import pycdf
-from MODULES.PSPops import data_quality as dq
+from MODULES.PSPops import data_quality as dq, data_turnaround as ta
 from MODULES.PSPops import data_transformation as dt
 from MODULES.PSPops import data_handling as dh
 from MODULES.PSPops import miscellaneous as misc
-from MODULES.Plotting import general_plotset as gp
-from MODULES.Statistics import data_binning as db
-from MODULES.Statistics import stats as st
-from MODULES.Statistics import turn_around as ta
+from MODULES.Plotting import plotset_general as gp
+from MODULES.Statistics import stats_databin as db
+from MODULES.Statistics import stats_general as st
 from MODULES.Miscellaneous import write_log
 from astropy.constants import R_sun
 

ingress_egress.py

@@ -7,10 +7,10 @@ import numpy as np
 from astropy.constants import R_sun
 from astropy import units as u
 from MODULES.PSPops import data_handling as dh
-from MODULES.Plotting import general_plotset as gp
-from MODULES.Plotting import obs_plotset as op
-from MODULES.Statistics import data_binning as db
-from MODULES.Statistics import stats as st
+from MODULES.Plotting import plotset_general as gp
+from MODULES.Plotting import plotset_observations as op
+from MODULES.Statistics import stats_databin as db
+from MODULES.Statistics import stats_general as st
 
 # DISTANCE BIN SIZE IN RSOL
 DISTANCE_BIN_SIZE = float(sys.argv[1])

obs_data_plots.py → observation_plots.py

 import sys
 from MODULES.PSPops import data_handling as dh
-from MODULES.Plotting import general_plotset as gp
-from MODULES.Plotting import obs_plotset as op
+from MODULES.Plotting import plotset_general as gp
+from MODULES.Plotting import plotset_observations as op
 
 
 # GLOBAL: STAT FILE NAME