Compare commits

...

4 Commits

Author SHA1 Message Date
Cal Wing
aeef8f9253 Finalise README 2024-10-25 16:21:42 +10:00
Cal Wing
247c503885 Correct 2024-10-25 16:05:59 +10:00
Cal Wing
c4b0dd1fef quick refactor and addition of README 2024-10-25 16:05:22 +10:00
Cal Wing
3398414b55 Finalise Thesis Codebase 2024-10-25 15:55:16 +10:00
4 changed files with 100 additions and 60 deletions

39
README.md Normal file
View File

@ -0,0 +1,39 @@
# Ionization Probe Analysis Code
> Written by Cal Wing (<c.wing@uq.net.au> - 45811953) in 2024 for his Thesis
## Installation
Run `pip install -r requirements.txt` or equivilient for your envoroment.
## Running
Run `main.py` it will then generate all the output graphs and put them in (a git untracked) folder called `./images`.
By default all data graphs will be generated - you need to change the functions called at the end in the `if '__name__ == '__main__':` section.
## `./data` Explanation
I like YAML files to store information. All the data shot folders have a file call `_info.yaml` this file contained all the info about the shot so that when it would load & be graphed it would be correct.
I hope the values are self explanatory but they may not - sorry
## Changes to [canny_shock_finder.py](./canny_shock_finder.py)
Basically I hacked in an extra argument `print_func`, that is used to override the `print` function in the file. It assumes its either a callable reference (like print, the default) or None (it then uses a no-operation function to silence all output)
I also removed the error catching around lines ~497 so the function blindly continues if it can't graph.
I also enabled grid lines on the graphs.
The UQ colours being used on the graphs is due to some funky ~~abuse~~ *utlitly* I built into my [`makeGraph`](#makegraph) library.
## MakeGraph
> This refers to the [makeGraph.py](./makeGraph.py) file/version *in this code base*, I would not trust the exact specifics for versions found elsewhere, but the general gist is the same.
A *long* time ago I wrote a wrapper for matplotlib that has expanded into a formatting tool. The crux of it is that I can throw a dictionary at the `makeGraph` function and it will do all the hard work for me.
There are a few fun things that loading the lib will do - like overriding the default colour cyclers to use only the UQ colours - so if your going to liberate / adapt it be wary.
All the graphing done in the [`main.py`](./main.py) uses makeGraph, its not the most scary thing in the world as its basically ~~AI~~ *a big if conditional*.

View File

@ -15,7 +15,7 @@ Chris James (c.james4@uq.edu.au) - 04/07/17
# Amened by Cal Wing to make the function not print # Amened by Cal Wing to make the function not print
VERSION_STRING = "17-Oct-2024" VERSION_STRING = "20-Oct-2024"
def canny_shock_finder(time_list, pressure_list, sigma = 4, derivative_threshold = 0.001, auto_derivative = False, post_suppression_threshold = 0.05, def canny_shock_finder(time_list, pressure_list, sigma = 4, derivative_threshold = 0.001, auto_derivative = False, post_suppression_threshold = 0.05,
post_shock_pressure = None, start_time = None, find_second_peak = None, plot = True, plot_scale_factor = None, post_shock_pressure = None, start_time = None, find_second_peak = None, plot = True, plot_scale_factor = None,
@ -116,6 +116,8 @@ def canny_shock_finder(time_list, pressure_list, sigma = 4, derivative_threshold
:param plot_time_scale: See plot_time_unit above. These two inputs allow the results plot to be in a different :param plot_time_scale: See plot_time_unit above. These two inputs allow the results plot to be in a different
time unit to the original data. The user must specify this scale to connect the original and plotted time unit to the original data. The user must specify this scale to connect the original and plotted
time units. Defaults to 1.0 (so no change if time units are seconds). time units. Defaults to 1.0 (so no change if time units are seconds).
:param print_func: Callable Reference to call when print is called in this function. Defaults to 'print' but if 'None'
is passed it uses a no-operation function to silence all output.
""" """
# Make the function silent / have print overridable # Make the function silent / have print overridable

View File

@ -56,7 +56,7 @@ probe-info:
- 2 - 2
locations: # In order of pulse locations: # In order of pulse
- "st2" - "st2"
- "st3" # - "st3"
data-records: data-records:
- type: "scope" - type: "scope"

41
main.py
View File

@ -29,22 +29,6 @@ CANNY_TIME_OFFSET = 50 #us
with open(TUNNEL_INFO_FILE, 'r') as file: with open(TUNNEL_INFO_FILE, 'r') as file:
TUNNEL_INFO = yaml.safe_load(file) TUNNEL_INFO = yaml.safe_load(file)
data_to_load = [
"x2s5823",
"x2s5824",
"x2s5827",
"x2s5829",
"x2s5830",
"x2s5831",
"x2s5832"
]
ref_data_to_load = [
"x2s5820",
"x2s5821",
"x2s5822"
]
# ==== Uncerts ==== # ==== Uncerts ====
# Taken from DOI: 10.1007/s00193-017-0763-3 (Implementation of a state-to-state analytical framework for the calculation of expansion tube flow properties) # Taken from DOI: 10.1007/s00193-017-0763-3 (Implementation of a state-to-state analytical framework for the calculation of expansion tube flow properties)
@ -319,7 +303,7 @@ def load_data(data_path: str, data={}) -> dict:
for i, probe in enumerate(dataInfo["probe-info"]["locations"]): for i, probe in enumerate(dataInfo["probe-info"]["locations"]):
# Get the canny-args # Get the canny-args
cArgs = dataInfo["canny-args"] cArgs = dataInfo["canny-args"]
doCannyPlot = x2_shot == "x2s5829" and probe == "st2" and False doCannyPlot = False # x2_shot == "x2s5829" and probe == "st2" # This condition was used to generate some graphs
plotValues = { plotValues = {
"plot_title": f"Canny Shock Finding Result for {x2_shot} - ST2 Gauge 1", "plot_title": f"Canny Shock Finding Result for {x2_shot} - ST2 Gauge 1",
"plot_time_unit": "$\\mu$s", "plot_time_unit": "$\\mu$s",
@ -1082,7 +1066,7 @@ def genX2CompGraphs(gData: dict, showPlot: bool = True):
graphData["plots"].append({ graphData["plots"].append({
"type": "axvLine", "type": "axvLine",
"x": gData["shock-point"]["x2"][f"{probe}-g1"][1],#[i], "x": gData["shock-point"]["x2"][f"{probe}-g1"][1],#[i],
"label": f"{probe}-Gauge 1 - X2 DAQ - Shock Point {gData['shock-point'][f'{probe}-g1'][1]:.2f}$\\mu$s", "label": f"{probe}-Gauge 1 - X2 DAQ - Shock Point {gData['shock-point']["x2"][f'{probe}-g1'][1]:.2f}$\\mu$s",
"colour": UQC["dark_grey"], "colour": UQC["dark_grey"],
"args":{"zorder":2, "linestyle":"--", "alpha":0.5} "args":{"zorder":2, "linestyle":"--", "alpha":0.5}
}) })
@ -1092,7 +1076,7 @@ def genX2CompGraphs(gData: dict, showPlot: bool = True):
graphData["plots"].append({ graphData["plots"].append({
"type": "axvLine", "type": "axvLine",
"x": gData["shock-point"]["x2"][f"{probe}-g2"][1],#[i], "x": gData["shock-point"]["x2"][f"{probe}-g2"][1],#[i],
"label": f"{probe}-Gauge 2 - X2 DAQ - Shock Point {gData['shock-point'][f'{probe}-g2'][1]:.2f}$\\mu$s", "label": f"{probe}-Gauge 2 - X2 DAQ - Shock Point {gData['shock-point']["x2"][f'{probe}-g2'][1]:.2f}$\\mu$s",
"colour": UQC["dark_grey"], "colour": UQC["dark_grey"],
"args":{"zorder":2, "linestyle":":", "alpha":0.5} "args":{"zorder":2, "linestyle":":", "alpha":0.5}
}) })
@ -1116,6 +1100,23 @@ def genX2CompGraphs(gData: dict, showPlot: bool = True):
pass pass
if __name__ == "__main__":
data_to_load = [
"x2s5823",
"x2s5824",
"x2s5827",
"x2s5829",
"x2s5830",
"x2s5831",
"x2s5832"
]
ref_data_to_load = [
"x2s5820",
"x2s5821",
"x2s5822"
]
print("Loading Data") print("Loading Data")
# My Shot Data # My Shot Data
@ -1133,8 +1134,6 @@ for refShot in ref_data_to_load:
print("Loaded Data") print("Loaded Data")
print("Graphing Data") print("Graphing Data")
# General Shot Graphing # General Shot Graphing