First commit

Author: Vincent Pillet

Dockerfile

@@ -0,0 +1,6 @@
+FROM python
+RUN apt-get update && apt-get -y install python3-pil python3-numpy python3-smbus
+WORKDIR /data
+COPY . .
+RUN pip install -r requirements.txt 
+CMD ["python", "example/OLED_1in5_rgb_test.py"]

example/OLED_1in5_rgb_test.py

@@ -0,0 +1,78 @@
+#!/usr/bin/python
+# -*- coding:utf-8 -*-
+
+import sys
+import os
+picdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'pic')
+libdir = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), 'lib')
+if os.path.exists(libdir):
+    sys.path.append(libdir)
+
+import logging    
+import time
+import traceback
+from waveshare_OLED import OLED_1in5_rgb
+from PIL import Image,ImageDraw,ImageFont
+logging.basicConfig(level=logging.DEBUG)
+
+try:
+    disp = OLED_1in5_rgb.OLED_1in5_rgb()
+
+    logging.info("\r 1.5inch rgb OLED ")
+    # Initialize library.
+    disp.Init()
+    # Clear display.
+    logging.info("clear display")
+    disp.clear()
+
+    # Create blank image for drawing.
+    image1 = Image.new('RGB', (disp.width, disp.height), 0)
+    draw = ImageDraw.Draw(image1)
+    font = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 12)
+    font1 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 18)
+    font2 = ImageFont.truetype(os.path.join(picdir, 'Font.ttc'), 24)
+    logging.info ("***draw line")
+    draw.line([(0,0),(127,0)], fill = "RED")
+    draw.line([(0,0),(0,127)], fill = "RED")
+    draw.line([(0,127),(127,127)], fill = "RED")
+    draw.line([(127,0),(127,127)], fill = "RED")
+    logging.info ("***draw text")
+    draw.text((20,0), 'Waveshare ', font = font1, fill = "BLUE")
+    draw.text((20,24), u'微雪电子 ', font = font2, fill = "MAGENTA")
+    draw.text((20,64), 'Waveshare ', font = font1, fill = "CYAN")
+    draw.text((20,92), u'微雪电子 ', font = font2, fill = "GREEN")
+    image1 = image1.rotate(0)
+    disp.ShowImage(disp.getbuffer(image1))
+    time.sleep(3)
+
+    logging.info ("***draw rectangle")
+    image1 = Image.new("RGB", (disp.width, disp.height), "BLACK")
+    draw = ImageDraw.Draw(image1)
+    draw.line([(0,8), (127,8)],   fill = "RED",    width = 16)
+    draw.line([(0,24),(127,24)],  fill = "YELLOW", width = 16)
+    draw.line([(0,40),(127,40)],  fill = "GREEN",  width = 16)
+    draw.line([(0,56),(127,56)],  fill = "CYAN",   width = 16)
+    draw.line([(0,72),(127,72)],  fill = "BLUE",   width = 16)
+    draw.line([(0,88),(127,88)],  fill = "MAGENTA",width = 16)
+    draw.line([(0,104),(127,104)],fill = "BLACK",  width = 16)
+    draw.line([(0,120),(127,120)],fill = "WHITE",  width = 16)
+    disp.ShowImage(disp.getbuffer(image1))
+    time.sleep(3)
+
+    logging.info ("***draw image")
+    Himage2 = Image.new('RGB', (disp.width, disp.height), 0)  # 0: clear the frame
+    bmp = Image.open(os.path.join(picdir, '1in5_rgb.bmp'))
+    Himage2.paste(bmp, (0,0))
+    Himage2=Himage2.rotate(0) 	
+    disp.ShowImage(disp.getbuffer(Himage2)) 
+    time.sleep(3)    
+
+    disp.clear()
+
+except IOError as e:
+    logging.info(e)
+    
+except KeyboardInterrupt:    
+    logging.info("ctrl + c:")
+    disp.module_exit()
+    exit()

example/setup.py

@@ -0,0 +1,10 @@
+import sys
+from setuptools import setup
+setup(
+    name='waveshare-OLED',
+    description='Waveshare OLED Display',
+    author='Waveshare',
+    package_dir={'': 'lib'},
+    packages=['waveshare_OLED'],
+)
+

