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) 2012: 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

    27

import wx

    28

    29

from util.BitmapLibrary import GetBitmap

    30

    31

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

    32

#                        Custom button for Graphic Viewer Class

    33

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

    34

    35

    36

class GraphButton(object):

    37

    """

    38

    Class that implements a custom button for graphic Viewer

    39

    """

    40

    41

    def __init__(self, x, y, bitmap, callback):

    42

        """

    43

        Constructor

    44

        @param x: X coordinate of Button in Graphic Viewer

    45

        @param y: Y coordinate of Button in Graphic Viewer

    46

        @param bitmap: Name of bitmap to use for button

    47

        @param callback: Reference to function to call when button is pressed

    48

        """

    49

        # Save button position

    50

        self.SetPosition(x, y)

    51

        # Set button bitmap

    52

        self.SetBitmap(bitmap)

    53

    54

        # By default button is hide and enabled

    55

        self.Shown = False

    56

        self.Enabled = True

    57

    58

        # Save reference to callback function

    59

        self.Callback = callback

    60

    61

    def SetBitmap(self, bitmap):

    62

        """

    63

        Set bitmap to use for button

    64

        @param bitmap: Name of bitmap to use for button

    65

        """

    66

        # Get wx.Bitmap object corresponding to bitmap

    67

        self.Bitmap = GetBitmap(bitmap)

    68

    69

    def GetSize(self):

    70

        """

    71

        Return size of button

    72

        @return: wx.Size object containing button size

    73

        """

    74

        # Button size is size of bitmap

    75

        return self.Bitmap.GetSize()

    76

    77

    def SetPosition(self, x, y):

    78

        """

    79

        Set button position

    80

        @param x: X coordinate of Button in Graphic Viewer

    81

        @param y: Y coordinate of Button in Graphic Viewer

    82

        """

    83

        self.Position = wx.Point(x, y)

    84

    85

    def Show(self, show=True):

    86

        """

    87

        Mark if button to be displayed in Graphic Viewer

    88

        @param show: True if button to be displayed in Graphic Viewer

    89

        (default True)

    90

        """

    91

        self.Shown = show

    92

    93

    def Hide(self):

    94

        """

    95

        Hide button from Graphic Viewer

    96

        """

    97

        self.Show(False)

    98

    99

    def IsShown(self):

   100

        """

   101

        Return if button is displayed in Graphic Viewer

   102

        @return: True if button is displayed in Graphic Viewer

   103

        """

   104

        return self.Shown

   105

   106

    def Enable(self, enable=True):

   107

        """

   108

        Mark if button is active in Graphic Viewer

   109

        @param enable: True if button is active in Graphic Viewer

   110

        (default True)

   111

        """

   112

        self.Enabled = enable

   113

   114

    def Disable(self):

   115

        """

   116

        Deactivate button in Graphic Viewer

   117

        """

   118

        self.Enabled = False

   119

   120

    def IsEnabled(self):

   121

        """

   122

        Return if button is active in Graphic Viewer

   123

        @return: True if button is active in Graphic Viewer

   124

        """

   125

        return self.Enabled

   126

   127

    def HitTest(self, x, y):

   128

        """

   129

        Test if point is inside button

   130

        @param x: X coordinate of point

   131

        @param y: Y coordinate of point

   132

        @return: True if button is active and displayed and point is inside

   133

        button

   134

        """

   135

        # Return immediately if button is hidden or inactive

   136

        if not (self.IsShown() and self.IsEnabled()):

   137

            return False

   138

   139

        # Test if point is inside button

   140

        w, h = self.Bitmap.GetSize()

   141

        rect = wx.Rect(self.Position.x, self.Position.y, w, h)

   142

        return rect.Contains(x, y)

   143

   144

    def ProcessCallback(self):

   145

        """

   146

        Call callback function if defined

   147

        """

   148

        if self.Callback is not None:

   149

            self.Callback()

   150

   151

    def Draw(self, dc):

   152

        """

   153

        Draw button in Graphic Viewer

   154

        @param dc: wx.DC object corresponding to Graphic Viewer device context

   155

        """

   156

        # Only draw button if button is active and displayed

   157

        if self.Shown and self.Enabled:

   158

            dc.DrawBitmap(self.Bitmap, self.Position.x, self.Position.y, True)