def test_two_streams(RE):
def pretty_print(name, doc):
pprint.pprint(name)
pprint.pprint(doc)
def serializer_factory(name, start_doc):
serializer = Serializer("xdi")
serializer("start", start_doc)
return [serializer], []
sd = SupplementalData()
RE.preprocessors.append(sd)
sd.baseline = [det1, motor1, motor2]
RE.subscribe(pretty_print)
RE.subscribe(RunRouter([serializer_factory]))
suitcase_meta_data = {"config": xdi_file_template}
xdi_meta_data = {
"Element_symbol": "A",
"Element_edge": "K",
"Mono_d_spacing": 10.0
}
nx_meta_data = {
"Source": {
"name": "NSLS-II"
},
"Instrument": {
"name": "BMM"
},
"Beam": {
"incident_energy": 1000.0
},
}
dets = [det1, det2]
RE(
count(dets, num=5),
md={
"suitcase-xdi": suitcase_meta_data,
"NX": nx_meta_data,
"XDI": xdi_meta_data,
},
)
def pretty_print(name, doc):
pprint(name)
pprint(doc)
def serializer_factory(name, start_doc):
serializer = Serializer("xdi")
serializer("start", start_doc)
return [serializer], []
RE = RunEngine({})
sd = SupplementalData()
RE.preprocessors.append(sd)
sd.baseline = [det3, motor1, motor2]
RE.subscribe(pretty_print)
RE.subscribe(RunRouter([serializer_factory]))
suitcase_meta_data = {"config-file-path": "XDI.toml"}
xdi_meta_data = {"Element_symbol": "A", "Element_edge": "K", "Mono_d_spacing": 10.0}
nx_meta_data = {
"Source": {"name": "NSLS-II"},
"Instrument": {"name": "BMM"},
"Beam": {"incident_energy": 1000.0},
}
dets = [det1, det2]
# Setup baseline measurements for scans
print(f'Loading {__file__}...')
from bluesky.preprocessors import SupplementalData
sd = SupplementalData()
sd.baseline = [
ring_current,
fe,
energy,
dcm,
hfm, # Front-end slits, Undulator/Bragg, HDCM, HFM
slt_wb,
slt_pb,
slt_ssa, # White-, Pink-Beam slits, SSA
jjslits,
attenuators, # JJ slits, Attenuator Box
nanoKB,
nano_vlm_stage,
nano_det,
temp_nanoKB, # nanoKBs, VLM, Detector, Temperatures
nano_stage,
nanoKB_interferometer
] # coarse/fine stages, sample interferometer
RE.preprocessors.append(sd)
bec.disable_baseline()
