-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathPiicoDev_SSD1306.py
355 lines (314 loc) · 12.5 KB
/
PiicoDev_SSD1306.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
# Sends data to the Core Electronics PiicoDev OLED SSD1306 Display
# Ported by Peter Johnston at Core Electronics October 2021
# Original Repos:
# 2021-09-12:
# https://github.com/micropython/micropython/blob/master/drivers/display/ssd1306.py
# 2021-08-01:
# https://github.com/fizban99/microbit_ssd1306/blob/master/ssd1306_text.py
# 2021-07-15:
# https://github.com/adafruit/Adafruit_Python_SSD1306/blob/master/Adafruit_SSD1306/SSD1306.py
# 2021 OCT 14 - Initial release
# 2022 JAN 04 - Remove dependency on PIL module. Improve compatibility with pbm files.
_SET_CONTRAST = 0x81
_SET_ENTIRE_ON = 0xA4
_SET_NORM_INV = 0xA6
_SET_DISP = 0xAE
_SET_MEM_ADDR = 0x20
_SET_COL_ADDR = 0x21
_SET_PAGE_ADDR = 0x22
_SET_DISP_START_LINE = 0x40
_SET_SEG_REMAP = 0xA0
_SET_MUX_RATIO = 0xA8
_SET_IREF_SELECT = 0xAD
_SET_COM_OUT_DIR = 0xC0
_SET_DISP_OFFSET = 0xD3
_SET_COM_PIN_CFG = 0xDA
_SET_DISP_CLK_DIV = 0xD5
_SET_PRECHARGE = 0xD9
_SET_VCOM_DESEL = 0xDB
_SET_CHARGE_PUMP = 0x8D
WIDTH = 128
HEIGHT = 64
from PiicoDev_Unified import *
from math import cos,sin,radians
compat_str = '\nUnified PiicoDev library out of date. Get the latest module: https://piico.dev/unified \n'
_SYSNAME = os.uname().sysname
if _SYSNAME == 'microbit':
from microbit import *
from utime import sleep_ms
from ustruct import pack_into
elif _SYSNAME == 'Linux':
from struct import pack_into
else:
import framebuf
if _SYSNAME == 'microbit' or _SYSNAME == 'Linux':
class framebuf:
class FrameBuffer():
#Framebuffer manipulation, used by Microbit and Linux
def _set_pos(self, col=0, page=0):
self.write_cmd(0xb0 | page) # page number
# take upper and lower value of col * 2
c1, c2 = col * 2 & 0x0F, col >> 3
self.write_cmd(0x00 | c1) # lower start column address
self.write_cmd(0x10 | c2) # upper start column address
def fill(self, c=0):
for i in range(0, 1024):
if c > 0:
self.buffer[i] = 0xFF
else:
self.buffer[i] = 0x00
def pixel(self, x, y, color):
x = x & (WIDTH - 1)
y = y & (HEIGHT - 1)
page, shift_page = divmod(y, 8)
ind = x + page * 128
b = self.buffer[ind] | (1 << shift_page) if color else self.buffer[ind] & ~ (1 << shift_page)
pack_into(">B", self.buffer, ind, b)
self._set_pos(x, page)
def line(self, x1, y1, x2, y2, c):
# bresenham
steep = abs(y2-y1) > abs(x2-x1)
if steep:
# Swap x/y
tmp = x1
x1 = y1
y1 = tmp
tmp = y2
y2 = x2
x2 = tmp
if x1 > x2:
# Swap start/end
tmp = x1
x1 = x2
x2 = tmp
tmp = y1
y1 = y2
y2 = tmp
dx = x2 - x1;
dy = abs(y2-y1)
err = dx/2
if(y1 < y2):
ystep = 1
else:
ystep = -1
while x1 <= x2:
if steep:
self.pixel(y1, x1, c)
else:
self.pixel(x1, y1, c)
err -= dy
if err < 0:
y1 += ystep
err += dx
x1 += 1
def hline(self, x, y, l, c):
self.line(x, y, x + l, y, c)
def vline(self, x, y, h, c):
self.line(x, y, x, y + h, c)
def rect(self, x, y, w, h, c):
self.hline(x, y, w, c)
self.hline(x, y+h, w, c)
self.vline(x, y, h, c)
self.vline(x+w, y, h, c)
def fill_rect(self, x, y, w, h, c):
for i in range(y, y + h):
self.hline(x, i, w, c)
def text(self, text, x, y, c=1):
fontFile = open("font-pet-me-128.dat", "rb")
font = bytearray(fontFile.read())
for text_index in range(0, len(text)):
ind = 0
for col in range(8):
fontDataPixelValues = font[(ord(text[text_index])-32)*8 + col]
#ind = text_index * 8 + x * 8 + y * 128 + col
for i in range(0,7):
if fontDataPixelValues & 1 << i != 0:
x_coordinate = x + col + text_index * 8
y_coordinate = y+i
if x_coordinate < WIDTH and y_coordinate < HEIGHT:
self.pixel(x_coordinate, y_coordinate, c)
class PiicoDev_SSD1306(framebuf.FrameBuffer):
def init_display(self):
self.width = WIDTH
self.height = HEIGHT
self.pages = HEIGHT // 8
self.buffer = bytearray(self.pages * WIDTH)
for cmd in (
_SET_DISP, # display off
# address setting
_SET_MEM_ADDR,
0x00, # horizontal
# resolution and layout
_SET_DISP_START_LINE, # start at line 0
_SET_SEG_REMAP | 0x01, # column addr 127 mapped to SEG0
_SET_MUX_RATIO,
HEIGHT - 1,
_SET_COM_OUT_DIR | 0x08, # scan from COM[N] to COM0
_SET_DISP_OFFSET,
0x00,
_SET_COM_PIN_CFG,
0x12,
# timing and driving scheme
_SET_DISP_CLK_DIV,
0x80,
_SET_PRECHARGE,
0xF1,
_SET_VCOM_DESEL,
0x30, # 0.83*Vcc
# display
_SET_CONTRAST,
0xFF,# maximum
_SET_ENTIRE_ON, # output follows RAM contents
_SET_NORM_INV, # not inverted
_SET_IREF_SELECT,
0x30, # enable internal IREF during display on
# charge pump
_SET_CHARGE_PUMP,
0x14,
_SET_DISP | 0x01, # display on
): # on
self.write_cmd(cmd)
def poweroff(self):
self.write_cmd(_SET_DISP)
def poweron(self):
self.write_cmd(_SET_DISP | 0x01)
def setContrast(self, contrast):
self.write_cmd(_SET_CONTRAST)
self.write_cmd(contrast)
def invert(self, invert):
self.write_cmd(_SET_NORM_INV | (invert & 1))
def rotate(self, rotate):
self.write_cmd(_SET_COM_OUT_DIR | ((rotate & 1) << 3))
self.write_cmd(_SET_SEG_REMAP | (rotate & 1))
def show(self):
x0 = 0
x1 = WIDTH - 1
self.write_cmd(_SET_COL_ADDR)
self.write_cmd(x0)
self.write_cmd(x1)
self.write_cmd(_SET_PAGE_ADDR)
self.write_cmd(0)
self.write_cmd(self.pages - 1)
self.write_data(self.buffer)
def write_cmd(self, cmd):
try:
self.i2c.writeto_mem(self.addr, int.from_bytes(b'\x80','big'), bytes([cmd]))
self.comms_err = False
except:
print(i2c_err_str.format(self.addr))
self.comms_err = True
def write_data(self, buf):
try:
self.write_list[1] = buf
self.i2c.writeto_mem(self.addr, int.from_bytes(self.write_list[0],'big'), self.write_list[1])
self.comms_err = False
except:
print(i2c_err_str.format(self.addr))
self.comms_err = True
def circ(self,x,y,r,t=1,c=1):
for i in range(x-r,x+r+1):
for j in range(y-r,y+r+1):
if t==1:
if((i-x)**2 + (j-y)**2 < r**2):
self.pixel(i,j,1)
else:
if((i-x)**2 + (j-y)**2 < r**2) and ((i-x)**2 + (j-y)**2 >= (r-r*t-1)**2):
self.pixel(i,j,c)
def arc(self,x,y,r,stAng,enAng,t=0,c=1):
for i in range(r*(1-t)-1,r):
for ta in range(stAng,enAng,1):
X = int(i*cos(radians(ta))+ x)
Y = int(i*sin(radians(ta))+ y)
self.pixel(X,Y,c)
def load_pbm(self, filename, c):
with open(filename, 'rb') as f:
line = f.readline()
if line.startswith(b'P4') is False:
print('Not a valid pbm P4 file')
return
line = f.readline()
while line.startswith(b'#') is True:
line = f.readline()
data_piicodev = bytearray(f.read())
for byte in range(WIDTH // 8 * HEIGHT):
for bit in range(8):
if data_piicodev[byte] & 1 << bit != 0:
x_coordinate = ((8-bit) + (byte * 8)) % WIDTH
y_coordinate = byte * 8 // WIDTH
if x_coordinate < WIDTH and y_coordinate < HEIGHT:
self.pixel(x_coordinate, y_coordinate, c)
class graph2D:
def __init__(self, originX = 0, originY = HEIGHT-1, width = WIDTH, height = HEIGHT, minValue=0, maxValue=255, c = 1, bars = False):
self.minValue = minValue
self.maxValue = maxValue
self.originX = originX
self.originY = originY
self.width = width
self.height = height
self.c = c
self.m = (1-height)/(maxValue-minValue)
self.offset = originY-self.m*minValue
self.bars = bars
self.data = []
def updateGraph2D(self, graph, value):
graph.data.insert(0,value)
if len(graph.data) > graph.width:
graph.data.pop()
x = graph.originX+graph.width-1
m = graph.c
for value in graph.data:
y = round(graph.m*value + graph.offset)
if graph.bars == True:
for idx in range(y, graph.originY+1):
if x >= graph.originX and x < graph.originX+graph.width and idx <= graph.originY and idx > graph.originY-graph.height:
self.pixel(x,idx, m)
else:
if x >= graph.originX and x < graph.originX+graph.width and y <= graph.originY and y > graph.originY-graph.height:
self.pixel(x,y, m)
x -= 1
class PiicoDev_SSD1306_MicroPython(PiicoDev_SSD1306):
def __init__(self, bus=None, freq=None, sda=None, scl=None, addr=0x3C):
self.i2c = create_unified_i2c(bus=bus, freq=freq, sda=sda, scl=scl)
self.addr = addr
self.temp = bytearray(2)
self.write_list = [b'\x40', None] # Co=0, D/C#=1
self.init_display()
super().__init__(self.buffer, WIDTH, HEIGHT, framebuf.MONO_VLSB)
self.fill(0)
self.show()
class PiicoDev_SSD1306_MicroBit(PiicoDev_SSD1306):
def __init__(self, bus=None, freq=None, sda=None, scl=None, addr=0x3C):
self.i2c = create_unified_i2c(bus=bus, freq=freq, sda=sda, scl=scl)
self.addr = addr
self.temp = bytearray(2)
self.write_list = [b'\x40', None] # Co=0, D/C#=1
self.init_display()
self.fill(0)
self.show()
class PiicoDev_SSD1306_Linux(PiicoDev_SSD1306):
def __init__(self, bus=None, freq=None, sda=None, scl=None, addr=0x3C):
self.i2c = create_unified_i2c(bus=bus, freq=freq, sda=sda, scl=scl)
self.addr = addr
self.temp = bytearray(2)
self.write_list = [b'\x40', None] # Co=0, D/C#=1
self.init_display()
self.fill(0)
self.show()
def create_PiicoDev_SSD1306(address=0x3C,bus=None, freq=None, sda=None, scl=None, asw=None):
if asw == 0: _a = 0x3C
elif asw == 1: _a = 0x3D
else: _a = address # parse desired address from direct address input or asw switch position (0 or 1)
try:
if compat_ind >= 1:
pass
else:
print(compat_str)
except:
print(compat_str)
if _SYSNAME == 'microbit':
display = PiicoDev_SSD1306_MicroBit(addr=_a, freq=freq)
elif _SYSNAME == 'Linux':
display = PiicoDev_SSD1306_Linux(addr=_a, freq=freq)
else:
display = PiicoDev_SSD1306_MicroPython(addr=_a, bus=bus, freq=freq, sda=sda, scl=scl)
return display