robotpy-cscore 2027.0.0a3__cp314-cp314-manylinux_2_39_x86_64.whl

cscore/imagewriter.py

@@ -0,0 +1,146 @@
+import os.path
+import time
+import threading
+
+import cv2
+import numpy as np
+
+import logging
+
+logger = logging.getLogger("cscore.storage")
+
+
+class ImageWriter:
+    """
+    Creates a thread that periodically writes images to a specified
+    directory. Useful for looking at images after a match has
+    completed.
+
+    The default location is ``/media/sda1/camera``. The folder
+    ``/media/sda1`` is the default location that USB drives inserted into
+    the RoboRIO are mounted at. The USB drive must have a directory in it
+    named ``camera``.
+
+    .. note:: It is recommended to only write images when something useful
+              (such as targeting) is happening, otherwise you'll end up
+              with a lot of images written to disk that you probably aren't
+              interested in.
+
+    Intended usage is::
+
+        self.image_writer = ImageWriter()
+
+        ..
+
+        while True:
+
+            img = ..
+
+            if self.logging_enabled:
+                self.image_writer.setImage(img)
+
+    """
+
+    def __init__(
+        self,
+        *,
+        location_root="/media/sda1/camera",
+        capture_period=0.5,
+        image_format="jpg"
+    ):
+        """
+        :param location_root: Directory to write images to. A subdirectory
+                              with the current time will be created, and
+                              timestamped images will be written to the
+                              subdirectory.
+        :param capture_period: How often to write images to disk
+        :param image_format: File extension of files to write
+        """
+
+        self.location_root = os.path.abspath(location_root)
+        self.capture_period = capture_period
+        self.image_format = image_format
+
+        self.active = True
+        self._location = None
+        self.has_image = False
+        self.size = None
+
+        self.lock = threading.Condition()
+        self._thread = threading.Thread(target=self._run, daemon=True)
+        self._thread.start()
+
+    def setImage(self, img):
+        """
+        Call this function when you wish to write the image to disk. Not
+        every image is written to disk. Makes a copy of the image.
+
+        :param img: A numpy array representing an OpenCV image
+        """
+
+        if not self.active:
+            return
+
+        if (
+            self.size is None
+            or self.size[0] != img.shape[0]
+            or self.size[1] != img.shape[1]
+        ):
+            h, w = img.shape[:2]
+            self.size = (h, w)
+
+            self.out1 = np.empty((h, w, 3), dtype=np.uint8)
+            self.out2 = np.empty((h, w, 3), dtype=np.uint8)
+
+        with self.lock:
+            cv2.copyMakeBorder(
+                img, 0, 0, 0, 0, cv2.BORDER_CONSTANT, value=(0, 0, 255), dst=self.out1
+            )
+            self.has_image = True
+            self.lock.notify()
+
+    @property
+    def location(self):
+        if self._location is None:
+            # This assures that we only log when a USB memory stick is plugged in
+            if not os.path.exists(self.location_root):
+                raise OSError(
+                    "Logging disabled, %s does not exist" % self.location_root
+                )
+
+            # Can't do this when program starts, time might be wrong. Ideally by now the DS
+            # has connected, so the time will be correct
+            self._location = self.location_root + "/%s" % time.strftime(
+                "%Y-%m-%d %H.%M.%S"
+            )
+            logger.info("Logging to %s", self._location)
+            os.makedirs(self._location, exist_ok=True)
+
+        return self._location
+
+    def _run(self):
+        last = time.time()
+
+        logger.info("Storage thread started")
+
+        try:
+            while True:
+                with self.lock:
+                    now = time.time()
+                    while (not self.has_image) or (now - last) < self.capture_period:
+                        self.lock.wait()
+                        now = time.time()
+
+                    self.out2, self.out1 = self.out1, self.out2
+                    self.has_image = False
+
+                fname = "%s/%.2f.%s" % (self.location, now, self.image_format)
+                cv2.imwrite(fname, self.out2)
+
+                last = now
+
+        except OSError as e:
+            logger.error("Error logging images: %s", e)
+
+        logger.warn("Storage thread exited")
+        self.active = False

cscore/src/main.cpp

