Source code for platypush.plugins.camera.ir.mlx90640

import os
import subprocess
from typing import Optional, Tuple

from platypush.plugins import action
from platypush.plugins.camera import CameraPlugin, Camera


[docs] class CameraIrMlx90640Plugin(CameraPlugin): """ Plugin to interact with a `ML90640 <https://shop.pimoroni.com/products/mlx90640-thermal-camera-breakout>`_ infrared thermal camera. In order to use this plugin you'll need to download and compile the `mlx90640 <https://github.com/pimoroni/mlx90640-library>`_ C++ bindings and examples for the device. Instructions on Raspbian:: # Install the dependencies $ [sudo] apt-get install libi2c-dev $ cd $PLATYPUSH_SRC_DIR $ git submodule init $ git submodule update $ cd platypush/plugins/camera/ir/mlx90640/lib $ make clean $ make bcm2835 $ make examples/rawrgb I2C_MODE=LINUX """
[docs] def __init__( self, rawrgb_path: Optional[str] = None, resolution: Tuple[int, int] = (32, 24), warmup_frames: Optional[int] = 5, **kwargs ): """ :param rawrgb_path: Specify it if the rawrgb executable compiled from https://github.com/pimoroni/mlx90640-library is in another folder than `<directory of this file>/lib/examples`. :param resolution: Device resolution (default: 32x24). :param warmup_frames: Number of frames to be skipped on sensor initialization/warmup (default: 2). :param kwargs: Extra parameters to be passed to :class:`platypush.plugins.camera.CameraPlugin`. """ super().__init__( device='mlx90640', resolution=resolution, warmup_frames=warmup_frames, **kwargs ) if not rawrgb_path: rawrgb_path = os.path.join( os.path.dirname(os.path.abspath(__file__)), 'lib', 'examples', 'rawrgb' ) rawrgb_path = os.path.abspath(os.path.expanduser(rawrgb_path)) assert os.path.isfile( rawrgb_path ), 'rawrgb executable not found. Please follow the documentation of this plugin to build it' self.rawrgb_path = rawrgb_path self._capture_proc = None
def _is_capture_running(self): return self._capture_proc is not None and self._capture_proc.poll() is None
[docs] def prepare_device(self, device: Camera, **_): if not self._is_capture_running(): self._capture_proc = subprocess.Popen( [self.rawrgb_path, '{}'.format(device.info.fps)], stdin=subprocess.PIPE, stdout=subprocess.PIPE, ) return self._capture_proc
[docs] def release_device(self, device: Camera): if not self._is_capture_running(): return self._capture_proc.terminate() self._capture_proc.kill() self._capture_proc.wait() self._capture_proc = None
[docs] def capture_frame(self, device: Camera, *args, **kwargs): from PIL import Image camera = self.prepare_device(device) frame = camera.stdout.read( device.info.resolution[0] * device.info.resolution[1] * 3 ) return Image.frombytes('RGB', device.info.resolution, frame)
[docs] def to_grayscale(self, image): from PIL import Image new_image = Image.new('L', image.size) for i in range(0, image.size[0]): for j in range(0, image.size[1]): r, g, b = image.getpixel((i, j)) value = int(2.0 * r - 1.125 * g - 1.75 * b) if value > 255: value = 255 if value < 0: value = 0 new_image.putpixel((i, j), value) return new_image
[docs] @action def capture(self, output_file=None, *args, **kwargs): """ Back-compatibility alias for :meth:`.capture_image`. """ return self.capture_image(image_file=output_file, *args, **kwargs)
# vim:sw=4:ts=4:et: