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)