1  # SPDX-FileCopyrightText: 2019 Kattni Rembor, written for Adafruit Industries
  2  #
  3  # SPDX-License-Identifier: Unlicense
  4  """
  5  Circuit Python BLE Color Patchwork
  6  This demo uses advertising to broadcast a color of the users choice.
  7  We will show a color "patch" on the screen for every unique device
  8  advertisement that we find.
  9  """
 10  
 11  import time
 12  import random
 13  import board
 14  import displayio
 15  from adafruit_display_text import label
 16  import terminalio
 17  from adafruit_ble import BLERadio
 18  from adafruit_ble.advertising.adafruit import AdafruitColor
 19  from adafruit_display_shapes.rect import Rect
 20  from adafruit_clue import clue
 21  
 22  MODE_COLOR_SELECT = 0
 23  MODE_SHOW_PATCHWORK = 1
 24  
 25  COLOR_TRANSPARENT_INDEX = 0
 26  COLOR_OFFWHITE_INDEX = 1
 27  
 28  PROXIMITY_LIMIT = 100
 29  
 30  current_mode = MODE_SHOW_PATCHWORK
 31  
 32  # The color pickers will cycle through this list with buttons A and B.
 33  color_options = [
 34      0xEE0000,
 35      0xEEEE00,
 36      0x00EE00,
 37      0x00EEEE,
 38      0x0000EE,
 39      0xEE00EE,
 40      0xCCCCCC,
 41      0xFF9999,
 42      0x99FF99,
 43      0x9999FF,
 44  ]
 45  
 46  
 47  # make all pixels in the patchwork bitmap transparent
 48  def make_transparent():
 49      palette_mapping.make_transparent(0)
 50      for i in range(0, 8):
 51          for j in range(0, 8):
 52              bitmap[i, j] = 0
 53  
 54  
 55  # make the pixels in the patchwork white by setting them in the palette
 56  def make_white():
 57      for i in range(2, 66):
 58          palette_mapping[i] = 0xFFFFFF
 59  
 60  
 61  # draw the patchwork grid based on nearby_colors
 62  def draw_grid():
 63      for i, color in enumerate(nearby_colors):
 64          if i < 64:
 65              palette_mapping[i + 2] = (
 66                  color & 0xFFFFFF
 67              )  # Mask 0xFFFFFF to avoid invalid color.
 68              print(i)
 69              print(color)
 70  
 71  
 72  # create a fake mac address and color for testing
 73  def add_fake():
 74      fake_mac = "".join([random.choice("0123456789abcdef") for _ in range(10)])
 75      fake_color = random.choice(color_options)
 76      nearby_addresses.append(fake_mac)
 77      nearby_colors.append(fake_color)
 78  
 79  
 80  # find nearby devices advertising colors
 81  def ble_scan():
 82      print("scanning")
 83      # loop over all found devices
 84      for entry in ble.start_scan(AdafruitColor, minimum_rssi=-100, timeout=1):
 85          # if this device is not in the list already
 86          if entry.color in color_options:
 87              print("new color")
 88          if entry.address.address_bytes not in nearby_addresses:
 89              # print(entry.color)
 90              # add the address and color to respective lists
 91              nearby_addresses.append(entry.address.address_bytes)
 92              nearby_colors.append(entry.color)
 93          else:  # address was already in the list
 94              # update the color to currently advertised value
 95              _index = nearby_addresses.index(entry.address.address_bytes)
 96              nearby_colors[_index] = entry.color
 97  
 98  
 99  # set a new color to be advertised
100  def change_advertisement(color):
101      ble.stop_advertising()
102      advertisement.color = color
103      ble.start_advertising(advertisement)
104      # set NeoPixel to selected advertised
105      clue.pixel.fill(color)
106      # update top left self patch
107      nearby_colors[0] = color
108  
109  
110  # BLE Setup
111  ble = BLERadio()
112  advertisement = AdafruitColor()
113  advertisement.color = color_options[0]
114  
115  # init neopixel
116  clue.pixel.fill(color_options[0])
117  clue.pixel.brightness = 0.05
118  
119  display = board.DISPLAY
120  
121  # Create a bitmap with two colors + 64 colors for the map
122  bitmap = displayio.Bitmap(8, 8, 64 + 2)
123  
124  # Create a 8*8 bitmap pre-filled with 64 colors (color 0 and 1 are reserved)
125  for _i in range(0, 8):
126      for _j in range(0, 8):
127          bitmap[_i, _j] = 2 + _i + _j * 8
128  
129  # Create an empty palette that will be used in one to one mapping
130  palette_mapping = displayio.Palette(64 + 2)
131  
132  palette_mapping[0] = 0x000000
133  palette_mapping[1] = 0xFFFFFF
134  
135  color_select_palette = displayio.Palette(len(color_options))
136  for _i, _color in enumerate(color_options):
137      color_select_palette[_i] = _color
138  
139  # Color Select Layout
140  color_select_group = displayio.Group(max_size=10)
141  color_select_text_group = displayio.Group(max_size=4, scale=3)
142  
143  # white backrground
144  background = Rect(0, 0, 240, 240, fill=0xFFFFFF)
145  center_line = Rect(119, 0, 2, 180, fill=0x000000)
146  
147  # box around the color preview
148  bottom_box = Rect(79, 174, 80, 80, fill=0xFFFFFF, outline=0x000000, stroke=2)
149  
150  left_btn_lbl = label.Label(terminalio.FONT, text="A", color=0x000000)
151  right_btn_lbl = label.Label(terminalio.FONT, text="B", color=0x000000)
152  
153  left_btn_action = label.Label(
154      terminalio.FONT, text="Next\nColor", color=0x000000, line_spacing=1
155  )
156  right_btn_action = label.Label(terminalio.FONT, text="Save", color=0x000000)
157  
158  color_select_text_group.append(left_btn_lbl)
159  color_select_text_group.append(right_btn_lbl)
160  
161  color_select_text_group.append(left_btn_action)
162  color_select_text_group.append(right_btn_action)
163  
164  # x position centered on 25% of screen
165  left_btn_lbl.anchor_point = (0.5, 0)
166  left_btn_lbl.anchored_position = ((240 / 4) // 3, 21 // 3)
167  
168  # x position centered on 75% of screen
169  right_btn_lbl.anchor_point = (0.5, 0)
170  right_btn_lbl.anchored_position = ((240 / 4), 21 // 3)
171  
172  # x position centered on 25% of screen
173  left_btn_action.anchor_point = (0.5, 0)
174  left_btn_action.anchored_position = ((240 / 4) // 3, 96 // 3)
175  
176  # x position centered on 75% of screen
177  right_btn_action.anchor_point = (0.5, 0)
178  right_btn_action.anchored_position = ((240 / 4), 96 // 3)
179  
180  color_select_group.append(background)
181  color_select_group.append(center_line)
182  color_select_group.append(bottom_box)
183  color_select_group.append(color_select_text_group)
184  
185  # color preview bmp
186  color_select_preview_bmp = displayio.Bitmap(1, 1, len(color_options))
187  color_preview_group = displayio.Group(scale=30 * 2)
188  
189  # centered horizontally near bottom on screen
190  color_preview_group.x = 240 // 2 - 60 // 2
191  color_preview_group.y = 240 - (60 + 2)
192  
193  color_preview_tilegrid = displayio.TileGrid(
194      color_select_preview_bmp, pixel_shader=color_select_palette
195  )
196  color_preview_group.append(color_preview_tilegrid)
197  
198  # Create a TileGrid using the Bitmap and Palette
199  tile_grid = displayio.TileGrid(bitmap, pixel_shader=palette_mapping)
200  patchwork_group = displayio.Group(scale=30)
201  patchwork_group.append(tile_grid)
202  
203  # Create main Group
204  group = displayio.Group()
205  
206  # Add the patchwork to the main
207  group.append(patchwork_group)
208  
209  # Add the Group to the Display
210  display.show(group)
211  
212  cur_color = 0
213  
214  prev_b = clue.button_b
215  prev_a = clue.button_a
216  
217  nearby_addresses = ["myself"]
218  nearby_colors = [color_options[cur_color]]
219  
220  make_white()
221  
222  last_scan_time = -30
223  SCAN_INTERVAL = 30  # seconds
224  
225  while True:
226      now = time.monotonic()
227      cur_a = clue.button_a
228      cur_b = clue.button_b
229      if current_mode == MODE_SHOW_PATCHWORK:
230          # a button was pressed
231          if cur_a and not prev_a:
232              current_mode = MODE_COLOR_SELECT
233  
234              # insert color select layout
235              group.append(color_select_group)
236              group.append(color_preview_group)
237  
238          # is it time to scan?
239          if last_scan_time + SCAN_INTERVAL < now:
240              ble_scan()
241              last_scan_time = now
242              print("after scan found {} results".format(len(nearby_colors)))
243              # print(nearby_addresses)
244              draw_grid()
245  
246          if clue.proximity >= PROXIMITY_LIMIT:
247              clue.white_leds = True
248              while clue.proximity >= PROXIMITY_LIMIT:
249                  r, g, b, w = clue.color
250                  clue.pixel.fill(((r >> 8) & 0xFF, (g >> 8) & 0xFF, (b >> 8) & 0xFF))
251                  change_advertisement(
252                      ((r & 0xFF00) << 8) + (g & 0xFF00) + ((b >> 8) & 0xFF)
253                  )
254                  time.sleep(0.1)
255              clue.white_leds = False
256  
257      elif current_mode == MODE_COLOR_SELECT:
258          # current selection preview
259          color_select_preview_bmp[0, 0] = cur_color
260  
261          # a button was pressed
262          if cur_a and not prev_a:
263              print("a button")
264              # increment currently selected color index
265              cur_color += 1
266              # reset to 0 if it's too big
267              if cur_color >= len(color_options):
268                  cur_color = 0
269              print(cur_color)
270          # b button was pressed
271          if cur_b and not prev_b:
272              print("b button")
273              # advertise new color selection
274              change_advertisement(color_options[cur_color])
275  
276              # go back to patchwork mode
277              current_mode = MODE_SHOW_PATCHWORK
278              # remove color select background
279              group.remove(color_select_group)
280              group.remove(color_preview_group)
281              make_white()
282              draw_grid()
283      prev_a = cur_a
284      prev_b = cur_b