Skip to content

Commit 2531312

Browse files
committed
Refactors CryspyCalculator and updates SPDX compliance
1 parent 8e29578 commit 2531312

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

50 files changed

+75
-104
lines changed

src/easydiffraction/analysis/calculators/cryspy.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,8 @@ def _recreate_cryspy_dict(
160160
cryspy_model_id = f'crystal_{sample_model.name}'
161161
cryspy_model_dict = cryspy_dict[cryspy_model_id]
162162

163+
# Update sample model parameters
164+
163165
# Cell
164166
cryspy_cell = cryspy_model_dict['unit_cell_parameters']
165167
cryspy_cell[0] = sample_model.cell.length_a.value
@@ -186,15 +188,18 @@ def _recreate_cryspy_dict(
186188
for idx, atom_site in enumerate(sample_model.atom_sites):
187189
cryspy_biso[idx] = atom_site.b_iso.value
188190

189-
# ---------- Update experiment parameters ----------
191+
# Update experiment parameters
192+
190193
if experiment.type.beam_mode.value == BeamModeEnum.CONSTANT_WAVELENGTH:
191194
cryspy_expt_name = f'pd_{experiment.name}'
192195
cryspy_expt_dict = cryspy_dict[cryspy_expt_name]
196+
193197
# Instrument
194198
cryspy_expt_dict['offset_ttheta'][0] = np.deg2rad(
195199
experiment.instrument.calib_twotheta_offset.value
196200
)
197201
cryspy_expt_dict['wavelength'][0] = experiment.instrument.setup_wavelength.value
202+
198203
# Peak
199204
cryspy_resolution = cryspy_expt_dict['resolution_parameters']
200205
cryspy_resolution[0] = experiment.peak.broad_gauss_u.value
@@ -206,13 +211,15 @@ def _recreate_cryspy_dict(
206211
elif experiment.type.beam_mode.value == BeamModeEnum.TIME_OF_FLIGHT:
207212
cryspy_expt_name = f'tof_{experiment.name}'
208213
cryspy_expt_dict = cryspy_dict[cryspy_expt_name]
214+
209215
# Instrument
210216
cryspy_expt_dict['zero'][0] = experiment.instrument.calib_d_to_tof_offset.value
211217
cryspy_expt_dict['dtt1'][0] = experiment.instrument.calib_d_to_tof_linear.value
212218
cryspy_expt_dict['dtt2'][0] = experiment.instrument.calib_d_to_tof_quad.value
213219
cryspy_expt_dict['ttheta_bank'] = np.deg2rad(
214220
experiment.instrument.setup_twotheta_bank.value
215221
)
222+
216223
# Peak
217224
cryspy_sigma = cryspy_expt_dict['profile_sigmas']
218225
cryspy_sigma[0] = experiment.peak.broad_gauss_sigma_0.value

src/easydiffraction/analysis/calculators/pdffit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1+
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
2+
# SPDX-License-Identifier: BSD-3-Clause
13
"""PDF calculation backend using diffpy.pdffit2 if available.
24
35
The class adapts the engine to EasyDiffraction calculator interface and
46
silences stdio on import to avoid noisy output in notebooks and logs.
57
"""
6-
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
7-
# SPDX-License-Identifier: BSD-3-Clause
88

99
import os
1010
import re

src/easydiffraction/analysis/categories/aliases.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1+
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
2+
# SPDX-License-Identifier: BSD-3-Clause
13
"""Alias category for mapping friendly names to parameter UIDs.
24
35
Defines a small record type used by analysis configuration to refer to
46
parameters via readable labels instead of raw unique identifiers.
57
"""
6-
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
7-
# SPDX-License-Identifier: BSD-3-Clause
88

99
from easydiffraction.core.category import CategoryCollection
1010
from easydiffraction.core.category import CategoryItem

src/easydiffraction/analysis/categories/constraints.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1+
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
2+
# SPDX-License-Identifier: BSD-3-Clause
13
"""Simple symbolic constraint between parameters.
24
35
Represents an equation of the form ``lhs_alias = rhs_expr`` where
46
``rhs_expr`` is evaluated elsewhere by the analysis engine.
57
"""
6-
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
7-
# SPDX-License-Identifier: BSD-3-Clause
88

99
from easydiffraction.core.category import CategoryCollection
1010
from easydiffraction.core.category import CategoryItem

src/easydiffraction/analysis/categories/joint_fit_experiments.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1+
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
2+
# SPDX-License-Identifier: BSD-3-Clause
13
"""Joint-fit experiment weighting configuration.
24
35
Stores per-experiment weights to be used when multiple experiments are
46
fitted simultaneously.
57
"""
6-
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
7-
# SPDX-License-Identifier: BSD-3-Clause
88

99
from easydiffraction.core.category import CategoryCollection
1010
from easydiffraction.core.category import CategoryItem

src/easydiffraction/experiments/categories/experiment_type.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1+
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
2+
# SPDX-License-Identifier: BSD-3-Clause
13
"""Experiment type descriptor (form, beam, probe, scattering).
24
35
This lightweight container stores the categorical attributes defining
46
an experiment configuration and handles CIF serialization via
57
``CifHandler``.
68
"""
7-
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
8-
# SPDX-License-Identifier: BSD-3-Clause
99

1010
from easydiffraction.core.category import CategoryItem
1111
from easydiffraction.core.parameters import StringDescriptor

src/easydiffraction/experiments/experiment/bragg_sc.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
1-
"""Single crystal experiment types and helpers."""
2-
31
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
42
# SPDX-License-Identifier: BSD-3-Clause
3+
"""Single crystal experiment types and helpers."""
54

65
from __future__ import annotations
76

src/easydiffraction/sample_models/sample_models.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
22
# SPDX-License-Identifier: BSD-3-Clause
33

4-
54
from typeguard import typechecked
65

76
from easydiffraction.core.datablock import DatablockCollection

src/easydiffraction/utils/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
22
# SPDX-License-Identifier: BSD-3-Clause
33

4-
import importlib
54
import io
65
import json
76
import os
@@ -10,6 +9,7 @@
109
import zipfile
1110
from importlib.metadata import PackageNotFoundError
1211
from importlib.metadata import version
12+
from importlib.util import find_spec
1313
from typing import List
1414
from typing import Optional
1515
from urllib.parse import urlparse
@@ -409,7 +409,7 @@ def is_colab() -> bool:
409409
bool: True if running in Google Colab PyCharm, False otherwise.
410410
"""
411411
try:
412-
return importlib.util.find_spec('google.colab') is not None
412+
return find_spec('google.colab') is not None
413413
except ModuleNotFoundError:
414414
return False
415415

tests/unit/easydiffraction/analysis/calculators/test_cryspy.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
# SPDX-FileCopyrightText: 2021-2025 EasyDiffraction contributors <https://github.com/easyscience/diffraction>
22
# SPDX-License-Identifier: BSD-3-Clause
33

4-
54
def test_module_import():
65
import easydiffraction.analysis.calculators.cryspy as MUT
76

0 commit comments

Comments
 (0)