larch_athena: larch_athena.py annotate

annotate larch_athena.py @ 1:2b3115342fef draft

planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6

author	muon-spectroscopy-computational-project
date	Wed, 06 Dec 2023 13:03:55 +0000
parents	ae2f265ecf8e
children	a1e26990131c

rev	line source
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	1 import gc
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	2 import json
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	3 import os
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	4 import re
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	5 import sys
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	6
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	7 from common import (
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	8 pre_edge_with_defaults, read_all_groups, read_group, xftf_with_defaults
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	9 )
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	10
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	11 from larch.io import (
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	12 create_athena,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	13 h5group,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	14 merge_groups,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	15 read_ascii,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	16 set_array_labels,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	17 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	18 from larch.symboltable import Group
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	19 from larch.xafs import rebin_xafs
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	20
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	21 import matplotlib
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	22 import matplotlib.pyplot as plt
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	23
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	24 import numpy as np
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	25
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	26
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	27 class Reader:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	28 def __init__(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	29 self,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	30 energy_column: str,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	31 mu_column: str,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	32 data_format: str,
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	33 extract_group: "dict[str, str]" = None,
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	34 ):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	35 self.energy_column = energy_column
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	36 self.mu_column = mu_column
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	37 self.data_format = data_format
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	38 self.extract_group = extract_group
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	39
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	40 def load_data(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	41 self,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	42 dat_file: str,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	43 merge_inputs: bool,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	44 is_zipped: bool,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	45 ) -> "dict[str, Group]":
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	46 if merge_inputs:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	47 out_group = self.merge_files(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	48 dat_files=dat_file, is_zipped=is_zipped
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	49 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	50 return {"out": out_group}
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	51 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	52 return self.load_single_file(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	53 filepath=dat_file, is_zipped=is_zipped
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	54 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	55
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	56 def merge_files(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	57 self,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	58 dat_files: str,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	59 is_zipped: bool,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	60 ) -> Group:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	61 if is_zipped:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	62 all_groups = list(self.load_zipped_files().values())
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	63 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	64 all_groups = []
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	65 for filepath in dat_files.split(","):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	66 group = self.load_single_file(filepath)["out"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	67 all_groups.append(group)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	68
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	69 return merge_groups(all_groups, xarray="energy", yarray="mu")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	70
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	71 def load_single_file(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	72 self,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	73 filepath: str,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	74 is_zipped: bool = False,
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	75 ) -> "tuple[dict, bool]":
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	76 if is_zipped:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	77 return self.load_zipped_files()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	78
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	79 print(f"Attempting to read from {filepath}")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	80 if self.data_format == "athena":
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	81 if self.extract_group["extract_group"] == "single":
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	82 group = read_group(filepath, self.extract_group["group_name"])
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	83 return {"out": group}
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	84 elif self.extract_group["extract_group"] == "multiple":
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	85 groups = {}
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	86 for repeat in self.extract_group["multiple"]:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	87 name = repeat["group_name"]
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	88 groups[name] = read_group(filepath, name)
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	89 return groups
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	90 else:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	91 return read_all_groups(filepath)
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	92
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	93 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	94 # Try ascii anyway
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	95 try:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	96 group = self.load_ascii(filepath)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	97 if not group.array_labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	98 # In later versions of larch, won't get a type error it
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	99 # will just fail to load any data
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	100 group = self.load_h5(filepath)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	101 except (UnicodeDecodeError, TypeError):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	102 # Indicates this isn't plaintext, try h5
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	103 group = self.load_h5(filepath)
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	104 pre_edge_with_defaults(group)
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	105 xftf_with_defaults(group)
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	106 return {"out": group}
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	107
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	108 def load_ascii(self, dat_file):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	109 with open(dat_file) as f:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	110 labels = None
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	111 last_line = None
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	112 line = f.readline()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	113 while line:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	114 if not line.startswith("#"):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	115 if last_line is not None and last_line.find("\t") > 0:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	116 labels = []
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	117 for label in last_line.split("\t"):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	118 labels.append(label.strip())
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	119 break
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	120
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	121 last_line = line
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	122 line = f.readline()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	123
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	124 xas_data = read_ascii(filename=dat_file, labels=labels)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	125 xas_data = self.rename_cols(xas_data)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	126 return xas_data
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	127
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	128 def load_h5(self, dat_file):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	129 h5_group = h5group(fname=dat_file, mode="r")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	130 energy = h5_group.entry1.instrument.qexafs_energy.qexafs_energy
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	131 mu = h5_group.entry1.instrument.qexafs_counterTimer01.lnI0It
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	132 xafs_group = Group(data=np.array([energy[:], mu[:]]))
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	133 set_array_labels(xafs_group, ["energy", "mu"])
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	134 return xafs_group
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	135
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	136 def load_zipped_files(self) -> "dict[str, Group]":
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	137 def sorting_key(filename: str) -> str:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	138 return re.findall(r"\d+", filename)[-1]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	139
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	140 all_paths = list(os.walk("dat_files"))
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	141 all_paths.sort(key=lambda x: x[0])
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	142 file_total = sum([len(f) for _, _, f in all_paths])
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	143 print(f"{file_total} files found")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	144 key_length = len(str(file_total))
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	145 i = 0
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	146 keyed_data = {}
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	147 for dirpath, _, filenames in all_paths:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	148 try:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	149 filenames.sort(key=sorting_key)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	150 except IndexError as e:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	151 print(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	152 "WARNING: Unable to sort files numerically, "
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	153 f"defaulting to sorting alphabetically:\n{e}"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	154 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	155 filenames.sort()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	156
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	157 for filename in filenames:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	158 key = str(i).zfill(key_length)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	159 filepath = os.path.join(dirpath, filename)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	160 xas_data = self.load_single_file(filepath)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	161 keyed_data[key] = xas_data["out"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	162 i += 1
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	163
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	164 return keyed_data
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	165
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	166 def rename_cols(self, xafs_group: Group) -> Group:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	167 labels = [label.lower() for label in xafs_group.array_labels]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	168 print(f"Read columns: {labels}")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	169
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	170 if "energy" in labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	171 print("'energy' present in column headers")
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	172 elif self.energy_column:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	173 if self.energy_column.lower() in labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	174 labels[labels.index(self.energy_column.lower())] = "energy"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	175 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	176 raise ValueError(f"{self.energy_column} not found in {labels}")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	177 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	178 for i, label in enumerate(labels):
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	179 if label in ("col1", "ef") or label.endswith("energy"):
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	180 labels[i] = "energy"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	181 break
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	182
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	183 if "mu" in labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	184 print("'mu' present in column headers")
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	185 elif self.mu_column:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	186 if self.mu_column.lower() in labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	187 labels[labels.index(self.mu_column.lower())] = "mu"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	188 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	189 raise ValueError(f"{self.mu_column} not found in {labels}")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	190 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	191 for i, label in enumerate(labels):
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	192 if label in ["col2", "xmu", "lni0it", "ffi0", "ff/i1"]:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	193 labels[i] = "mu"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	194 break
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	195
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	196 if labels != xafs_group.array_labels:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	197 print(f"Renaming columns to: {labels}")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	198 return set_array_labels(xafs_group, labels)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	199 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	200 return xafs_group
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	201
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	202
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	203 def calibrate_energy(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	204 xafs_group: Group,
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	205 calibration_e0: float = None,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	206 energy_min: float = None,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	207 energy_max: float = None,
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	208 ):
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	209 if calibration_e0 is not None:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	210 print(f"Recalibrating edge from {xafs_group.e0} to {calibration_e0}")
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	211 xafs_group.energy = xafs_group.energy + calibration_e0 - xafs_group.e0
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	212 xafs_group.e0 = calibration_e0
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	213
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	214 if not (energy_min or energy_max):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	215 return xafs_group
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	216
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	217 if energy_min is not None:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	218 index_min = np.searchsorted(xafs_group.energy, energy_min)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	219 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	220 index_min = 0
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	221
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	222 if energy_max is not None:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	223 index_max = np.searchsorted(xafs_group.energy, energy_max)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	224 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	225 index_max = len(xafs_group.energy)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	226
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	227 print(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	228 f"Cropping energy range from {energy_min} to {energy_max}, "
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	229 f"index {index_min} to {index_max}"
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	230 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	231 try:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	232 xafs_group.dmude = xafs_group.dmude[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	233 xafs_group.pre_edge = xafs_group.pre_edge[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	234 xafs_group.post_edge = xafs_group.post_edge[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	235 xafs_group.flat = xafs_group.flat[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	236 except AttributeError:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	237 pass
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	238
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	239 xafs_group.energy = xafs_group.energy[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	240 xafs_group.mu = xafs_group.mu[index_min:index_max]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	241
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	242 # Sanity check
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	243 if len(xafs_group.energy) == 0:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	244 raise ValueError("Energy cropping led to an empty array")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	245
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	246 return xafs_group
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	247
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	248
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	249 def main(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	250 xas_data: Group,
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	251 do_calibrate: bool,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	252 calibrate_settings: dict,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	253 do_rebin: bool,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	254 do_pre_edge: bool,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	255 pre_edge_settings: dict,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	256 do_xftf: bool,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	257 xftf_settings: dict,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	258 plot_graph: bool,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	259 annotation: str,
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	260 path_key: str = "out",
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	261 ):
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	262 if do_calibrate:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	263 print(f"Calibrating energy with {calibrate_settings}")
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	264 xas_data = calibrate_energy(xas_data, **calibrate_settings)
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	265 # After re-calibrating, will need to redo pre-edge with new range
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	266 do_pre_edge = True
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	267
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	268 if do_rebin:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	269 print("Re-binning data")
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	270 rebin_xafs(
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	271 energy=xas_data.energy,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	272 mu=xas_data.mu,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	273 group=xas_data,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	274 **pre_edge_settings,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	275 )
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	276 xas_data = xas_data.rebinned
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	277 # After re-bin, will need to redo pre-edge
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	278 do_pre_edge = True
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	279
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	280 if do_pre_edge:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	281 pre_edge_with_defaults(xas_data, pre_edge_settings)
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	282
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	283 if do_xftf:
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	284 xftf_with_defaults(xas_data, xftf_settings)
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	285
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	286 if plot_graph:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	287 plot_edge_fits(f"edge/{path_key}.png", xas_data)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	288 plot_flattened(f"flat/{path_key}.png", xas_data)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	289 plot_derivative(f"derivative/{path_key}.png", xas_data)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	290
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	291 xas_project = create_athena(f"prj/{path_key}.prj")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	292 xas_project.add_group(xas_data)
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	293 if annotation:
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	294 group = next(iter(xas_project.groups.values()))
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	295 group.args["annotation"] = annotation
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	296 xas_project.save()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	297
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	298 # Ensure that we do not run out of memory when running on large zips
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	299 gc.collect()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	300
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	301
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	302 def plot_derivative(plot_path: str, xafs_group: Group):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	303 plt.figure()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	304 plt.plot(xafs_group.energy, xafs_group.dmude)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	305 plt.grid(color="r", linestyle=":", linewidth=1)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	306 plt.xlabel("Energy (eV)")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	307 plt.ylabel("Derivative normalised to x$\mu$(E)") # noqa: W605
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	308 plt.savefig(plot_path, format="png")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	309 plt.close("all")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	310
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	311
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	312 def plot_edge_fits(plot_path: str, xafs_group: Group):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	313 plt.figure()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	314 plt.plot(xafs_group.energy, xafs_group.pre_edge, "g", label="pre-edge")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	315 plt.plot(xafs_group.energy, xafs_group.post_edge, "r", label="post-edge")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	316 plt.plot(xafs_group.energy, xafs_group.mu, "b", label="fit data")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	317 plt.grid(color="r", linestyle=":", linewidth=1)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	318 plt.xlabel("Energy (eV)")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	319 plt.ylabel("x$\mu$(E)") # noqa: W605
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	320 plt.title("pre-edge and post_edge fitting to $\mu$") # noqa: W605
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	321 plt.legend()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	322 plt.savefig(plot_path, format="png")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	323 plt.close("all")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	324
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	325
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	326 def plot_flattened(plot_path: str, xafs_group: Group):
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	327 plt.figure()
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	328 plt.plot(xafs_group.energy, xafs_group.flat)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	329 plt.grid(color="r", linestyle=":", linewidth=1)
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	330 plt.xlabel("Energy (eV)")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	331 plt.ylabel("normalised x$\mu$(E)") # noqa: W605
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	332 plt.savefig(plot_path, format="png")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	333 plt.close("all")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	334
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	335
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	336 if __name__ == "__main__":
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	337 # larch imports set this to an interactive backend, so need to change it
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	338 matplotlib.use("Agg")
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	339
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	340 dat_file = sys.argv[1]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	341 input_values = json.load(open(sys.argv[2], "r", encoding="utf-8"))
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	342 merge_inputs = input_values["merge_inputs"]["merge_inputs"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	343 data_format = input_values["merge_inputs"]["format"]["format"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	344 if "is_zipped" in input_values["merge_inputs"]["format"]:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	345 is_zipped = bool(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	346 input_values["merge_inputs"]["format"]["is_zipped"]["is_zipped"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	347 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	348 else:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	349 is_zipped = False
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	350
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	351 extract_group = None
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	352 if "extract_group" in input_values["merge_inputs"]["format"]:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	353 extract_group = input_values["merge_inputs"]["format"]["extract_group"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	354
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	355 energy_column = None
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	356 mu_column = None
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	357 if "energy_column" in input_values["merge_inputs"]["format"]:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	358 energy_column = input_values["merge_inputs"]["format"]["energy_column"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	359 if "mu_column" in input_values["merge_inputs"]["format"]:
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	360 mu_column = input_values["merge_inputs"]["format"]["mu_column"]
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	361
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	362 reader = Reader(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	363 energy_column=energy_column,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	364 mu_column=mu_column,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	365 data_format=data_format,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	366 extract_group=extract_group,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	367 )
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	368 keyed_data = reader.load_data(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	369 dat_file=dat_file,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	370 merge_inputs=merge_inputs,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	371 is_zipped=is_zipped,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	372 )
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	373
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	374 calibrate_items = input_values["processing"]["calibrate"].items()
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	375 calibrate_settings = {k: v for k, v in calibrate_items if v is not None}
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	376 do_calibrate = calibrate_settings.pop("calibrate") == "true"
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	377
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	378 do_rebin = input_values["processing"].pop("rebin")
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	379
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	380 pre_edge_items = input_values["processing"]["pre_edge"].items()
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	381 pre_edge_settings = {k: v for k, v in pre_edge_items if v is not None}
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	382 do_pre_edge = pre_edge_settings.pop("pre_edge") == "true"
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	383
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	384 xftf_items = input_values["processing"]["xftf"].items()
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	385 xftf_settings = {k: v for k, v in xftf_items if v is not None}
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	386 do_xftf = xftf_settings.pop("xftf") == "true"
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	387
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	388 plot_graph = input_values["plot_graph"]
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	389 annotation = input_values["annotation"]
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	390
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	391 for key, group in keyed_data.items():
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	392 main(
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	393 group,
1 2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	394 do_calibrate=do_calibrate,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	395 calibrate_settings=calibrate_settings,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	396 do_rebin=do_rebin,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	397 do_pre_edge=do_pre_edge,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	398 pre_edge_settings=pre_edge_settings,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	399 do_xftf=do_xftf,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	400 xftf_settings=xftf_settings,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	401 plot_graph=plot_graph,
2b3115342fef planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 1cf6d7160497ba58fe16a51f00d088a20934eba6 muon-spectroscopy-computational-project parents: 0 diff changeset	402 annotation=annotation,
0 ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	403 path_key=key,
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	404 )

Mercurial > repos > muon-spectroscopy-computational-project > larch_athena

annotate larch_athena.py @ 1:2b3115342fef draft