From 2daf4274ecba6e096bcd41d1c14c85cde0fe70f4 Mon Sep 17 00:00:00 2001
From: alexanderchernenko <alexanderchernenko@gmail.com>
Date: Mon, 6 Apr 2020 00:10:44 +0200
Subject: [PATCH] console version of heat map added

---
 .gitignore                                  |   3 +-
 examples/amg88xx_rpi_thermal_cam_console.py | 112 ++++++++++++++++++++
 2 files changed, 114 insertions(+), 1 deletion(-)
 create mode 100644 examples/amg88xx_rpi_thermal_cam_console.py

diff --git a/.gitignore b/.gitignore
index c83f8b7..a278099 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,4 +8,5 @@ bundles
 *.DS_Store
 .eggs
 dist
-**/*.egg-info
\ No newline at end of file
+**/*.egg-info
+/build
diff --git a/examples/amg88xx_rpi_thermal_cam_console.py b/examples/amg88xx_rpi_thermal_cam_console.py
new file mode 100644
index 0000000..8a89dac
--- /dev/null
+++ b/examples/amg88xx_rpi_thermal_cam_console.py
@@ -0,0 +1,112 @@
+"""This example is for Raspberry Pi (Linux) only!
+   It will not work on microcontrollers running CircuitPython!"""
+
+import os
+import sys
+import math
+import time
+import atexit
+import busio
+import board
+
+import numpy as np
+from scipy.interpolate import griddata
+
+from colour import Color
+from os import system, name
+import adafruit_amg88xx
+
+i2c_bus = busio.I2C(board.SCL, board.SDA)
+
+# low range of the sensor (this will be blue on the screen)
+MINTEMP = 26.0
+
+# high range of the sensor (this will be red on the screen)
+MAXTEMP = 32.0
+
+# how many color values we can have
+COLORDEPTH = 1024
+
+#os.putenv("SDL_FBDEV", "/dev/fb1")
+#pygame.init()
+
+# initialize the sensor
+sensor = adafruit_amg88xx.AMG88XX(i2c_bus)
+
+# pylint: disable=invalid-slice-index
+points = [(math.floor(ix / 8), (ix % 8)) for ix in range(0, 64)]
+grid_x, grid_y = np.mgrid[0:7:32j, 0:7:32j]
+# pylint: enable=invalid-slice-index
+
+# sensor is an 8x8 grid so lets do a square
+height = 240
+width = 240
+
+# the list of colors we can choose from
+blue = Color("indigo")
+colors = list(blue.range_to(Color("red"), COLORDEPTH))
+
+# create the array of colors
+colors = [(int(c.red * 255), int(c.green * 255), int(c.blue * 255)) for c in colors]
+console_colors = [17,18,19,20,21,57,93,129,165,201,200,199,198,197,196,202,208,214,220]
+
+displayPixelWidth = width / 30
+displayPixelHeight = height / 30
+
+# some utility functions
+def constrain(val, min_val, max_val):
+    return min(max_val, max(min_val, val))
+
+
+def map_value(x, in_min, in_max, out_min, out_max):
+    return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min
+
+def clear(): 
+    _ = system('clear') 
+
+def print_there(x, y, text, color):
+	sys.stdout.write("\x1b7\x1b[48;5;%dm" % (color))
+	sys.stdout.write("\x1b7\x1b[%d;%df%s\x1b8" % (x, y, text))
+maxpixel = 0
+minpixel = 0
+color_range = 1
+def exit_handler():
+    print("maxpixel:" + str(maxpixel))
+    print("minpixel:" + str(minpixel))
+
+atexit.register(exit_handler)
+# let the sensor initialize
+time.sleep(0.1)
+
+while True:
+
+    # read the pixels
+    pixels = []
+    for row in sensor.pixels:
+        pixels = pixels + row
+    pixels = [map_value(p, MINTEMP, MAXTEMP, 0, COLORDEPTH - 1) for p in pixels]
+
+    # perform interpolation
+    bicubic = griddata(points, pixels, (grid_x, grid_y), method="cubic")
+
+    # draw everything
+    y=2
+    for ix, row in enumerate(bicubic):
+        x=2
+        for jx, pixel in enumerate(row):
+            color_index = 0
+            if (color_range != 0):
+                color_index = int(round((pixel-minpixel)/color_range))
+            if color_index < 0:
+                color_index = 0
+            if color_index > len(console_colors)-1:
+                color_index = len(console_colors)-1
+            print_there(x, y*2-2, '  ', console_colors[color_index])
+            if pixel > maxpixel:
+                maxpixel = pixel
+            if pixel < minpixel:
+                minpixel = pixel
+            x+=1
+        y+=1
+    heat_range = maxpixel - minpixel
+    color_range = heat_range / len(console_colors)