aboutsummaryrefslogtreecommitdiffstats
path: root/src/plugins/image.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/plugins/image.py114
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)
+