def init_display(self): self.display = Display(['xy', 'yz']) self.gdetector = GDetector(self.detector) self.display.register(self.gdetector, layer=0, clearable=False) self.is_display = True
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()