Laurent@814: #!/usr/bin/env python Laurent@814: # -*- coding: utf-8 -*- Laurent@814: Laurent@814: #This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor Laurent@814: #based on the plcopen standard. Laurent@814: # Laurent@814: #Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD Laurent@814: # Laurent@814: #See COPYING file for copyrights details. Laurent@814: # Laurent@814: #This library is free software; you can redistribute it and/or Laurent@814: #modify it under the terms of the GNU General Public Laurent@814: #License as published by the Free Software Foundation; either Laurent@814: #version 2.1 of the License, or (at your option) any later version. Laurent@814: # Laurent@814: #This library is distributed in the hope that it will be useful, Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Laurent@814: #General Public License for more details. Laurent@814: # Laurent@814: #You should have received a copy of the GNU General Public Laurent@814: #License along with this library; if not, write to the Free Software Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA Laurent@814: Laurent@814: import re Laurent@814: Laurent@814: import wx Laurent@814: Laurent@814: RE_ESCAPED_CHARACTERS = ".*+()[]?:|{}^$<>=-," Laurent@814: Laurent@814: def EscapeText(text): Laurent@814: text = text.replace('\\', '\\\\') Laurent@814: for c in RE_ESCAPED_CHARACTERS: Laurent@814: text = text.replace(c, '\\' + c) Laurent@814: return text Laurent@814: Laurent@814: #------------------------------------------------------------------------------- Laurent@814: # Search In Project Dialog Laurent@814: #------------------------------------------------------------------------------- Laurent@814: Laurent@814: def GetElementsChoices(): Laurent@814: _ = lambda x: x Laurent@814: return [("datatype", _("Data Type")), Laurent@814: ("function", _("Function")), Laurent@814: ("functionBlock", _("Function Block")), Laurent@814: ("program", _("Program")), Laurent@814: ("configuration", _("Configuration"))] Laurent@814: Laurent@814: class SearchInProjectDialog(wx.Dialog): Laurent@814: Laurent@814: def __init__(self, parent): Laurent@814: wx.Dialog.__init__(self, parent, title=_('Search in Project'), Laurent@814: size=wx.Size(600, 350)) Laurent@814: Laurent@814: main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=10) Laurent@814: main_sizer.AddGrowableCol(0) Laurent@814: main_sizer.AddGrowableRow(1) Laurent@814: Laurent@814: pattern_sizer = wx.FlexGridSizer(cols=2, hgap=5, rows=2, vgap=5) Laurent@814: pattern_sizer.AddGrowableCol(0) Laurent@814: main_sizer.AddSizer(pattern_sizer, border=20, Laurent@814: flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT) Laurent@814: Laurent@814: pattern_label = wx.StaticText(self, label=_('Pattern to search:')) Laurent@814: pattern_sizer.AddWindow(pattern_label, flag=wx.ALIGN_BOTTOM) Laurent@814: Laurent@814: self.CaseSensitive = wx.CheckBox(self, label=_('Case sensitive')) Laurent@814: pattern_sizer.AddWindow(self.CaseSensitive, flag=wx.GROW) Laurent@814: Laurent@814: self.Pattern = wx.TextCtrl(self) Laurent@814: pattern_sizer.AddWindow(self.Pattern, flag=wx.GROW) Laurent@814: Laurent@814: self.RegularExpression = wx.CheckBox(self, label=_('Regular expression')) Laurent@814: pattern_sizer.AddWindow(self.RegularExpression, flag=wx.GROW) Laurent@814: Laurent@814: scope_staticbox = wx.StaticBox(self, label=_('Scope')) Laurent@814: scope_sizer = wx.StaticBoxSizer(scope_staticbox, wx.HORIZONTAL) Laurent@814: main_sizer.AddSizer(scope_sizer, border=20, Laurent@814: flag=wx.GROW|wx.LEFT|wx.RIGHT) Laurent@814: Laurent@814: scope_selection_sizer = wx.BoxSizer(wx.VERTICAL) Laurent@814: scope_sizer.AddSizer(scope_selection_sizer, 1, border=5, Laurent@814: flag=wx.GROW|wx.TOP|wx.LEFT|wx.BOTTOM) Laurent@814: Laurent@814: self.WholeProject = wx.RadioButton(self, label=_('Whole Project'), Laurent@814: size=wx.Size(0, 24), style=wx.RB_GROUP) Laurent@814: self.WholeProject.SetValue(True) Laurent@814: self.Bind(wx.EVT_RADIOBUTTON, self.OnScopeChanged, self.WholeProject) Laurent@814: scope_selection_sizer.AddWindow(self.WholeProject, border=5, Laurent@814: flag=wx.GROW|wx.BOTTOM) Laurent@814: Laurent@814: self.OnlyElements = wx.RadioButton(self, Laurent@814: label=_('Only Elements'), size=wx.Size(0, 24)) Laurent@814: self.Bind(wx.EVT_RADIOBUTTON, self.OnScopeChanged, self.OnlyElements) Laurent@814: self.OnlyElements.SetValue(False) Laurent@814: scope_selection_sizer.AddWindow(self.OnlyElements, flag=wx.GROW) Laurent@814: Laurent@814: self.ElementsList = wx.CheckListBox(self) Laurent@814: self.ElementsList.Enable(False) Laurent@814: scope_sizer.AddWindow(self.ElementsList, 1, border=5, Laurent@814: flag=wx.GROW|wx.TOP|wx.RIGHT|wx.BOTTOM) Laurent@814: Laurent@814: self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE) Laurent@814: ok_button = self.ButtonSizer.GetAffirmativeButton() Laurent@814: ok_button.SetLabel(_('Search')) Laurent@814: self.Bind(wx.EVT_BUTTON, self.OnOK, ok_button) Laurent@814: main_sizer.AddSizer(self.ButtonSizer, border=20, Laurent@814: flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT) Laurent@814: Laurent@814: self.SetSizer(main_sizer) Laurent@814: Laurent@814: for name, label in GetElementsChoices(): Laurent@814: self.ElementsList.Append(_(label)) Laurent@814: Laurent@814: self.Pattern.SetFocus() Laurent@814: Laurent@814: def GetCriteria(self): Laurent@814: raw_pattern = pattern = self.Pattern.GetValue() Laurent@814: if not self.CaseSensitive.GetValue(): Laurent@814: pattern = pattern.upper() Laurent@814: if not self.RegularExpression.GetValue(): Laurent@814: pattern = EscapeText(pattern) Laurent@814: criteria = { Laurent@814: "raw_pattern": raw_pattern, Laurent@814: "pattern": re.compile(pattern), Laurent@814: "case_sensitive": self.CaseSensitive.GetValue(), Laurent@814: "regular_expression": self.RegularExpression.GetValue(), Laurent@814: } Laurent@814: if self.WholeProject.GetValue(): Laurent@814: criteria["filter"] = "all" Laurent@814: elif self.OnlyElements.GetValue(): Laurent@814: criteria["filter"] = [] Laurent@814: for index, (name, label) in enumerate(GetElementsChoices()): Laurent@814: if self.ElementsList.IsChecked(index): Laurent@814: criteria["filter"].append(name) Laurent@814: return criteria Laurent@814: Laurent@814: def OnScopeChanged(self, event): Laurent@814: self.ElementsList.Enable(self.OnlyElements.GetValue()) Laurent@814: event.Skip() Laurent@814: Laurent@814: def OnOK(self, event): Laurent@814: message = None Laurent@814: if self.Pattern.GetValue() == "": Laurent@814: message = _("Form isn't complete. Pattern to search must be filled!") Laurent@814: else: Laurent@814: wrong_pattern = False Laurent@814: if self.RegularExpression.GetValue(): Laurent@814: try: Laurent@814: re.compile(self.Pattern.GetValue()) Laurent@814: except: Laurent@814: wrong_pattern = True Laurent@814: if wrong_pattern: Laurent@814: message = _("Syntax error in regular expression of pattern to search!") Laurent@814: Laurent@814: if message is not None: Laurent@814: dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR) Laurent@814: dialog.ShowModal() Laurent@814: dialog.Destroy() Laurent@814: else: Laurent@814: self.EndModal(wx.ID_OK)