57 lines
1.8 KiB
Python
Executable File
57 lines
1.8 KiB
Python
Executable File
from diffcalc import settings
|
|
from diffcalc.dc.common import energy_to_wavelength
|
|
from diffcalc.dc.help import compile_extra_motion_commands_for_help
|
|
|
|
import diffcalc.hkl.you.calc
|
|
settings.ubcalc_strategy = diffcalc.hkl.you.calc.YouUbCalcStrategy()
|
|
settings.angles_to_hkl_function = diffcalc.hkl.you.calc.youAnglesToHkl
|
|
settings.include_reference = True
|
|
|
|
# reload to aid testing only
|
|
from diffcalc.ub import ub as _ub
|
|
|
|
reload(_ub)
|
|
from diffcalc import hardware as _hardware
|
|
#reload(_hardware)
|
|
from diffcalc.hkl.you import hkl as _hkl
|
|
reload(_hkl)
|
|
|
|
from diffcalc.ub.ub import * # @UnusedWildImport
|
|
from diffcalc.hardware import * # @UnusedWildImport
|
|
from diffcalc.hkl.you.hkl import * # @UnusedWildImport
|
|
|
|
|
|
def hkl_to_angles(h, k, l, energy=None):
|
|
"""Convert a given hkl vector to a set of diffractometer angles
|
|
|
|
return angle tuple and params dictionary
|
|
|
|
"""
|
|
if energy is None:
|
|
energy = settings.hardware.get_energy() # @UndefinedVariable
|
|
|
|
(pos, params) = hklcalc.hklToAngles(h, k, l, energy_to_wavelength(energy))
|
|
angle_tuple = settings.geometry.internal_position_to_physical_angles(pos) # @UndefinedVariable
|
|
angle_tuple = settings.hardware.cut_angles(angle_tuple) # @UndefinedVariable
|
|
|
|
return angle_tuple, params
|
|
|
|
|
|
def angles_to_hkl(angleTuple, energy=None):
|
|
"""Converts a set of diffractometer angles to an hkl position
|
|
|
|
Return hkl tuple and params dictionary
|
|
|
|
"""
|
|
if energy is None:
|
|
energy = settings.hardware.get_energy() # @UndefinedVariable
|
|
i_pos = settings.geometry.physical_angles_to_internal_position(angleTuple) # @UndefinedVariable
|
|
return hklcalc.anglesToHkl(i_pos, energy_to_wavelength(energy))
|
|
|
|
|
|
|
|
|
|
|
|
ub_commands_for_help = _ub.commands_for_help
|
|
hkl_commands_for_help = _hkl.commands_for_help + _hardware.commands_for_help + compile_extra_motion_commands_for_help()
|