@@ -0,0 +1,19 @@
+
+#include <semiwrap_init.cscore._cscore.hpp>
+
+#include "cscore_cpp.h"
+
+SEMIWRAP_PYBIND11_MODULE(m) {
+    initWrapper(m);
+
+    static int unused; // the capsule needs something to reference
+    py::capsule cleanup(&unused, [](void *) {
+        // don't release gil until after calling this
+        cs::SetDefaultLogger(20 /* WPI_LOG_INFO */);
+        
+        // but this MUST release the gil, or deadlock may occur
+        py::gil_scoped_release __release;
+        CS_Shutdown();
+    });
+    m.add_object("_cleanup", cleanup);
+}

cscore/trampolines/cs__AxisCamera.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::AxisCamera does not have a trampoline"

cscore/trampolines/cs__CvSink.hpp

@@ -0,0 +1,13 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_cv.h>
+
+// from extra_includes
+#include <opencv2/core/core.hpp>
+#include <cvnp/cvnp.h>
+#error "cs::CvSink does not have a trampoline"

cscore/trampolines/cs__CvSource.hpp

@@ -0,0 +1,13 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_cv.h>
+
+// from extra_includes
+#include <opencv2/core/core.hpp>
+#include <cvnp/cvnp.h>
+#error "cs::CvSource does not have a trampoline"

cscore/trampolines/cs__HttpCamera.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::HttpCamera does not have a trampoline"

cscore/trampolines/cs__ImageSink.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::ImageSink does not have a trampoline"

cscore/trampolines/cs__ImageSource.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::ImageSource does not have a trampoline"

cscore/trampolines/cs__MjpegServer.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::MjpegServer does not have a trampoline"

cscore/trampolines/cs__RawEvent.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_cpp.h>
+#error "cs::RawEvent does not have a trampoline"

cscore/trampolines/cs__UsbCamera.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::UsbCamera does not have a trampoline"

cscore/trampolines/cs__UsbCameraInfo.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_cpp.h>
+#error "cs::UsbCameraInfo does not have a trampoline"

cscore/trampolines/cs__VideoCamera.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoCamera does not have a trampoline"

cscore/trampolines/cs__VideoEvent.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoEvent does not have a trampoline"

cscore/trampolines/cs__VideoListener.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoListener does not have a trampoline"

cscore/trampolines/cs__VideoMode.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_cpp.h>
+#error "cs::VideoMode does not have a trampoline"

cscore/trampolines/cs__VideoProperty.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoProperty does not have a trampoline"

cscore/trampolines/cs__VideoSink.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoSink does not have a trampoline"

cscore/trampolines/cs__VideoSource.hpp

@@ -0,0 +1,9 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cscore_oo.h>
+#error "cs::VideoSource does not have a trampoline"

cscore/trampolines/frc__CameraServer.hpp

@@ -0,0 +1,12 @@
+// This file is autogenerated. DO NOT EDIT
+
+#pragma once
+#include <semiwrap.h>
+
+// wrapped header
+
+#include <cameraserver/CameraServer.h>
+
+// from extra_includes
+#include <optional>
+#error "frc::CameraServer does not have a trampoline"

cscore/version.py

@@ -0,0 +1,3 @@
+# This file is automatically generated, DO NOT EDIT
+
+version = __version__ = "2027.0.0a3"

robotpy_cscore-2027.0.0a3.dist-info/METADATA

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: robotpy-cscore
+Version: 2027.0.0a3
+Summary: RobotPy bindings for cscore image processing library
+Project-URL: Source code, https://github.com/robotpy/mostrobotpy
+Author-email: RobotPy Development Team <robotpy@googlegroups.com>
+License-Expression: BSD-3-Clause
+License-File: LICENSE
+Requires-Dist: pyntcore==2027.0.0a3
+Requires-Dist: robotpy-wpinet==2027.0.0a3
+Requires-Dist: robotpy-wpiutil==2027.0.0a3

robotpy_cscore-2027.0.0a3.dist-info/RECORD

