From bf61d4881ec4cb7ab2e73a21da2349c681e371a5 Mon Sep 17 00:00:00 2001 From: Lorenz Diener Date: Sat, 11 May 2019 00:55:40 +0200 Subject: Add blurhash code --- mastodon/Mastodon.py | 38 +++++++++++++++++++++++++++++++++++++- 1 file changed, 37 insertions(+), 1 deletion(-) (limited to 'mastodon') diff --git a/mastodon/Mastodon.py b/mastodon/Mastodon.py index 27acbaa..d725dd9 100644 --- a/mastodon/Mastodon.py +++ b/mastodon/Mastodon.py @@ -25,6 +25,7 @@ from cryptography.hazmat.primitives.asymmetric import ec import http_ece import base64 import json +import blurhash try: from urllib.parse import urlparse @@ -2315,7 +2316,42 @@ class Mastodon: ) return json.loads(decrypted.decode('utf-8'), object_hook = Mastodon.__json_hooks) - + + ### + # Blurhash utilities + ### + def decode_blurhash(self, media_dict, out_size = (16, 16), size_per_component = True, return_linear = True): + """ + Basic media-dict blurhash decoding. + + out_size is the desired result size in pixels, either absolute or per blurhash + component (this is the default). + + By default, this function will return the image as linear RGB, ready for further + scaling operations. If you want to display the image directly, set return_linear + to False. + + Returns the decoded blurhash image as a three-dimensional list: [height][width][3], + with the last dimension being RGB colours. + + For further info and tips for advanced usage, refer to the documentation for the + blurhash module: https://github.com/halcy/blurhash-python + """ + # Figure out what size to decode to + decode_components_x, decode_components_y = blurhash.components(media_dict["blurhash"]) + if size_per_component == False: + decode_size_x = out_size[0] + decode_size_y = out_size[1] + else: + decode_size_x = decode_components_x * out_size[0] + decode_size_y = decode_components_y * out_size[1] + + # Decode + decoded_image = blurhash.decode(media_dict["blurhash"], decode_size_x, decode_size_y, linear = return_linear) + + # And that's pretty much it. + return decoded_image + ### # Pagination ### -- cgit v1.2.3