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

import os

    26

    27

import wx

    28

    29

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

    30

#                            Library Structures

    31

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

    32

    33

BitmapLibrary = {}

    34

BitmapFolders = []

    35

    36

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

    37

#                             Library Helpers

    38

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

    39

    40

    41

def AddBitmapFolder(path):

    42

    if os.path.exists(path) and os.path.isdir(path) and path not in BitmapFolders:

    43

        BitmapFolders.append(path)

    44

    45

    46

def SearchBitmap(bmp_name):

    47

    for folder in BitmapFolders:

    48

        bmp_path = os.path.join(folder, bmp_name + ".png")

    49

        if os.path.isfile(bmp_path):

    50

            return wx.Bitmap(bmp_path)

    51

    return None

    52

    53

    54

def GetBitmap(bmp_name1, bmp_name2=None, size=None):

    55

    bmp = BitmapLibrary.get((bmp_name1, bmp_name2, size))

    56

    if bmp is not None:

    57

        return bmp

    58

    59

    if bmp_name2 is None:

    60

        bmp = SearchBitmap(bmp_name1)

    61

    else:

    62

        # Bitmap with two icon

    63

        bmp1 = SearchBitmap(bmp_name1)

    64

        bmp2 = SearchBitmap(bmp_name2)

    65

    66

        if bmp1 is not None and bmp2 is not None:

    67

            # Calculate bitmap size

    68

            width = bmp1.GetWidth() + bmp2.GetWidth() - 1

    69

            height = max(bmp1.GetHeight(), bmp2.GetHeight())

    70

    71

            # Create bitmap with both icons

    72

            bmp = wx.EmptyBitmap(width, height)

    73

            dc = wx.MemoryDC()

    74

            dc.SelectObject(bmp)

    75

            dc.Clear()

    76

            dc.DrawBitmap(bmp1, 0, 0)

    77

            dc.DrawBitmap(bmp2, bmp1.GetWidth() - 1, 0)

    78

            dc.Destroy()

    79

    80

        elif bmp1 is not None:

    81

            bmp = bmp1

    82

        elif bmp2 is not None:

    83

            bmp = bmp2

    84

    85

    if bmp is not None:

    86

        BitmapLibrary[(bmp_name1, bmp_name2, size)] = bmp

    87

    88

    return bmp