box.Draw('samel')
elif 'thetaphi' in projection:
pass
else:
raise ValueError('implement drawing for projection ' + projection )
class GDetector(object):
def __init__(self, description):
self.desc = description
self.elements = [GDetectorElement(elem) for elem in self.desc.elements.values()]
def draw(self, projection):
for elem in self.elements:
elem.draw(projection)
if __name__ == '__main__':
from ROOT import TCanvas, TH2F
from heppy_fcc.fastsim.detectors.CMS import CMS
from heppy_fcc.display.core import Display
cms = CMS()
gcms = GDetector(cms)
display = Display()
display.register(gcms, 0)
display.draw()
