-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathLHC_BBQ.py
More file actions
119 lines (89 loc) · 4.52 KB
/
LHC_BBQ.py
File metadata and controls
119 lines (89 loc) · 4.52 KB
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
import numpy as np
from . import TimberManager as tm
class BBQ(object):
def __init__(self, timber_variable_bbq, beam=0):
if not (beam == 1 or beam == 2):
raise ValueError('You need to specify which beam! (1 or 2)')
self.beam = beam
if type(timber_variable_bbq) is dict:
dict_timber = timber_variable_bbq
else:
dict_timber = tm.parse_timber_file(timber_variable_bbq, verbose=True)
self.t_stamps = np.squeeze(np.array(
dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_AMPL_1'.format(beam)].t_stamps))
self.amp_1 = np.squeeze(np.array(
dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_AMPL_1'.format(beam)].values))
self.amp_2 = np.squeeze(np.array(
dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_AMPL_2'.format(beam)].values))
self.xamp_1 = np.squeeze(np.array(
dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_X_AMPL_1'.format(beam)].values))
self.xamp_2 = np.squeeze(np.array(
dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_X_AMPL_2'.format(beam)].values))
def nearest_older_sample(self, t_obs, flag_return_time=False):
ind_min = np.argmin(np.abs(self.t_stamps - t_obs))
if self.t_stamps[ind_min] > t_obs:
ind_min -= 1
if flag_return_time:
if ind_min == -1:
return 0.*self.values[ind_min], -1
else:
return self.values[ind_min], self.t_stamps[ind_min]
else:
if ind_min == -1:
return 0.*self.values[ind_min]
else:
return self.values[ind_min]
def convert_to_array(self, v):
v.t_stamps = np.squeeze(np.array(v.t_stamps, dtype='float_'))
v.values = np.squeeze(np.array(v.values, dtype='float_'))
class BBQ_FFT(object):
def __init__(self, timber_variable_bbq, beam=0):
if not (beam == 1 or beam == 2):
raise ValueError('You need to specify which beam! (1 or 2)')
self.beam = beam
if type(timber_variable_bbq) is dict:
dict_timber = timber_variable_bbq
else:
dict_timber = tm.parse_timber_file(timber_variable_bbq, verbose=True)
self.fft_h = dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:FFT_DATA_H'.format(beam)]
self.fft_v = dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:FFT_DATA_V'.format(beam)]
self.tune_h = dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_FREQ_1'.format(beam)]
self.tune_v = dict_timber['LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_FREQ_2'.format(beam)]
# self.fft_h = self.convert_to_array(self.fft_h)
# self.fft_v = self.convert_to_array(self.fft_v)
def nearest_older_sample(self, t_obs, flag_return_time=False):
ind_min = np.argmin(np.abs(self.t_stamps - t_obs))
if self.t_stamps[ind_min] > t_obs:
ind_min -= 1
if flag_return_time:
if ind_min == -1:
return 0.*self.values[ind_min], -1
else:
return self.values[ind_min], self.t_stamps[ind_min]
else:
if ind_min == -1:
return 0.*self.values[ind_min]
else:
return self.values[ind_min]
def convert_to_array(self, v):
v.t_stamps = np.squeeze(np.array(v.t_stamps, dtype='float_'))
v.values = np.squeeze(np.array(v.values, dtype='float_'))
def get_variable_dict(beam):
print(beam)
var_dict = {}
var_dict['BBQ_HS_AMPL_1'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_AMPL_1'.format(beam)
var_dict['BBQ_HS_AMPL_2'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_AMPL_2'.format(beam)
var_dict['BBQ_HS_FREQ_1'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_FREQ_1'.format(beam)
var_dict['BBQ_HS_FREQ_2'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_FREQ_2'.format(beam)
var_dict['BBQ_HS_WIDTH_1'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_WIDTH_1'.format(beam)
var_dict['BBQ_HS_WIDTH_2'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_WIDTH_2'.format(beam)
var_dict['BBQ_HS_X_AMPL_1'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_X_AMPL_1'.format(beam)
var_dict['BBQ_HS_X_AMPL_2'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:EIGEN_X_AMPL_2'.format(beam)
var_dict['BBQ_HS_FFT_H'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:FFT_DATA_H'.format(beam)
var_dict['BBQ_HS_FFT_V'] = 'LHC.BQBBQ.CONTINUOUS_HS.B{:d}:FFT_DATA_V'.format(beam)
return var_dict
def variable_list(beams = [1, 2]):
var_list = []
for beam in beams:
var_list += list(get_variable_dict(beam).values())
return var_list