Inheritance diagram for CDCLayerConstraints:

Public Member Functions
	__init__ (self, filename='cdc-constraints.txt', rigid=True, twist=True, z_offset=False, r_scale=False, z_scale=False)

	generate (self)

	configure_collector (self, collector)

Public Attributes
	rigid = rigid
	6D CDC constraints

	twist = twist
	CDC twist constraint.

	z_offset = z_offset
	Constraint for z-offset.

	r_scale = r_scale
	Constraint for r-scale.

	z_scale = z_scale
	Constraint for z-scale.

	filename = filename
	File name.

Static Public Attributes
list	cdc
	CDC geometry hard-code -> TODO.

list	parameter
	layer parameters

Detailed Description

Various constraints for CDC layers

(Code from Claus Kleinwort)

Definition at line 173 of file constraints.py.

Constructor & Destructor Documentation

◆ init()

__init__	(	self,
		filename = 'cdc-constraints.txt',
		rigid = True,
		twist = True,
		z_offset = False,
		r_scale = False,
		z_scale = False )

filename : str
  Filename with constraints (probably generated later)
rigid : bool
  5D CDC constraints: x+y+rot + dx+dy
twist : bool
  CDC twist constraint: drot
z_offset : bool
  Constraint for Z-offset
r_scale : bool
  Constraint for R-scale
z_scale : bool
  Constraint for Z-scale

Definition at line 195 of file constraints.py.

    def __init__(self, filename='cdc-constraints.txt', rigid=True, twist=True, z_offset=False, r_scale=False, z_scale=False):
        """
        filename : str
          Filename with constraints (probably generated later)
        rigid : bool
          5D CDC constraints: x+y+rot + dx+dy
        twist : bool
          CDC twist constraint: drot
        z_offset : bool
          Constraint for Z-offset
        r_scale : bool
          Constraint for R-scale
        z_scale : bool
          Constraint for Z-scale
        """
        super().__init__(filename)
        
        self.rigid = rigid
        
        self.twist = twist
        
        self.z_offset = z_offset
        
        self.r_scale = r_scale
        
        self.z_scale = z_scale
 

Member Function Documentation

◆ configure_collector()

configure_collector	(		self,
			collector )

inherited

Can be overridden be child classes to pass additional configuration to the
MillepedeCollector (activated by the use of the constraints)

Reimplemented in CDCWireConstraints, and VXDHierarchyConstraints.

Definition at line 83 of file constraints.py.

    def configure_collector(self, collector):
        """
        Can be overridden be child classes to pass additional configuration to the
        MillepedeCollector (activated by the use of the constraints)
        """
        pass
 
 

◆ generate()

generate ( self )

Generate constraints from CDC geometry

Reimplemented from Constraints.

