1

#!/usr/bin/env python

     2

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

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2013: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

    26

from datetime import datetime

    27

from time import time as gettime

    28

from weakref import proxy

    29

    30

import numpy

    31

import wx

    32

    33

from controls.CustomToolTip import CustomToolTip, TOOLTIP_WAIT_PERIOD

    34

from editors.DebugViewer import DebugViewer, REFRESH_PERIOD

    35

from targets.typemapping import LogLevelsCount, LogLevels

    36

from util.BitmapLibrary import GetBitmap

    37

    38

    39

THUMB_SIZE_RATIO = 1. / 8.

    40

    41

    42

def ArrowPoints(direction, width, height, xoffset, yoffset):

    43

    if direction == wx.TOP:

    44

        return [wx.Point(xoffset + 1, yoffset + height - 2),

    45

                wx.Point(xoffset + width / 2, yoffset + 1),

    46

                wx.Point(xoffset + width - 1, yoffset + height - 2)]

    47

    else:

    48

        return [wx.Point(xoffset + 1, yoffset - height + 1),

    49

                wx.Point(xoffset + width / 2, yoffset - 2),

    50

                wx.Point(xoffset + width - 1, yoffset - height + 1)]

    51

    52

    53

class LogScrollBar(wx.Panel):

    54

    55

    def __init__(self, parent, size):

    56

        wx.Panel.__init__(self, parent, size=size)

    57

        self.Bind(wx.EVT_LEFT_DOWN, self.OnLeftDown)

    58

        self.Bind(wx.EVT_LEFT_UP, self.OnLeftUp)

    59

        self.Bind(wx.EVT_MOTION, self.OnMotion)

    60

        self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)

    61

        self.Bind(wx.EVT_PAINT, self.OnPaint)

    62

        self.Bind(wx.EVT_SIZE, self.OnResize)

    63

    64

        self.ThumbPosition = 0.  # -1 <= ThumbPosition <= 1

    65

        self.ThumbScrollingStartPos = None

    66

    67

    def GetRangeRect(self):

    68

        width, height = self.GetClientSize()

    69

        return wx.Rect(0, width, width, height - 2 * width)

    70

    71

    def GetThumbRect(self):

    72

        width, _height = self.GetClientSize()

    73

        range_rect = self.GetRangeRect()

    74

        thumb_size = range_rect.height * THUMB_SIZE_RATIO

    75

        thumb_range = range_rect.height - thumb_size

    76

        thumb_center_position = (thumb_size + (self.ThumbPosition + 1) * thumb_range) / 2.

    77

        thumb_start = int(thumb_center_position - thumb_size / 2.)

    78

        thumb_end = int(thumb_center_position + thumb_size / 2.)

    79

        return wx.Rect(0, range_rect.y + thumb_start, width, thumb_end - thumb_start)

    80

    81

    def RefreshThumbPosition(self, thumb_position=None):

    82

        if thumb_position is None:

    83

            thumb_position = self.ThumbPosition

    84

        if self.Parent.IsMessagePanelTop():

    85

            thumb_position = max(0., thumb_position)

    86

        if self.Parent.IsMessagePanelBottom():

    87

            thumb_position = min(0., thumb_position)

    88

        if thumb_position != self.ThumbPosition:

    89

            self.ThumbPosition = thumb_position

    90

            self.Parent.SetScrollSpeed(self.ThumbPosition)

    91

        self.Refresh()

    92

    93

    def OnLeftDown(self, event):

    94

        self.CaptureMouse()

    95

        posx, posy = event.GetPosition()

    96

        width, height = self.GetClientSize()

    97

        range_rect = self.GetRangeRect()

    98

        thumb_rect = self.GetThumbRect()

    99

        if range_rect.InsideXY(posx, posy):

   100

            if thumb_rect.InsideXY(posx, posy):

   101

                self.ThumbScrollingStartPos = wx.Point(posx, posy)

   102

            elif posy < thumb_rect.y:

   103

                self.Parent.ScrollToLast()

   104

            elif posy > thumb_rect.y + thumb_rect.height:

   105

                self.Parent.ScrollToFirst()

   106

        elif posy < width:

   107

            self.Parent.ScrollMessagePanelByPage(1)

   108

        elif posy > height - width:

   109

            self.Parent.ScrollMessagePanelByPage(-1)

   110

        event.Skip()

   111

   112

    def OnLeftUp(self, event):

   113

        self.ThumbScrollingStartPos = None

   114

        self.RefreshThumbPosition(0.)

   115

        if self.HasCapture():

   116

            self.ReleaseMouse()

   117

        event.Skip()

   118

   119

    def OnMotion(self, event):

   120

        if event.Dragging() and self.ThumbScrollingStartPos is not None:

   121

            _posx, posy = event.GetPosition()

   122

            range_rect = self.GetRangeRect()

   123

            thumb_size = range_rect.height * THUMB_SIZE_RATIO

   124

            thumb_range = range_rect.height - thumb_size

   125

            self.RefreshThumbPosition(

   126

                max(-1., min((posy - self.ThumbScrollingStartPos.y) * 2. / thumb_range, 1.)))

   127

        event.Skip()

   128

   129

    def OnResize(self, event):

   130

        self.Refresh()

   131

        event.Skip()

   132

   133

    def OnEraseBackground(self, event):

   134

        pass

   135

   136

    def OnPaint(self, event):

   137

        dc = wx.BufferedPaintDC(self)

   138

        dc.Clear()

   139

        dc.BeginDrawing()

   140

   141

        gc = wx.GCDC(dc)

   142

   143

        width, height = self.GetClientSize()

   144

   145

        gc.SetPen(wx.Pen(wx.NamedColour("GREY"), 3))

   146

        gc.SetBrush(wx.GREY_BRUSH)

   147

   148

        gc.DrawLines(ArrowPoints(wx.TOP, width * 0.75, width * 0.5, 2, (width + height) / 4 - 3))

   149

        gc.DrawLines(ArrowPoints(wx.TOP, width * 0.75, width * 0.5, 2, (width + height) / 4 + 3))

   150

   151

        gc.DrawLines(ArrowPoints(wx.BOTTOM, width * 0.75, width * 0.5, 2, (height * 3 - width) / 4 + 3))

   152

        gc.DrawLines(ArrowPoints(wx.BOTTOM, width * 0.75, width * 0.5, 2, (height * 3 - width) / 4 - 3))

   153

   154

        thumb_rect = self.GetThumbRect()

   155

        exclusion_rect = wx.Rect(thumb_rect.x, thumb_rect.y,

   156

                                 thumb_rect.width, thumb_rect.height)

   157

        if self.Parent.IsMessagePanelTop():

   158

            exclusion_rect.y, exclusion_rect.height = width, exclusion_rect.y + exclusion_rect.height - width

   159

        if self.Parent.IsMessagePanelBottom():

   160

            exclusion_rect.height = height - width - exclusion_rect.y

   161

        if exclusion_rect != thumb_rect:

   162

            colour = wx.NamedColour("LIGHT GREY")

   163

            gc.SetPen(wx.Pen(colour))

   164

            gc.SetBrush(wx.Brush(colour))

   165

   166

            gc.DrawRectangle(exclusion_rect.x, exclusion_rect.y,

   167

                             exclusion_rect.width, exclusion_rect.height)

   168

   169

        gc.SetPen(wx.GREY_PEN)

   170

        gc.SetBrush(wx.GREY_BRUSH)

   171

   172

        gc.DrawPolygon(ArrowPoints(wx.TOP, width, width, 0, 0))

   173

   174

        gc.DrawPolygon(ArrowPoints(wx.BOTTOM, width, width, 0, height))

   175

   176

        gc.DrawRectangle(thumb_rect.x, thumb_rect.y,

   177

                         thumb_rect.width, thumb_rect.height)

   178

   179

        dc.EndDrawing()

   180

        event.Skip()

   181

   182

   183

BUTTON_SIZE = (70, 15)

   184

   185

   186

class LogButton(object):

   187

   188

    def __init__(self, label, callback):

   189

        self.Position = wx.Point(0, 0)

   190

        self.Size = wx.Size(*BUTTON_SIZE)

   191

        self.Label = label

   192

        self.Shown = True

   193

        self.Callback = callback

   194

   195

    def __del__(self):

   196

        self.callback = None

   197

   198

    def GetSize(self):

   199

        return self.Size

   200

   201

    def SetPosition(self, x, y):

   202

        self.Position = wx.Point(x, y)

   203

   204

    def HitTest(self, x, y):

   205

        rect = wx.Rect(self.Position.x, self.Position.y,

   206

                       self.Size.width, self.Size.height)

   207

        if rect.InsideXY(x, y):

   208

            return True

   209

        return False

   210

   211

    def ProcessCallback(self):

   212

        if self.Callback is not None:

   213

            wx.CallAfter(self.Callback)

   214

   215

    def Draw(self, dc):

   216

        dc.SetPen(wx.TRANSPARENT_PEN)

   217

        dc.SetBrush(wx.Brush(wx.NamedColour("LIGHT GREY")))

   218

   219

        dc.DrawRectangle(self.Position.x, self.Position.y,

   220

                         self.Size.width, self.Size.height)

   221

   222

        w, h = dc.GetTextExtent(self.Label)

   223

        dc.DrawText(self.Label,

   224

                    self.Position.x + (self.Size.width - w) / 2,

   225

                    self.Position.y + (self.Size.height - h) / 2)

   226

   227

   228

DATE_INFO_SIZE = 10

   229

MESSAGE_INFO_SIZE = 18

   230

   231

   232

class LogMessage(object):

   233

   234

    def __init__(self, tv_sec, tv_nsec, level, level_bitmap, msg):

   235

        self.Date = datetime.utcfromtimestamp(tv_sec)

   236

        self.Seconds = self.Date.second + tv_nsec * 1e-9

   237

        self.Date = self.Date.replace(second=0)

   238

        self.Timestamp = tv_sec + tv_nsec * 1e-9

   239

        self.Level = level

   240

        self.LevelBitmap = level_bitmap

   241

        self.Message = msg

   242

        self.DrawDate = True

   243

   244

    def __cmp__(self, other):

   245

        if self.Date == other.Date:

   246

            return cmp(self.Seconds, other.Seconds)

   247

        return cmp(self.Date, other.Date)

   248

   249

    def GetFullText(self):

   250

        date = self.Date.replace(second=int(self.Seconds))

   251

        nsec = (self.Seconds % 1.) * 1e9

   252

        return "%s at %s.%9.9d:\n%s" % (

   253

            LogLevels[self.Level],

   254

            str(date), nsec,

   255

            self.Message)

   256

   257

    def Draw(self, dc, offset, width, draw_date):

   258

        if draw_date:

   259

            datetime_text = self.Date.strftime("%d/%m/%y %H:%M")

   260

            dw, dh = dc.GetTextExtent(datetime_text)

   261

            dc.DrawText(datetime_text, (width - dw) / 2, offset + (DATE_INFO_SIZE - dh) / 2)

   262

            offset += DATE_INFO_SIZE

   263

   264

        seconds_text = "%12.9f" % self.Seconds

   265

        sw, sh = dc.GetTextExtent(seconds_text)

   266

        dc.DrawText(seconds_text, 5, offset + (MESSAGE_INFO_SIZE - sh) / 2)

   267

   268

        bw, bh = self.LevelBitmap.GetWidth(), self.LevelBitmap.GetHeight()

   269

        dc.DrawBitmap(self.LevelBitmap, 10 + sw, offset + (MESSAGE_INFO_SIZE - bh) / 2)

   270

   271

        text = self.Message.replace("\n", " ")

   272

        _mw, mh = dc.GetTextExtent(text)

   273

        dc.DrawText(text, 15 + sw + bw, offset + (MESSAGE_INFO_SIZE - mh) / 2)

   274

   275

    def GetHeight(self, draw_date):

   276

        if draw_date:

   277

            return DATE_INFO_SIZE + MESSAGE_INFO_SIZE

   278

        return MESSAGE_INFO_SIZE

   279

   280

   281

