HeiligeListe/SwissFEL-HL
1
1
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Added several fixes and features

Browse Source
This commit is contained in:
reiche
2024-02-27 14:16:18 +01:00
parent 68ec24c40b
commit 13cea1421c
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
Python_Code/AssemblyList.py
View File

@@ -6,12 +6,14 @@ import numpy as np
class AssemblyList:
def __init__(self):
self.assembly={}
self.assemblylist = {}
self.components = ComponentList.ComponentList()
self.check={}
def readFiles(self,filedir,folder,phase):
self.components.readFiles(folder,phase)
self.assembly.clear()
self.assemblylist.clear()
for subdir, dirs, files in os.walk(filedir):
for file in files:
filepath = subdir + os.sep + file
@@ -39,7 +41,11 @@ class AssemblyList:
soff = tab['S Offset'][i]
part3d = tab['3D-Part'][i]
type = tab['Type'][i]
self.assembly[parent].append({'Name':name,'soff':soff,'3DPart':part3d,'Type':type})
if name in self.assemblylist.keys():
print('*** Error: Duplicated generation of assembly element:',name,'for parent',parent)
else:
self.assemblylist[name] = 1
self.assembly[parent].append({'Name':name,'soff':soff,'3DPart':part3d,'Type':type})
def generateLayout(self,proto):
@@ -56,7 +62,7 @@ class AssemblyList:
pele['Type']=''
pele['Category']=''
if len(pele['Suffix']) < 2:
name = '%s-ZERO' % pele['Prefix']
name = '%s-CELL' % pele['Prefix']
if len(DLsave)> 0:
esec=copy.deepcopy(DLsave[0])
sname=esec['Name'][0:7]
@@ -155,10 +161,11 @@ class AssemblyList:
def generateEntry(self,name,data,parent,soff):
if 'ZERO' in name:
name=name[0:7]
sec = self.components.find_section(name)
if sec is None:
return
data['3DX_Section']=sec['3DX-Section UG'].to_string(index=False,na_rep='').strip()
if 'RKLY' in name:
data['3DX_Section'] = sec['3DX-Section OG'].to_string(index=False, na_rep='').strip()