Definition at line 222 of file constraints.py.

    def generate(self):
        """Generate constraints from CDC geometry
        """
        print("Generating constraints for CDC layers...")
 
        def cdc_layer_label(layer, param):
            wire = 511
            wireid = Belle2.WireID(layer, wire).getEWire()
            label = Belle2.GlobalLabel()
            label.construct(Belle2.CDCAlignment.getGlobalUniqueID(), wireid, param)
            return label.label()
 
        def cmp(a, b):
            return (a > b) - (a < b)
 
        consts = []
 
        for par in self.parameter:
            nTot = 0
            for sl in self.cdc:
                nlyr = sl[1]
                for lyr in range(nlyr):
                    nTot += nlyr
        nLayer = nTot
 
        if self.rigid:
            #  all layers
            for par in self.parameter:
                if par[0] == 16:
                    continue
                const = Constraint()
 
                nTot = 0
                for sl in self.cdc:
                    nlyr = sl[1]
                    for lyr in range(nlyr):
                        label = cdc_layer_label(nTot + lyr, par[0])
                        const.add(label, 1.0)
                    nTot += nlyr
 
                consts.append(const)
 
        if self.twist:
            #  all layers
            for par in self.parameter:
                if par[0] != 16:
                    continue
                const = Constraint()
 
                nTot = 0
                for sl in self.cdc:
                    nlyr = sl[1]
                    for lyr in range(nlyr):
                        label = cdc_layer_label(nTot + lyr, par[0])
                        const.add(label, 1.0)
                    nTot += nlyr
 
                #  f.write('\n')
                consts.append(const)
 
        if self.z_offset:
            #  stereo layers (Z offset)
            par = self.parameter[2]
            const = Constraint()
 
            nTot = 0
            for sl in self.cdc:
                nlyr = sl[1]
                rInner = sl[3]
                rOuter = sl[4]
                stereo = sl[5]
                if stereo != 0.:
                    dr = (rOuter - rInner) / (nlyr - 1)
                    rWire = rInner
                    for lyr in range(nlyr):
                        label = cdc_layer_label(nTot + lyr, par[0])
                        const.add(label, stereo * rWire)
                        rWire += dr
                nTot += nlyr
 
            #  f.write('\n')
            consts.append(const)
 
        if self.r_scale:
            #  all layers 2nd
            for par in self.parameter[:2]:
                const = Constraint()
 
                nTot = 0
                for sl in self.cdc:
                    nlyr = sl[1]
                    for lyr in range(nlyr):
                        label = cdc_layer_label(nTot + lyr, par[0])
                        der = cmp(2. * float(nTot + lyr) + 0.5, float(nLayer))
                        const.add(label, der)
                    nTot += nlyr
 
                consts.append(const)
 
        if self.z_scale:
            #  stereo layers (Z -scale)
            par = self.parameter[5]
            const = Constraint()
            nTot = 0
            for sl in self.cdc:
                nlyr = sl[1]
                stereo = sl[5]
                if stereo != 0.:
                    for lyr in range(nlyr):
                        label = cdc_layer_label(nTot + lyr, par[0])
                        const.add(label, stereo)
                nTot += nlyr
 
            #  f.write('\n')
            consts.append(const)
 
        return consts
 
 

Member Data Documentation

◆ cdc

list cdc

static

Initial value:

=  [['A1', 8, 160, 16.80, 23.80, 0., -35.9, 67.9],
           ['U2', 6, 160, 25.70, 34.80, 0.068, -51.4, 98.6],
           ['A3', 6, 192, 36.52, 45.57, 0., -57.5, 132.9],
           ['V4', 6, 224, 47.69, 56.69, -0.060, -59.6, 144.7],
           ['A5', 6, 256, 58.41, 67.41, 0., -61.8, 146.8],
           ['U6', 6, 288, 69.53, 78.53, 0.064, -63.9, 148.9],
           ['A7', 6, 320, 80.25, 89.25, 0., -66.0, 151.0],
           ['V8', 6, 352, 91.37, 100.37, -0.072, -68.2, 153.2],
           ['A9', 6, 384, 102.00, 111.14, 0., -70.2, 155.3]]

CDC geometry hard-code -> TODO.

Definition at line 181 of file constraints.py.

◆ filename

filename = filename

inherited

File name.

Definition at line 72 of file constraints.py.

◆ parameter

list parameter

static

Initial value:

= [(1, 'x-offset bwd'), (2, 'y-offset bwd'), (6, 'z-rotation bwd'),

(11, 'x-offset fwd-bwd'), (12, 'y-offset fwd-bwd'), (16, 'z-rotation fwd-bwd')]

layer parameters

Definition at line 192 of file constraints.py.

◆ r_scale

r_scale = r_scale

Constraint for r-scale.

Definition at line 218 of file constraints.py.

◆ rigid

rigid = rigid

6D CDC constraints

Definition at line 212 of file constraints.py.

◆ twist

twist = twist

CDC twist constraint.

Definition at line 214 of file constraints.py.

◆ z_offset

z_offset = z_offset

Constraint for z-offset.

Definition at line 216 of file constraints.py.

◆ z_scale

z_scale = z_scale

Constraint for z-scale.

Definition at line 220 of file constraints.py.

The documentation for this class was generated from the following file:

alignment/scripts/alignment/constraints.py

Public Member Functions

Public Attributes

Static Public Attributes