def __init__(self):
self.linac = Linac()
self.ring = Ring()
self.tDeviceManager = None
self.svgFile = None
def __init__(self):
self.linac = Linac()
self.ring = Ring()
self.tDeviceManager = None
self.svgFile = None
class SvgDrawer:
def __init__(self):
self.linac = Linac()
self.ring = Ring()
self.tDeviceManager = None
self.svgFile = None
def loadSvg(self, svgPath):
svgTree = etree.parse(svgPath)
self.svgFile = svg.SVG()
self.svgFile.setSvg(svgTree.getroot())
def saveSvg(self, svgPath):
svgRoot = self.svgFile.getSvg()
toWrite = etree.tostring(svgRoot, pretty_print=True)
fd = open(svgPath, 'w')
fd.write(toWrite)
def addSectionToLinac(self, name, colour, width):
return self.linac.addSection(name, colour, width)
def addSectionToRing(self, name, colour, angle):
return self.ring.addSection(name, colour, angle)
def addSubsection(self, section, subsectionName, colour, widthOrAngle):
section.addSubsection(subsectionName, colour, widthOrAngle)
def drawAll(self):
self.drawAccelerator()
if TangoDeviceManager.canConnectToTango():
self.tDeviceManager = TangoDeviceManager()
if self.tDeviceManager.initialized:
devices = self.tDeviceManager.getDevices()
self.drawIcons(devices)
self.drawDevices(devices)
def initializeBlankSvg(self):
blankSVGpath = '../src/blank.svg'
self.loadSvg(blankSVGpath)
def drawAccelerator(self):
self.linac.updateSvg()
self.ring.updateSvg()
def drawIcons(self, devices):
icons = set()
for device in devices:
icons.add(device.icon)
for icon in icons:
icon.updateSvg()
def drawDevices(self, devices):
self.addDeviceToAccelerator(devices)
devices = self.sortDevices()
self.drawDevicesOnSvg(devices)
def addDeviceToAccelerator(self, devices):
for device in devices:
if device.isLinacElement():
self.linac.addDevice(device)
elif device.isRingElement():
self.ring.addDevice(device)
self.ring.assignDevicesBeforeDrawing()
self.linac.assignDevicesBeforeDrawing()
def sortDevices(self):
sortedDevices = list()
sortedDevices.extend(self.linac.getAllDevicesSorted())
sortedDevices.extend(self.ring.getAllDevicesSorted())
return sortedDevices
def drawDevicesOnSvg(self, devices):
for device in devices:
device.updateSvg()
def getAllSections(self):
sections = list()
sections.extend(self.linac.sections)
sections.extend(self.ring.sections)
return sections
class SvgDrawer:
def __init__(self):
self.linac = Linac()
self.ring = Ring()
self.tDeviceManager = None
self.svgFile = None
def loadSvg(self, svgPath):
svgTree = etree.parse(svgPath)
self.svgFile = svg.SVG()
self.svgFile.setSvg(svgTree.getroot())
def saveSvg(self, svgPath):
svgRoot = self.svgFile.getSvg()
toWrite = etree.tostring(svgRoot, pretty_print=True)
fd = open(svgPath, 'w')
fd.write(toWrite)
def addSectionToLinac(self, name, colour, width):
return self.linac.addSection(name, colour, width)
def addSectionToRing(self, name, colour, angle):
return self.ring.addSection(name, colour, angle)
def addSubsection(self, section, subsectionName, colour, widthOrAngle):
section.addSubsection(subsectionName, colour, widthOrAngle)
def drawAll(self):
self.drawAccelerator()
if TangoDeviceManager.canConnectToTango():
self.tDeviceManager = TangoDeviceManager()
if self.tDeviceManager.initialized:
devices = self.tDeviceManager.getDevices()
self.drawIcons(devices)
self.drawDevices(devices)
def initializeBlankSvg(self):
blankSVGpath = '../src/blank.svg'
self.loadSvg(blankSVGpath)
def drawAccelerator(self):
self.linac.updateSvg()
self.ring.updateSvg()
def drawIcons(self, devices):
icons = set()
for device in devices:
icons.add(device.icon)
for icon in icons:
icon.updateSvg()
def drawDevices(self, devices):
self.addDeviceToAccelerator(devices)
devices = self.sortDevices()
self.drawDevicesOnSvg(devices)
def addDeviceToAccelerator(self, devices):
for device in devices:
if device.isLinacElement():
self.linac.addDevice(device)
elif device.isRingElement():
self.ring.addDevice(device)
self.ring.assignDevicesBeforeDrawing()
self.linac.assignDevicesBeforeDrawing()
def sortDevices(self):
sortedDevices = list()
sortedDevices.extend(self.linac.getAllDevicesSorted())
sortedDevices.extend(self.ring.getAllDevicesSorted())
return sortedDevices
def drawDevicesOnSvg(self, devices):
for device in devices:
device.updateSvg()
def getAllSections(self):
sections = list()
sections.extend(self.linac.sections)
sections.extend(self.ring.sections)
return sections