class StripChartFrame(wx.Frame):
def __init__(self, parent, ID, **kws):
kws["style"] = wx.DEFAULT_FRAME_STYLE | wx.RESIZE_BORDER | wx.TAB_TRAVERSAL
wx.Frame.__init__(self, parent, ID, '', wx.DefaultPosition,
wx.Size(-1, -1), **kws)
self.SetTitle("wxmplot StripChart Demo")
self.tmin = 15.0
self.SetFont(wx.Font(12, wx.SWISS, wx.NORMAL, wx.BOLD, False))
menu = wx.Menu()
ID_EXIT = wx.NewId()
ID_TIMER = wx.NewId()
menu_exit = menu.Append(ID_EXIT, "E&xit", "Terminate the program")
menuBar = wx.MenuBar()
menuBar.Append(menu, "&File")
self.SetMenuBar(menuBar)
self.Bind(wx.EVT_MENU, self.OnExit, menu_exit)
self.Bind(wx.EVT_CLOSE, self.OnExit)
sbar = self.CreateStatusBar(2, wx.CAPTION)
sfont = sbar.GetFont()
sfont.SetWeight(wx.BOLD)
sfont.SetPointSize(11)
sbar.SetFont(sfont)
self.SetStatusWidths([-3, -1])
self.SetStatusText('', 0)
mainsizer = wx.BoxSizer(wx.VERTICAL)
btnpanel = wx.Panel(self, -1)
btnsizer = wx.BoxSizer(wx.HORIZONTAL)
b_on = wx.Button(btnpanel, -1, 'Start', size=(-1, -1))
b_off = wx.Button(btnpanel, -1, 'Stop', size=(-1, -1))
b_on.Bind(wx.EVT_BUTTON, self.onStartTimer)
b_off.Bind(wx.EVT_BUTTON, self.onStopTimer)
tlabel = wx.StaticText(btnpanel, -1, ' Time range:')
self.time_range = FloatCtrl(btnpanel,
size=(100, -1),
value=abs(self.tmin),
precision=1)
btnsizer.Add(b_on, 0, wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
btnsizer.Add(b_off, 0, wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
btnsizer.Add(
tlabel, 1,
wx.GROW | wx.ALL | wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
btnsizer.Add(self.time_range, 0,
wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
btnpanel.SetSizer(btnsizer)
btnsizer.Fit(btnpanel)
self.plotpanel = PlotPanel(self, messenger=self.write_message)
self.plotpanel.BuildPanel()
self.plotpanel.set_xlabel('Time from Present (s)')
mainsizer.Add(btnpanel, 0,
wx.GROW | wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
mainsizer.Add(
self.plotpanel, 1,
wx.GROW | wx.ALL | wx.ALIGN_LEFT | wx.ALIGN_CENTER | wx.LEFT, 0)
self.SetSizer(mainsizer)
mainsizer.Fit(self)
self.Bind(wx.EVT_TIMER, self.onTimer)
self.timer = wx.Timer(self)
self.count = 0
self.Refresh()
wx.CallAfter(self.onStartTimer)
def write_message(self, msg, panel=0):
"""write a message to the Status Bar"""
self.SetStatusText(msg, panel)
def onStartTimer(self, event=None):
self.count = 0
t0, y0 = next_data()
self.ylist = [y0]
self.tlist = [t0]
self.tmin_last = -10000
self.time0 = time.time()
self.timer.Start(50)
def onStopTimer(self, event=None):
self.timer.Stop()
def onTimer(self, event):
self.count += 1
etime = time.time() - self.time0
self.tmin = float(self.time_range.GetValue())
t1, y1 = next_data()
self.tlist.append(t1)
self.ylist.append(y1)
tdat = np.array(self.tlist) - t1
mask = np.where(tdat > -abs(self.tmin))
ydat = np.array(self.ylist)
n = len(self.ylist)
if n <= 2:
self.plotpanel.plot(tdat, ydat)
else:
self.plotpanel.update_line(0, tdat, ydat, draw=True)
self.write_message(" %i points in %8.4f s" % (n, etime))
lims = self.plotpanel.get_viewlimits()
try:
ymin, ymax = ydat[mask].min(), ydat[mask].max()
except:
ymin, ymax = ydat.min(), ydat.max()
tmin = max(int(min(tdat)) - 1.0, -self.tmin)
if (ymin < lims[2] or ymax > lims[3] or tmin != self.tmin_last
or time.time() - self.last_update > 2):
self.tmin_last = tmin
self.last_update = time.time()
self.plotpanel.set_xylims((tmin, 0, ymin, ymax))
def OnAbout(self, event):
dlg = wx.MessageDialog(self, "wxmplot example: stripchart app",
"About WXMPlot test",
wx.OK | wx.ICON_INFORMATION)
dlg.ShowModal()
dlg.Destroy()
def OnExit(self, event):
self.Destroy()
class SectoinParamsDefsPanel(SectionParamsDefsPanelBase):
def __init__(self, parent):
SectionParamsDefsPanelBase.__init__(self, parent)
# ÓÃÓÚ´æ·Å solveÖ®ºóµÄÖµ
self._args = dict()
self.plotpanel = PlotPanel(self.m_panel_canvas,
size=(300, 300),
fontsize=5)
self.plotpanel.BuildPanel()
# ĬÈϽØÃæÐÎ×´
self._section_type = u'Ö±½Ç¸Ö'
# ½ØÃæ²ÎÊýµÚÒ»ÁÐÖ»¶Á
for irow in range(self.m_grid_params_defs.GetNumberRows()):
self.m_grid_params_defs.SetReadOnly(irow, 0, True)
self.grid_value = {
u'Ö±½Ç¸Ö':
['±ß¿í¶È1', '±ß¿í¶È2', '±ßºñ¶È1', '±ßºñ¶È2', 'ÄÚÔ²»¡°ë¾¶'],
u'¹¤×Ö¸Ö': [
'¸ß¶È', 'ÍÈ¿í¶È', 'Ñüºñ¶È', 'ƽ¾ùÍȺñ¶È', 'ÄÚÔ²»¡°ë¾¶',
'½Ç¶ËÔ²»¡°ë¾¶'
],
u'²Û¸Ö': [
'¸ß¶È', 'ÍÈ¿í¶È', 'Ñüºñ¶È', 'ƽ¾ùÍȺñ¶È', 'ÄÚÔ²»¡°ë¾¶',
'½Ç¶ËÔ²»¡°ë¾¶'
],
#u'CÐ͸Ö': ['¸ß¶È', '±ß¿í¶È1', '±ßºñ¶È1', 'Ñüºñ¶È', '±ßºñ¶È2', '±ß¿í¶È2'],
u'TÐ͸Ö': ['¸ß¶È', '±ß¿í¶È1', '±ß¿í¶È2', '±ßºñ¶È1', '±ßºñ¶È2'],
u'ñÐ͸Ö': [
'¸ß¶È', '½Å±ß¿í¶È', 'Í·±ß¿í¶È', 'ÍÈÇã½Ç', '½Å±ßºñ¶È', 'ÍȺñ¶È',
'Í·ºñ¶È'
],
u'JÐ͸Ö': [
'¸ß¶È', '±ß¿í¶È1', '±ß¿í¶È2', '±ß¿í¶È3', '±ßºñ¶È1', '±ßºñ¶È2',
'±ßºñ¶È3'
]
}
for num in range(len(self.grid_value[self._section_type])):
self.m_grid_params_defs.SetCellValue(
num, 0, self.grid_value[self._section_type][num])
def OnSelectSectionType(self, event):
self._section_type = self.m_choice_section_type.GetStringSelection()
section_parameter_name = self.grid_value[self._section_type]
self.m_grid_params_defs.ClearGrid()
for name_num in range(len(section_parameter_name)):
self.m_grid_params_defs.SetCellValue(
name_num, 0, section_parameter_name[name_num])
def m_btn_calculationOnButtonClick(self, event):
# µã»÷btnÖ®ºó£¬»ñµÃ´«ÈëµÄ²ÎÊý¡£
# ͨ¹ý»ñµÃµÄ²ÎÊýÉú³É³ö½ØÃæ¶ÔÏó¡£
# ͨ¹ý½ØÃæ¼ÆË㣬µÃµ½¼¸ºÎÊýÖµ
# ͨ¹ýDraw¶ÔÏó»Í¼£¬»³ö¼¸ºÎͼÏñ¡£
_args = list()
for num in range(len(self.grid_value[self._section_type])):
_args.append(float(self.m_grid_params_defs.GetCellValue(num, 1)))
sectionType = self._section_type
if sectionType == u"¹¤×Ö¸Ö":
section = ISection(*_args)
elif sectionType == u"Ö±½Ç¸Ö":
section = rightAngleSection(*_args)
elif sectionType == u"²Û¸Ö":
section = grooveSection(*_args)
#elif sectionType == u"CÐ͸Ö":
#pass
elif sectionType == u"TÐ͸Ö":
section = TshapeSection(*_args)
elif sectionType == u"JÐ͸Ö":
section = JshapeSection(*_args)
elif sectionType == u"ñÐ͸Ö":
section = NshapeSection(*_args)
geo = GeoCalculator(section)
geo.Solve()
self._args = geo._args
if "Area" in self._args:
if self._args['Area'] < 0.0000001:
res = 0
else:
res = self._args['Area']
self.m_propertyGridItem2.SetValue(str(res))
if "Sx" in self._args:
if self._args['Sx'] < 0.0000001:
res = 0
else:
res = self._args['Sx']
self.m_propertyGridItem3.SetValue(str(res))
if "Sy" in self._args:
if self._args['Sy'] < 0.0000001:
res = 0
else:
res = self._args['Sy']
self.m_propertyGridItem42.SetValue(str(res))
if "Ix" in self._args:
if self._args['Ix'] < 0.0000001:
res = 0
else:
res = self._args['Ix']
self.m_propertyGridItem4.SetValue(str(res))
if "Iy" in self._args:
if self._args['Iy'] < 0.0000001:
res = 0
else:
res = self._args['Iy']
self.m_propertyGridItem5.SetValue(str(res))
if "Ixy" in self._args:
if self._args['Ixy'] < 0.0000001:
res = 0
else:
res = self._args['Ixy']
self.m_propertyGridItem7.SetValue(str(res))
if "centroid" in self._args:
res = self._args['centroid']
if res[0] < 0.0000001:
res = [0, res[1]]
if res[1] < 0.0000001:
res = [res[0], 0]
self.m_propertyGridItem8.SetValue(str(res))
if "tan_alfa" in self._args:
if self._args['tan_alfa'] < 0.0000001:
res = 0
else:
res = self._args['tan_alfa']
self.m_propertyGridItem9.SetValue(str(res))
if "ix" in self._args:
if self._args['ix'] < 0.0000001:
res = 0
else:
res = self._args['ix']
self.m_propertyGridItem61.SetValue(str(res))
if "iy" in self._args:
if self._args['iy'] < 0.0000001:
res = 0
else:
res = self._args['iy']
self.m_propertyGridItem62.SetValue(str(res))
Path = DrawGeometry(section)
Path.Draw()
self.plotpanel.clear()
for i in Path._paths:
m, n = zip(*i)
self.plotpanel.oplot(m, n, fullbox=False, axes_style='open')
for i in Path._dimen:
m, n = zip(*i)
self.plotpanel.oplot(m,
n,
fullbox=False,
axes_style='open',
linewidth=1,
color='green')
# ±éÀú×ֵ䣬»Éϱê×¢
for key, value in Path._text.items():
x = key[0]
y = key[1]
self.plotpanel.add_text(str(value), x, y, size=4)
def m_btn_calculationOnSetFocus(self, event):
print "on btn_Set Focus!"
class SectoinPointDefsPanel (SectionPointDefsPanelBase):
def __init__(self, parent):
SectionPointDefsPanelBase.__init__(self, parent)
# 用于存放 solve之后的值
self._args = dict()
self.point_list = list()
# 这是一个复连通的全局变量
self.MP = MultiConnectPoly()
# 再来一个复合截面的全局变量
self.Comp = compoundSection()
# 仅仅只是初始化了画布
self.plotpanel = PlotPanel(self.m_panel1, size=(300, 300), fontsize=5)
self.plotpanel.BuildPanel()
def m_btn_InsertOnButtonClick( self, event ):
# 将input进来的点填入复连通截面中,并填入buffet数据中
x = self.m_textCtrl8.Value
y = self.m_textCtrl9.Value
p = Point(x, y)
self.point_list.append(p)
string = '(' + x + ',' + y + ')'
self.m_listBox2.Append(string)
def btn_addMultiPropOnButtonClick( self, event ):
# 默认第一个是放进Outer的，后面来的就放在inner里
if len(self.MP.outerLoop) == 0 :
_args = self.point_list
self.MP.setOuterLoop(*_args)
else:
_args = self.point_list
self.MP.addInnerLoop(*_args)
# procedure -- remove all items of list，as well as the data of point list
self.m_listBox2.Clear()
self.point_list = []
Path = DrawGeometry(self.MP)
Path.Draw()
self.plotpanel.clear()
for i in Path._paths:
m,n=zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open')
for i in Path._dimen:
m, n = zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open', linewidth=1, color='green')
# 遍历字典，画上标注
for key, value in Path._text.items():
x = key[0]
y = key[1]
self.plotpanel.add_text(str(value), x, y, size=4)
def btn_genCompOnButtonClick( self, event ):
# 如果有2个复连通的话就把他们同时画出来。然后直接计算，边上再添加一个数据清空的btn
self.ShowResult(self.Comp)
Comp_Path = DrawGeometry(self.Comp)
Comp_Path.Draw()
self.plotpanel.clear()
for i in Comp_Path._paths:
m,n=zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open')
for i in Comp_Path._dimen:
m, n = zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open', linewidth=1, color='green')
# 遍历字典，画上标注
for key, value in Comp_Path._text.items():
x = key[0]
y = key[1]
self.plotpanel.add_text(str(value), x, y, size=4)
def ShowResult(self, section):
geo = GeoCalculator(section)
geo.Solve()
self._args = geo._args
if "Area" in self._args:
if abs(self._args['Area']) < 0.0000001:
res = 0
else:
res = self._args['Area']
self.m_propertyGridItem2.SetValue(str(res))
if "Sx" in self._args:
#if abs(self._args['Sx']) < 0.0000001:
# res = 0
#else:
res = self._args['Sx']
self.m_propertyGridItem3.SetValue(str(res))
if "Sy" in self._args:
#if abs(self._args['Sy']) < 0.0000001:
#res = 0
#else:
res = self._args['Sy']
self.m_propertyGridItem42.SetValue(str(res))
if "Ix" in self._args:
if abs(self._args['Ix']) < 0.0000001:
res = 0
else:
res = self._args['Ix']
self.m_propertyGridItem4.SetValue(str(res))
if "Iy" in self._args:
if abs(self._args['Iy']) < 0.0000001:
res = 0
else:
res = self._args['Iy']
self.m_propertyGridItem5.SetValue(str(res))
if "Ixy" in self._args:
#if abs(self._args['Ixy']) < 0.0000001:
#res = 0
#else:
res = self._args['Ixy']
self.m_propertyGridItem7.SetValue(str(res))
if "centroid" in self._args:
res = self._args['centroid']
#if abs(res[0]) < 0.0000001:
#res = [0, res[1]]
#if abs(res[1]) < 0.0000001:
#res = [res[0], 0]
self.m_propertyGridItem8.SetValue(str(res))
if "tan_alfa" in self._args:
#if abs(self._args['tan_alfa']) < 0.0000001:
#res = 0
#else:
res = self._args['tan_alfa']
self.m_propertyGridItem9.SetValue(str(res))
if "ix" in self._args:
if abs(self._args['ix']) < 0.0000001:
res = 0
else:
res = self._args['ix']
self.m_propertyGridItem61.SetValue(str(res))
if "iy" in self._args:
if abs(self._args['iy']) < 0.0000001:
res = 0
else:
res = self._args['iy']
self.m_propertyGridItem62.SetValue(str(res))
def showItemOnDClick( self, event ):
print 'double _click'
n = self.m_listBox3.GetSelection()
multisection = self.Comp.dataResource[n]
self.ShowResult(multisection)
Path = DrawGeometry(multisection)
Path.Draw()
self.plotpanel.clear()
for i in Path._paths:
m,n=zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open')
for i in Path._dimen:
m, n = zip(*i)
self.plotpanel.oplot(m,n,fullbox=False,axes_style='open', linewidth=1, color='green')
# 遍历字典，画上标注
for key, value in Path._text.items():
x = key[0]
y = key[1]
self.plotpanel.add_text(str(value), x, y, size=4)
def btn_genMultiPropOnButtonClick( self, event ):
# 将MP放进来，然后清空，图像也清空
self.Comp.addToSections(self.MP)
n = len(self.Comp.dataResource)
string = '第' + str(n) + '个截面'
self.m_listBox3.Append(string)
self.plotpanel.clear()
self.MP = MultiConnectPoly()
def btn_DataClearOnClickButton( self, event ):
self.MP = MultiConnectPoly()
self.Comp = compoundSection()
self.m_listBox3.Clear()
self.plotpanel.BuildPanel()
self.m_textCtrl8.SetValue("")
self.m_textCtrl9.SetValue("")
self.m_propertyGridItem2.SetValue('0')
self.m_propertyGridItem3.SetValue('0')
self.m_propertyGridItem42.SetValue('0')
self.m_propertyGridItem4.SetValue('0')
self.m_propertyGridItem5.SetValue('0')
self.m_propertyGridItem7.SetValue('0')
self.m_propertyGridItem8.SetValue('[0,0]')
self.m_propertyGridItem9.SetValue('0')
self.m_propertyGridItem61.SetValue('0')
self.m_propertyGridItem62.SetValue('0')
def del_OnRightClick( self, event ):
n = self.m_listBox2.GetSelection()
self.m_listBox2.Delete(n)
self.point_list.pop(n)
# 还要把listpoint里的去掉
def btn_PrintDataOnclick( self, event ):
# procedure 1:生成UI2.png,2:
path = os.path.abspath(os.curdir)+'//section.png'
if hasattr(self, 'fig'):
self.plotpanel.fig.savefig(path, transparent=False, dpi=300)
else:
self.plotpanel.canvas.print_figure(path, transparent=False, dpi=300)
if (path.find(self.plotpanel.launch_dir) == 0):
path = path[len(self.plotpanel.launch_dir)+1:]
self.plotpanel.write_message('Saved plot to %s' % path)
self.filePrint()
def filePrint(self):
# Create an new Excel file and add a worksheet.
workbook = xlsxwriter.Workbook('reportData.xlsx')
worksheet = workbook.add_worksheet()
# Widen the first column to make the text clearer.
worksheet.set_column('A:A', 15)
worksheet.set_column('B:B', 15)
ID = self.__getcurrentTimeID()
worksheet.write('A2', 'ID:')
worksheet.write('B2', ID)
worksheet.write('A3', 'Area:')
worksheet.write('B3', self._args['Area'])
worksheet.write('A4', 'Sx')
worksheet.write('B4', self._args['Sx'])
worksheet.write('A5', 'Sy')
worksheet.write('B5', self._args['Sy'])
worksheet.write('A6', 'Ix')
worksheet.write('B6', self._args['Ix'])
worksheet.write('A7', 'Iy')
worksheet.write('B7', self._args['Iy'])
worksheet.write('A8', 'Ixy')
worksheet.write('B8', self._args['Ixy'])
worksheet.write('A9', 'centroid')
worksheet.write('B9', str(self._args['centroid']))
worksheet.write('A10', 'tan_alfa')
worksheet.write('B10', str(self._args['tan_alfa']))
worksheet.write('A11', 'ix')
worksheet.write('B11', self._args['ix'])
worksheet.write('A12', 'iy')
worksheet.write('B12', self._args['iy'])
worksheet.insert_image('F2', 'section.png')
workbook.close()
def __getcurrentTimeID(self):
currenttime = datetime.datetime.now()
currenttime = str(currenttime.isoformat())
currenttime = currenttime[0:len(currenttime)-6]
for c in '-:.T': # "
currenttime = currenttime.replace(c, '')
return currenttime
def btn_insertToLibOnButtonClick( self, event ):
# 将该信息存放到库文件中。
# workbook = xlsxwriter.Workbook('Library.xlsx')
# 两个步骤：1把section的图片save到lib中，并改名为ID.png
# 存入数据库。只有打印的时候才有文件。一次性打印出来。
path = os.path.abspath(os.curdir)+'//section.png'
if hasattr(self, 'fig'):
self.plotpanel.fig.savefig(path, transparent=False, dpi=300)
else:
self.plotpanel.canvas.print_figure(path, transparent=False, dpi=300)
if (path.find(self.plotpanel.launch_dir) == 0):
path = path[len(self.plotpanel.launch_dir)+1:]
self.plotpanel.write_message('Saved plot to %s' % path)
ID = self.__getcurrentTimeID()
ImageURL = ID+'.png'
lib = SectionLibrary()
lib.addSection(ID,ImageURL,self._args)
# 然后再把图片存到ImageLIB中
shutil.copy('section.png',"..\\ImageLib\\"+ImageURL)
# 然后再update一下存放着所有的信息的文件。
# self.__updateLibFile(lib)
def __updateLibFile(self,lib):
workbook = xlsxwriter.Workbook('LibInfo.xlsx')
worksheet = workbook.add_worksheet()
worksheet.set_column('A:A', 15)
worksheet.set_column('B:B', 15)
# 我要获取info
info = lib.selectAllSection()
self.__worksheet_write(worksheet,info)
workbook.close()
def __worksheet_write(self,ws,info):
count = 0
for singleSec in info:
# 计数
ws.write('A'+ str(count*15+1), 'ID')
ws.write('B'+ str(count*15+1), singleSec[0])
ws.insert_image('F'+str(count*15+1), "..\\ImageLib\\" + singleSec[1])
ws.write('A'+ str(count*15+2), 'Area')
ws.write('B'+ str(count*15+2), singleSec[2])
ws.write('A'+ str(count*15+3), 'Sx')
ws.write('B'+ str(count*15+3), singleSec[3])
ws.write('A'+ str(count*15+4), 'Sy')
ws.write('B'+ str(count*15+4),singleSec[4])
ws.write('A'+ str(count*15+5), 'Iy')
ws.write('B'+ str(count*15+5),singleSec[5])
ws.write('A'+ str(count*15+6), 'Ix')
ws.write('B'+ str(count*15+6),singleSec[6])
ws.write('A'+ str(count*15+7), 'Ixy')
ws.write('B'+ str(count*15+7),singleSec[7])
ws.write('A'+ str(count*15+8), 'Centroid')
ws.write('B'+ str(count*15+8),singleSec[8])
ws.write('A'+ str(count*15+9), 'Angle')
ws.write('B'+ str(count*15+9),singleSec[9])
ws.write('A'+ str(count*15+10), 'ix')
ws.write('B'+ str(count*15+10),singleSec[10])
ws.write('A'+ str(count*15+11), 'iy')
ws.write('B'+ str(count*15+11), singleSec[11])
count = count+1
def checkLibOnButtonClick( self, event ):
lib = SectionLibrary()
self.__updateLibFile(lib)
os.system("LibInfo.xlsx")
