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

from __future__ import absolute_import

    27

import os

    28

    29

import wx

    30

    31

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

    32

#                            Library Structures

    33

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

    34

    35

BitmapLibrary = {}

    36

BitmapFolders = []

    37

    38

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

    39

#                             Library Helpers

    40

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

    41

    42

    43

def AddBitmapFolder(path):

    44

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

    45

        BitmapFolders.append(path)

    46

    47

    48

def SearchBitmap(bmp_name):

    49

    for folder in BitmapFolders:

    50

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

    51

        if os.path.isfile(bmp_path):

    52

            return wx.Bitmap(bmp_path)

    53

    return None

    54

    55

    56

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

    57

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

    58

    if bmp is not None:

    59

        return bmp

    60

    61

    if bmp_name2 is None:

    62

        bmp = SearchBitmap(bmp_name1)

    63

    else:

    64

        # Bitmap with two icon

    65

        bmp1 = SearchBitmap(bmp_name1)

    66

        bmp2 = SearchBitmap(bmp_name2)

    67

    68

        if bmp1 is not None and bmp2 is not None:

    69

            # Calculate bitmap size

    70

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

    71

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

    72

    73

            # Create bitmap with both icons

    74

            bmp = wx.Bitmap(width, height)

    75

            dc = wx.MemoryDC()

    76

            dc.SelectObject(bmp)

    77

            dc.Clear()

    78

            dc.DrawBitmap(bmp1, 0, 0)

    79

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

    80

            dc.Destroy()

    81

    82

        elif bmp1 is not None:

    83

            bmp = bmp1

    84

        elif bmp2 is not None:

    85

            bmp = bmp2

    86

    87

    if bmp is not None:

    88

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

    89

    90

    return bmp