Commit 0f34b522 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add run this test

parent e7b14e23
Loading
Loading
Loading
Loading
+21 −0
Original line number Diff line number Diff line
import numpy as np
from PIL import Image

from imgutils.data import istack
from plot import image_plot

if __name__ == '__main__':
    width, height = Image.open('nian.png').size
    hs1 = (1 - np.abs(np.linspace(-1 / 3, 1, height))) ** 0.5
    ws1 = (1 - np.abs(np.linspace(-1, 1, width))) ** 0.5
    nian_mask = hs1[..., None] * ws1  # HxW

    image_plot(
        'nian.png',
        (istack('lime', 'nian.png'), 'nian_lime.png'),
        (istack(('yellow', 0.5), ('nian.png', 0.9)), 'nian_trans.png'),
        (istack(('nian.png', nian_mask)), 'nian_mask.png'),
        save_as='grid_istack.dat.svg',
        columns=2,
        figsize=(12, 12),
    )
+1 −0
Original line number Diff line number Diff line
@@ -13,3 +13,4 @@ imgutils.data
    encode
    decode
    image
    layer
+15 −0
Original line number Diff line number Diff line
imgutils.data.layer
==========================

.. currentmodule:: imgutils.data.layer

.. automodule:: imgutils.data.layer


istack
---------------------------

.. autofunction:: istack


+1 −0
Original line number Diff line number Diff line
@@ -2,3 +2,4 @@ from .background import *
from .decode import *
from .encode import *
from .image import *
from .layer import *
+3 −4
Original line number Diff line number Diff line
@@ -3,7 +3,8 @@ from typing import Optional
import numpy as np
from PIL import ImageColor, Image

from .image import ImageTyping, load_image
from .image import ImageTyping
from .layer import istack

__all__ = [
    'grid_background',
@@ -62,6 +63,4 @@ def grid_transparent(image: ImageTyping, step: Optional[int] = None,

    """
    retval = grid_background(image.height, image.width, step, forecolor, backcolor)
    image = load_image(image, force_background=None, mode='RGBA')
    retval.paste(image, mask=image)
    return retval.convert('RGB')
    return istack(retval, image).convert('RGB')
Loading