rdagger / pi-st7565 Goto Github PK

	y-coordinate: {0} above maximum of {1}.
	Traceback (most recent call last):
	  File "/scratch.py", line 30, in <module>
	    is_off_grid(100, 50, 10, 50)
	  File "/scratch.py", line 26, in is_off_grid
	    print('y-coordinate: {0} above maximum of {1}.').format(ymax, LCD_HEIGHT - 1)
	AttributeError: 'NoneType' object has no attribute 'format'

# new imports: 
import os
import numpy as np
from converter import get_average_pixel_intensity, load_image, get_pixel_intensity

# new cache for the images
imagecache = dict()


class Glcd(object):
    # ... 

    # new method for drawing a png
    def draw_png(self, name, x, y, invert_colors=False, w=16, h=16):
        # TODO: find the propper way to get a relative path (in package and expanded)
        filename = os.path.join(os.path.dirname(__file__), "icons/%s.png" % name)
        bitmap = self._create_bitmap_from_image(filename, invert_colors, w, h)
        self.draw_bitmap(bitmap, x, y)



    @classmethod
    def _create_bitmap_from_image(cls, filename, invert_colors, width, height):
        if imagecache.has_key(filename):
            bitmap = imagecache[filename]
        else:
            width, height, image_data = load_image(filename, width, height)
            crossover_intensity = get_average_pixel_intensity(width, height, image_data, invert_colors)
            bitmap = cls._convert_pixeldata_to_bitmap(width, height, image_data, crossover_intensity, invert_colors)
            imagecache[filename] = bitmap
        return bitmap

    @classmethod
    def _convert_pixeldata_to_bitmap(cls, width, height, pixel_data, crossover, invert):
        """
        Outputs the data in a C bitmap array format.
        """
        bitmap = np.zeros((height, width), dtype='uint8')

        for y_idx in range(0, height):
            for x_idx in range(0, width):
                value = get_pixel_intensity(pixel_data[x_idx, y_idx], invert) > crossover
                bitmap[y_idx, x_idx] = value

        return bitmap