def setUp(self):
blankSVGpath = './blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
self.svgFile = svg.SVG()
self.svgFile.setSvg(svgRoot)
self.linac = Linac()
def testFoundElement(self):
linac = Linac()
linac.addSection("TEST-abc")
linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
sectionRectNode = self.svgFile.getElementById("section1bottomRect", zoomNode)
self.assertEqual(sectionRectNode.attrib["id"], "section1bottomRect")
self.assertEqual(sectionRectNode.attrib["width"], "630")
def testFoundElement(self):
linac = Linac()
linac.addSection("TEST-abc")
linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
sectionRectNode = self.svgFile.getElementById("section1bottomRect",
zoomNode)
self.assertEqual(sectionRectNode.attrib["id"], "section1bottomRect")
self.assertEqual(sectionRectNode.attrib["width"], "630")
def testAssigningSvgCoordinate(self):
device = Device("abc/testDevice", None, None)
self.assertEqual(device.svgCoordinateX, None)
linac = Linac()
linac.addSection("abc")
linac.addDevice(device)
device.assignSvgCoordinatesForLinacDevice()
testSection = linac.getSection("abc")
sectionStartCoordinate = testSection.startCoordinate
distanceBetweenDevices = testSection.getDistanceBetweenDevices()
self.assertEqual(device.svgCoordinateX,
sectionStartCoordinate + distanceBetweenDevices)
def testDrawDevice(self):
icon = Icon("symbol-quadrupole.svg")
device = Device("I-K01/VAC/I-K01CAB05-VAC-IPCU1", icon, (113, 122))
linac = Linac()
linac.addSection("I-K01")
linac.addDevice(device)
blankSVGpath = 'blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
svgFile = svg.SVG()
svgFile.setSvg(svgRoot)
device.updateSvg()
vacNode = svgFile.getSubsystemZoomNode("VAC")
deviceNode = svgFile.getElementById("ik01vacik01cab05vacipcu1",
vacNode)
self.assertEqual(
deviceNode.attrib["{http://www.w3.org/1999/xlink}href"],
"#symbol-quadrupole")
descriptionNode = svgFile.getElementById(
"ik01vacik01cab05vacipcu1Desc", deviceNode)
self.assertEqual(descriptionNode.text,
"device=I-K01/VAC/I-K01CAB05-VAC-IPCU1")
#!/usr/bin/python
import src.svg as svg
from src.Linac import Linac
from lxml import etree
from src.TangoDeviceManager import TangoDeviceManager
blankSVGpath = './src/blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
svgFile = svg.SVG()
svgFile.setSvg(svgRoot)
print "SVG file loaded "
L = Linac()
L.addSection("AB-01")
L.addSection("BB02")
L.addSection("CC03")
L.addLastLongSection()
L.updateSvg()
svgRoot = svgFile.getSvg()
toWrite = etree.tostring(svgRoot, pretty_print=True)
fd = open("new.svg", 'w')
fd.write(toWrite)
class TestLinac(unittest.TestCase):
def setUp(self):
blankSVGpath = './blank.svg'
svgTree = etree.parse(blankSVGpath)
svgRoot = svgTree.getroot()
self.svgFile = svg.SVG()
self.svgFile.setSvg(svgRoot)
self.linac = Linac()
def testSectionColour(self):
firstColour = self.linac.getNextSectionColour()
self.assertEqual(firstColour, "#55ffdd")
self.linac.addSection("AB01")
nextSectionColour = self.linac.getNextSectionColour()
self.assertEqual(nextSectionColour, "#b3b3b3")
self.linac.addSection("TEST")
nextSectionColour = self.linac.getNextSectionColour()
self.assertEqual(nextSectionColour, "grey")
def testNumberOfSections(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.linac.addSection("A")
self.assertEqual(self.linac.numberOfSections(), 1)
self.linac.addSection("B")
self.linac.addSection("C")
self.assertEqual(self.linac.numberOfSections(), 3)
def testGetSection(self):
self.assertEqual(self.linac.getSection("A"), None)
self.linac.addSection("A")
self.assertNotEqual(self.linac.getSection("A"), None)
sectionA = self.linac.sections[0]
self.assertEqual(self.linac.getSection("A"), sectionA)
self.linac.addSection("TEST-01")
sectionTest = self.linac.sections[1]
self.assertEqual(self.linac.getSection("TEST-01"), sectionTest)
def testAddSection(self):
self.assertEqual(self.linac.numberOfSections(), 0)
addedSection = self.linac.addSection("test002")
self.linac.updateSvg()
newSection = self.linac.getSection("test002")
self.assertEqual(newSection.longName, "test002")
self.assertEqual(addedSection, newSection)
zoomNode = self.svgFile.getZoom2Background()
bottomRect = self.svgFile.getElementById("section1bottomRect",
zoomNode)
self.assertEqual(bottomRect.attrib["x"], "1885")
self.assertEqual(bottomRect.attrib["y"], "3685")
self.assertEqual(self.linac.numberOfSections(), 1)
def testAddLastLongSection(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.linac.addSection("test001")
self.linac.addLastLongSection()
self.assertEqual(self.linac.numberOfSections(), 2)
firstSection = self.linac.getSection("test001")
lastSection = self.linac.getSection("")
self.assertNotEqual(lastSection, None)
self.assertEqual(firstSection.width, 630)
self.assertEqual(lastSection.width, self.linac.width - 630)
self.linac.updateSvg()
zoomNode = self.svgFile.getZoom2Background()
bottomRect = self.svgFile.getElementById("section2bottomRect",
zoomNode)
self.assertEqual(bottomRect.attrib["x"], "2515")
def testAddDevice(self):
device = Device("ABC/MAG/device123", None, None)
anotherDevice = Device("ABC/VAC/dev7", None, None)
self.linac.addSection("ABC")
section = self.linac.getSection("ABC")
self.assertEqual(section.numberOfDevices(), 0)
self.assertEqual(section.getDevice(0), None)
self.linac.addDevice(device)
self.assertEqual(section.getDevice(0), device)
self.assertEqual(section.getDevice(1), None)
self.linac.addDevice(anotherDevice)
self.assertEqual(section.getDevice(1), anotherDevice)
self.assertEqual(section.numberOfDevices(), 2)
def testAddDeviceToSubsection(self):
section = self.linac.addSection("I-S01", None, 200)
subsection = section.addSubsection("I-S01B", None, 50)
device = Device("I-S01B/ab/cd", None, [0, 20])
self.linac.addDevice(device)
self.assertEqual(device.section, subsection)
def testComputeCoordinate(self):
self.assertEqual(self.linac.numberOfSections(), 0)
self.assertEqual(self.linac.computeNewSectionCoordinate(), 1885)
self.linac.addSection("FOO")
self.assertEqual(self.linac.computeNewSectionCoordinate(), 2515)
def testAssignDevicesBeforeDrawing(self):
section = self.linac.addSection("I-K01", None, 200)
subsection = section.addSubsection("I-K01A", None, 50)
firstDevice = Device("I-K01/ab/cd", None, [12, 20])
secondDevice = Device("I-K01A/ef/gh", None, [5, 20])
thirdDevice = Device("I-K01A/ij/kl", None, [15, 20])
self.linac.addDevice(firstDevice)
self.linac.addDevice(secondDevice)
self.linac.addDevice(thirdDevice)
self.assertEqual(firstDevice.section, section)
self.linac.assignDevicesBeforeDrawing()
self.assertEqual(firstDevice.section, subsection)
def testGetAllDevicesSorted(self):
section = self.linac.addSection("A", None, 100)
subsection = section.addSubsection("A-1", None, 20)
firstDevice = Device("first", None, [301, 222])
secondDevice = Device("second", None, [154, 222])
thirdDevice = Device("third", None, [101, 222])
fourthDevice = Device("fourth", None, [55, 30])
subsection.addDevice(firstDevice)
subsection.addDevice(secondDevice)
section.addDevice(thirdDevice)
section.addDevice(fourthDevice)
devices = self.linac.getAllDevicesSorted()
self.assertEqual(
devices, [fourthDevice, thirdDevice, secondDevice, firstDevice])
self.assertEqual(devices[0].numberInSection, 0)
self.assertEqual(devices[2].numberInSection, 0)
self.assertEqual(devices[3].numberInSection, 1)
def tearDown(self):
pass