example/waveshare_OLED/OLED_1in5_rgb.py

@@ -0,0 +1,172 @@
+# /*****************************************************************************
+# * | File        :	  OLED_1in5_rgb.py
+# * | Author      :   Waveshare team
+# * | Function    :   Driver for OLED_1in5_rgb
+# * | Info        :
+# *----------------
+# * | This version:   V2.0
+# * | Date        :   2020-08-17
+# * | Info        :   
+# ******************************************************************************/
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documnetation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to  whom the Software is
+# furished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+from . import config
+import time
+import numpy as np
+
+Device_SPI = config.Device_SPI
+Device_I2C = config.Device_I2C
+
+OLED_WIDTH   = 128  #OLED width
+OLED_HEIGHT  = 128  #OLED height
+
+class OLED_1in5_rgb(config.RaspberryPi):
+        
+    """    Write register address and data     """
+    def command(self, cmd):
+        self.digital_write(self.DC_PIN,False)
+        self.spi_writebyte([cmd])
+
+    """    Write data     """
+    def data(self, data):
+        self.digital_write(self.DC_PIN,True)
+        self.spi_writebyte([data])
+
+    def Init(self):
+        if (self.module_init() != 0):
+            return -1
+
+        self.width = OLED_WIDTH
+        self.height = OLED_HEIGHT
+
+        """Initialize dispaly"""    
+        self.reset()
+
+        if(self.Device == Device_I2C):
+            print ("Only Device_SPI, Please revise config.py !!!")
+            exit()  
+            
+        self.command(0xfd) # command lock
+        self.data(0x12)
+        self.command(0xfd) # command lock
+        self.data(0xB1)
+
+        self.command(0xae) # display off
+        self.command(0xa4) # Normal Display mode
+
+        self.command(0x15) # set column address
+        self.data(0x00)    # column address start 00
+        self.data(0x7f)    # column address end 127
+        self.command(0x75) # set row address
+        self.data(0x00)    # row address start 00
+        self.data(0x7f)    # row address end 127   
+
+        self.command(0xB3)
+        self.data(0xF1)
+
+        self.command(0xCA)
+        self.data(0x7F)
+
+        self.command(0xa0) # set re-map & data format
+        self.data(0x74)    # Horizontal address increment
+
+        self.command(0xa1) # set display start line
+        self.data(0x00)    # start 00 line
+
+        self.command(0xa2) # set display offset
+        self.data(0x00)
+
+        self.command(0xAB)
+        self.command(0x01)
+
+        self.command(0xB4)
+        self.data(0xA0)
+        self.data(0xB5)
+        self.data(0x55)
+
+        self.command(0xC1)
+        self.data(0xC8)
+        self.data(0x80)
+        self.data(0xC0)
+
+        self.command(0xC7)
+        self.data(0x0F)
+
+        self.command(0xB1)
+        self.data(0x32)
+
+        self.command(0xB2)
+        self.data(0xA4)
+        self.data(0x00)
+        self.data(0x00)
+
+        self.command(0xBB)
+        self.data(0x17)
+
+        self.command(0xB6)
+        self.data(0x01)
+
+        self.command(0xBE)
+        self.data(0x05)
+
+        self.command(0xA6)
+
+        time.sleep(0.1)
+        self.command(0xAF);#--turn on oled panel
+        
+   
+    def reset(self):
+        """Reset the display"""
+        self.digital_write(self.RST_PIN,True)
+        time.sleep(0.1)
+        self.digital_write(self.RST_PIN,False)
+        time.sleep(0.1)
+        self.digital_write(self.RST_PIN,True)
+        time.sleep(0.1)
+
+    def clear(self):
+        _buffer = [0x00]*(self.width * self.height * 2)
+        self.ShowImage(_buffer)             
+    
+    def getbuffer(self, image):
+        buf = [0x00] * ((self.width*2) * self.height)
+        imwidth, imheight = image.size
+        pixels = image.load()
+        for y in range(imheight):
+            for x in range(imwidth):
+                # Set the bits for the column of pixels at the current position.
+                buf[x*2 + y*imwidth*2] = ((pixels[x,y][0] & 0xF8) | (pixels[x,y][1] >> 5))
+                buf[x*2+1 + y*imwidth*2] = (((pixels[x,y][1]<<3) & 0xE0) | (pixels[x,y][2] >> 3))
+        return buf   
+
+    def ShowImage(self, pBuf):
+        self.command(0x15) # set column address
+        self.data(0x00)    # column address start 00
+        self.data(0x7f)    # column address end 127
+        self.command(0x75) # set row address
+        self.data(0x00)    # row address start 00
+        self.data(0x7f)    # row address end 127   
+        self.command(0x5C); 
+        for i in range(0, self.height):
+            for j in range(0, self.width*2):
+                self.data(pBuf[j + self.width*2*i])
+        return
+
+       

