larch_athena: larch_athena.py annotate

annotate larch_athena.py @ 2:a1e26990131c draft

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

author	muon-spectroscopy-computational-project
date	Mon, 04 Mar 2024 11:43:19 +0000
parents	2b3115342fef
children	82e9dd980916

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(","):
2 a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	66 for group in self.load_single_file(filepath).values():
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	67 all_groups.append(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	68
2 a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	69 merged_group = merge_groups(all_groups, xarray="energy", yarray="mu")
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	70 pre_edge_with_defaults(merged_group)
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	71 return merged_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	72
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 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	74 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	75 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	76 is_zipped: bool = False,
2 a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	77 ) -> dict:
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	78 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	79 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	80
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	81 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	82 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	83 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	84 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	85 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	86 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	87 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	88 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	89 name = repeat["group_name"]
2 a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	90 print(f"\nExtracting group {name}")
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	91 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	92 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	93 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	94 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	95
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	96 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	97 # 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	98 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	99 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	100 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	101 # 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	102 # 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	103 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	104 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	105 # 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	106 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	107 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	108 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	109 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	110
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 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	112 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	113 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	114 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	115 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	116 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	117 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	118 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	119 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	120 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	121 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	122 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	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 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	125 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	126
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 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	128 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	129 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	130
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 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	132 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	133 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	134 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	135 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	136 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	137 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	138
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 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	140 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	141 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	142
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 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	144 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	145 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	146 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	147 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	148 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	149 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	150 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	151 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	152 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	153 "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	154 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	155 )
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 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	157
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 for filename in filenames:
2 a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	159 if len(all_paths) > 1:
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	160 key = f"{dirpath}_{filename}"
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	161 else:
a1e26990131c planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 0f66842e802430e887d1c6cb7be1cc5436408fd2 muon-spectroscopy-computational-project parents: 1 diff changeset	162 key = filename
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	163 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	164 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	165 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	166
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 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	168
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 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	170 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	171 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	172
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 "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	174 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	175 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	176 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	177 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	178 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	179 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	180 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	181 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	182 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	183 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	184 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	185
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 "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	187 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	188 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	189 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	190 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	191 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	192 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	193 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	194 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	195 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	196 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	197 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	198
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 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	200 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	201 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	202 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	203 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	204
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	205
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	206 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	207 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	208 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	209 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	210 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	211 ):
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	212 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	213 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	214 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	215 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	216
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	217 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	218 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	219
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	220 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	221 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	222 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	223 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	224
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	225 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	226 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	227 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	228 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	229
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 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	231 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	232 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	233 )
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 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	235 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	236 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	237 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	238 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	239 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	240 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	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 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	243 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	244
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 # 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	246 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	247 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	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 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	250
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	251
ae2f265ecf8e planemo upload for repository https://github.com/MaterialsGalaxy/larch-tools/tree/main/larch_athena commit 5be486890442dedfb327289d597e1c8110240735 muon-spectroscopy-computational-project parents: diff changeset	252 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	253 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	254 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	255 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	256 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	257 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	258 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	259 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	260 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	261 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	262 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	263 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	264 ):
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	265 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	266 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	267 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	268 # 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	269 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	270
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	271 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	272 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	273 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	274 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	275 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	276 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	277 **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	278 )
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	279 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	280 # 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	281 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	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_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	284 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	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 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	287 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	288
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	289 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	290 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	291 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	292 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	293
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 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	295 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	296 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	297 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	298 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	299 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	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 # 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	302 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	303
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
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 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	306 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	307 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	308 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	309 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	310 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	311 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	312 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	313
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
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 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	316 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	317 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	318 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	319 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	320 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	321 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	322 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	323 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	324 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	325 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	326 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	327
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
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 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	330 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	331 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	332 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	333 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	334 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	335 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	336 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	337
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
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 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	340 # 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	341 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	342
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 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	344 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	345 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	346 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	347 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	348 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	349 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	350 )
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 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	352 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	353
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	354 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	355 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	356 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	357
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 = 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	359 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	360 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	361 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	362 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	363 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	364
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 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	366 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	367 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	368 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	369 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	370 )
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 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	372 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	373 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	374 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	375 )
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	376
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 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	378 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	379 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	380
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 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	382
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 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	384 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	385 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	386
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 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	388 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	389 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	390
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	391 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	392 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	393
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	394 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	395 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	396 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	397 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	398 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	399 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	400 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	401 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	402 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	403 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	404 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	405 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	406 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	407 )

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

annotate larch_athena.py @ 2:a1e26990131c draft