Merge pull request adafruit#1 from caternuson/master

added amg88xx_still.py to examples

Author: ladyada

examples/amg88xx_still.py

@@ -0,0 +1,84 @@
+#!/usr/bin/python
+# Copyright (c) 2017 Adafruit Industries
+# Author: Carter Nelson
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation 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
+# furnished 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 FOR 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 argparse
+from time import sleep
+from colour import Color
+
+from Adafruit_AMG88xx import Adafruit_AMG88xx
+from PIL import Image, ImageDraw
+
+# parse command line arguments
+parser = argparse.ArgumentParser(description='Take a still image.')
+parser.add_argument('-o','--output', metavar='filename', default='amg88xx_still.jpg', help='specify output filename')
+parser.add_argument('-s','--scale', type=int, default=2, help='specify image scale')
+parser.add_argument('--min', type=float, help='specify minimum temperature')
+parser.add_argument('--max', type=float, help='specify maximum temperature')
+parser.add_argument('--report', action='store_true', default=False, help='show sensor information without saving image')
+
+args = parser.parse_args()
+    
+# sensor setup
+NX = 8
+NY = 8
+sensor = Adafruit_AMG88xx()
+
+# wait for it to boot
+sleep(.1)
+
+# get sensor readings  
+pixels = sensor.readPixels()
+
+if args.report:
+    print "Min Pixel = {0} C".format(min(pixels))
+    print "Max Pixel = {0} C".format(max(pixels))
+    print "Thermistor = {0} C".format(sensor.readThermistor())
+    exit()
+
+# output image buffer
+image = Image.new("RGB", (NX, NY), "white")
+draw = ImageDraw.Draw(image)
+
+# color map
+COLORDEPTH = 256
+colors = list(Color("indigo").range_to(Color("red"), COLORDEPTH))
+colors = [(int(c.red * 255), int(c.green * 255), int(c.blue * 255)) for c in colors]
+
+#some utility functions
+def constrain(val, min_val, max_val):
+    return min(max_val, max(min_val, val))
+
+def map(x, in_min, in_max, out_min, out_max):
+    return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min
+
+# map sensor readings to color map
+MINTEMP = min(pixels) if args.min == None else args.min
+MAXTEMP = max(pixels) if args.max == None else args.max
+pixels = [map(p, MINTEMP, MAXTEMP, 0, COLORDEPTH - 1) for p in pixels]
+
+# create the image
+for ix in xrange(NX):
+    for iy in xrange(NY):
+        draw.point([(ix,iy%NX)], fill=colors[constrain(int(pixels[ix+NX*iy]), 0, COLORDEPTH- 1)])
+
+# scale and save
+image.resize((NX*args.scale, NY*args.scale), Image.BICUBIC).save(args.output)