example/waveshare_OLED/__init__.py

@@ -0,0 +1,99 @@
+# /*****************************************************************************
+# * | File        :	  config.py
+# * | Author      :   Waveshare team
+# * | Function    :   Hardware underlying interface,for Raspberry pi
+# * | Info        :
+# *----------------
+# * | This version:   V1.0
+# * | Date        :   2020-06-17
+# * | Info        :   
+# ******************************************************************************/
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documnetation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to  whom the Software is
+# furished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+
+import time
+from smbus import SMBus
+import spidev
+import ctypes
+from gpiozero import *
+
+Device_SPI = 1
+Device_I2C = 0
+
+class RaspberryPi:
+    def __init__(self,spi=spidev.SpiDev(0,0),spi_freq=10000000,rst = 27,dc = 25,bl = 18,bl_freq=1000,i2c=None):
+        self.INPUT = False
+        self.OUTPUT = True
+        
+        self.SPEED  =spi_freq
+
+        if(Device_SPI == 1):
+            self.Device = Device_SPI
+            self.spi = spi
+        else :
+            self.Device = Device_I2C
+            self.address = 0x3c
+            self.bus = SMBus(1)
+        
+        self.RST_PIN = self.gpio_mode(rst,self.OUTPUT)
+        self.DC_PIN = self.gpio_mode(dc,self.OUTPUT)
+
+
+    def delay_ms(self,delaytime):
+        time.sleep(delaytime / 1000.0)
+
+    def gpio_mode(self,Pin,Mode,pull_up = None,active_state = True):
+        if Mode:
+            return DigitalOutputDevice(Pin,active_high = True,initial_value =False)
+        else:
+            return DigitalInputDevice(Pin,pull_up=pull_up,active_state=active_state)
+
+    def digital_write(self, Pin, value):
+        if value:
+            Pin.on()
+        else:
+            Pin.off()
+
+    def digital_read(self, Pin):
+        return Pin.value
+
+    def spi_writebyte(self,data):
+        self.spi.writebytes([data[0]])
+
+    def i2c_writebyte(self,reg, value):
+        self.bus.write_byte_data(self.address, reg, value)
+    
+    def module_init(self): 
+        self.digital_write(self.RST_PIN,False)
+        if(self.Device == Device_SPI):
+            self.spi.max_speed_hz = self.SPEED
+            self.spi.mode = 0b11  
+        self.digital_write(self.DC_PIN,False)
+        return 0
+
+    def module_exit(self):
+        if(self.Device == Device_SPI):
+            self.spi.close()
+        else :
+            self.bus.close()
+        self.digital_write(self.RST_PIN,False)
+        self.digital_write(self.DC_PIN,False)
+
+### END OF FILE ###

example/waveshare_OLED/__pycache__/OLED_0in49.cpython-311.pyc

@@ -0,0 +1,10 @@
+import sys
+from setuptools import setup
+setup(
+    name='waveshare-OLED',
+    description='Waveshare OLED Display',
+    author='Waveshare',
+    package_dir={'': 'lib'},
+    packages=['waveshare_OLED'],
+)
+

example/waveshare_OLED/__pycache__/__init__.cpython-311.pyc

@@ -0,0 +1,2 @@
+#!/bin/bash
+docker build -t oled .

example/waveshare_OLED/__pycache__/config.cpython-311.pyc

@@ -0,0 +1,7 @@
+smbus
+spidev
+gpiozero
+numpy
+pillow
+lgpio
+RPi.GPIO