from aaredaqlib.beamline import MXBeamline
from mxlibs3.non_standard import NonStandard

BEAMLINE = MXBeamline.X10SA.value.upper()
# print(BEAMLINE)
# gmx = NonStandard(
#                 name="GMX",
#                 setpv=f"{BEAMLINE}-ES-DF1:TRX1.VAL",
#                 getpv=f"{BEAMLINE}-ES-DF1:TRX1.RBV",
#                 speed=(f"{BEAMLINE}-ES-DF1:TRX1.VELO", f"{BEAMLINE}-ES-DF1:TRX1.VELO"),
#             )
#
# print(gmx.readback)
# gmx.move(5,True)
# print(gmx.readback)

collimator = NonStandard(
    name="Collimator",
    setpv=f"{BEAMLINE}-ES-COL:TRY.VAL",
    getpv=f"{BEAMLINE}-ES-COL:TRY.RBV",
    tolerance=0.01,
    predefs={
        "parking": 1.0,
        "measure": 41.2,
        "down": 20.0}
)

collimator.move("down")