1

#!/usr/bin/env python

     2

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

     3

     4

#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor

     5

#based on the plcopen standard. 

     6

#

     7

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

     8

#

     9

#See COPYING file for copyrights details.

    10

#

    11

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

    12

#modify it under the terms of the GNU General Public

    13

#License as published by the Free Software Foundation; either

    14

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

    15

#

    16

#This library 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 GNU

    19

#General Public License for more details.

    20

#

    21

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

    22

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

    23

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    24

    25

import wx

    26

    27

from graphics import *

    28

from controls.LibraryPanel import LibraryPanel

    29

    30

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

    31

#                          Create New Block Dialog

    32

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

    33

    34

class FBDBlockDialog(wx.Dialog):

    35

    36

    def __init__(self, parent, controller):

    37

        wx.Dialog.__init__(self, parent,

    38

              size=wx.Size(600, 450), title=_('Block Properties'))

    39

    40

        main_sizer = wx.FlexGridSizer(cols=1, hgap=0, rows=4, vgap=10)

    41

        main_sizer.AddGrowableCol(0)

    42

        main_sizer.AddGrowableRow(0)

    43

    44

        column_sizer = wx.BoxSizer(wx.HORIZONTAL)

    45

        main_sizer.AddSizer(column_sizer, border=20, 

    46

              flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)

    47

    48

        type_staticbox = wx.StaticBox(self, label=_('Type:'))

    49

        left_staticboxsizer = wx.StaticBoxSizer(type_staticbox, wx.VERTICAL)

    50

        column_sizer.AddSizer(left_staticboxsizer, 1, border=5, 

    51

              flag=wx.GROW|wx.RIGHT)

    52

    53

        self.LibraryPanel = LibraryPanel(self)

    54

        setattr(self.LibraryPanel, "_OnTreeItemSelected", 

    55

              self.OnLibraryTreeItemSelected)

    56

        left_staticboxsizer.AddWindow(self.LibraryPanel, 1, border=5, 

    57

              flag=wx.GROW|wx.TOP)

    58

    59

        right_gridsizer = wx.FlexGridSizer(cols=1, hgap=0, rows=3, vgap=5)

    60

        right_gridsizer.AddGrowableCol(0)

    61

        right_gridsizer.AddGrowableRow(2)

    62

        column_sizer.AddSizer(right_gridsizer, 1, border=5, 

    63

              flag=wx.GROW|wx.LEFT)

    64

    65

        top_right_gridsizer = wx.FlexGridSizer(cols=2, hgap=0, rows=4, vgap=5)

    66

        top_right_gridsizer.AddGrowableCol(1)

    67

        right_gridsizer.AddSizer(top_right_gridsizer, flag=wx.GROW)

    68

    69

        name_label = wx.StaticText(self, label=_('Name:'))

    70

        top_right_gridsizer.AddWindow(name_label, 

    71

              flag=wx.ALIGN_CENTER_VERTICAL)

    72

    73

        self.BlockName = wx.TextCtrl(self)

    74

        self.Bind(wx.EVT_TEXT, self.OnNameChanged, self.BlockName)

    75

        top_right_gridsizer.AddWindow(self.BlockName, flag=wx.GROW)

    76

    77

        inputs_label = wx.StaticText(self, label=_('Inputs:'))

    78

        top_right_gridsizer.AddWindow(inputs_label, 

    79

              flag=wx.ALIGN_CENTER_VERTICAL)

    80

    81

        self.Inputs = wx.SpinCtrl(self, min=2, max=20,

    82

              style=wx.SP_ARROW_KEYS)

    83

        self.Bind(wx.EVT_SPINCTRL, self.OnInputsChanged, self.Inputs)

    84

        top_right_gridsizer.AddWindow(self.Inputs, flag=wx.GROW)

    85

    86

        execution_order_label = wx.StaticText(self, label=_('Execution Order:'))

    87

        top_right_gridsizer.AddWindow(execution_order_label, 

    88

              flag=wx.ALIGN_CENTER_VERTICAL)

    89

    90

        self.ExecutionOrder = wx.SpinCtrl(self, min=0, style=wx.SP_ARROW_KEYS)

    91

        self.Bind(wx.EVT_SPINCTRL, self.OnExecutionOrderChanged, self.ExecutionOrder)

    92

        top_right_gridsizer.AddWindow(self.ExecutionOrder, flag=wx.GROW)

    93

    94

        execution_control_label = wx.StaticText(self, label=_('Execution Control:'))

    95

        top_right_gridsizer.AddWindow(execution_control_label, 

    96

              flag=wx.ALIGN_CENTER_VERTICAL)

    97

    98

        self.ExecutionControl = wx.CheckBox(self)

    99

        self.Bind(wx.EVT_CHECKBOX, self.OnExecutionOrderChanged, self.ExecutionControl)

   100

        top_right_gridsizer.AddWindow(self.ExecutionControl, flag=wx.GROW)

   101

   102

        preview_label = wx.StaticText(self, label=_('Preview:'))

   103

        right_gridsizer.AddWindow(preview_label, flag=wx.GROW)

   104

   105

        self.Preview = wx.Panel(self,

   106

              style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   107

        self.Preview.SetBackgroundColour(wx.Colour(255,255,255))

   108

        setattr(self.Preview, "GetDrawingMode", lambda:FREEDRAWING_MODE)

   109

        setattr(self.Preview, "GetScaling", lambda:None)

   110

        setattr(self.Preview, "GetBlockType", controller.GetBlockType)

   111

        setattr(self.Preview, "IsOfType", controller.IsOfType)

   112

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

   113

        right_gridsizer.AddWindow(self.Preview, flag=wx.GROW)

   114

   115

        button_sizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)

   116

        self.Bind(wx.EVT_BUTTON, self.OnOK, button_sizer.GetAffirmativeButton())

   117

        main_sizer.AddSizer(button_sizer, border=20, 

   118

              flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)

   119

   120

        self.SetSizer(main_sizer)

   121

   122

        self.Controller = controller

   123

   124

        self.BlockName.SetValue("")

   125

        self.BlockName.Enable(False)

   126

        self.Inputs.Enable(False)

   127

        self.Block = None

   128

        self.MinBlockSize = None

   129

        self.First = True

   130

   131

        self.PouNames = []

   132

        self.PouElementNames = []

   133

   134

        self.LibraryPanel.SetFocus()

   135

   136

    def __del__(self):

   137

        self.Controller = None

   138

   139

    def SetBlockList(self, blocklist):

   140

        self.LibraryPanel.SetBlockList(blocklist)

   141

   142

    def SetPreviewFont(self, font):

   143

        self.Preview.SetFont(font)

   144

   145

    def OnOK(self, event):

   146

        message = None

   147

        selected = self.LibraryPanel.GetSelectedBlock()

   148

        block_name = self.BlockName.GetValue()

   149

        name_enabled = self.BlockName.IsEnabled()

   150

        if selected is None:

   151

            message = _("Form isn't complete. Valid block type must be selected!")

   152

        elif name_enabled and block_name == "":

   153

            message = _("Form isn't complete. Name must be filled!")

   154

        elif name_enabled and not TestIdentifier(block_name):

   155

            message = _("\"%s\" is not a valid identifier!") % block_name

   156

        elif name_enabled and block_name.upper() in IEC_KEYWORDS:

   157

            message = _("\"%s\" is a keyword. It can't be used!") % block_name

   158

        elif name_enabled and block_name.upper() in self.PouNames:

   159

            message = _("\"%s\" pou already exists!") % block_name

   160

        elif name_enabled and block_name.upper() in self.PouElementNames:

   161

            message = _("\"%s\" element for this pou already exists!") % block_name

   162

        if message is not None:

   163

            dialog = wx.MessageDialog(self, message, _("Error"), wx.OK|wx.ICON_ERROR)

   164

            dialog.ShowModal()

   165

            dialog.Destroy()

   166

        else:

   167

            self.EndModal(wx.ID_OK)

   168

   169

    def SetMinBlockSize(self, size):

   170

        self.MinBlockSize = size

   171

   172

    def SetPouNames(self, pou_names):

   173

        self.PouNames = [pou_name.upper() for pou_name in pou_names]

   174

   175

    def SetPouElementNames(self, element_names):

   176

        self.PouElementNames = [element_name.upper() for element_name in element_names]

   177

   178

    def SetValues(self, values):

   179

        blocktype = values.get("type", None)

   180

        if blocktype is not None:

   181

            self.LibraryPanel.SelectTreeItem(blocktype, values.get("inputs", None))

   182

        for name, value in values.items():

   183

            if name == "name":

   184

                self.BlockName.SetValue(value)

   185

            elif name == "extension":

   186

                self.Inputs.SetValue(value)

   187

            elif name == "executionOrder":

   188

                self.ExecutionOrder.SetValue(value)

   189

            elif name == "executionControl":

   190

                   self.ExecutionControl.SetValue(value)

   191

        self.RefreshPreview()

   192

   193

    def GetValues(self):

   194

        values = self.LibraryPanel.GetSelectedBlock()

   195

        if self.BlockName.IsEnabled() and self.BlockName.GetValue() != "":

   196

            values["name"] = self.BlockName.GetValue()

   197

        values["width"], values["height"] = self.Block.GetSize()

   198

        values["extension"] = self.Inputs.GetValue()

   199

        values["executionOrder"] = self.ExecutionOrder.GetValue()

   200

        values["executionControl"] = self.ExecutionControl.GetValue()

   201

        return values

   202

   203

    def OnLibraryTreeItemSelected(self, event):

   204

        values = self.LibraryPanel.GetSelectedBlock()

   205

        if values is not None:

   206

            blocktype = self.Controller.GetBlockType(values["type"], values["inputs"])

   207

        else:

   208

            blocktype = None

   209

        if blocktype is not None:

   210

            self.Inputs.SetValue(len(blocktype["inputs"]))

   211

            self.Inputs.Enable(blocktype["extensible"])

   212

            self.BlockName.Enable(blocktype["type"] != "function")

   213

            wx.CallAfter(self.RefreshPreview)

   214

        else:

   215

            self.BlockName.Enable(False)

   216

            self.Inputs.Enable(False)

   217

            self.Inputs.SetValue(2)

   218

            wx.CallAfter(self.ErasePreview)

   219

   220

    def OnNameChanged(self, event):

   221

        if self.BlockName.IsEnabled():

   222

            self.RefreshPreview()

   223

        event.Skip()

   224

   225

    def OnInputsChanged(self, event):

   226

        if self.Inputs.IsEnabled():

   227

            self.RefreshPreview()

   228

        event.Skip()

   229

   230

    def OnExecutionOrderChanged(self, event):

   231

        self.RefreshPreview()

   232

        event.Skip()

   233

   234

    def OnExecutionControlChanged(self, event):

   235

        self.RefreshPreview()

   236

        event.Skip()

   237

   238

    def ErasePreview(self):

   239

        dc = wx.ClientDC(self.Preview)

   240

        dc.Clear()

   241

        self.Block = None

   242

   243

    def RefreshPreview(self):

   244

        dc = wx.ClientDC(self.Preview)

   245

        dc.SetFont(self.Preview.GetFont())

   246

        dc.Clear()

   247

        values = self.LibraryPanel.GetSelectedBlock()

   248

        if values is not None:

   249

            if self.BlockName.IsEnabled():

   250

                blockname = self.BlockName.GetValue()

   251

            else:

   252

                blockname = ""

   253

            self.Block = FBD_Block(self.Preview, values["type"], 

   254

                    blockname, 

   255

                    extension = self.Inputs.GetValue(), 

   256

                    inputs = values["inputs"], 

   257

                    executionControl = self.ExecutionControl.GetValue(), 

   258

                    executionOrder = self.ExecutionOrder.GetValue())

   259

            width, height = self.MinBlockSize

   260

            min_width, min_height = self.Block.GetMinSize()

   261

            width, height = max(min_width, width), max(min_height, height)

   262

            self.Block.SetSize(width, height)

   263

            clientsize = self.Preview.GetClientSize()

   264

            x = (clientsize.width - width) / 2

   265

            y = (clientsize.height - height) / 2

   266

            self.Block.SetPosition(x, y)

   267

            self.Block.Draw(dc)

   268

        else:

   269

            self.Block = None        

   270

   271

    def OnPaint(self, event):

   272

        if self.Block is not None:

   273

            self.RefreshPreview()

   274

        event.Skip()