@@ -0,0 +1,42 @@
+cscore/__init__.py,sha256=xWxHJBSUnPL21pKSgOSup8k5NClz8xUb-fANf4VGPLI,1056
+cscore/__main__.py,sha256=lg7P4Dq6g5sJOcgDICENfjQp43AC-tg6IY99SfCvbcc,3984
+cscore/_logging.py,sha256=FBz3zqXozXiMVB27nXH_65gN9eziUyrhso9pQhcIBdw,321
+cscore/grip.py,sha256=CG4yAHNzAn1JbXoChKO_wblbnNoAuDDaLA9ipp0Tclo,1001
+cscore/imagewriter.py,sha256=ZKvK0zLDj_OBl1KkGT79w_rBiI-bEAeetG6FEnTpHV8,4391
+cscore/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cscore/version.py,sha256=HhBxA53rMU6cMc_GaF2D5zBRbhBLuGI9pyDJ1WV4SDE,90
+cscore/cvnp/README.md,sha256=tjM2hm7INpdQZRPHRmpxTPj23-k4vkGbTlbaBTunN1Q,1164
+cscore/cvnp/cvnp.cpp,sha256=9q5YWv53QyG5pRcOyJCOMEFQJSsUeYRklfgLi9amELU,8974
+cscore/cvnp/cvnp.h,sha256=YgTLCL73KZU3kNUBDUxxYie_kP9nMniMPkNnBrTgPwU,14503
+cscore/cvnp/cvnp_synonyms.cpp,sha256=CdsCtg3Vpl7-kpjpPONhPvsa_9giTvns_xTgtnby4WQ,2350
+cscore/cvnp/cvnp_synonyms.h,sha256=g1oNmv1OTA3TXgg_54JqAwCg43slzenrAet3M-ipEv4,565
+cscore/src/main.cpp,sha256=q4TR_AA_DNmRdPQuljjgLLbrx_CHLMr06qp37oTON4w,529
+cscore/_cscore.cpython-314-x86_64-linux-gnu.so,sha256=NvjX_Azjzq5aUQpsBa3JRqg-QFdtmc9KDEWDwuuJs5g,25325760
+cscore/_cscore.pyi,sha256=xgNao_3Z0rmHedEgzPTVF2X4bp1pht91MQfODoKHW4k,55761
+cscore/_init__cscore.py,sha256=YO5GPjYpB9XMx4NwdPef9WccC-ENQDaV6j3vFPB4R8w,150
+cscore/cscore-casters.pc,sha256=SW2qtGa0plJ64Tu9BA8uDStrQzRGZ1g0GdbDvW0CtG8,192
+cscore/cscore-casters.pybind11.json,sha256=OhrB1ZkzUlHqaL5_tPr0hvifkq13qZZ2pbXVSprxLho,89
+cscore/cscore.pc,sha256=Q4h5Dxj9rSaf2Mrhn3rV0Vm0fgiQITeGyf64bSfbxyk,284
+cscore/trampolines/cs__AxisCamera.hpp,sha256=o7uVmn8X0933zdNSocRON1hAUDm0UC2g6e3AyYknpnk,173
+cscore/trampolines/cs__CvSink.hpp,sha256=TII8RcbzjH8j6if-9LwiUi8iaVSupp9r618CyA2Z7no,249
+cscore/trampolines/cs__CvSource.hpp,sha256=il9Kc_GVmxpz2xcY16DgzY8QbqtLyZiFYuPAKOGWg0o,251
+cscore/trampolines/cs__HttpCamera.hpp,sha256=7kltdI-FLhaw4Ovk_YCFo1BByCCmvyxWiZH3BP-gJPg,173
+cscore/trampolines/cs__ImageSink.hpp,sha256=dqhelXoMeyhn3k1ahZpbjKjARE0HVSrwZqP5495WYvw,172
+cscore/trampolines/cs__ImageSource.hpp,sha256=DnwYvzjm3gbyBP32NHzfY8WcUcQP0qxZwJCAuzSd6R8,174
+cscore/trampolines/cs__MjpegServer.hpp,sha256=rwJ-c9rg62ga2fTet1dnFqFXlStDjEgtuU0AJ5owflA,174
+cscore/trampolines/cs__RawEvent.hpp,sha256=6tID7fzfs99lEplF2Ih3mr66MbnCR3ujL6EIeFiGs9s,172
+cscore/trampolines/cs__UsbCamera.hpp,sha256=0pybyew_LeArqaJuGiAVPc9Ko9RhfYlG3v94b7zyv5s,172
+cscore/trampolines/cs__UsbCameraInfo.hpp,sha256=da-_-ITi6u83gmf_Y17ZbjaKGUyA_CA4MgHCyr-_C6Y,177
+cscore/trampolines/cs__VideoCamera.hpp,sha256=PoKi0HdLNdDQrPy5aTcmWGzZXGM6TYmHptr-95uF6cw,174
+cscore/trampolines/cs__VideoEvent.hpp,sha256=y2HgZExCuRPiUrAAM3Wf7BfYGfMsZOkil9nOHEqi3Yg,173
+cscore/trampolines/cs__VideoListener.hpp,sha256=bj91DPXOebMKluv1Guo69sO8RR49XbwojQekR4CmAR8,176
+cscore/trampolines/cs__VideoMode.hpp,sha256=5aZCNutaJ3xc2EuoRd_XQzKMpWixfXi82qNplRfE5aA,173
+cscore/trampolines/cs__VideoProperty.hpp,sha256=6ypbrQIzLhZSLhBqnfxGchuBbPP7QBWn_m0y2iof4lE,176
+cscore/trampolines/cs__VideoSink.hpp,sha256=nq3nr47D8OCOYYl9XMCD1lDzUKtOKb8qZPYJHTGCFhY,172
+cscore/trampolines/cs__VideoSource.hpp,sha256=txLjYiPmd1PQXm5WZ11j_CbJZcRAE9hcQVWGL8yc-Vs,174
+cscore/trampolines/frc__CameraServer.hpp,sha256=QOD0jYRW4_9PCJLXIcIDTKevnFj9-iO2MxLki0uvdH8,236
+robotpy_cscore-2027.0.0a3.dist-info/METADATA,sha256=R2tXv1rniJjWfY7wL8KhxFVNirlYu8ptrgkC4xMHMLw,432
+robotpy_cscore-2027.0.0a3.dist-info/WHEEL,sha256=bEAjktRbE_07UBQ6mHuKME1IbS2HwPj_aqXCz81sWEw,100
+robotpy_cscore-2027.0.0a3.dist-info/entry_points.txt,sha256=MTVv-2rIw51vWKhX4bVj581sChRKQvZp9WTUlp3mGiY,53
+robotpy_cscore-2027.0.0a3.dist-info/licenses/LICENSE,sha256=sH4JB8OjTOdUoAPcybggUhhjphTHQWR9M8uoT3v8u5w,3087
+robotpy_cscore-2027.0.0a3.dist-info/RECORD,,

