drivers/display/ssd1306: Make poweron() work the same with SSD1306_SPI.

dpgeorge · dpgeorge · commit ca2427c31399 · 2017-10-06T12:48:44.000+11:00
The poweroff() and poweron() methods are used to do soft power control of
the display, and this patch makes these methods work the same for both I2C
and SPI interfaces.
diff --git a/drivers/display/ssd1306.py b/drivers/display/ssd1306.py
@@ -1,7 +1,6 @@
 # MicroPython SSD1306 OLED driver, I2C and SPI interfaces
 
 from micropython import const
-import time
 import framebuf
 
 
@@ -47,7 +46,6 @@ def __init__(self, width, height, external_vcc):
         self.text = fb.text
         self.scroll = fb.scroll
         self.blit = fb.blit
-        self.poweron()
         self.init_display()
 
     def init_display(self):
@@ -80,6 +78,9 @@ def init_display(self):
     def poweroff(self):
         self.write_cmd(SET_DISP | 0x00)
 
+    def poweron(self):
+        self.write_cmd(SET_DISP | 0x01)
+
     def contrast(self, contrast):
         self.write_cmd(SET_CONTRAST)
         self.write_cmd(contrast)
@@ -123,9 +124,6 @@ def write_data(self, buf):
         self.i2c.write(buf)
         self.i2c.stop()
 
-    def poweron(self):
-        self.write_cmd(SET_DISP | 0x01)
-
 
 class SSD1306_SPI(SSD1306):
     def __init__(self, width, height, spi, dc, res, cs, external_vcc=False):
@@ -137,6 +135,12 @@ def __init__(self, width, height, spi, dc, res, cs, external_vcc=False):
         self.dc = dc
         self.res = res
         self.cs = cs
+        import time
+        self.res(1)
+        time.sleep_ms(1)
+        self.res(0)
+        time.sleep_ms(10)
+        self.res(1)
         super().__init__(width, height, external_vcc)
 
     def write_cmd(self, cmd):
@@ -154,10 +158,3 @@ def write_data(self, buf):
         self.cs(0)
         self.spi.write(buf)
         self.cs(1)
-
-    def poweron(self):
-        self.res(1)
-        time.sleep_ms(1)
-        self.res(0)
-        time.sleep_ms(10)
-        self.res(1)
