profiler/yiwinframe.py

import wx
import os
import time
import threading
from pathlib import PurePath,Path
from pubsub import pub
 

from listctrl import Listctrl
from mypanel import MyPanel,Plot,LineColor

from uiconfig.uisensor import UISensor
# from uiconfig.uilogging import UILogging

from uiconfig.uiabout import About
from uiconfig.uihelp import Help
from uiconfig.ui_plot_setting import UI_Plot_Setting

from myconfig import TOKEN,DeviceType,YAML_FILE_NAME,RETRIEVE_CFG_FILE,DATA_DIR
from myconfig import MyConfig,RamsesFunc
from mythread import Mythead
from configuration import Configuration
# from handheld import HandHeld,HandHeldBuf
from gatherdata import GatherData
  
from tools.myexception import MyException
from tools.mylogger import log

from awrams import AwramsHandle,HandheldHandle
from profiler import ProfilerHandle

# -定义菜单ID,关联Event-------------------------
ID_MEASURE          = 1
 
ID_SENSOR_SETTING   = 11
ID_LOGGING_SETTING  = 12
ID_CAL_INFO         = 13
ID_DISPLAY_PARA     = 14

ID_PLOT_LAST        = 15
ID_PLOT_LAST_7      = 16
ID_PLOT_7           = 17
ID_PLOT_SETTING     = 18
ID_GATHER_DATA       = 19

ID_HELP             = 21
ID_ABOUT            = 22


