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

        self.LibraryPanel.SetController(controller)

    55

        setattr(self.LibraryPanel, "_OnTreeItemSelected", 

    56

              self.OnLibraryTreeItemSelected)

    57

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

    58

              flag=wx.GROW|wx.TOP)

    59

    60

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

    61

        right_gridsizer.AddGrowableCol(0)

    62

        right_gridsizer.AddGrowableRow(2)

    63

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

    64

              flag=wx.GROW|wx.LEFT)

    65

    66

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

    67

        top_right_gridsizer.AddGrowableCol(1)

    68

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

    69

    70

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

    71

        top_right_gridsizer.AddWindow(name_label, 

    72

              flag=wx.ALIGN_CENTER_VERTICAL)

    73

    74

        self.BlockName = wx.TextCtrl(self)

    75

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

    76

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

    77

    78

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

    79

        top_right_gridsizer.AddWindow(inputs_label, 

    80

              flag=wx.ALIGN_CENTER_VERTICAL)

    81

    82

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

    83

              style=wx.SP_ARROW_KEYS)

    84

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

    85

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

    86

    87

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

    88

        top_right_gridsizer.AddWindow(execution_order_label, 

    89

              flag=wx.ALIGN_CENTER_VERTICAL)

    90

    91

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

    92

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

    93

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

    94

    95

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

    96

        top_right_gridsizer.AddWindow(execution_control_label, 

    97

              flag=wx.ALIGN_CENTER_VERTICAL)

    98

    99

        self.ExecutionControl = wx.CheckBox(self)

   100

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

   101

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

   102

   103

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

   104

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

   105

   106

        self.Preview = wx.Panel(self,

   107

              style=wx.TAB_TRAVERSAL|wx.SIMPLE_BORDER)

   108

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

   109

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

   110

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

   111

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

   112

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

   113

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

   114

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

   115

   116

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

   117

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

   118

        main_sizer.AddSizer(button_sizer, border=20, 

   119

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

   120

   121

        self.SetSizer(main_sizer)

   122

   123

        self.Controller = controller

   124

   125

        self.BlockName.SetValue("")

   126

        self.BlockName.Enable(False)

   127

        self.Inputs.Enable(False)

   128

        self.Block = None

   129

        self.MinBlockSize = None

   130

        self.First = True

   131

   132

        self.PouNames = []

   133

        self.PouElementNames = []

   134

   135

        self.LibraryPanel.SetFocus()

   136

   137

    def SetBlockList(self, blocklist):

   138

        self.LibraryPanel.SetBlockList(blocklist)

   139

   140

    def SetPreviewFont(self, font):

   141

        self.Preview.SetFont(font)

   142

   143

    def OnOK(self, event):

   144

        message = None

   145

        selected = self.LibraryPanel.GetSelectedBlock()

   146

        block_name = self.BlockName.GetValue()

   147

        name_enabled = self.BlockName.IsEnabled()

   148

        if selected is None:

   149

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

   150

        elif name_enabled and block_name == "":

   151

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

   152

        elif name_enabled and not TestIdentifier(block_name):

   153

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

   154

        elif name_enabled and block_name.upper() in IEC_KEYWORDS:

   155

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

   156

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

   157

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

   158

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

   159

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

   160

        if message is not None:

   161

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

   162

            dialog.ShowModal()

   163

            dialog.Destroy()

   164

        else:

   165

            self.EndModal(wx.ID_OK)

   166

   167

    def SetMinBlockSize(self, size):

   168

        self.MinBlockSize = size

   169

   170

    def SetPouNames(self, pou_names):

   171

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

   172

   173

    def SetPouElementNames(self, element_names):

   174

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

   175

   176

    def SetValues(self, values):

   177

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

   178

        if blocktype is not None:

   179

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

   180

        for name, value in values.items():

   181

            if name == "name":

   182

                self.BlockName.SetValue(value)

   183

            elif name == "extension":

   184

                self.Inputs.SetValue(value)

   185

            elif name == "executionOrder":

   186

                self.ExecutionOrder.SetValue(value)

   187

            elif name == "executionControl":

   188

                   self.ExecutionControl.SetValue(value)

   189

        self.RefreshPreview()

   190

   191

    def GetValues(self):

   192

        values = self.LibraryPanel.GetSelectedBlock()

   193

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

   194

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

   195

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

   196

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

   197

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

   198

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

   199

        return values

   200

   201

    def OnLibraryTreeItemSelected(self, event):

   202

        values = self.LibraryPanel.GetSelectedBlock()

   203

        if values is not None:

   204

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

   205

        else:

   206

            blocktype = None

   207

        if blocktype is not None:

   208

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

   209

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

   210

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

   211

            wx.CallAfter(self.RefreshPreview)

   212

        else:

   213

            self.BlockName.Enable(False)

   214

            self.Inputs.Enable(False)

   215

            self.Inputs.SetValue(2)

   216

            wx.CallAfter(self.ErasePreview)

   217

   218

    def OnNameChanged(self, event):

   219

        if self.BlockName.IsEnabled():

   220

            self.RefreshPreview()

   221

        event.Skip()

   222

   223

    def OnInputsChanged(self, event):

   224

        if self.Inputs.IsEnabled():

   225

            self.RefreshPreview()

   226

        event.Skip()

   227

   228

    def OnExecutionOrderChanged(self, event):

   229

        self.RefreshPreview()

   230

        event.Skip()

   231

   232

    def OnExecutionControlChanged(self, event):

   233

        self.RefreshPreview()

   234

        event.Skip()

   235

   236

    def ErasePreview(self):

   237

        dc = wx.ClientDC(self.Preview)

   238

        dc.Clear()

   239

        self.Block = None

   240

   241

    def RefreshPreview(self):

   242

        dc = wx.ClientDC(self.Preview)

   243

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

   244

        dc.Clear()

   245

        values = self.LibraryPanel.GetSelectedBlock()

   246

        if values is not None:

   247

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

   248

                    self.BlockName.GetValue(), 

   249

                    extension = self.Inputs.GetValue(), 

   250

                    inputs = values["inputs"], 

   251

                    executionControl = self.ExecutionControl.GetValue(), 

   252

                    executionOrder = self.ExecutionOrder.GetValue())

   253

            width, height = self.MinBlockSize

   254

            min_width, min_height = self.Block.GetMinSize()

   255

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

   256

            self.Block.SetSize(width, height)

   257

            clientsize = self.Preview.GetClientSize()

   258

            x = (clientsize.width - width) / 2

   259

            y = (clientsize.height - height) / 2

   260

            self.Block.SetPosition(x, y)

   261

            self.Block.Draw(dc)

   262

        else:

   263

            self.Block = None        

   264

   265

    def OnPaint(self, event):

   266

        if self.Block is not None:

   267

            self.RefreshPreview()

   268

        event.Skip()