Update magnetAddition.py

scripts/magnetAddition.py

@@ -4,15 +4,8 @@ root = ET.Element("lccdd")
 dets = ET.SubElement(root, "detectors")
 flds = ET.SubElement(root, "fields")
 
-data = open('./compton/compact/beamline.text', 'r').readlines()
-
-# def type(i):
-#     switcher = {
-#         'QF': []
-#         'DB': []
-#     }
-
-#     return swtitcher.get(i, "Object doesn't exist")
+file1 = open('./compton/compact/beamline.text', 'r')
+data = file1.readlines()
 
 # looping over every object in magnet file
 for x in data[3:]:
@@ -27,10 +20,18 @@ for x in data[3:]:
     # creating detector elements and subelements
     a = ET.SubElement(root[0], 'detector', id='100', name=line[1], vis="FFMagnetVis", type="compton_CylindricalDipoleMagnet")
     ET.SubElement(a, 'placement', x=line[2]+"*m", y=line[3]+"*m", z=line[4]+"*m", theta=line[9]+"*rad")
-    ET.SubElement(a, 'dimensions', x=line[5]+"*m", y=line[5]+"*m", z=line[8]+"*m", r= line[4]+"*2.0*m")
-    ET.SubElement(a, 'apperture', x=line[5]+"*m", y=line[5]+"*m", z=line[5]+"*m")
+    ET.SubElement(a, 'dimensions', x=line[5]+"*m", y=line[5]+"*m", z=line[8]+"*m", r= line[5]+"*2.0*m")
+    ET.SubElement(a, 'apperture', x=line[5]+"*m", y=line[5]+"*m", r=line[5]+"*m")
     ET.SubElement(a, 'coil', vis="MagnetCoilVis", dx="2*cm", dy="1.5*cm")
 
+    # creating field elements and subelements 
+    b = ET.SubElement(root[1], 'fields', name=line[1], type="MultipoleMagnet")
+    ET.SubElement(b, 'position', x=line[2]+"*m", y=line[3]+"*m", z=line[4]+"*m")
+    ET.SubElement(b, 'rotation', x="0", y=line[9]+"*rad", z="0")
+    ET.SubElement(b, 'shape', type="Tube", rmin="0.0", rmax=line[4]+"*2.0*m", dz=line[8]+"*0.5")
+    ET.SubElement(b, 'coefficient', coefficient=line[10]+"*tesla", skew="0.0*tesla")
+    ET.SubElement(b, 'coefficient', coefficient=line[11]+"*tesla/m", skew="0.0*tesla")
+
 # prettifying
 ET.indent(root)
 b_xml = ET.tostring(root)