Source code for shapeflow.plugins.Area_mm2

import numpy as np

from shapeflow.config import extend
from shapeflow.maths.images import area_pixelsum
from shapeflow.video import MaskFunction, FeatureType


@extend(FeatureType, __name__.split('.')[-1])
class _Feature(MaskFunction):
    """Convert :mod:`~shapeflow.plugins.PixelSum` to an area in mm²,
    taking into account the DPI of the design file.
    """
    _label = "Area"
    _unit = "mm²"

    def _function(self, frame: np.ndarray) -> float:
        return self.pxsq2mmsq(area_pixelsum(frame))