geometryDiff Namespace Reference

Functions
def	loopover
def	matrixmult
def	rotFromEuler
def	rotFromPhi
def	transpose

Variables
int	epsilon = 1
tuple	geom1 = MuonGeometry(file(args[0]))
tuple	geom2 = MuonGeometry(file(args[1]))
tuple	opts = dict(opts)
tuple	sqrtepsilon = sqrt(epsilon)
string	usage = "Usage: geometryDiff.py [-h|--help] [-e|--epsilon epsilon] geometry1.xml geometry2.xml"

Function Documentation

def geometryDiff.loopover

(

which

)

Definition at line 60 of file geometryDiff.py.

References abs, matrixmult(), rotFromEuler(), rotFromPhi(), and transpose().

def loopover(which):
  if which == "DT":
    keys = geom1.dt.keys()
    keys.sort(dtorder)

  elif which == "CSC":
    keys = geom1.csc.keys()
    keys.sort(cscorder)

  else: raise Exception

  for key in keys:
    if which == "DT":
      g1 = geom1.dt[key]
      g2 = geom2.dt[key]
    else:
      g1 = geom1.csc[key]
      g2 = geom2.csc[key]

    if g1.relativeto != g2.relativeto:
      print "%s %s relativeto=\"%s\" versus relativeto=\"%s\"" % (which, str(key), g1.relativeto, g2.relativeto)

    if abs(g1.x - g2.x) > epsilon or abs(g1.y - g2.y) > epsilon or abs(g1.z - g2.z) > epsilon:
      print "%s %s position difference: (%g, %g, %g) - (%g, %g, %g) = (%g, %g, %g)" % \
            (which, str(key), g1.x, g1.y, g1.z, g2.x, g2.y, g2.z, g1.x - g2.x, g1.y - g2.y, g1.z - g2.z)

    if "phix" in g1.__dict__:
      g1type = "phi"
      g1a, g1b, g1c = g1.phix, g1.phiy, g1.phiz
      g1rot = rotFromPhi(g1)
    else:
      g1type = "euler"
      g1a, g1b, g1c = g1.alpha, g1.beta, g1.gamma
      g1rot = rotFromEuler(g1)

    if "phix" in g2.__dict__:
      g2type = "phi"
      g2a, g2b, g2c = g2.phix, g2.phiy, g2.phiz
      g2rot = rotFromPhi(g2)
    else:
      g2type = "euler"
      g2a, g2b, g2c = g2.alpha, g2.beta, g2.gamma
      g2rot = rotFromEuler(g2)
    
    diff = matrixmult(g1rot, transpose(g2rot))
    if abs(diff[0][0] - 1.) > sqrtepsilon or abs(diff[1][1] - 1.) > sqrtepsilon or abs(diff[2][2] - 1.) > sqrtepsilon or \
       abs(diff[0][1]) > epsilon or abs(diff[0][2]) > epsilon or abs(diff[1][2]) > epsilon:
      print "%s %s rotation difference: %s(%g, %g, %g) - %s(%g, %g, %g) = %s" % \
            (which, str(key), g1type, g1a, g1b, g1c, g2type, g2a, g2b, g2c, str(diff))

loopover("DT")
loopover("CSC")

def geometryDiff.matrixmult	(		a,
			b
	)

Definition at line 34 of file geometryDiff.py.

Referenced by loopover(), and rotFromPhi().

def matrixmult(a, b):
  return [[sum([i*j for i, j in zip(row, col)]) for col in zip(*b)] for row in a]