Skip to content
Snippets Groups Projects
Commit 41799d43 authored by Elliot Oram's avatar Elliot Oram
Browse files

WIP: basic total scattering analysis

Refs #21294
parent 276ba1e0
No related branches found
No related tags found
No related merge requests found
...@@ -94,6 +94,14 @@ class FocusTest(stresstesting.MantidStressTest): ...@@ -94,6 +94,14 @@ class FocusTest(stresstesting.MantidStressTest):
config['datasearch.directories'] = self.existing_config config['datasearch.directories'] = self.existing_config
mantid.mtd.clear() mantid.mtd.clear()
class TotalScatteringTest(stresstesting.MantidStressTest):
def runTest(self):
def validate(self):
def cleanup(self):
def _gen_required_files(): def _gen_required_files():
required_run_numbers = ["98531", "98532", # create_van : PDF mode required_run_numbers = ["98531", "98532", # create_van : PDF mode
......
...@@ -42,6 +42,14 @@ class Polaris(AbstractInst): ...@@ -42,6 +42,14 @@ class Polaris(AbstractInst):
output_path=run_details.unsplined_vanadium_file_path) output_path=run_details.unsplined_vanadium_file_path)
return vanadium_d return vanadium_d
def create_total_scattering_pdf(self, **kwargs):
self._inst_settings.update_attributes(kwargs=kwargs)
# Validate that required files exist
# generate pdf
pdf_output = self._generate_ts_pdf(focused_ws=self._inst_settings.focused_ws,
merge_banks=self._inst_settings.merge_banks)
return pdf_output
def set_sample_details(self, **kwargs): def set_sample_details(self, **kwargs):
self._switch_mode_specific_inst_settings(kwargs.get("mode")) self._switch_mode_specific_inst_settings(kwargs.get("mode"))
kwarg_name = "sample" kwarg_name = "sample"
......
...@@ -76,6 +76,14 @@ def save_unsplined_vanadium(vanadium_ws, output_path): ...@@ -76,6 +76,14 @@ def save_unsplined_vanadium(vanadium_ws, output_path):
mantid.DeleteWorkspace(converted_group) mantid.DeleteWorkspace(converted_group)
def generate_ts_pdf(tof_ws, merge_banks=False):
q_ws = mantid.ConvertUnits(InputWorkspace=tof_ws, Target="MomentumTransfer")
if merge_banks:
raise RuntimeError("Merging banks is currently not supported")
pdf_output = mantid.PDFFourierTransform(Inputworkspace=q_ws, InputSofQType="S(Q)", PDFType="G(r)")
return pdf_output
def _apply_bragg_peaks_masking(workspaces_to_mask, mask_list): def _apply_bragg_peaks_masking(workspaces_to_mask, mask_list):
output_workspaces = list(workspaces_to_mask) output_workspaces = list(workspaces_to_mask)
......
...@@ -16,8 +16,10 @@ attr_mapping = \ ...@@ -16,8 +16,10 @@ attr_mapping = \
ParamMapEntry(ext_name="first_cycle_run_no", int_name="run_in_range"), ParamMapEntry(ext_name="first_cycle_run_no", int_name="run_in_range"),
ParamMapEntry(ext_name="focused_cropping_values", int_name="focused_cropping_values"), ParamMapEntry(ext_name="focused_cropping_values", int_name="focused_cropping_values"),
ParamMapEntry(ext_name="focused_bin_widths", int_name="focused_bin_widths"), ParamMapEntry(ext_name="focused_bin_widths", int_name="focused_bin_widths"),
ParamMapEntry(ext_name="focused_workspace", int_name="focused_ws"),
ParamMapEntry(ext_name="grouping_file_name", int_name="grouping_file_name"), ParamMapEntry(ext_name="grouping_file_name", int_name="grouping_file_name"),
ParamMapEntry(ext_name="input_mode", int_name="input_mode", enum_class=INPUT_BATCHING), ParamMapEntry(ext_name="input_mode", int_name="input_mode", enum_class=INPUT_BATCHING),
ParamMapEntry(ext_name="merge_banks", int_name="merge_banks"),
ParamMapEntry(ext_name="mode", int_name="mode", enum_class=POLARIS_CHOPPER_MODES), ParamMapEntry(ext_name="mode", int_name="mode", enum_class=POLARIS_CHOPPER_MODES),
ParamMapEntry(ext_name="multiple_scattering", int_name="multiple_scattering"), ParamMapEntry(ext_name="multiple_scattering", int_name="multiple_scattering"),
ParamMapEntry(ext_name="raw_data_cropping_values", int_name="raw_data_crop_values"), ParamMapEntry(ext_name="raw_data_cropping_values", int_name="raw_data_crop_values"),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment