pve, pygl

pve/pixmap.py

@@ -0,0 +1,131 @@
+# People's Video Editor: high quality, GPU accelerated mp4 editor
+# Copyright (C) 2025 Roz K <roz@rozk.net>
+#
+# This file is part of People's Video Editor.
+#
+# People's Video Editor is free software: you can redistribute it and/or modify it under the terms of the
+# GNU General Public License as published by the Free Software Foundation, either version 3 of the License,
+# or (at your option) any later version.
+#
+# People's Video Editor is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
+# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+# See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with People's Video Editor.
+# If not, see <https://www.gnu.org/licenses/>.
+
+from fractions import Fraction
+
+class Geometry:
+    __slots__ = '_width', '_height', '_ratio', '_size'
+
+    def __init__(self, width = 0, height = 0, ratio = None):
+        if width > 0 and height > 0:
+            assert ratio is None
+            self._width = width
+            self._height = height
+            self._ratio = Fraction(width, height).limit_denominator()
+            self._size = width * height
+        elif width > 0 and ratio:
+            self._width = width
+            self._height = round((width * ratio.numerator) / ratio.denominator)
+            self._ratio = ratio
+            self._size = width * self._height
+        elif height > 0 and ratio:
+            self._width = round((height * ratio.denominator) / ratio.numerator)
+            self._height = height
+            self._ratio = ratio
+            self._size = self._width * height
+        else:
+            raise ValueError
+
+    def __ge__(self, other):
+        return (self._width >= other._width and self._height >= other._height)
+
+    @property
+    def width(self):
+        return self._width
+
+    @property
+    def height(self):
+        return self._height
+
+    @property
+    def ratio(self):
+        return self._ratio
+
+class Format:
+    __slots__ = '_channels', '_depth', '_type', '_size'
+
+    def __init__(self, channels, depth):
+        self._channels = channels
+        self._depth = depth
+        if depth is 8:
+            self._type = 'B'
+            self._size = channels
+        elif depth is 16:
+            self._type = 'H'
+            self._size = channels * 2
+        else:
+            raise ValueError
+
+    def __eq__(self, other):
+        return (self._channels == other._channels and self._depth == other._depth)
+
+    @property
+    def channels(self):
+        return self._channels
+
+    @property
+    def depth(self):
+        return self._depth
+
+class Pixmap:
+    __slots__ = '_geometry', '_format', '_line_size', '_bytes', '_view'
+
+    def __init__(self, geometry, format):
+        self._geometry = geometry
+        self._format = format
+        self._line_size = geometry._width * format._size
+        self._bytes = bytearray(geometry._size * format._size)
+        self._view = memoryview(self._bytes)
+
+    def __del__(self):
+        del self._view
+        del self._bytes
+
+    @property
+    def geometry(self):
+        return self._geometry
+
+    @property
+    def width(self):
+        return self._geometry._width
+
+    @property
+    def height(self):
+        return self._geometry._height
+
+    @property
+    def ratio(self):
+        return self._geometry._ratio
+
+    @property
+    def format(self):
+        return self._format
+
+    @property
+    def channels(self):
+        return self._format._channels
+
+    @property
+    def depth(self):
+        return self._format._depth
+
+    @property
+    def line_size(self):
+        return self._line_size
+
+    @property
+    def view(self):
+        return self._view