From patchwork Fri May 27 14:44:44 2022 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Tomi Valkeinen X-Patchwork-Id: 16098 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 1A3A2C3271 for ; Fri, 27 May 2022 14:45:41 +0000 (UTC) Received: from lancelot.ideasonboard.com (localhost [IPv6:::1]) by lancelot.ideasonboard.com (Postfix) with ESMTP id 863A7656A5; Fri, 27 May 2022 16:45:40 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=libcamera.org; s=mail; t=1653662740; bh=QUnYvFxz4Ygo344WFDS1YECj2E5+A8bUHAIf/hS7j6o=; 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=DQvr4xJ/Vq7TusWU4aiv5cQrjMO2JjIzQtNxWXOXy5jg0iDx2vWZzfG5COSs3Y+g0 u2qFwkIef5SjT5WWaJ8S/2Ieo5xbMB9O8e25U+vQnHSFQy8iMO+1agK8xw8pRYUlRJ 6Zey2XS2YxlHb0xu7v8QoSvhMYxCYZJQpp4+28bGWCl76/0YRnkVKtquDYBtj0mS0q f2NIrCaKAgxftcS+yMLFklUWWP48afiAK/yBhk5e6TOzktUKyTnjo9CBTFgFxmeRvx U7GQ6pLWArMsFtmt/F5VcNFsNXLOe4II+3KsklZvrAdwM2BgZptfcUfvGbtO46u0fH TiQf4MBdMOruQ== Received: from perceval.ideasonboard.com (perceval.ideasonboard.com [213.167.242.64]) by lancelot.ideasonboard.com (Postfix) with ESMTPS id 79C8B65648 for ; Fri, 27 May 2022 16:45:18 +0200 (CEST) Authentication-Results: lancelot.ideasonboard.com; dkim=pass (1024-bit key; unprotected) header.d=ideasonboard.com header.i=@ideasonboard.com header.b="TEtV9mo/"; 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 DF5C74D18; Fri, 27 May 2022 16:45:17 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=ideasonboard.com; s=mail; t=1653662718; bh=QUnYvFxz4Ygo344WFDS1YECj2E5+A8bUHAIf/hS7j6o=; h=From:To:Cc:Subject:Date:In-Reply-To:References:From; b=TEtV9mo/tW0vAXvy0raevSM68gfHE57m0DpWtz6JfsN1edSuzeAXXjFCUrlHmqUyb vlxsB4or4YvK/a2JeSgAGmrFz4MVBX2420m3FKO/p74k7zDTUqbS1rCXW+Xm3XBv8R ymb8Ptz+c/zQggzMRPEt/eGznNS8e1Soy4+OeOH0= To: libcamera-devel@lists.libcamera.org, David Plowman , Kieran Bingham , Laurent Pinchart , Jacopo Mondi Date: Fri, 27 May 2022 17:44:44 +0300 Message-Id: <20220527144447.94891-28-tomi.valkeinen@ideasonboard.com> X-Mailer: git-send-email 2.34.1 In-Reply-To: <20220527144447.94891-1-tomi.valkeinen@ideasonboard.com> References: <20220527144447.94891-1-tomi.valkeinen@ideasonboard.com> MIME-Version: 1.0 Subject: [libcamera-devel] [PATCH v3 27/30] py: examples: Add simple-capture.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 an example to showcase the more-or-less minimal capture case. Signed-off-by: Tomi Valkeinen --- src/py/examples/simple-capture.py | 159 ++++++++++++++++++++++++++++++ 1 file changed, 159 insertions(+) create mode 100755 src/py/examples/simple-capture.py diff --git a/src/py/examples/simple-capture.py b/src/py/examples/simple-capture.py new file mode 100755 index 00000000..4b0c3f0c --- /dev/null +++ b/src/py/examples/simple-capture.py @@ -0,0 +1,159 @@ +#!/usr/bin/env python3 + +# SPDX-License-Identifier: BSD-3-Clause +# Copyright (C) 2022, Tomi Valkeinen + +# A simple minimal capture example showing: +# - How to setup the camera +# - Capture frames in a blocking manner +# - Memory map the frames +# - How to stop the camera +# +# This simple example is, in many ways, too simple. The purpose of the example +# is to introduce the concepts. A more realistic example is given in +# simple-continuous-capture.py. + +import argparse +import binascii +import libcamera as libcam +import libcamera.utils +import sys + + +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 {num_bufs} frames with {stream_config}') + + # 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 + + # Queue the requests to the camera + + for req in reqs: + ret = cam.queue_request(req) + assert ret == 0 + + # Wait until the requests are finished + + reqs = [] + + while True: + # cm.read_event() blocks until there is an event + cm.read_event() + + # Get all the ready requests + ready_reqs = cm.get_ready_requests() + + reqs += ready_reqs + + if len(reqs) >= num_bufs: + break + + # Process the captured frames + + for i, req in enumerate(reqs): + assert i == req.cookie + + buffers = req.buffers + + # A ready Request could contain multiple buffers if multiple streams + # were being used. Here we know we only have a single stream, + # and we use next(iter()) to get the first and only buffer. + + assert len(buffers) == 1 + + stream, fb = next(iter(buffers.items())) + + # Use MappedFrameBuffer to access the pixel data with CPU. We calculate + # the crc for each plane. + + with libcamera.utils.MappedFrameBuffer(fb) as mfb: + crcs = [binascii.crc32(p) for p in mfb.planes] + + meta = fb.metadata + + print('seq {}, bytes {}, CRCs {}' + .format(meta.sequence, + '/'.join([str(p.bytes_used) for p in meta.planes]), + crcs)) + + # 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())