vkit.mechanism.distortion.geometric.grid_rendering.grid_creator

  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 List
 15from itertools import chain
 16
 17from vkit.element import Point, PointList
 18from .type import ImageGrid
 19from .point_projector import PointProjector
 20
 21
 22def create_src_image_grid(height: int, width: int, grid_size: int):
 23    ys = list(range(0, height, grid_size))
 24    if ys[-1] != height - 1:
 25        ys.append(height - 1)
 26
 27    xs = list(range(0, width, grid_size))
 28    if xs[-1] != width - 1:
 29        xs.append(width - 1)
 30
 31    points_2d = []
 32    for y in ys:
 33        points = []
 34        for x in xs:
 35            points.append(Point.create(y=y, x=x))
 36        points_2d.append(points)
 37
 38    return ImageGrid(
 39        points_2d=points_2d,
 40        grid_size=grid_size,
 41    )
 42
 43
 44def create_dst_image_grid_and_shift_amounts_and_resize_ratios(
 45    src_image_grid: ImageGrid,
 46    point_projector: PointProjector,
 47    resize_as_src: bool = True,
 48):
 49    dst_points_2d = []
 50
 51    src_flatten_points = src_image_grid.flatten_points
 52    num_src_flatten_points = len(src_flatten_points)
 53
 54    dst_flatten_points = point_projector.project_points(src_flatten_points)
 55
 56    assert len(dst_flatten_points) == num_src_flatten_points
 57    dst_points_2d: List[PointList] = []
 58    for begin in range(0, num_src_flatten_points, src_image_grid.num_cols):
 59        dst_points_2d.append(PointList(dst_flatten_points[begin:begin + src_image_grid.num_cols]))
 60
 61    y_min = dst_points_2d[0][0].y
 62    y_max = y_min
 63    x_min = dst_points_2d[0][0].x
 64    x_max = x_min
 65
 66    for point in chain.from_iterable(dst_points_2d):
 67        y_min = min(y_min, point.y)
 68        y_max = max(y_max, point.y)
 69        x_min = min(x_min, point.x)
 70        x_max = max(x_max, point.x)
 71
 72    shift_amount_y = y_min
 73    shift_amount_x = x_min
 74
 75    for row_idx in range(src_image_grid.num_rows):
 76        for col_idx in range(src_image_grid.num_cols):
 77            point = dst_points_2d[row_idx][col_idx]
 78            dst_points_2d[row_idx][col_idx] = point.to_shifted_point(
 79                offset_y=-shift_amount_y,
 80                offset_x=-shift_amount_x,
 81            )
 82
 83    src_image_height = src_image_grid.image_height
 84    src_image_width = src_image_grid.image_width
 85
 86    resize_ratio_y = 1.0
 87    resize_ratio_x = 1.0
 88
 89    if resize_as_src:
 90        raw_dst_image_grid = ImageGrid(points_2d=dst_points_2d)
 91
 92        resize_ratio_y = src_image_height / raw_dst_image_grid.image_height
 93        resize_ratio_x = src_image_width / raw_dst_image_grid.image_width
 94
 95        raw_dst_image_grid_image_shape = raw_dst_image_grid.image_shape
 96        del raw_dst_image_grid
 97
 98        for row_idx in range(src_image_grid.num_rows):
 99            for col_idx in range(src_image_grid.num_cols):
100                point = dst_points_2d[row_idx][col_idx]
101                dst_points_2d[row_idx][col_idx] = point.to_conducted_resized_point(
102                    raw_dst_image_grid_image_shape,
103                    resized_height=src_image_height,
104                    resized_width=src_image_width,
105                )
106
107    dst_image_grid = ImageGrid(points_2d=dst_points_2d)
108
109    if resize_as_src:
110        assert dst_image_grid.image_height == src_image_height
111        assert dst_image_grid.image_width == src_image_width
112
113    shift_amounts = (shift_amount_y, shift_amount_x)
114    resize_ratios = (resize_ratio_y, resize_ratio_x)
115    return dst_image_grid, shift_amounts, resize_ratios
116
117
118def create_dst_image_grid(
119    src_image_grid: ImageGrid,
120    point_projector: PointProjector,
121    resize_as_src: bool = True,
122):
123    dst_image_grid, _, _ = create_dst_image_grid_and_shift_amounts_and_resize_ratios(
124        src_image_grid=src_image_grid,
125        point_projector=point_projector,
126        resize_as_src=resize_as_src,
127    )
128    return dst_image_grid