rotate ssd

gfxlcd/demos/ssd_3.py

@@ -0,0 +1,60 @@
+import RPi.GPIO
+import sys
+import random
+sys.path.append("../../")
+from gfxlcd.driver.ssd1306.spi import SPI
+from gfxlcd.driver.ssd1306.ssd1306 import SSD1306
+RPi.GPIO.setmode(RPi.GPIO.BCM)
+
+
+def hole(o, x, y):
+    o.draw_pixel(x+1, y)
+    o.draw_pixel(x+2, y)
+    o.draw_pixel(x+3, y)
+    o.draw_pixel(x+1, y + 4)
+    o.draw_pixel(x+2, y + 4)
+    o.draw_pixel(x+3, y + 4)
+    o.draw_pixel(x, y + 1)
+    o.draw_pixel(x+4, y + 1)
+    o.draw_pixel(x, y + 2)
+    o.draw_pixel(x+4, y + 2)
+    o.draw_pixel(x, y + 3)
+    o.draw_pixel(x+4, y + 3)
+
+
+def draw_points(o):
+    for _ in range(0, 50):
+        hole(o,
+             random.randint(2, o.width - 10),
+             random.randint(2, o.height - 10)
+        )
+
+
+def draw_net(o):
+    s = 0
+    while s < o.width-1:
+        o.draw_line(s, 0, s, o.height-1)
+        s += 10
+    s = 0
+    while s < o.height-1:
+        o.draw_line(0, s, o.width-1, s)
+        s += 10
+
+
+lcd_oled = SSD1306(128, 64, SPI())
+lcd_oled.rotation = 90
+lcd_oled.init()
+lcd_oled.auto_flush = False
+
+x, y = lcd_oled.width // 2, lcd_oled.height // 2
+lcd_oled.draw_circle(x, y, 31)
+lcd_oled.draw_circle(x-12, y-10, 7)
+lcd_oled.draw_circle(x+12, y-10, 7)
+lcd_oled.draw_arc(x, y, 20, 45, 135)
+lcd_oled.draw_line(x, y-5, x-4, y+6)
+lcd_oled.draw_line(x, y-5, x+4, y+6)
+lcd_oled.draw_arc(x, y+3, 5, 45, 135)
+
+lcd_oled.fill_rect(0, 0, 10, 10)
+
+lcd_oled.flush(True)

gfxlcd/demos/ssd_image_1.py

@@ -0,0 +1,19 @@
+import RPi.GPIO
+import sys
+from PIL import Image
+sys.path.append("../../")
+from gfxlcd.driver.ssd1306.spi import SPI
+from gfxlcd.driver.ssd1306.ssd1306 import SSD1306
+RPi.GPIO.setmode(RPi.GPIO.BCM)
+
+lcd_oled = SSD1306(128, 64, SPI())
+lcd_oled.rotation = 270
+lcd_oled.init()
+lcd_oled.auto_flush = False
+
+image_file = Image.open("assets/20x20.png")
+lcd_oled.threshold = 0
+
+lcd_oled.draw_image(10, 0, image_file)
+
+lcd_oled.flush(True)

gfxlcd/drawing/page.py

@@ -11,10 +11,16 @@ class Page(Pixel, metaclass=abc.ABCMeta):
 
     def init(self):
         """init page"""
