vkit.mechanism.distortion_policy.photometric.effect

  1# Copyright 2022 vkit-x Administrator. All Rights Reserved.
  2#
  3# This project (vkit-x/vkit) is dual-licensed under commercial and SSPL licenses.
  4#
  5# The commercial license gives you the full rights to create and distribute software
  6# on your own terms without any SSPL license obligations. For more information,
  7# please see the "LICENSE_COMMERCIAL.txt" file.
  8#
  9# This project is also available under Server Side Public License (SSPL).
 10# The SSPL licensing is ideal for use cases such as open source projects with
 11# SSPL distribution, student/academic purposes, hobby projects, internal research
 12# projects without external distribution, or other projects where all SSPL
 13# obligations can be met. For more information, please see the "LICENSE_SSPL.txt" file.
 14from typing import Tuple
 15
 16import attrs
 17from numpy.random import Generator as RandomGenerator
 18
 19from vkit.mechanism import distortion
 20from ..type import DistortionConfigGenerator, DistortionPolicyFactory
 21from ..opt import sample_int, sample_float
 22
 23
 24@attrs.define
 25class JpegQualityConfigGeneratorConfig:
 26    quality_min: int = 1
 27    quality_max: int = 50
 28
 29
 30class JpegQualityConfigGenerator(
 31    DistortionConfigGenerator[
 32        JpegQualityConfigGeneratorConfig,
 33        distortion.JpegQualityConfig,
 34    ]
 35):  # yapf: disable
 36
 37    def __call__(self, shape: Tuple[int, int], rng: RandomGenerator):
 38        quality = sample_int(
 39            level=self.level,
 40            value_min=self.config.quality_min,
 41            value_max=self.config.quality_max,
 42            prob_negative=None,
 43            rng=rng,
 44            inverse_level=True
 45        )
 46
 47        return distortion.JpegQualityConfig(quality=quality)
 48
 49
 50jpeg_quality_policy_factory = DistortionPolicyFactory(
 51    distortion.jpeg_quality,
 52    JpegQualityConfigGenerator,
 53)
 54
 55
 56@attrs.define
 57class PixelationConfigGeneratorConfig:
 58    ratio_min: float = 0.3
 59    ratio_max: float = 1.0
 60
 61
 62class PixelationConfigGenerator(
 63    DistortionConfigGenerator[
 64        PixelationConfigGeneratorConfig,
 65        distortion.PixelationConfig,
 66    ]
 67):  # yapf: disable
 68
 69    def __call__(self, shape: Tuple[int, int], rng: RandomGenerator):
 70        ratio = sample_float(
 71            level=self.level,
 72            value_min=self.config.ratio_min,
 73            value_max=self.config.ratio_max,
 74            prob_reciprocal=None,
 75            rng=rng,
 76            inverse_level=True,
 77        )
 78
 79        return distortion.PixelationConfig(ratio=ratio)
 80
 81
 82pixelation_policy_factory = DistortionPolicyFactory(
 83    distortion.pixelation,
 84    PixelationConfigGenerator,
 85)
 86
 87
 88@attrs.define
 89class FogConfigGeneratorConfig:
 90    roughness_min: float = 0.2
 91    roughness_max: float = 0.85
 92    ratio_max_min: float = 0.2
 93    ratio_max_max: float = 0.75
 94
 95
 96class FogConfigGenerator(
 97    DistortionConfigGenerator[
 98        FogConfigGeneratorConfig,
 99        distortion.FogConfig,
100    ]
101):  # yapf: disable
102
103    def __call__(self, shape: Tuple[int, int], rng: RandomGenerator):
104        roughness = sample_float(
105            level=self.level,
106            value_min=self.config.roughness_min,
107            value_max=self.config.roughness_max,
108            prob_reciprocal=None,
109            rng=rng,
110        )
111        ratio_max = sample_float(
112            level=self.level,
113            value_min=self.config.ratio_max_min,
114            value_max=self.config.ratio_max_max,
115            prob_reciprocal=None,
116            rng=rng,
117        )
118
119        return distortion.FogConfig(
120            roughness=roughness,
121            ratio_max=ratio_max,
122        )
123
124
125fog_policy_factory = DistortionPolicyFactory(
126    distortion.fog,
127    FogConfigGenerator,
128)