From patchwork Mon May 30 14:27:21 2022 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Tomi Valkeinen X-Patchwork-Id: 16122 Return-Path: X-Original-To: parsemail@patchwork.libcamera.org Delivered-To: parsemail@patchwork.libcamera.org Received: from lancelot.ideasonboard.com (lancelot.ideasonboard.com [92.243.16.209]) by patchwork.libcamera.org (Postfix) with ESMTPS id 28816C3279 for ; Mon, 30 May 2022 14:27:56 +0000 (UTC) Received: from lancelot.ideasonboard.com (localhost [IPv6:::1]) by lancelot.ideasonboard.com (Postfix) with ESMTP id 885B165699; Mon, 30 May 2022 16:27:55 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=libcamera.org; s=mail; t=1653920875; bh=Dqbd2jWSxJoWb7CMOMAO8z/GpiNtZdD+FqW7WPAxi3c=; h=To:Date:In-Reply-To:References:Subject:List-Id:List-Unsubscribe: List-Archive:List-Post:List-Help:List-Subscribe:From:Reply-To: From; b=z6wK3r1yOvuxvFG+OD61SWygsGLvERHnTa57flEUnlRLzlRA5O6bx/wGCof29i2jH maYBRMpTftLnD0zS0K5g7gvLDMePPLv9cmLdepaPSdSa47juFTcWL6t7cZ83tEwMnd DGXLGX402+nW1NBZawUsQ8YunrRLRzjbtJIbAM+W9A9RdzKkkC7atcFH3tLoJLIleI ZGwRQnFO5gZe0JEZ/Sau/KZyseA+FYyHKpSnQ5oAtNCDh4hmA/e2gB/HQYPYE/yxPf hOYmK4hJVMjXxkD92snqhjNN0U4Tek5EXQs5gjSYhN6pyKqKy92kEaoz0T/GiTPa4d SNBE/Ia33M7vA== Received: from perceval.ideasonboard.com (perceval.ideasonboard.com [IPv6:2001:4b98:dc2:55:216:3eff:fef7:d647]) by lancelot.ideasonboard.com (Postfix) with ESMTPS id EEC6A6563B for ; Mon, 30 May 2022 16:27:46 +0200 (CEST) Authentication-Results: lancelot.ideasonboard.com; dkim=pass (1024-bit key; unprotected) header.d=ideasonboard.com header.i=@ideasonboard.com header.b="hV2abj3x"; dkim-atps=neutral Received: from deskari.lan (91-156-85-209.elisa-laajakaista.fi [91.156.85.209]) by perceval.ideasonboard.com (Postfix) with ESMTPSA id 604581918; Mon, 30 May 2022 16:27:46 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=ideasonboard.com; s=mail; t=1653920866; bh=Dqbd2jWSxJoWb7CMOMAO8z/GpiNtZdD+FqW7WPAxi3c=; h=From:To:Cc:Subject:Date:In-Reply-To:References:From; b=hV2abj3xlK2uHgBShu00YOf06XVwNySpnMh63NQg7Ya73HPm+i75QsNy75s8S0UHY umyM9C3BvEZpQDkCAi/zpwqzx+EHpKUuMsVKFXZeKZ4cZWaEetllAjIaSF5r6th4+R bxudVSPvaR20DP02aNDx1oA6sjhsDn7QYg/6nbSo= To: libcamera-devel@lists.libcamera.org, David Plowman , Kieran Bingham , Laurent Pinchart , Jacopo Mondi Date: Mon, 30 May 2022 17:27:21 +0300 Message-Id: <20220530142722.57618-16-tomi.valkeinen@ideasonboard.com> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20220530142722.57618-1-tomi.valkeinen@ideasonboard.com> References: <20220530142722.57618-1-tomi.valkeinen@ideasonboard.com> MIME-Version: 1.0 Subject: [libcamera-devel] [PATCH v4 15/16] py: examples: Add itest.py X-BeenThere: libcamera-devel@lists.libcamera.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-Patchwork-Original-From: Tomi Valkeinen via libcamera-devel From: Tomi Valkeinen Reply-To: Tomi Valkeinen Errors-To: libcamera-devel-bounces@lists.libcamera.org Sender: "libcamera-devel" Add a small tool which gives the user an IPython based interactive shell with libcamera capturing in the background. Can be used to study and test small things with libcamera. Signed-off-by: Tomi Valkeinen --- src/py/examples/itest.py | 195 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 195 insertions(+) create mode 100755 src/py/examples/itest.py diff --git a/src/py/examples/itest.py b/src/py/examples/itest.py new file mode 100755 index 00000000..9c97649d --- /dev/null +++ b/src/py/examples/itest.py @@ -0,0 +1,195 @@ +#!/usr/bin/env python3 + +# SPDX-License-Identifier: BSD-3-Clause +# Copyright (C) 2022, Tomi Valkeinen + +# This sets up a single camera, starts a capture loop in a background thread, +# and starts an IPython shell which can be used to study and test libcamera. +# +# Note: This is not an example as such, but more of a developer tool to try +# out things. + +import argparse +import IPython +import libcamera as libcam +import selectors +import sys +import threading + + +def handle_camera_event(ctx): + cm = ctx['cm'] + cam = ctx['cam'] + + # cm.get_ready_requests() will not block here, as we know there is an event + # to read. + + reqs = cm.get_ready_requests() + + assert len(reqs) > 0 + + # Process the captured frames + + for req in reqs: + buffers = req.buffers + + assert len(buffers) == 1 + + # We want to re-queue the buffer we just handled. Instead of creating + # a new Request, we re-use the old one. We need to call req.reuse() + # to re-initialize the Request before queuing. + + req.reuse() + cam.queue_request(req) + + scope = ctx['scope'] + + if 'num_frames' not in scope: + scope['num_frames'] = 0 + + scope['num_frames'] += 1 + + +def capture(ctx): + cm = ctx['cm'] + cam = ctx['cam'] + reqs = ctx['reqs'] + + # Queue the requests to the camera + + for req in reqs: + ret = cam.queue_request(req) + assert ret == 0 + + # Use Selector to wait for events from the camera and from the keyboard + + sel = selectors.DefaultSelector() + sel.register(cm.event_fd, selectors.EVENT_READ, lambda fd: handle_camera_event(ctx)) + + reqs = [] + + while ctx['running']: + events = sel.select() + for key, mask in events: + key.data(key.fileobj) + + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument('-c', '--camera', type=str, default='1', + help='Camera index number (starting from 1) or part of the name') + parser.add_argument('-f', '--format', type=str, help='Pixel format') + parser.add_argument('-s', '--size', type=str, help='Size ("WxH")') + args = parser.parse_args() + + cm = libcam.CameraManager.singleton() + + try: + if args.camera.isnumeric(): + cam_idx = int(args.camera) + cam = next((cam for i, cam in enumerate(cm.cameras) if i + 1 == cam_idx)) + else: + cam = next((cam for cam in cm.cameras if args.camera in cam.id)) + except Exception: + print(f'Failed to find camera "{args.camera}"') + return -1 + + # Acquire the camera for our use + + ret = cam.acquire() + assert ret == 0 + + # Configure the camera + + cam_config = cam.generate_configuration([libcam.StreamRole.Viewfinder]) + + stream_config = cam_config.at(0) + + if args.format: + fmt = libcam.PixelFormat(args.format) + stream_config.pixel_format = fmt + + if args.size: + w, h = [int(v) for v in args.size.split('x')] + stream_config.size = libcam.Size(w, h) + + ret = cam.configure(cam_config) + assert ret == 0 + + stream = stream_config.stream + + # Allocate the buffers for capture + + allocator = libcam.FrameBufferAllocator(cam) + ret = allocator.allocate(stream) + assert ret > 0 + + num_bufs = len(allocator.buffers(stream)) + + print(f'Capturing {stream_config} with {num_bufs} buffers from {cam.id}') + + # Create the requests and assign a buffer for each request + + reqs = [] + for i in range(num_bufs): + # Use the buffer index as the "cookie" + req = cam.create_request(i) + + buffer = allocator.buffers(stream)[i] + ret = req.add_buffer(stream, buffer) + assert ret == 0 + + reqs.append(req) + + # Start the camera + + ret = cam.start() + assert ret == 0 + + # Create a scope for the IPython shell + scope = { + 'cm': cm, + 'cam': cam, + 'libcam': libcam, + } + + # Create a simple context shared between the background thread and the main + # thread. + ctx = { + 'running': True, + 'cm': cm, + 'cam': cam, + 'reqs': reqs, + 'scope': scope, + } + + # Note that "In CPython, due to the Global Interpreter Lock, only one thread + # can execute Python code at once". We rely on that here, which is not very + # nice. I am sure an fd-based polling loop could be integrated with IPython, + # somehow, which would allow us to drop the threading. + + t = threading.Thread(target=capture, args=(ctx,)) + t.start() + + IPython.embed(banner1='', exit_msg='', confirm_exit=False, user_ns=scope) + + print('Exiting...') + + ctx['running'] = False + t.join() + + # Stop the camera + + ret = cam.stop() + assert ret == 0 + + # Release the camera + + ret = cam.release() + assert ret == 0 + + return 0 + + +if __name__ == '__main__': + sys.exit(main())