Belle II Software  release-05-01-25
utils.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
6  """ Class to generate data for AlignmentGenerator module
7  Can be used to train VXD misalignment by storing wrong alignment
8  and trying to reach nominal values (zeros)
9  """
10 
11  def __init__(self):
12  """ Constructor """
13 
14  self.alignment = []
15 
16  def get_name(self):
17  """ get object name """
18  return self.name
19 
20  def get_data(self):
21  """ get alignment data """
22  return self.alignment
23 
24  def align_sensors(self, vxdid, param_id, distribution, value):
25  """ align sensors by VxdID pattern """
26  self.alignment.append(str(vxdid + ',' + str(param_id) + ',' + distribution + ',' + str(value)))
27 
28  def gen_u_gaus(self, vxdid, value):
29  """ generate u coordinate shift for VxdID mask with gausian distribution of given width """
30  self.align_sensors(vxdid, 1, 'gaus', value)
31 
32  def gen_v_gaus(self, vxdid, value):
33  """ generate v coordinate shift for VxdID mask with gausian distribution of given width """
34  self.align_sensors(vxdid, 2, 'gaus', value)
35 
36  def gen_w_gaus(self, vxdid, value):
37  """ generate w coordinate shift for VxdID mask with gausian distribution of given width """
38  self.align_sensors(vxdid, 3, 'gaus', value)
39 
40  def gen_alpha_gaus(self, vxdid, value):
41  """ generate alpha rotation for VxdID mask with gausian distribution of given width """
42  self.align_sensors(vxdid, 4, 'gaus', value)
43 
44  def gen_beta_gaus(self, vxdid, value):
45  """ generate beta rotation for VxdID mask with gausian distribution of given width """
46  self.align_sensors(vxdid, 5, 'gaus', value)
47 
48  def gen_gamma_gaus(self, vxdid, value):
49  """ generate gamma rotation coordinate shift for VxdID mask with gausian distribution of given width """
50  self.align_sensors(vxdid, 6, 'gaus', value)
51 
52  def gen_u_uniform(self, vxdid, value):
53  """ generate u coordinate shift for VxdID mask with uniform distribution of given width """
54  self.align_sensors(vxdid, 1, 'gaus', value)
55 
56  def gen_v_uniform(self, vxdid, value):
57  """ generate v coordinate shift for VxdID mask with uniform distribution of given width """
58  self.align_sensors(vxdid, 2, 'gaus', value)
59 
60  def gen_w_uniform(self, vxdid, value):
61  """ generate w coordinate shift for VxdID mask with uniform distribution of given width """
62  self.align_sensors(vxdid, 3, 'gaus', value)
63 
64  def gen_alpha_uniform(self, vxdid, value):
65  """ generate alpha rotation for VxdID mask with uniform distribution of given width """
66  self.align_sensors(vxdid, 4, 'gaus', value)
67 
68  def gen_beta_uniform(self, vxdid, value):
69  """ generate beta rotation for VxdID mask with uniform distribution of given width """
70  self.align_sensors(vxdid, 5, 'gaus', value)
71 
72  def gen_gamma_uniform(self, vxdid, value):
73  """ generate gamma rotation coordinate shift for VxdID mask with uniform distribution of given width """
74  self.align_sensors(vxdid, 6, 'gaus', value)
75 
76  def set_u(self, vxdid, value):
77  """ set u coordinate shift for VxdID """
78  self.align_sensors(vxdid, 1, 'fix', value)
79 
80  def set_v(self, vxdid, value):
81  """ set v coordinate shift for VxdID """
82  self.align_sensors(vxdid, 2, 'fix', value)
83 
84  def set_w(self, vxdid, value):
85  """ set w coordinate shift for VxdID """
86  self.align_sensors(vxdid, 3, 'fix', value)
87 
88  def set_alpha(self, vxdid, value):
89  """ set alpha angle for VxdID mask """
90  self.align_sensors(vxdid, 4, 'fix', value)
91 
92  def set_beta(self, vxdid, value):
93  """ set beta angle for VxdID mask """
94  self.align_sensors(vxdid, 5, 'fix', value)
95 
96  def set_gamma(self, vxdid, value):
97  """ set gamma angle for VxdID mask """
98  self.align_sensors(vxdid, 6, 'fix', value)
99 
100  def set_all(self, vxdid, value):
101  """ Set all shifts/rotation to given value (usually 0.) for VxdID mask """
102  self.set_u(vxdid, value)
103  self.set_v(vxdid, value)
104  self.set_w(vxdid, value)
105  self.set_alpha(vxdid, value)
106  self.set_beta(vxdid, value)
107  self.set_gamma(vxdid, value)
alignment.utils.AlignmentGeneratorConfig.set_w
def set_w(self, vxdid, value)
Definition: utils.py:84
alignment.utils.AlignmentGeneratorConfig.gen_beta_gaus
def gen_beta_gaus(self, vxdid, value)
Definition: utils.py:44
alignment.utils.AlignmentGeneratorConfig.alignment
alignment
alignment data = list of VxdID mask, param id, generation distribution and value
Definition: utils.py:14
alignment.utils.AlignmentGeneratorConfig.set_all
def set_all(self, vxdid, value)
Definition: utils.py:100
alignment.utils.AlignmentGeneratorConfig.set_gamma
def set_gamma(self, vxdid, value)
Definition: utils.py:96
alignment.utils.AlignmentGeneratorConfig.gen_v_gaus
def gen_v_gaus(self, vxdid, value)
Definition: utils.py:32
alignment.utils.AlignmentGeneratorConfig.gen_alpha_uniform
def gen_alpha_uniform(self, vxdid, value)
Definition: utils.py:64
alignment.utils.AlignmentGeneratorConfig.gen_gamma_gaus
def gen_gamma_gaus(self, vxdid, value)
Definition: utils.py:48
alignment.utils.AlignmentGeneratorConfig.set_alpha
def set_alpha(self, vxdid, value)
Definition: utils.py:88
alignment.utils.AlignmentGeneratorConfig.set_u
def set_u(self, vxdid, value)
Definition: utils.py:76
alignment.utils.AlignmentGeneratorConfig.gen_gamma_uniform
def gen_gamma_uniform(self, vxdid, value)
Definition: utils.py:72
alignment.utils.AlignmentGeneratorConfig.get_name
def get_name(self)
Definition: utils.py:16
alignment.utils.AlignmentGeneratorConfig.gen_beta_uniform
def gen_beta_uniform(self, vxdid, value)
Definition: utils.py:68
alignment.utils.AlignmentGeneratorConfig.__init__
def __init__(self)
Definition: utils.py:11
alignment.utils.AlignmentGeneratorConfig.gen_u_uniform
def gen_u_uniform(self, vxdid, value)
Definition: utils.py:52
alignment.utils.AlignmentGeneratorConfig.gen_u_gaus
def gen_u_gaus(self, vxdid, value)
Definition: utils.py:28
alignment.utils.AlignmentGeneratorConfig
Definition: utils.py:5
alignment.utils.AlignmentGeneratorConfig.gen_alpha_gaus
def gen_alpha_gaus(self, vxdid, value)
Definition: utils.py:40
alignment.utils.AlignmentGeneratorConfig.set_beta
def set_beta(self, vxdid, value)
Definition: utils.py:92
alignment.utils.AlignmentGeneratorConfig.gen_w_gaus
def gen_w_gaus(self, vxdid, value)
Definition: utils.py:36
alignment.utils.AlignmentGeneratorConfig.gen_w_uniform
def gen_w_uniform(self, vxdid, value)
Definition: utils.py:60
alignment.utils.AlignmentGeneratorConfig.set_v
def set_v(self, vxdid, value)
Definition: utils.py:80
alignment.utils.AlignmentGeneratorConfig.align_sensors
def align_sensors(self, vxdid, param_id, distribution, value)
Definition: utils.py:24
alignment.utils.AlignmentGeneratorConfig.get_data
def get_data(self)
Definition: utils.py:20
alignment.utils.AlignmentGeneratorConfig.gen_v_uniform
def gen_v_uniform(self, vxdid, value)
Definition: utils.py:56