class YiwinFrame(wx.Frame):
    '''将buf类传进来'''
    def __init__(self, title, parent, size=(900, 750)):
        self.device_id      = 2
        self.device_type    = DeviceType.SURFACE.name
        self.syscfg = {}
        self.sensor_cfg = {}
        self.retrieve = {}
        self.calinfo_is_ok = False
        self.title = title
        self.path = PurePath()
        # self.save_fname:Path = Path(self.path.joinpath(DATA_FNAME))
        self.result = ''
        self.displayData:list =[]
        self.interval = 0

        super(YiwinFrame, self).__init__( parent, title=self.title, size = size )

        self.BoxSizer = wx.BoxSizer(wx.HORIZONTAL)  # 建立一个Boxsizer
        self.SetSizer(self.BoxSizer)
        self.Centre(wx.BOTH)

        # 建立 listctrl panel 放到BoxSizer
        self.listctrl_panel = Listctrl( self  )  # 调用自己建立的 Listctrl panel 类
        self.BoxSizer.Add( self.listctrl_panel, proportion =-10, border = 2, flag = wx.ALL | wx.EXPAND)
        self.list_ctrl = self.listctrl_panel.list_ctrl
        # 隐藏 等待show
        self.listctrl_panel.Hide()

        self.mypanel = MyPanel( self  )  # 调用自己建立的 Listctrl panel 类
        self.BoxSizer.Add( self.mypanel, proportion =-10, border = 0, flag = wx.ALL | wx.EXPAND)
        self.static_text = self.mypanel.staticText1 
        # 隐藏 等待show
        self.mypanel.Hide()

        self.plotpanel          = Plot( self  )  # 调用自己建立的 Listctrl panel 类
        self.BoxSizer.Add( self.plotpanel, proportion =-10, border = 0, flag = wx.ALL | wx.EXPAND)
        self.plotpanel.set_title_x_y(*self.plotpanel.purewater_legend)
        self.plotpanel.Show()

        self.statusBar = self.CreateStatusBar()  # 创建状态栏
        self.statusBar.SetFieldsCount(2)  # 状态栏分成3个区域
        self.statusBar.SetStatusWidths([-1, -1])  # 区域宽度比列，用负数
        self.statusBar.SetStatusText(u" 等待接收消息......", 0)

        self.__set_menu()       # 添加菜单
        self.__attach_events()  # 菜单事件
        self.__set_properties()
        # self.__read_config()
        self.__setTimer()
        log.info(f"system init....",__name__, "__init__")
        self.mycfg = MyConfig()
        # self.hh = HandHeld()
        self.mythread = Mythead()
        pub.subscribe( self.updateDisplay, "update")   
        
        pass

    def __set_menu(self):
        '''
        # 设置菜单
        '''
        self.menubar = wx.MenuBar()

        settingMenu = wx.Menu()
        settingMenu.Append(ID_SENSOR_SETTING, u'&设置', '...') 
        settingMenu.AppendSeparator()
        settingMenu.Append(ID_CAL_INFO, u'&获取标定信息', ' ')
        settingMenu.AppendSeparator()
        settingMenu.Append(ID_MEASURE, u'&处理数据', ' ')
        # settingMenu.AppendSeparator()
        # settingMenu.Append(ID_LOGGING_SETTING, u'&采集设置', ' ')
        self.menubar.Append(settingMenu, u'&系统  ')

        plotMenu = wx.Menu()
        plotMenu.Append(ID_PLOT_LAST, u'&最后一条', '...') 
        plotMenu.AppendSeparator()
        plotMenu.Append(ID_PLOT_LAST_7, u'&最后七条', ' ')
        plotMenu.AppendSeparator()
        plotMenu.Append(ID_PLOT_7, u'&指定七条', ' ')
        plotMenu.AppendSeparator()
        plotMenu.Append(ID_PLOT_SETTING, u'&指定设置', ' ')
        plotMenu.AppendSeparator()
        plotMenu.Append(ID_GATHER_DATA, u'&汇总数据', ' ')
        self.menubar.Append(plotMenu, u'&绘图  ')

        aboutMenu = wx.Menu()
        aboutMenu.Append(ID_HELP, u'&帮助', 'help...')
        aboutMenu.AppendSeparator()
        aboutMenu.Append(ID_ABOUT, u'&关于我们', '关于我们...')
        self.menubar.Append(aboutMenu, u'&帮助  ')
        
        self.SetMenuBar(self.menubar)
        pass

    def __set_properties(self):
        self.SetSize((800, 600))
        # self.SetTitle(u'传感器数据采集--上海奕枫仪器设备有限公司')
        self.Centre()

    def __attach_events(self):
        '''
        # 绑定菜单事件
        '''
        self.Bind(wx.EVT_MENU, self.OnSensorSetting, id=ID_SENSOR_SETTING )
        self.Bind(wx.EVT_MENU, self.OnCalInfo, id=ID_CAL_INFO )
        # self.Bind(wx.EVT_MENU, self.OnStart, id=ID_MEASURE)
        self.Bind(wx.EVT_MENU, self.OnDealData, id=ID_MEASURE)
        # self.Bind(wx.EVT_MENU, self.OnLoggingSetting, id=ID_LOGGING_SETTING)

        self.Bind(wx.EVT_MENU, self.OnPlotLast, id=ID_PLOT_LAST )
        self.Bind(wx.EVT_MENU, self.OnPlotLast7, id=ID_PLOT_LAST_7 )
        self.Bind(wx.EVT_MENU, self.OnPlot7, id=ID_PLOT_7)
        self.Bind(wx.EVT_MENU, self.OnPlotSetting, id=ID_PLOT_SETTING)
        self.Bind(wx.EVT_MENU, self.OnGatherData, id=ID_GATHER_DATA)

        self.Bind(wx.EVT_MENU, self.OnHelpConfig, id=ID_HELP)
        self.Bind(wx.EVT_MENU, self.OnAboutConfig, id=ID_ABOUT)
        pass

    def __setTimer(self):
        self.timer = wx.Timer(self)   #创建定时器,菜单以后
        self.Bind(wx.EVT_TIMER, self.OnTimer, self.timer) #绑定一个定时器事件
        pass

    def updateDisplay(self, msg):
        log.warning(f" updateDisplay ..  {msg['data']}",__name__,"updateDisplay","")
        if msg['flag']  == "notice":
            self.__update_notice(msg['data'])
            pass
        if msg['flag']  == "data":
            # log.info(f" ...  update .{msg['data']}........... " )
            self.__update_data(msg['data'])
            pass
        pass

    def __update_notice(self,msg):
        log.info(f" .......{msg}... ",__name__,'__update_notice')
        self.alterStatus_0( msg)
        # self.popDialog( msg )
        pass

    def __update_data(self,data ):
        log.info(f"  .......... ", __name__, '__update_data')
        self.plotpanel.set_axes_title( data[0])
        self.plotpanel.clear_past()
        self.plotpanel.plot_one( "Rs", data[1], data[2], color= LineColor(1).name  )


    def OnStart( self, event  ):
        log.info(f"OnStart....interval: {self.interval} min, port: {self.port}")   
        # self.m = SerialThread()  
        interval_ms = int(self.interval) * 10 * 1000
        self.timer.Start( interval_ms )
    
    def OnStop( self ,event  ):
        # self.kh.disconnect()
        # self.timer.Stop()
        pass
    
    def OnTimer(self, event):
        '''
        定时器实现
        '''
        # log.info( f"OnTimer...." )
        # self.OnDisplaySave()   
        pass 

    def onNotify(self, msg:str):
        self.mypanel.setStaticText(msg)
        pass 

    def __read_config(self,):
        # 读取配置文件
        self.mycfg.setDeviceType(DeviceType.SURFACE)
        self.sensor_cfg = self.mycfg.read_yaml()

    def OnDealData(self, e) -> None:
        ''' SD卡的文件结构 '''
        log.info(f"OnDealData: 处理测量数据", __name__, "OnDealData", "")
        if not self.calinfo_is_ok:
            self.onNotify("Pls get the cal info, and try again" )
            log.warning(f"Pls get the cal info, and try again", __name__, "OnDealData")
            self.statusBar.SetStatusText("Pls get the cal info, and try again",0)
            pass

        # 调用 ProfilerHandle 的函数
        self.myhandle:ProfilerHandle = self.profiler_handle.get(self.device_id )      

        # 处理数据
        self.alterStatus_0(" 正在处理数据...." )
        try:
            self.mythread.set_task( self.read_folders_from_sd )
            self.mythread.start()
        except Exception as e:
            log.error(e,__name__,"OnDealData")
        pass 

    def read_folders_from_sd(self,):
        # myhandle:AwramsHandle = self.awrams_handle[self.device_id]  
        self.myhandle.read_folders_from_SD( DATA_DIR )     
        pass

       
    def OnCalInfo(self,e):
        '''依据传感器获取标定文件信息'''
        #######################################
        self.onNotify("正在获取标定信息")
        self.device_type = DeviceType.PROFILER.name
        self.mycfg = MyConfig()         # 传入  cfg  retrieve 的yml文件

        # 配置反演需要的参数  波长  间隔 , rowFactor
        self.retrieve = self.mycfg.read_rtv_yaml()    # 读retrieve.yml
        log.info(f"Retrieve: {self.retrieve}", __name__, "", "")

        # log.info(f"syscfg: {device}", __name__, "", "")
        self.mycfg.setDeviceType( DeviceType.PROFILER )
        self.sensor_cfg  = self.mycfg.read_yaml()            # 读config.yml, 多个device_id 配置
        log.info(f"Current Device: {self.device_type}  ", __name__, "OnCalInfo", "")
        log.info(f"Sensor cfg:  {self.sensor_cfg}", __name__, "OnCalInfo", "")
    
        self.cal_cfg = {}
        
        cfgr = Configuration( )
        cfgr.setDeviceType(self.device_type)
        # cfgr.setSystemCfgDict(self.sensor_cfg)
        
        for k,v in self.sensor_cfg.items(): 
            cfgr.setSystemCfgDict(v)
            try:
                cfgr.getCalConfiguration()
            except Exception as e:
                log.error(f"读取配置文件失败. \n  {e}",__name__, "OnCalInfo", "" )
                raise
            log.info(f"v:  {cfgr.configuration}", __name__, "OnCalInfo", "")       
            self.cal_cfg.update({k:cfgr.cal_configuration})

        log.debug(f"cal_cfg: {self.cal_cfg}", __name__, "OnCalInfo", "")
        log.warning(f"cal_cfg 2: {self.cal_cfg[2].keys()}", __name__, "OnCalInfo", "")
        # log.warning(f"cal_cfg 3: {self.cal_cfg[3].keys()}", __name__, "", "")
        log.info(f"传感器配置文件读取成功", __name__, "", "")
        
        #  配置初始化 ProfilerHandle
        self.init_data_process()
        self.calinfo_is_ok =True
    
        pass

    def init_data_process(self, mode = 0):
        ''' 多个handle '''
        log.info(f" 配置设备处理数据... ", __name__, "init_data_process", "")
        self.profiler_handle = { }
        for k,v in self.cal_cfg.items():
            cal_cfg = v
            ph  = ProfilerHandle( deviceid=k, cfg=self.sensor_cfg, calcfg=cal_cfg, rtv=self.retrieve )
            self.profiler_handle.update( {k:ph}  )

    def alterStatus_0(self,msg):
        self.statusBar.SetStatusText( msg, 0 )

    def popDialog(self, msg, msg_type=u"错误提示"):
        with wx.MessageDialog( self, msg, msg_type, wx.OK )as dlg:
            dlg.ShowModal()

    def OnPlotLast(self,e):
        # 如何取数据 需要先合并数据？
        self.popDialog(" 功能开发中...")
        pass

    def OnPlotLast7(self,e):
        self.popDialog(" 功能开发中...")
        pass

    def OnPlot7(self,e):
        self.popDialog(" 功能开发中...")
        pass  

    def OnGatherData(self,e):
        gd = GatherData()
        gd.gather_handheld()
        self.alterStatus_0(" 数据汇总完成")
        pass
    
    def OnPlotSetting(self,e):
        with UI_Plot_Setting(
                self,
                -1   ) as Dialog_Sensor_Setting:
            Dialog_Sensor_Setting.CenterOnParent()
            resultLog = Dialog_Sensor_Setting.ShowModal()

            if resultLog == wx.ID_OK:
                log.info( " Plot config dialog confirm, call back " )
        self.__read_config()  
        pass

    def OnSensorSetting(self,e):
        with UISensor(
                self,
                -1 ,
                did = self.device_id ) as Dialog_Sensor_Setting:
            Dialog_Sensor_Setting.CenterOnParent()
            resultLog = Dialog_Sensor_Setting.ShowModal()

            if resultLog == wx.ID_OK:
                print( " Sensor config dialog confirm, call back " )
        self.__read_config()  
        pass    
    
    def OnHelpConfig(self, e):
        with Help(
                self,
                -1,
                "") as Dialog_Help:
            resultLog = Dialog_Help.ShowModal()

            if resultLog == wx.ID_OK:
                print("Help info")

        pass

    def OnAboutConfig(self, e):
        with About(
                self,
                -1,
                "") as Dialog_About:
            resultLog = Dialog_About.ShowModal()

            if resultLog == wx.ID_OK:
                print("Aboutus")
          
        pass

    def OnQuit(self, e):
        self.Close()