images.py

Go to the documentation of this file.

#  Copyright 2008-2015 Nokia Networks
#  Copyright 2016-     Robot Framework Foundation
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.
 
import os
 
import wx
 
 
class ImageList(wx.ImageList):
 
    def __init__(self, img_width, img_height):
        wx.ImageList.__init__(self, img_width, img_height)
 
    def add(self, image):
        self.Add(image)
 
 
class ImageProvider():
    
    _BASE = os.path.dirname(__file__)
 
    def __init__(self, size=(16, 16)):
        self._size_size = size
        self.TESTCASEIMGTESTCASEIMG = self._load_image_load_image('robot.png')
        self.KEYWORDIMGKEYWORDIMG = self._load_image_load_image('process.png')
        self.DATADIRIMGDATADIRIMG = self._img_from_art_provider_img_from_art_provider(wx.ART_FOLDER)
        self.DATAFILEIMGDATAFILEIMG = self._img_from_art_provider_img_from_art_provider(wx.ART_NORMAL_FILE)
        self.PROGICONSPROGICONS = self._load_prog_icons_load_prog_icons()
        self.REPORTIMGREPORTIMG = self._load_image_load_image('report.png')
        self.REFRESH_ALLREFRESH_ALL = self._load_image_load_image('database_refresh.png')
        self.KW_SEARCH_ICONKW_SEARCH_ICON = self._load_image_load_image('kw_search_button.png')
        self.TEST_SEARCH_ICONTEST_SEARCH_ICON = self._load_image_load_image('test_search_button.png')
        self.TOOLBAR_PLAYTOOLBAR_PLAY = self._load_image_load_image('control_play.png')
        self.TOOLBAR_STOPTOOLBAR_STOP = self._load_image_load_image('control_stop.png')
        self.TOOLBAR_PAUSETOOLBAR_PAUSE = self._load_image_load_image('control_pause.png')
        self.TOOLBAR_CONTINUETOOLBAR_CONTINUE = self._load_image_load_image('control_play.png')
        self.TOOLBAR_NEXTTOOLBAR_NEXT = self._load_image_load_image('control_fastforward.png')
        self.SWITCH_FIELDS_ICONSWITCH_FIELDS_ICON = self._load_image_load_image('switch.png')
        self.RIDE_ICONRIDE_ICON = self._load_image_load_image('robot.ico')
        self._build_icons_build_icons()
 
    def _build_icons(self):
        self._icons_icons = {}
        for name in dir(self):
            value = getattr(self, name)
            if isinstance(value, wx.Bitmap):
                self._icons_icons[name] = getattr(self, name)
 
    def get_image_by_name(self, name):
        return self._icons_icons.get(name, None)
 
    def _load_image(self, name):
        path = self._get_img_path_get_img_path(name)
        noLog = wx.LogNull()
        # Suppress warnings generated by recent libpng versions.
        # The png files themselves are most likely ok.
        img = wx.Image(path, wx.BITMAP_TYPE_ANY).ConvertToBitmap()
        del noLog
        return img
 
    def _get_img_path(self, name):
        return os.path.join(self._BASE_BASE, name)
 
    def _img_from_art_provider(self, source):
        return wx.ArtProvider.GetBitmap(source, wx.ART_OTHER, self._size_size)
 
    def _load_prog_icons(self):
        icons = wx.IconBundle()
        icons.AddIcon(self._get_img_path_get_img_path('robot.ico'), wx.BITMAP_TYPE_ANY)
        return icons