diff options
Diffstat (limited to '')
| -rw-r--r-- | src/plugins/image.py | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/src/plugins/image.py b/src/plugins/image.py new file mode 100644 index 0000000..6d40179 --- /dev/null +++ b/src/plugins/image.py @@ -0,0 +1,114 @@ +# -*- python -*- +''' +xpybar – xmobar replacement written in python +Copyright © 2014, 2015 Mattias Andrée (maandree@member.fsf.org) + +This program is free software: you can redistribute it and/or modify +it under the terms of the GNU General Public License as published by +the Free Software Foundation, either version 3 of the License, or +(at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program. If not, see <http://www.gnu.org/licenses/>. +''' + + +class Image: + ''' + Images and icons + ''' + + + def __init__(self, file, background = 'black', width = None, height = None): + ''' + Constructor + + @param file:str The pathname of the image + @param background:str ImageMagick understandable string for the background colour + @param width:int? The width the image should have, `None` to not resize or use `height` + @param height:int? The height the image should have, `None` to not resize or use `width` + ''' + import Xlib.X, sys + from subprocess import Popen, PIPE + + self.format = Xlib.X.ZPixmap + self.depth = 24 + + height = width if height is None else height + width = height if width is None else width + raster = None + + convert = ['file', '-'] + convert = Popen(convert, stdin = open(file, 'rb'), stdout = PIPE, stderr = sys.stderr) + if 'Scalable Vector Graphics' in convert.communicate()[0].decode('utf-8', 'replace'): + convert = ['rsvg-convert', '-f', 'png'] + if width is not None: + convert += ['-w', str(width), '-h', str(height)] + convert = Popen(convert, stdin = open(file, 'rb'), stdout = PIPE, stderr = sys.stderr) + raster = convert.communicate()[0] + if not convert.wait() == 0: + raise Exception('Image could not be converted') + + convert = ['convert', '-', '-background', background, '-alpha', 'remove', '-depth', '8'] + if width is not None: + convert += ['-resize', '%ix%i!' % (width, height)] + convert += ['ppm:-'] + convert = Popen(convert, stdin = PIPE if raster is not None else open(file, 'rb'), + stdout = PIPE, stderr = sys.stderr) + if raster is not None: + self.data = list(convert.communicate(raster)[0]) + else: + self.data = list(convert.communicate()[0]) + if not convert.wait() == 0: + raise Exception('Image could not be converted') + + self.width, self.height, state, comment = [], [], 0, False + for i in range(len(self.data)): + b = self.data[i] + if comment: + if b == ord('\n'): + comment = False + elif b == ord('#'): + comment = True + elif b in (ord(' '), ord('\n')): + if (state & 1) == 0: + continue + state += 1 + if state == 4 * 2: + break + else: + state += 1 - (state & 1) + if state == 2 * 1 + 1: + self.width.append(b) + elif state == 2 * 2 + 1: + self.height.append(b) + + self.data, buf = [], self.data[i:] + self.width = int(bytes(self.width).decode('utf-8', 'strict')) + self.height = int(bytes(self.height).decode('utf-8', 'strict')) + + i = 0 + for _i in range(len(buf) // 3): + self.data.append(buf[i]) + self.data.append(buf[i + 1]) + self.data.append(buf[i + 2]) + self.data.append(0) + i += 3 + self.data = bytes(self.data) + + + def draw(self, bar, x, y): + ''' + Draw the image on a bar + + @param bar:Bar The bar to draw the image on + @param x:int The left position of the image + @param y:int The top position of the image + ''' + bar.window.put_image(bar.gc, x, y, self.width, self.height, self.format, self.depth, 0, self.data) + |