robotpy_cscore-2027.0.0a3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: false
+Tag: cp314-cp314-linux_x86_64

robotpy_cscore-2027.0.0a3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[pkg_config]
+cscore-casters = cscore
+cscore = cscore

robotpy_cscore-2027.0.0a3.dist-info/licenses/LICENSE

@@ -0,0 +1,51 @@
+
+Copyright (c) 2017 Dustin Spicuzza
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+    * Redistributions of source code must retain the above copyright
+      notice, this list of conditions and the following disclaimer.
+    * Redistributions in binary form must reproduce the above copyright
+      notice, this list of conditions and the following disclaimer in the
+      documentation and/or other materials provided with the distribution.
+    * Neither the name of RobotPy nor the
+      names of its contributors may be used to endorse or promote products
+      derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY ROBOTPY AND CONTRIBUTORS``AS IS'' AND ANY
+EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY NONINFRINGEMENT AND FITNESS FOR A PARTICULAR 
+PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL ROBOTPY OR CONTRIBUTORS BE LIABLE FOR 
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+Code for for cscore and other FIRST libraries use the following license:
+
+* Copyright (c) 2016 FIRST
+* All rights reserved.
+*
+* Redistribution and use in source and binary forms, with or without
+* modification, are permitted provided that the following conditions are met:
+*     * Redistributions of source code must retain the above copyright
+*       notice, this list of conditions and the following disclaimer.
+*     * Redistributions in binary form must reproduce the above copyright
+*       notice, this list of conditions and the following disclaimer in the
+*       documentation and/or other materials provided with the distribution.
+*     * Neither the name of the FIRST nor the
+*       names of its contributors may be used to endorse or promote products
+*       derived from this software without specific prior written permission.
+*
+* THIS SOFTWARE IS PROVIDED BY FIRST AND CONTRIBUTORS``AS IS'' AND ANY
+* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+* WARRANTIES OF MERCHANTABILITY NONINFRINGEMENT AND FITNESS FOR A PARTICULAR
+* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL FIRST OR CONTRIBUTORS BE LIABLE FOR
+* ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.