-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathre_config.py
42 lines (31 loc) · 1.03 KB
/
re_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import datetime
import numpy as np
import bluesky.preprocessors as bpp
import bluesky.plan_stubs as bps
import bluesky.plans as bp
from bluesky.run_engine import RunEngine
from bluesky.callbacks import best_effort
from bluesky.simulators import summarize_plan
from bluesky.utils import install_kicker
from bluesky.utils import ProgressBarManager
import databroker
from databroker import Broker, temp_config
from ophyd.utils import make_dir_tree
from srw_handler import SRWFileHandler
import matplotlib.pyplot as plt
RE = RunEngine({})
bec = best_effort.BestEffortCallback()
RE.subscribe(bec)
# MongoDB backend:
db = Broker.named('local') # mongodb backend
try:
databroker.assets.utils.install_sentinels(db.reg.config, version=1)
except:
pass
RE.subscribe(db.insert)
db.reg.register_handler('srw', SRWFileHandler, overwrite=True)
db.reg.register_handler('SIREPO_FLYER', SRWFileHandler, overwrite=True)
plt.ion()
install_kicker()
ROOT_DIR = '/tmp/sirepo_flyer_data'
_ = make_dir_tree(datetime.datetime.now().year, base_path=ROOT_DIR)