Coverage for src/robotide/widgets/images.py: 94%

1# Copyright 2008-2015 Nokia Networks 

2# Copyright 2016- Robot Framework Foundation 

3# 

4# Licensed under the Apache License, Version 2.0 (the "License"); 

5# you may not use this file except in compliance with the License. 

6# You may obtain a copy of the License at 

7# 

8# http://www.apache.org/licenses/LICENSE-2.0 

9# 

10# Unless required by applicable law or agreed to in writing, software 

11# distributed under the License is distributed on an "AS IS" BASIS, 

12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

13# See the License for the specific language governing permissions and 

14# limitations under the License. 

15 

16import os 2 ctx1ab

17 

18import wx 2 ctx1ab

19 

20 

21class ImageList(wx.ImageList): 2 ctx1ab

22 

23 def __init__(self, img_width, img_height): 2 ctx1ab

24 wx.ImageList.__init__(self, img_width, img_height) 

25 

26 def add_image(self, image): 2 ctx1ab

27 self.Add(image) 

28 

29 

30class ImageProvider(object): 2 ctx1ab

31 _BASE = os.path.dirname(__file__) 2 ctx1ab

32 

33 def __init__(self, size=(16, 16)): 2 ctx1ab

34 self._size = size (empty)

35 self.TESTCASEIMG = self._load_image('robot.png') (empty)

36 self.KEYWORDIMG = self._load_image('process.png') (empty)

37 self.DATADIRIMG = self._img_from_art_provider(wx.ART_FOLDER) (empty)

38 self.DATAFILEIMG = self._img_from_art_provider(wx.ART_NORMAL_FILE) (empty)

39 self.PROGICONS = self._load_prog_icons() (empty)

40 self.REPORTIMG = self._load_image('report.png') (empty)

41 self.REFRESH_ALL = self._load_image('database_refresh.png') (empty)

42 self.KW_SEARCH_ICON = self._load_image('kw_search_button.png') (empty)

43 self.TEST_SEARCH_ICON = self._load_image('test_search_button.png') (empty)

44 self.TOOLBAR_PLAY = self._load_image('control_play.png') (empty)

45 self.TOOLBAR_STOP = self._load_image('control_stop.png') (empty)

46 self.TOOLBAR_PAUSE = self._load_image('control_pause.png') (empty)

47 self.TOOLBAR_CONTINUE = self._load_image('control_play.png') (empty)

48 self.TOOLBAR_NEXT = self._load_image('control_fastforward.png') (empty)

49 self.SWITCH_FIELDS_ICON = self._load_image('switch.png') (empty)

50 self.RIDE_ICON = self._load_image('robot.ico') (empty)

51 self._build_icons() (empty)

52 

53 def _build_icons(self): 2 ctx1ab

54 self._icons = {} (empty)

55 for name in dir(self): (empty)

56 value = getattr(self, name) (empty)

57 if isinstance(value, wx.Bitmap): (empty)

58 self._icons[name] = getattr(self, name) (empty)

59 

60 def get_image_by_name(self, name): 2 ctx1ab

61 return self._icons.get(name, None) 

62 

63 def _load_image(self, name): 2 ctx1ab

64 path = self._get_img_path(name) (empty)

65 no_log = wx.LogNull() (empty)

66 # Suppress warnings generated by recent libpng versions. 

67 # The png files themselves are most likely ok. 

68 img = wx.Image(path, wx.BITMAP_TYPE_ANY).ConvertToBitmap() (empty)

69 del no_log (empty)

70 return img (empty)

71 

72 def _get_img_path(self, name): 2 ctx1ab

73 return os.path.join(self._BASE, name) (empty)

74 

75 def _img_from_art_provider(self, source): 2 ctx1ab

76 return wx.ArtProvider.GetBitmap(source, wx.ART_OTHER, self._size) (empty)

77 

78 def _load_prog_icons(self): 2 ctx1ab

79 icons = wx.IconBundle() (empty)

80 icons.AddIcon(self._get_img_path('robot.ico'), wx.BITMAP_TYPE_ANY) (empty)

81 return icons (empty)