dddd

# -*- coding: utf-8 -*-

import wx
import modelmngr_execute

##########################################################################################
class MyFrame1 ( wx.Frame ):
    
    def __init__( self, parent ):
        wx.Frame.__init__ ( self, parent, id = wx.ID_ANY, title = u"模型管家", pos = wx.DefaultPosition, size = wx.Size( 779,521 ), style = wx.DEFAULT_FRAME_STYLE|wx.TAB_TRAVERSAL )
	
        self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize )

	####################################################
        bSizer6 = wx.BoxSizer( wx.VERTICAL )

	####################################################
        bSizer7 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText6 = wx.StaticText( self, wx.ID_ANY, u"模型文件1路径 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText6.Wrap( -1 )
        bSizer7.Add( self.m_staticText6, 0, wx.ALL, 5 )
		
        self.m_textCtrl6 = wx.TextCtrl( self, wx.ID_ANY, u"/system/etc/models/fsrcn.cambricon", wx.DefaultPosition, wx.Size( 650,-1 ), 0 )
        bSizer7.Add( self.m_textCtrl6, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer7, 1, wx.EXPAND, 5 )

        ########################
        bSizer90 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_checkBox2 = wx.CheckBox( self, wx.ID_ANY, u"是否加密 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        bSizer90.Add( self.m_checkBox2, 0, wx.ALL, 5 )
        
        self.Bind(wx.EVT_CHECKBOX, self.on_model1_check, self.m_checkBox2)
		
        self.m_staticText19 = wx.StaticText( self, wx.ID_ANY, u"模型文件1密钥", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText19.Wrap( -1 )
        bSizer90.Add( self.m_staticText19, 0, wx.ALL, 5 )
		
        self.m_textCtrl19 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 300,-1 ), 0 )
        bSizer90.Add( self.m_textCtrl19, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer90, 1, wx.EXPAND, 5 )

        ####################################################
        bSizer24 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText20 = wx.StaticText( self, wx.ID_ANY, u"模型文件2路径", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText20.Wrap( -1 )
        bSizer24.Add( self.m_staticText20, 0, wx.ALL, 5 )
		
        self.m_textCtrl20 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 650,-1 ), 0 )
        bSizer24.Add( self.m_textCtrl20, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer24, 1, wx.EXPAND, 5 )

        ########################
        bSizer12 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_checkBox3 = wx.CheckBox( self, wx.ID_ANY, u"是否加密", wx.DefaultPosition, wx.DefaultSize, 0 )
        bSizer12.Add( self.m_checkBox3, 0, wx.ALL, 5 )

        self.Bind(wx.EVT_CHECKBOX, self.on_model2_check, self.m_checkBox3)
		
        self.m_staticText21 = wx.StaticText( self, wx.ID_ANY, u"模型文件2密钥", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText21.Wrap( -1 )
        bSizer12.Add( self.m_staticText21, 0, wx.ALL, 5 )
		
        self.m_textCtrl21 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 300,-1 ), 0 )
        bSizer12.Add( self.m_textCtrl21, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer12, 1, wx.EXPAND, 5 )

	####################################################
        bSizer31 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText22 = wx.StaticText( self, wx.ID_ANY, u"模型类型 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText22.Wrap( -1 )
        bSizer31.Add( self.m_staticText22, 0, wx.ALL, 5 )
		
        m_choice1Choices = [ u"在线模型（0）", u"离线模型（1）" ]
        self.m_choice1 = wx.Choice( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, m_choice1Choices, 0 )
        self.m_choice1.SetSelection( 0 )
        bSizer31.Add( self.m_choice1, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer31, 1, wx.EXPAND, 5 )

	####################################################
        bSizer32 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText29 = wx.StaticText( self, wx.ID_ANY, u"框架类型 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText29.Wrap( -1 )
        bSizer32.Add( self.m_staticText29, 0, wx.ALL, 5 )
		
        m_choice3Choices = [ u"无（0）", u"TensorFlow（1）", u"KalDi（2）", u"Caffe（3）"]
        self.m_choice3 = wx.Choice( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, m_choice3Choices, 0 )
        self.m_choice3.SetSelection( 0 )
        bSizer32.Add( self.m_choice3, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer32, 1, wx.EXPAND, 5 )

	####################################################
        bSizer33 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText30 = wx.StaticText( self, wx.ID_ANY, u"源数据（图片）路径 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText30.Wrap( -1 )
        bSizer33.Add( self.m_staticText30, 0, wx.ALL, 5 )
		
        self.m_textCtrl27 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 500,-1 ), 0 )
        bSizer33.Add( self.m_textCtrl27, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer33, 1, wx.EXPAND, 5 )

	####################################################
        bSizer45 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText31 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸N *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText31.Wrap( -1 )
        bSizer45.Add( self.m_staticText31, 0, wx.ALL, 5 )
		
        self.m_textCtrl28 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer45.Add( self.m_textCtrl28, 0, wx.ALL, 5 )
	
        self.m_staticText32 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸C *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText32.Wrap( -1 )
        bSizer45.Add( self.m_staticText32, 0, wx.ALL, 5 )
		
        self.m_textCtrl29 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer45.Add( self.m_textCtrl29, 0, wx.ALL, 5 )
		
        self.m_staticText33 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸H *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText33.Wrap( -1 )
        bSizer45.Add( self.m_staticText33, 0, wx.ALL, 5 )
		
        self.m_textCtrl30 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer45.Add( self.m_textCtrl30, 0, wx.ALL, 5 )
		
        self.m_staticText34 = wx.StaticText( self, wx.ID_ANY, u"输入尺寸W *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText34.Wrap( -1 )
        bSizer45.Add( self.m_staticText34, 0, wx.ALL, 5 )
		
        self.m_textCtrl31 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer45.Add( self.m_textCtrl31, 0, wx.ALL, 5 )
		
		
        bSizer6.Add( bSizer45, 1, wx.EXPAND, 5 )

        ####################################################		
        bSizer47 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText35 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸N *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText35.Wrap( -1 )
        bSizer47.Add( self.m_staticText35, 0, wx.ALL, 5 )
		
        self.m_textCtrl32 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer47.Add( self.m_textCtrl32, 0, wx.ALL, 5 )
		
        self.m_staticText36 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸C *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText36.Wrap( -1 )
        bSizer47.Add( self.m_staticText36, 0, wx.ALL, 5 )
		
        self.m_textCtrl33 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer47.Add( self.m_textCtrl33, 0, wx.ALL, 5 )
		
        self.m_staticText37 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸H *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText37.Wrap( -1 )
        bSizer47.Add( self.m_staticText37, 0, wx.ALL, 5 )
		
        self.m_textCtrl34 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer47.Add( self.m_textCtrl34, 0, wx.ALL, 5 )
		
        self.m_staticText38 = wx.StaticText( self, wx.ID_ANY, u"输出尺寸W *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText38.Wrap( -1 )
        bSizer47.Add( self.m_staticText38, 0, wx.ALL, 5 )
	
        self.m_textCtrl35 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.Size( 50,-1 ), 0 )
        bSizer47.Add( self.m_textCtrl35, 0, wx.ALL, 5 )
	
	
        bSizer6.Add( bSizer47, 1, wx.EXPAND, 5 )

	####################################################
        bSizer48 = wx.BoxSizer( wx.HORIZONTAL )
		
        self.m_staticText39 = wx.StaticText( self, wx.ID_ANY, u"超时时限 *", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.m_staticText39.Wrap( -1 )
        bSizer48.Add( self.m_staticText39, 0, wx.ALL, 5 )
	
        self.m_textCtrl36 = wx.TextCtrl( self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0 )
        bSizer48.Add( self.m_textCtrl36, 0, wx.ALL, 5 )
	
	
        bSizer6.Add( bSizer48, 1, wx.EXPAND, 5 )

	####################################################
        bSizer49 = wx.BoxSizer( wx.VERTICAL )
	
        self.m_button2 = wx.Button( self, wx.ID_ANY, u"模型执行-异步-非级联", wx.DefaultPosition, wx.DefaultSize, 0 )
        bSizer49.Add( self.m_button2, 0, wx.ALL, 5 )
	
	
        bSizer6.Add( bSizer49, 1, wx.EXPAND, 5 )

        self.Bind(wx.EVT_BUTTON, self.on_async_single, self.m_button2)
	
	####################################################
        self.SetSizer( bSizer6 )
        self.Layout()
	
        self.Centre( wx.BOTH )

        ###############################################################################################################################################################################
        if self.m_checkBox2.GetValue() == True:
            self.m_textCtrl19.Enable()
        else:
            self.m_textCtrl19.Disable()
            
        if self.m_checkBox3.GetValue() == True:
            self.m_textCtrl21.Enable()
        else:
            self.m_textCtrl21.Disable()            

        
        
    def on_async_single(self, event):
        model1Path = modelmngr_execute._getTextCtrl(self.m_textCtrl6.GetValue()) #1
        model1Encr = modelmngr_execute._getCheckBox(self.m_checkBox2.GetValue()) #2
        model1Key = modelmngr_execute._getTextCtrl(self.m_textCtrl19.GetValue()) #3
        print model1Path, model1Encr, model1Key
        model2Path = modelmngr_execute._getTextCtrl(self.m_textCtrl20.GetValue()) #4
        model2Encr = modelmngr_execute._getCheckBox(self.m_checkBox3.GetValue()) #5
        model2Key = modelmngr_execute._getTextCtrl(self.m_textCtrl21.GetValue()) #6
        print model2Path, model2Encr, model2Key
        modelType = str(self.m_choice1.GetSelection()) #7
        frameworkType = str(self.m_choice3.GetSelection()) #8
        jpgPath = modelmngr_execute._getTextCtrl(self.m_textCtrl27.GetValue()) #9
        print modelType,frameworkType,jpgPath
        srcN = str(self.m_textCtrl28.GetValue()) #10
        srcC = str(self.m_textCtrl29.GetValue()) #11
        srcH = str(self.m_textCtrl30.GetValue()) #12
        srcW = str(self.m_textCtrl31.GetValue()) #13
        print srcN,srcC,srcH,srcW

        dstN = str(self.m_textCtrl32.GetValue()) #14
        dstC = str(self.m_textCtrl33.GetValue()) #15
        dstH = str(self.m_textCtrl34.GetValue()) #16
        dstW = str(self.m_textCtrl35.GetValue()) #17
        print dstN,dstC,dstH,dstW
        timeout = str(self.m_textCtrl36.GetValue()) #18
        print timeout
        
        modelmngr_execute.singleRun_async(model1Path, model1Encr, model1Key, model2Path, model2Encr, model2Key, modelType, frameworkType, jpgPath, srcN, srcC, srcH, srcW, dstN, dstC, dstH, dstW, timeout)
        
        event.Skip()

    def on_model1_check(self, event):
        cb = event.GetEventObject()
        if cb.GetValue() == True:
            self.m_textCtrl19.Enable()
            self.m_textCtrl19.SetValue("")
        else:
            self.m_textCtrl19.Disable()
            self.m_textCtrl19.SetValue("")
        event.Skip()

    def on_model2_check(self, event):
        cb = event.GetEventObject()
        if cb.GetValue() == True:
            self.m_textCtrl21.Enable()
            self.m_textCtrl21.SetValue("")
        else:
            self.m_textCtrl21.Disable()
            self.m_textCtrl21.SetValue("")
        event.Skip()        
	
    def __del__( self ):
        pass