SECOND = 1

   282

MINUTE = 60 * SECOND

   283

HOUR = 60 * MINUTE

   284

DAY = 24 * HOUR

   285

   286

CHANGE_TIMESTAMP_BUTTONS = [(_("1d"), DAY),

   287

                            (_("1h"), HOUR),

   288

                            (_("1m"), MINUTE),

   289

                            (_("1s"), SECOND)]

   290

   291

   292

class LogViewer(DebugViewer, wx.Panel):

   293

   294

    def __init__(self, parent, window):

   295

        wx.Panel.__init__(self, parent, style=wx.TAB_TRAVERSAL | wx.SUNKEN_BORDER)

   296

        DebugViewer.__init__(self, None, False, False)

   297

   298

        main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=5)

   299

        main_sizer.AddGrowableCol(0)

   300

        main_sizer.AddGrowableRow(1)

   301

   302

        filter_sizer = wx.BoxSizer(wx.HORIZONTAL)

   303

        main_sizer.AddSizer(filter_sizer, border=5, flag=wx.TOP | wx.LEFT | wx.RIGHT | wx.GROW)

   304

   305

        self.MessageFilter = wx.ComboBox(self, style=wx.CB_READONLY)

   306

        self.MessageFilter.Append(_("All"))

   307

        levels = LogLevels[:3]

   308

        levels.reverse()

   309

        for level in levels:

   310

            self.MessageFilter.Append(_(level))

   311

        self.Bind(wx.EVT_COMBOBOX, self.OnMessageFilterChanged, self.MessageFilter)

   312

        filter_sizer.AddWindow(self.MessageFilter, 1, border=5, flag=wx.RIGHT | wx.ALIGN_CENTER_VERTICAL)

   313

   314

        self.SearchMessage = wx.SearchCtrl(self, style=wx.TE_PROCESS_ENTER)

   315

        self.SearchMessage.ShowSearchButton(True)

   316

        self.SearchMessage.ShowCancelButton(True)

   317

        self.Bind(wx.EVT_TEXT_ENTER, self.OnSearchMessageChanged, self.SearchMessage)

   318

        self.Bind(wx.EVT_SEARCHCTRL_SEARCH_BTN,

   319

                  self.OnSearchMessageSearchButtonClick, self.SearchMessage)

   320

        self.Bind(wx.EVT_SEARCHCTRL_CANCEL_BTN,

   321

                  self.OnSearchMessageCancelButtonClick, self.SearchMessage)

   322

        filter_sizer.AddWindow(self.SearchMessage, 3, border=5, flag=wx.RIGHT | wx.ALIGN_CENTER_VERTICAL)

   323

   324

        self.CleanButton = wx.lib.buttons.GenBitmapButton(self, bitmap=GetBitmap("Clean"),

   325

                                                          size=wx.Size(28, 28), style=wx.NO_BORDER)

   326

        self.CleanButton.SetToolTipString(_("Clean log messages"))

   327

        self.Bind(wx.EVT_BUTTON, self.OnCleanButton, self.CleanButton)

   328

        filter_sizer.AddWindow(self.CleanButton)

   329

   330

        message_panel_sizer = wx.FlexGridSizer(cols=2, hgap=0, rows=1, vgap=0)

   331

        message_panel_sizer.AddGrowableCol(0)

   332

        message_panel_sizer.AddGrowableRow(0)

   333

        main_sizer.AddSizer(message_panel_sizer, border=5, flag=wx.LEFT | wx.RIGHT | wx.BOTTOM | wx.GROW)

   334

   335

        self.MessagePanel = wx.Panel(self)

   336

        if wx.Platform == '__WXMSW__':

   337

            self.Font = wx.Font(8, wx.SWISS, wx.NORMAL, wx.NORMAL, faceName='Courier New')

   338

        else:

   339

            self.Font = wx.Font(10, wx.SWISS, wx.NORMAL, wx.NORMAL, faceName='Courier')

   340

        self.MessagePanel.Bind(wx.EVT_LEFT_UP, self.OnMessagePanelLeftUp)

   341

        self.MessagePanel.Bind(wx.EVT_RIGHT_UP, self.OnMessagePanelRightUp)

   342

        self.MessagePanel.Bind(wx.EVT_LEFT_DCLICK, self.OnMessagePanelLeftDCLick)

   343

        self.MessagePanel.Bind(wx.EVT_MOTION, self.OnMessagePanelMotion)

   344

        self.MessagePanel.Bind(wx.EVT_LEAVE_WINDOW, self.OnMessagePanelLeaveWindow)

   345

        self.MessagePanel.Bind(wx.EVT_MOUSEWHEEL, self.OnMessagePanelMouseWheel)

   346

        self.MessagePanel.Bind(wx.EVT_ERASE_BACKGROUND, self.OnMessagePanelEraseBackground)

   347

        self.MessagePanel.Bind(wx.EVT_PAINT, self.OnMessagePanelPaint)

   348

        self.MessagePanel.Bind(wx.EVT_SIZE, self.OnMessagePanelResize)

   349

        message_panel_sizer.AddWindow(self.MessagePanel, flag=wx.GROW)

   350

   351

        self.MessageScrollBar = LogScrollBar(self, wx.Size(16, -1))

   352

        message_panel_sizer.AddWindow(self.MessageScrollBar, flag=wx.GROW)

   353

   354

        self.SetSizer(main_sizer)

   355

   356

        self.LeftButtons = []

   357

        for label, callback in [("+" + text, self.GenerateOnDurationButton(duration))

   358

                                for text, duration in CHANGE_TIMESTAMP_BUTTONS]:

   359

            self.LeftButtons.append(LogButton(label, callback))

   360

   361

        self.RightButtons = []

   362

        for label, callback in [("-" + text, self.GenerateOnDurationButton(-duration))

   363

                                for text, duration in CHANGE_TIMESTAMP_BUTTONS]:

   364

            self.RightButtons.append(LogButton(label, callback))

   365

   366

        self.MessageFilter.SetSelection(0)

   367

        self.LogSource = None

   368

        self.ResetLogMessages()

   369

        self.ParentWindow = window

   370

   371

        self.LevelIcons = [GetBitmap("LOG_" + level) for level in LogLevels]

   372

        self.LevelFilters = [range(i) for i in xrange(4, 0, -1)]

   373

        self.CurrentFilter = self.LevelFilters[0]

   374

        self.CurrentSearchValue = ""

   375

   376

        self.ScrollSpeed = 0.

   377

        self.LastStartTime = None

   378

        self.ScrollTimer = wx.Timer(self, -1)

   379

        self.Bind(wx.EVT_TIMER, self.OnScrollTimer, self.ScrollTimer)

   380

   381

        self.LastMousePos = None

   382

        self.MessageToolTip = None

   383

        self.MessageToolTipTimer = wx.Timer(self, -1)

   384

        self.Bind(wx.EVT_TIMER, self.OnMessageToolTipTimer, self.MessageToolTipTimer)

   385

   386

    def __del__(self):

   387

        self.ScrollTimer.Stop()

   388

   389

    def ResetLogMessages(self):

   390

        self.ResetLogCounters()

   391

        self.OldestMessages = []

   392

        self.LogMessages = []

   393

        self.LogMessagesTimestamp = numpy.array([])

   394

        self.CurrentMessage = None

   395

        self.HasNewData = False

   396

   397

    def SetLogSource(self, log_source):

   398

        self.LogSource = proxy(log_source) if log_source else None

   399

        self.CleanButton.Enable(self.LogSource is not None)

   400

        if log_source is not None:

   401

            self.ResetLogMessages()

   402

            self.RefreshView()

   403

   404

    def GetLogMessageFromSource(self, msgidx, level):

   405

        if self.LogSource is not None:

   406

            answer = self.LogSource.GetLogMessage(level, msgidx)

   407

            if answer is not None:

   408

                msg, _tick, tv_sec, tv_nsec = answer

   409

                return LogMessage(tv_sec, tv_nsec, level, self.LevelIcons[level], msg)

   410

        return None

   411

   412

    def ResetLogCounters(self):

   413

        self.previous_log_count = [None]*LogLevelsCount

   414

   415

    def SetLogCounters(self, log_count):

   416

        new_messages = []

   417

        for level, count, prev in zip(xrange(LogLevelsCount), log_count, self.previous_log_count):

   418

            if count is not None and prev != count:

   419

                if prev is None:

   420

                    dump_end = max(-1, count - 10)

   421

                    oldest_message = (-1, None)

   422

                else:

   423

                    dump_end = prev - 1

   424

                for msgidx in xrange(count-1, dump_end, -1):

   425

                    new_message = self.GetLogMessageFromSource(msgidx, level)

   426

                    if new_message is None:

   427

                        if prev is None:

   428

                            oldest_message = (-1, None)

   429

                        break

   430

                    if prev is None:

   431

                        oldest_message = (msgidx, new_message)

   432

                        if len(new_messages) == 0:

   433

                            new_messages = [new_message]

   434

                        else:

   435

                            new_messages.insert(0, new_message)

   436

                    else:

   437

                        new_messages.insert(0, new_message)

   438

                if prev is None and len(self.OldestMessages) <= level:

   439

                    self.OldestMessages.append(oldest_message)

   440

                self.previous_log_count[level] = count

   441

        new_messages.sort()

   442

        if len(new_messages) > 0:

   443

            self.HasNewData = True

   444

            if self.CurrentMessage is not None:

   445

                current_is_last = self.GetNextMessage(self.CurrentMessage)[0] is None

   446

            else:

   447

                current_is_last = True

   448

            for new_message in new_messages:

   449

                self.LogMessages.append(new_message)

   450

                self.LogMessagesTimestamp = numpy.append(self.LogMessagesTimestamp, [new_message.Timestamp])

   451

            if current_is_last:

   452

                self.ScrollToLast(False)

   453

                self.ResetMessageToolTip()

   454

                self.MessageToolTipTimer.Stop()

   455

                self.ParentWindow.SelectTab(self)

   456

            self.NewDataAvailable(None)

   457

   458

    def FilterLogMessage(self, message, timestamp=None):

   459

        return (message.Level in self.CurrentFilter and

   460

                message.Message.find(self.CurrentSearchValue) != -1 and

   461

                (timestamp is None or message.Timestamp < timestamp))

   462

   463

    def GetMessageByTimestamp(self, timestamp):

   464

        if self.CurrentMessage is not None:

   465

            msgidx = numpy.argmin(abs(self.LogMessagesTimestamp - timestamp))

   466

            message = self.LogMessages[msgidx]

   467

            if self.FilterLogMessage(message) and message.Timestamp > timestamp:

   468

                return self.GetPreviousMessage(msgidx, timestamp)

   469

            return message, msgidx

   470

        return None, None

   471

   472

    def GetNextMessage(self, msgidx):

   473

        while msgidx < len(self.LogMessages) - 1:

   474

            message = self.LogMessages[msgidx + 1]

   475

            if self.FilterLogMessage(message):

   476

                return message, msgidx + 1

   477

            msgidx += 1

   478

        return None, None

   479

   480

    def GetPreviousMessage(self, msgidx, timestamp=None):

   481

        message = None

   482

        while 0 < msgidx < len(self.LogMessages):

   483

            message = self.LogMessages[msgidx - 1]

   484

            if self.FilterLogMessage(message, timestamp):

   485

                return message, msgidx - 1

   486

            msgidx -= 1

   487

        if len(self.LogMessages) > 0:

   488

            message = self.LogMessages[0]

   489

            for _idx, msg in self.OldestMessages: