예제 #1
0
파일: root.py 프로젝트: hcsullivan12/gegede
def make_shape(bucket, shape):
    '''
    Make a TGeo shape out of the object
    '''
    typename = type(shape).__name__
    shape, _ = wash_units(shape)

    obj = None
    if typename == 'Box':
        obj = bucket.make(ROOT.TGeoBBox, shape.name, 
                          shape.dx, shape.dy, shape.dz)

    if typename == 'Tubs':
        obj = bucket.make(ROOT.TGeoTubeSeg, shape.name,
                          shape.rmin, shape.rmax, shape.dz,
                          shape.sphi, shape.sphi+shape.dphi)
    
    if typename == 'Sphere':
        obj = bucket.make(ROOT.TGeoSphere, shape.name,
                          shape.rmin, shape.rmax,
                          shape.stheta, shape.stheta+shape.dtheta,
                          shape.sphi, shape.sphi+shape.dphi)

    # etc....  Grow this as gegede.schema.Schema.shapes grows....

    if obj:                     # for generic look up later
        bucket.add(obj, 'SHAPE')

    return obj
예제 #2
0
def test_wash_units():
    Thing = namedtuple("Thing", "name l a")
    t1 = Thing('t1', Q("1mm"), Q("1degree"))
    n1, u1 = wash_units(t1, lunit='meter', aunit='radian')
    assert n1.l == 0.001, n1
    assert abs(n1.a - 3.14159 / 180) < 0.01, n1
    assert 'meter' == u1['lunit'], u1
    assert 'radian' == u1['aunit'], u1

    OtherThing = namedtuple("OtherThing", "name l1 l2")
    ot2 = OtherThing('ot2', Q("1mm"), Q("2meter"))
    n2, u2 = wash_units(ot2, lunit='meter', aunit='radian')
    assert n2.l1 == 0.001, n2
    assert n2.l2 == 2.0, n2
    assert 'meter' == u2['lunit'], u2
    assert 'aunit' not in u2, u2
예제 #3
0
파일: test_util.py 프로젝트: DUNE/gegede
def test_wash_units():
    Thing = namedtuple("Thing", "name l a")
    t1 = Thing('t1', Q("1mm"), Q("1degree"))
    n1,u1 = wash_units(t1, lunit='meter', aunit='radian')
    assert n1.l == 0.001, n1
    assert abs(n1.a - 3.14159/180) < 0.01, n1
    assert 'meter' == u1['lunit'], u1
    assert 'radian' == u1['aunit'], u1


    OtherThing = namedtuple("OtherThing", "name l1 l2")
    ot2 = OtherThing('ot2', Q("1mm"), Q("2meter"))
    n2,u2 = wash_units(ot2, lunit='meter', aunit='radian')
    assert n2.l1 == 0.001, n2
    assert n2.l2 == 2.0, n2
    assert 'meter' == u2['lunit'], u2
    assert 'aunit' not in u2, u2