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) 2007: 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 __future__ import absolute_import

    27

from functools import reduce

    28

    29

import wx

    30

import wx.lib.buttons

    31

import wx.lib.agw.customtreectrl as CT

    32

from pprint import pprint

    33

    34

from PLCControler import *

    35

from util.BitmapLibrary import GetBitmap

    36

from plcopen.types_enums import GetElementType

    37

    38

    39

def GenerateName(infos):

    40

    if infos[0] in ["input", "output", "value"]:

    41

        return "%s %d:" % (infos[0], infos[1])

    42

    elif infos[0] == "range":

    43

        return "%s %d %s" % (infos[0], infos[1], infos[2])

    44

    elif infos[0] == "struct":

    45

        return "element %d %s" % (infos[1], infos[2])

    46

    return "%s:" % infos[0]

    47

    48

    49

# -------------------------------------------------------------------------------

    50

#                            Search Result Panel

    51

# -------------------------------------------------------------------------------

    52

    53

    54

[

    55

    ID_SEARCHRESULTPANEL, ID_SEARCHRESULTPANELHEADERLABEL,

    56

    ID_SEARCHRESULTPANELSEARCHRESULTSTREE, ID_SEARCHRESULTPANELRESETBUTTON,

    57

] = [wx.NewId() for _init_ctrls in range(4)]

    58

    59

    60

class SearchResultPanel(wx.Panel):

    61

    62

    def _init_coll_MainSizer_Items(self, parent):

    63

        parent.AddSizer(self.HeaderSizer, 0, border=0, flag=wx.GROW)

    64

        parent.AddWindow(self.SearchResultsTree, 1, border=0, flag=wx.GROW)

    65

    66

    def _init_coll_MainSizer_Growables(self, parent):

    67

        parent.AddGrowableCol(0)

    68

        parent.AddGrowableRow(1)

    69

    70

    def _init_coll_HeaderSizer_Items(self, parent):

    71

        parent.AddWindow(self.HeaderLabel, 1, border=5, flag=wx.LEFT | wx.RIGHT | wx.ALIGN_CENTER_VERTICAL)

    72

        parent.AddWindow(self.ResetButton, 0, border=0, flag=0)

    73

    74

    def _init_coll_HeaderSizer_Growables(self, parent):

    75

        parent.AddGrowableCol(0)

    76

    77

    def _init_sizers(self):

    78

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

    79

        self.HeaderSizer = wx.BoxSizer(wx.HORIZONTAL)

    80

    81

        self._init_coll_MainSizer_Items(self.MainSizer)

    82

        self._init_coll_MainSizer_Growables(self.MainSizer)

    83

        self._init_coll_HeaderSizer_Items(self.HeaderSizer)

    84

    85

        self.SetSizer(self.MainSizer)

    86

    87

    def _init_ctrls(self, prnt):

    88

        self.HeaderLabel = wx.StaticText(id=ID_SEARCHRESULTPANELHEADERLABEL,

    89

                                         name='HeaderLabel', parent=self,

    90

                                         pos=wx.Point(0, 0), size=wx.Size(0, 17), style=0)

    91

    92

        search_results_tree_style = CT.TR_HAS_BUTTONS | CT.TR_NO_LINES | CT.TR_HAS_VARIABLE_ROW_HEIGHT

    93

        self.SearchResultsTree = CT.CustomTreeCtrl(id=ID_SEARCHRESULTPANELSEARCHRESULTSTREE,

    94

                                                   name="SearchResultsTree", parent=self,

    95

                                                   pos=wx.Point(0, 0), style=search_results_tree_style)

    96

        if wx.VERSION >= (2, 8, 11):

    97

            self.SearchResultsTree.SetAGWWindowStyleFlag(search_results_tree_style)

    98

        self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnSearchResultsTreeItemActivated,

    99

                  id=ID_SEARCHRESULTPANELSEARCHRESULTSTREE)

   100

   101

        self.ResetButton = wx.lib.buttons.GenBitmapButton(

   102

            self, bitmap=GetBitmap("reset"),

   103

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

   104

        self.ResetButton.SetToolTipString(_("Reset search result"))

   105

        self.Bind(wx.EVT_BUTTON, self.OnResetButton, self.ResetButton)

   106

   107

        self._init_sizers()

   108

   109

    def __init__(self, parent, window):

   110

        wx.Panel.__init__(self, id=ID_SEARCHRESULTPANEL,

   111

                          name='SearchResultPanel', parent=parent,

   112

                          pos=wx.Point(0, 0),

   113

                          size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)

   114

   115

        self.ParentWindow = window

   116

   117

        self._init_ctrls(parent)

   118

   119

        # Define Tree item icon list

   120

        self.TreeImageList = wx.ImageList(16, 16)

   121

        self.TreeImageDict = {}

   122

   123

        # Icons for other items

   124

        for imgname, itemtype in [

   125

                # editables

   126

                ("PROJECT",        ITEM_PROJECT),

   127

                ("TRANSITION",     ITEM_TRANSITION),

   128

                ("ACTION",         ITEM_ACTION),

   129

                ("CONFIGURATION",  ITEM_CONFIGURATION),

   130

                ("RESOURCE",       ITEM_RESOURCE),

   131

                ("DATATYPE",       ITEM_DATATYPE),

   132

                ("ACTION",         "action_block"),

   133

                ("IL",             "IL"),

   134

                ("ST",             "ST"),

   135

                ("FILE",           ITEM_CONFNODE),

   136

                ]:

   137

            self.TreeImageDict[itemtype] = self.TreeImageList.Add(GetBitmap(imgname))

   138

   139

        for itemtype in ["function", "functionBlock", "program",

   140

                         "comment", "block", "io_variable",

   141

                         "connector", "contact", "coil",

   142

                         "step", "transition", "jump",

   143

                         "var_local", "var_input",

   144

                         "var_inout", "var_output"]:

   145

            self.TreeImageDict[itemtype] = self.TreeImageList.Add(GetBitmap(itemtype.upper()))

   146

   147

        # Assign icon list to TreeCtrl

   148

        self.SearchResultsTree.SetImageList(self.TreeImageList)

   149

   150

        self.ResetSearchResults()

   151

   152

    def SetSearchResults(self, criteria, search_results):

   153

        self.Criteria = criteria

   154

        self.SearchResults = {}

   155

        self.ElementsOrder = []

   156

   157

        for infos, start, end, text in search_results:

   158

            if infos[0] not in self.ElementsOrder:

   159

                self.ElementsOrder.append(infos[0])

   160

   161

            results = self.SearchResults.setdefault(infos[0], [])

   162

            results.append((infos, start, end, text))

   163

   164

        self.RefreshView()

   165

   166

    def ResetSearchResults(self):

   167

        self.Criteria = None

   168

        self.ElementsOrder = []

   169

        self.SearchResults = {}

   170

        self.RefreshView()

   171

   172

    def RefreshView(self):

   173

        self.SearchResultsTree.DeleteAllItems()

   174

        if self.Criteria is None:

   175

            self.SearchResultsTree.AddRoot("")

   176

            root = self.SearchResultsTree.GetRootItem()

   177

            root.SetHilight(False)

   178

            self.HeaderLabel.SetLabel(_("No search results available."))

   179

            self.ResetButton.Enable(False)

   180

        else:

   181

            matches_number = 0

   182

            search_results_tree_infos = {

   183

                "name": _("Project '%s':") % self.ParentWindow.Controler.GetProjectName(),

   184

                "type": ITEM_PROJECT,

   185

                "data": None,

   186

                "text": None,

   187

                "matches": None,

   188

            }

   189

            search_results_tree_children = search_results_tree_infos.setdefault("children", [])

   190

            for tagname in self.ElementsOrder:

   191

                results = self.SearchResults.get(tagname, [])

   192

                matches_number += len(results)

   193

   194

                words = tagname.split("::")

   195

   196

                element_type = GetElementType(tagname)

   197

                if element_type == ITEM_POU:

   198

                    element_type = self.ParentWindow.Controler.GetPouType(words[1])

   199

   200

                element_infos = {"name": words[-1],

   201

                                 "type": element_type,

   202

                                 "data": tagname,

   203

                                 "text": None,

   204

                                 "matches": len(results)}

   205

   206

                children = element_infos.setdefault("children", [])

   207

                for infos, start, end, text in results:

   208

                    if len(words) == 1:  # CTN match

   209

                        child_name = {"body":str(start[0])+":",

   210

                                      "var_inout":_("Variable:")}[infos[1]]

   211

                        child_type = {"body":ITEM_CONFNODE,

   212

                                      "var_inout":"var_inout"}[infos[1]]

   213

                    elif infos[1] == "name" or element_type == ITEM_DATATYPE:

   214

                        child_name = GenerateName(infos[1:])

   215

                        child_type = element_type

   216

                    else:

   217

                        if element_type == ITEM_RESOURCE:

   218

                            child_type = element_type

   219

                        else:

   220

                            child_type = infos[1]

   221

                        if child_type == "name":

   222

                            child_name = "name"

   223

                        elif child_type == "body":

   224

                            child_name = "body"

   225

                            if element_type == ITEM_TRANSITION:

   226

                                child_type = self.ParentWindow.Controler.GetTransitionBodyType(words[1], words[2])

   227

                            elif element_type == ITEM_ACTION:

   228

                                child_type = self.ParentWindow.Controler.GetActionBodyType(words[1], words[2])

   229

                            else:

   230

                                child_type = self.ParentWindow.Controler.GetPouBodyType(words[1])

   231

                        else:

   232

                            child_name = GenerateName(infos[3:])

   233

                    child_infos = {

   234

                        "name": child_name,

   235

                        "type": child_type,

   236

                        "data": (infos, start, end, None),

   237

                        "text": text,

   238

                        "matches": 1,

   239

                        "children": [],

   240

                    }

   241

                    children.append(child_infos)

   242

   243

                # not Project node

   244

                if len(words) > 2:

   245

                    for _element_infos in search_results_tree_children:

   246

                        if _element_infos["name"] == words[1]:

   247

                            _element_infos["matches"] += len(children)

   248

                            _element_infos["children"].append(element_infos)

   249

                            break

   250

                    if element_type == ITEM_RESOURCE:

   251

                        search_results_tree_children.append(element_infos)

   252

                else:  # Project node or CTN

   253

                    search_results_tree_children.append(element_infos)

   254

   255

            if matches_number < 2:

   256

                header_format = _("'{a1}' - {a2} match in project")

   257

            else:

   258

                header_format = _("'{a1}' - {a2} matches in project")

   259

   260

            self.HeaderLabel.SetLabel(header_format.format(a1=self.Criteria["find_pattern"], a2=matches_number))

   261

            self.ResetButton.Enable(True)

   262

   263

            if matches_number > 0:

   264

                root = self.SearchResultsTree.GetRootItem()

   265

                if root is None:

   266

                    root = self.SearchResultsTree.AddRoot(search_results_tree_infos["name"])

   267

                self.GenerateSearchResultsTreeBranch(root, search_results_tree_infos)

   268

                self.SearchResultsTree.Expand(root)

   269

   270

    def GetTextCtrlClickFunction(self, item):

   271

        def OnTextCtrlClick(event):

   272

            self.SearchResultsTree.SelectItem(item)

   273

            event.Skip()

   274

        return OnTextCtrlClick

   275

   276

    def GetTextCtrlDClickFunction(self, item):

   277

        def OnTextCtrlDClick(event):

   278

            self.ShowSearchResults(item)

   279

            event.Skip()

   280

        return OnTextCtrlDClick

   281

   282

    def GenerateSearchResultsTreeBranch(self, root, infos):

   283

        if infos["name"] == "body":

   284

            item_name = "%d:" % infos["data"][1][0]

   285

        else:

   286

            item_name = infos["name"]

   287

   288

        self.SearchResultsTree.SetItemText(root, item_name)

   289

        self.SearchResultsTree.SetPyData(root, infos["data"])

   290

        self.SearchResultsTree.SetItemBackgroundColour(root, wx.WHITE)

   291

        self.SearchResultsTree.SetItemTextColour(root, wx.BLACK)

   292

        if infos["type"] is not None:

   293

            if infos["type"] == ITEM_POU:

   294

                self.SearchResultsTree.SetItemImage(root, self.TreeImageDict[self.ParentWindow.Controler.GetPouType(infos["name"])])

   295

            else:

   296

                self.SearchResultsTree.SetItemImage(root, self.TreeImageDict[infos["type"]])

   297

   298

        text = None

   299

        print("XXXXXXXXXXXXXX")

   300

        pprint(infos)

   301

        if infos["text"] is not None:

   302

            text = infos["text"]

   303

            start, end = infos["data"][1:3]

   304

            text_lines = infos["text"].splitlines()

   305

            start_idx = start[1]

   306

            end_idx = reduce(lambda x, y: x + y, map(lambda x: len(x) + 1, text_lines[:end[0] - start[0]]), end[1] + 1)

   307

            style = wx.TextAttr(wx.BLACK, wx.Colour(206, 204, 247))

   308

        elif infos["type"] is not None and infos["matches"] > 1:

   309

            text = _("(%d matches)") % infos["matches"]

   310

            start_idx, end_idx = 0, len(text)

   311

            style = wx.TextAttr(wx.Colour(0, 127, 174))

   312

   313

        if text is not None:

   314

            text_ctrl_style = wx.BORDER_NONE | wx.TE_READONLY | wx.TE_RICH2

   315

            if wx.Platform != '__WXMSW__' or len(text.splitlines()) > 1:

   316

                text_ctrl_style |= wx.TE_MULTILINE | wx.TE_NO_VSCROLL

   317

            text_ctrl = wx.TextCtrl(id=-1, parent=self.SearchResultsTree, pos=wx.Point(0, 0),

   318

                                    value=text, style=text_ctrl_style)

   319

            width, height = text_ctrl.GetTextExtent(text)

   320

            text_ctrl.SetClientSize(wx.Size(width + 1, height))

   321

            text_ctrl.SetBackgroundColour(self.SearchResultsTree.GetBackgroundColour())

   322

            text_ctrl.Bind(wx.EVT_LEFT_DOWN, self.GetTextCtrlClickFunction(root))

   323

            text_ctrl.Bind(wx.EVT_LEFT_DCLICK, self.GetTextCtrlDClickFunction(root))

   324

            text_ctrl.SetInsertionPoint(0)

   325

            text_ctrl.SetStyle(start_idx, end_idx, style)

   326

            self.SearchResultsTree.SetItemWindow(root, text_ctrl)

   327

   328

        item, root_cookie = self.SearchResultsTree.GetFirstChild(root)

   329

        for child in infos["children"]:

   330

            if item is None:

   331

                item = self.SearchResultsTree.AppendItem(root, "")

   332

                item, root_cookie = self.SearchResultsTree.GetNextChild(root, root_cookie)

   333

            self.GenerateSearchResultsTreeBranch(item, child)

   334

            item, root_cookie = self.SearchResultsTree.GetNextChild(root, root_cookie)

   335

   336

    def ShowSearchResults(self, item):

   337

        data = self.SearchResultsTree.GetPyData(item)

   338

        if isinstance(data, tuple):

   339

            search_results = [data]

   340

        else:

   341

            search_results = self.SearchResults.get(data, [])

   342

        self.ParentWindow.ClearHighlights(SEARCH_RESULT_HIGHLIGHT)

   343

        for infos, start, end, _text in search_results:

   344

            self.ParentWindow.ShowSearchResult(infos, start, end)

   345

   346

    def OnSearchResultsTreeItemActivated(self, event):

   347

        self.ShowSearchResults(event.GetItem())

   348

        event.Skip()

   349

   350

    def OnResetButton(self, event):

   351

        self.ResetSearchResults()

   352

        self.ParentWindow.ClearSearchResults()

   353

        event.Skip()