forked from NSLS-II/lsdc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
daq_main_common.py
executable file
·183 lines (158 loc) · 5.73 KB
/
daq_main_common.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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
import string
import sys
import os
import time
import _thread
import daq_lib
import daq_utils
import det_lib
import beamline_support
import beamline_lib
from start_bs import plt
import atexit
import traceback
#imports to get useful things into namespace for server
from daq_macros import *
from daq_lib import *
from robot_lib import *
from beamline_lib import *
import logging
logger = logging.getLogger(__name__)
sitefilename = ""
global command_list,immediate_command_list,z
command_list = []
immediate_command_list = []
z = 25
def execute_command(command_s):
logger.info('executing command: %s' % command_s)
exec(command_s)
def pybass_init():
global message_string_pv
daq_utils.init_environment()
daq_lib.init_var_channels()
det_lib.init_detector()
daq_lib.message_string_pv = beamline_support.pvCreate(daq_utils.beamlineComm + "message_string")
daq_lib.gui_popup_message_string_pv = beamline_support.pvCreate(daq_utils.beamlineComm + "gui_popup_message_string")
beamline_lib.read_db()
logger.info("init mots")
beamline_lib.init_mots() #for now
logger.info("init done mots")
daq_lib.init_diffractometer()
try:
sitefilename = os.environ["LSDC_SITE_FILE"]
except KeyError:
gui_message("\$LSDC_SITE_FILE not defined. Questionable defaults in effect.")
if (sitefilename != ""):
if (os.path.exists(sitefilename) == 0):
error_msg = "\$LSDC_SITE_FILE: %s does not exist. Questionable defaults in effect." % sitefilename
gui_message(error_msg)
else:
process_command_file(sitefilename)
def process_command_file(command_file_name):
echo_s = "reading %s\n" % command_file_name
logger.info(echo_s)
command_file = open(command_file_name,"r")
while 1:
command = command_file.readline()
if not command:
break
else:
input_tokens = string.split(command)
if (len(input_tokens)>0):
command_string = "%s(" % input_tokens[0]
for i in range(1,len(input_tokens)):
command_string = command_string + "\"" + input_tokens[i] + "\""
if (i != (len(input_tokens)-1)):
command_string = command_string + ","
command_string = command_string + ")"
logger.info(command_string)
try:
exec(command_string);
except NameError as e:
error_string = "Unknown command in file: %s Error: %s" % (command_string, e)
logger.error(error_string)
except SyntaxError:
logger.error("Syntax error")
except KeyError as e:
logger.error("Key error. Error: %s" % e)
command_file.close()
def process_immediate_commands(frequency):
while (1):
if (len(immediate_command_list) > 0):
command = immediate_command_list.pop(0)
logger.info('immediate command: %s' % command)
process_input(command)
time.sleep(frequency)
def process_commands(frequency):
while (1):
if (len(command_list) > 0):
command = command_list.pop(0)
logger.info('command: %s' % command)
process_input(command)
def print_status_thread(frequency):
global count_list,ring_intensity
previous_image_started = 0
percent_done = 0
shutter_dead_time = .6
while 1:
time.sleep(frequency)
current_percent_done = daq_lib.get_field("state_percent")
if (daq_lib.image_started > 0):
if (start_time == 0 or daq_lib.image_started != previous_image_started):
previous_image_started = daq_lib.image_started
start_time = time.time()
now = time.time()
total_time = float(daq_lib.image_started)
if (total_time>0.0):
percent_done = int(.5+((now-start_time)/total_time*100))
else:
percent_done = 0
else:
start_time = 0
percent_done = 0
if (percent_done != current_percent_done):
daq_lib.set_field("state_percent",percent_done)
def comm_cb(value=None, char_value=None, **kw):
command = char_value
command_list.append(command)
def comm2_cb(value=None, char_value=None, **kw):
command = char_value
if not (command == "\n"):
immediate_command_list.append(command)
def process_input(command_string):
if (command_string == ""):
return
if (command_string == "q"):
sys.exit()
daq_lib.broadcast_output(time.ctime(time.time()) + "\n" + command_string)
try:
daq_lib.set_field("program_state","Program Busy")
execute_command(command_string)
except NameError as e:
error_string = "Unknown command in queue: %s Error: %s" % (command_string, e)
logger.error(error_string)
exc_type, exc_value, exc_traceback = sys.exc_info()
print("*** print_tb:")
traceback.print_tb(exc_traceback, limit=1, file=sys.stdout)
except SyntaxError:
logger.exception("Syntax error")
except KeyError as e:
logger.exception("Key error. Error: %s. Command was: %s" % (e, command_string))
except TypeError as e:
logger.exception("Type error. Error: %s" % e)
except AttributeError as e:
logger.error("Attribute Error: %s" % e)
except KeyboardInterrupt:
abort_data_collection()
logger.info("Interrupt caught by daq server\n")
if (command_string != "pause_data_collection()" and command_string != "continue_data_collection()" and command_string != "abort_data_collection()" and command_string != "unmount_after_abort()" and command_string != "no_unmount_after_abort()"):
daq_lib.set_field("program_state","Program Ready")
def run_server():
_thread.start_new_thread(process_immediate_commands,(.25,))
comm_pv = beamline_support.pvCreate(daq_utils.beamlineComm + "command_s")
beamline_support.pvPut(comm_pv,"\n")
immediate_comm_pv = beamline_support.pvCreate(daq_utils.beamlineComm + "immediate_command_s")
beamline_support.pvPut(immediate_comm_pv,"\n")
comm_pv.add_callback(comm_cb)
immediate_comm_pv.add_callback(comm2_cb)
process_commands(0.5)