from __future__ import (nested_scopes, generators, division, absolute_import,
print_function, unicode_literals)
from six import iteritems
from pyNastran.op2.tables.oes_stressStrain.real.oes_objects import StressObject
from pyNastran.f06.f06_formatting import writeImagFloats13E
[docs]class ComplexBush1DStress(StressObject):
"""
# s_code=0
C O M P L E X S T R E S S E S I N B A R E L E M E N T S ( C B A R )
(MAGNITUDE/PHASE)
ELEMENT LOCATION LOCATION LOCATION LOCATION AVERAGE
ID. 1 2 3 4 AXIAL STRESS
1 ENDA 9.331276E+04 9.331276E+04 9.331276E+04 9.331276E+04 0.0
180.0000 0.0 0.0 180.0000 0.0
"""
def __init__(self, data_code, is_sort1, isubcase, dt):
StressObject.__init__(self, data_code, isubcase)
self.eType = {}
self.code = [self.format_code, self.sort_code, self.s_code]
self.element_force = {}
self.axial_displacement = {}
self.axial_stress = {}
self.axial_strain = {}
self.dt = dt
if is_sort1:
if dt is not None:
#self.add = self.add_sort1
self.add_new_eid = self.add_new_eid_sort1
else:
assert dt is not None
#self.add = self.addSort2
self.add_new_eid = self.add_new_eid_sort2
[docs] def get_stats(self):
nelements = len(self.eType)
msg = self.get_data_code()
if self.dt is not None: # transient
ntimes = len(self.element_force)
msg.append(' type=%s ntimes=%s nelements=%s\n'
% (self.__class__.__name__, ntimes, nelements))
else:
msg.append(' imaginary type=%s nelements=%s\n' % (self.__class__.__name__,
nelements))
msg.append(' eType, element_force, axial_displacement, axial_velocity, axial_stress, axial_strain\n')
return msg
[docs] def add_f06_data(self, data, transient):
raise NotImplementedError('CBUSH1D')
if transient is None:
for line in data:
(eType, eid, fe, ue, ao, ae) = line
self.eType[eid] = 'CBUSH1D'
self.element_force[eid] = fe
self.axial_displacement[eid] = ue
self.axial_stress[eid] = ao
self.axial_strain[eid] = ae
return
(dtName, dt) = transient
self.data_code['name'] = dtName
if dt not in self.element_force:
self.update_dt(self.data_code, dt)
for line in data:
(eType, eid, fe, ue, ao, ae) = line
self.eType[eid] = 'CBUSH1D'
self.element_force[dt][eid] = fe
self.axial_displacement[dt][eid] = ue
self.axial_stress[dt][eid] = ao
self.axial_strain[dt][eid] = ae
[docs] def delete_transient(self, dt):
del self.element_force[dt]
del self.axial_displacement[dt]
del self.axial_stress[dt]
del self.axial_strain[dt]
[docs] def get_transients(self):
k = self.element_force.keys()
k.sort()
return k
[docs] def add_new_transient(self, dt):
"""
initializes the transient variables
"""
#print "****add new transient****"
self.dt = dt
self.element_force[dt] = {}
self.axial_displacement[dt] = {}
self.axial_stress[dt] = {}
self.axial_strain[dt] = {}
[docs] def add_new_eid(self, eType, dt, eid, fe, ue, ao, ae):
#print "Bush1d Stress add..."
self.eType[eid] = eType
self.element_force[eid] = fe
self.axial_displacement[eid] = ue
self.axial_stress[eid] = ao
self.axial_strain[eid] = ae
[docs] def add_new_eid_sort1(self, eType, dt, eid, fe, ue, ao, ae):
#msg = "dt=%s eid=%s fe=%s" % (dt, eid, fe)
#print msg
if dt not in self.element_force:
self.add_new_transient(dt)
self.eType[eid] = eType
self.element_force[dt][eid] = fe
self.axial_displacement[dt][eid] = ue
self.axial_stress[dt][eid] = ao
self.axial_strain[dt][eid] = ae
[docs] def write_f06(self, header, page_stamp, page_num=1, f=None, is_mag_phase=False):
raise NotImplementedError('CBUSH1D')
if self.nonlinear_factor is not None:
return self._write_f06_transient(header, page_stamp, page_num, f, is_mag_phase)
msg = header + [
' S T R E S S E S I N B A R E L E M E N T S ( C B A R )\n',
' ELEMENT SA1 SA2 SA3 SA4 AXIAL SA-MAX SA-MIN M.S.-T\n',
' ID. SB1 SB2 SB3 SB4 STRESS SB-MAX SB-MIN M.S.-C\n',
]
for eid, S1s in sorted(iteritems(self.s1)):
eType = self.eType[eid]
axial = self.axial[eid]
s1 = self.s1[eid]
s2 = self.s2[eid]
s3 = self.s3[eid]
s4 = self.s4[eid]
vals = [s1[0], s2[0], s3[0], s4[0], axial,
s1[1], s2[1], s3[1], s4[1], ]
(vals2, is_all_zeros) = writeImagFloats13E(vals, is_mag_phase)
[s1ar, s2ar, s3ar, s4ar, axialr,
s1br, s2br, s3br, s4br,
s1ai, s2ai, s3ai, s4ai, axiali,
s1bi, s2bi, s3bi, s4bi, ] = vals2
msg.append('0%8i %-13s %-13s %-13s %-13s %s\n' % (eid, s1ar, s2ar, s3ar, s4ar, axialr))
msg.append(' %8s %-13s %-13s %-13s %-13s %s\n' % ('', s1ai, s2ai, s3ai, s4ai, axiali))
msg.append(' %8s %-13s %-13s %-13s %s\n' % ('', s1br, s2br, s3br, s4br))
msg.append(' %8s %-13s %-13s %-13s %s\n' % ('', s1bi, s2bi, s3bi, s4bi))
msg.append(page_stamp % page_num)
f.write(''.join(msg))
return page_num
def __write_f06_transient(self, header, page_stamp, page_num=1, f=None, is_mag_phase=False):
raise NotImplementedError('CBUSH1D')
words = [
' S T R E S S E S I N B A R E L E M E N T S ( C B A R )\n',
' ELEMENT SA1 SA2 SA3 SA4 AXIAL SA-MAX SA-MIN M.S.-T\n',
' ID. SB1 SB2 SB3 SB4 STRESS SB-MAX SB-MIN M.S.-C\n',
]
msg = []
for dt, S1s in sorted(iteritems(self.s1)):
header[1] = ' %s = %10.4E\n' % (self.data_code['name'], dt)
msg += header + words
for eid, S1 in sorted(iteritems(S1s)):
eType = self.eType[eid]
axial = self.axial[dt][eid]
s1 = self.s1[dt][eid]
s2 = self.s2[dt][eid]
s3 = self.s3[dt][eid]
s4 = self.s4[dt][eid]
vals = [s1[0], s2[0], s3[0], s4[0], axial,
s1[1], s2[1], s3[1], s4[1], ]
(vals2, is_all_zeros) = writeImagFloats13E(vals, is_mag_phase)
[s1ar, s2ar, s3ar, s4ar, axialr,
s1br, s2br, s3br, s4br,
s1ai, s2ai, s3ai, s4ai, axiali,
s1bi, s2bi, s3bi, s4bi, ] = vals2
msg.append('0%8i %-13s %-13s %-13s %-13s %s\n' % (eid, s1ar, s2ar, s3ar, s4ar, axialr))
msg.append(' %8s %-13s %-13s %-13s %-13s %s\n' % ('', s1ai, s2ai, s3ai, s4ai, axiali))
msg.append(' %8s %-13s %-13s %-13s %s\n' % ('', s1br, s2br, s3br, s4br))
msg.append(' %8s %-13s %-13s %-13s %s\n' % ('', s1bi, s2bi, s3bi, s4bi))
msg.append(page_stamp % page_num)
f.write(''.join(msg))
msg = ['']
page_num += 1
return page_num - 1