|
|
|
|
@@ -39,6 +39,9 @@ class SingleDipoleDiagnostic():
|
|
|
|
|
self.s_result_x = self.result_dipole.s[0]
|
|
|
|
|
self.s_result_y = self.result_dipole.s[1]
|
|
|
|
|
self.s_result_z = self.result_dipole.s[2]
|
|
|
|
|
self.w_actual = self.actual_dipole.w
|
|
|
|
|
self.w_result = self.result_dipole.w
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Diagnostic():
|
|
|
|
|
@@ -65,7 +68,7 @@ class Diagnostic():
|
|
|
|
|
self.discretisations_with_names = discretisations_with_names
|
|
|
|
|
self.model_count = len(self.discretisations_with_names)
|
|
|
|
|
|
|
|
|
|
self.csv_fields = ["model", "index", "bounds", "p_actual_x", "p_actual_y", "p_actual_z", "s_actual_x", "s_actual_y", "s_actual_z", "actual_dipole_freq", "success", "p_result_x", "p_result_y", "p_result_z", "s_result_x", "s_result_y", "s_result_z"]
|
|
|
|
|
self.csv_fields = ["model", "index", "bounds", "p_actual_x", "p_actual_y", "p_actual_z", "s_actual_x", "s_actual_y", "s_actual_z", "w_actual", "success", "p_result_x", "p_result_y", "p_result_z", "s_result_x", "s_result_y", "s_result_z", "w_result"]
|
|
|
|
|
|
|
|
|
|
timestamp = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
|
|
|
|
|
self.filename = f"{timestamp}-{filename_slug}.diag.csv"
|
|
|
|
|
@@ -84,14 +87,14 @@ class Diagnostic():
|
|
|
|
|
results = pool.starmap(get_a_result, zip(itertools.repeat(discretisation), itertools.repeat(self.dots), discretisation.all_indices()))
|
|
|
|
|
|
|
|
|
|
with open(self.filename, "a", newline='') as outfile:
|
|
|
|
|
writer = csv.DictWriter(outfile, fieldnames=self.csv_fields, dialect='unix')
|
|
|
|
|
writer = csv.DictWriter(outfile, fieldnames=self.csv_fields, dialect='unix', extrasaction="ignore")
|
|
|
|
|
|
|
|
|
|
for idx, result in results:
|
|
|
|
|
|
|
|
|
|
bounds = discretisation.bounds(idx)
|
|
|
|
|
|
|
|
|
|
actual_success = result.success and result.cost <= 1e-10
|
|
|
|
|
diag_row = SingleDipoleDiagnostic(name, idx, bounds, self.dipoles.dipoles[0], discretisation.model.solution_as_dipoles(result.normalised_x), actual_success)
|
|
|
|
|
diag_row = SingleDipoleDiagnostic(name, idx, bounds, self.dipoles.dipoles[0], discretisation.model.solution_as_dipoles(result.normalised_x)[0], actual_success)
|
|
|
|
|
row = vars(diag_row)
|
|
|
|
|
_logger.debug(f"Writing result {row}")
|
|
|
|
|
writer.writerow(row)
|
|
|
|
|
|