+        #self.rotate = rotate
+        if self.rotation == 0 or self.rotation == 180:
             self.buffer = [[0] * (self.height // 8) for x in range(self.width)]
+        else:
+            self.buffer = [[0] * (self.width // 8) for x in range(self.height)]
 
     def draw_pixel(self, pos_x, pos_y):
         """draw a pixel at x,y"""
+        if self.rotation == 90 or self.rotation == 270:
+            pos_x, pos_y = pos_y, pos_x
         self.buffer[pos_x][pos_y//8] |= 1 << (pos_y % 8)
         self.flush()
 

gfxlcd/drawing/pixel.py

@@ -6,6 +6,7 @@ class Pixel(object):
     """Pixel class"""
     def __init__(self, driver):
         self.driver = driver
+        self.rotation = 0
         self.options['color'] = {
             'R': 255, 'G': 255, 'B': 255
         }
@@ -36,11 +37,11 @@ class Pixel(object):
         self.options['transparency_color'] = transparency_color
 
     def draw_pixel(self, pos_x, pos_y):
-        """dummy fuction"""
+        """dummy function"""
         pass
 
     def draw_line(self, pos_x1, pos_y1, pos_x2, pos_y2):
-        """dummy fuction"""
+        """dummy function"""
         pass
 
     def draw_rect(self, pos_x1, pos_y1, pos_x2, pos_y2):

gfxlcd/driver/ssd1306/ssd1306.py

@@ -5,6 +5,25 @@ from gfxlcd.abstract.chip import Chip
 
 class SSD1306(Page, Chip):
     """Class for an LCD with SSD306 chip"""
+    rotations = {
+        0: {
+            'sgmt': 0xa1,
+            'com': 0xc8
+        },
+        90: {
+            'sgmt': 0xa0,
+            'com': 0xc8
+        },
+        180: {
+            'sgmt': 0xa0,
+            'com': 0xc0
+        },
+        270: {
+            'sgmt': 0xa1,
+            'com': 0xc0
+        }
+    }
+
     def __init__(self, width, height, driver, auto_flush=True):
         Chip.__init__(self, width, height, driver, auto_flush)
         Page.__init__(self, driver)
@@ -13,7 +32,7 @@ class SSD1306(Page, Chip):
     def init(self):
         """inits a device"""
         self.driver.init()
-        Page.init(self)
+        Page.init(self)#, self.rotate)
         Chip.init(self)
         self.driver.reset()
         self.driver.cmd(0xae)  # turn off panel
@@ -27,8 +46,12 @@ class SSD1306(Page, Chip):
         self.driver.cmd(0xb0)  # set page address
         self.driver.cmd(0x81)  # set contrast control register
         self.driver.cmd(0xff)
-        self.driver.cmd(0xa1)  # a0/a1, a1 = segment 127 to 0, a0:0 to seg127
+        # a0/a1, a1 = segment 127 to 0, a0:0 to seg127
-        self.driver.cmd(0xc8)  # c8/c0 set com(N-1)to com0  c0:com0 to com(N-1)
+        self.driver.cmd(self.rotations[self.rotation]['sgmt'])
+
+        # c8/c0 set com(N-1)to com0  c0:com0 to com(N-1)
+        self.driver.cmd(self.rotations[self.rotation]['com'])
+
         self.driver.cmd(0xa6)  # set normal display, a6 - normal, a7 - inverted
 
         self.driver.cmd(0xa8)  # set multiplex ratio(16to63)
@@ -66,15 +89,19 @@ class SSD1306(Page, Chip):
         :force - boolean|None"""
         if force is None:
             force = self.options['auto_flush']
-
         if force:
-            for j in range(0, self.height//8):
+            if self.rotation == 0 or self.rotation == 180:
-                self.set_area(0, j, self.width-1, j+1)
+                height, width = self.height, self.width
-                for i in range(0, self.width):
+            else:
+                width, height = self.height, self.width
+            for j in range(0, height//8):
+                self.set_area(0, j, width-1, j+1)
+                for i in range(0, width):
                     self.driver.data(self.get_page_value(i, j))
 
     def set_area(self, pos_x1, pos_y1, pos_x2, pos_y2):
         """set area to work on"""
+
         self.driver.cmd(0x22)
         self.driver.cmd(0xb0 + pos_y1)
         self.driver.cmd(0xb0 + pos_y2)

gfxlcd/tests/test_chip.py

@@ -0,0 +1,31 @@
+__author__ = 'kosci'
+
+import sys
+from nose.tools import assert_equal
+sys.path.append("../../")
+from gfxlcd.driver.null.null_page import NullPage
+
+
+class TestChip(object):
+    def setUp(self):
+        self.lcd = NullPage(10, 16, None, False)
+
+    def test_rotate_by_0(self):
+        self.lcd.rotation = 0
+        assert_equal(10, self.lcd.width)
+        assert_equal(16, self.lcd.height)
+
+    def test_rotate_by_90(self):
+        self.lcd.rotation = 90
+        assert_equal(16, self.lcd.width)
+        assert_equal(10, self.lcd.height)
+
+    def test_rotate_by_180(self):
+        self.lcd.rotation = 180
+        assert_equal(10, self.lcd.width)
+        assert_equal(16, self.lcd.height)
+
+    def test_rotate_by_270(self):
+        self.lcd.rotation = 270
+        assert_equal(16, self.lcd.width)
+        assert_equal(10, self.lcd.height)