from ffmpegio import image, probe, transcode, FFmpegError

import tempfile, re

from os import path

outext = ".png"

def test_create():

# make sure rgb channels are mapped properly

A = image.create("color", c="red", s="100x100", d=1)

assert A["shape"] == (100, 100, 3)

assert A["dtype"] == ("|u1")

# A = image.create("color", c="green", s="100x100", d=1)

# A = image.create("color", c="blue", s="100x100", d=1)

# B = image.create("cellauto")

# B = image.create("allrgb", d=1)

# B = image.create("allyuv", d=1)

# B = image.create("gradients=s=1920x1080:x0=0:x1=0:y0=0:y1=1080", d=1)

# B = image.create("gradients", d=1)

# B = image.create("mandelbrot")

# B = image.create("mptestsrc=t=dc_luma", d=1)

# B = image.create("mptestsrc", t="ring1", d=1)

# B = image.create("life", life_color="#00ff00")

# B = image.create("haldclutsrc", 16, d=1)

# A = image.create("testsrc", d=1)

# B = image.create("testsrc2", alpha=0, d=1)

# B = image.create("rgbtestsrc", d=1)

# B = image.create("smptebars", d=1)

# B = image.create("smptehdbars", d=1)

# B = image.create("pal100bars", d=1)

# B = image.create("pal75bars", d=1)

# B = image.create("yuvtestsrc", d=1)

# B = image.create("sierpinski")

# print(A['dtype'], A['shape'])

# plt.subplot(1,2,1)

# plt.imshow(A)

# plt.subplot(1,2,2)

# plt.imshow(B)

# plt.show()

def test_read_write():

url = "tests/assets/ffmpeg-logo.png"

outext = ".jpg"

A = image.read(url)

print(A["dtype"] == "|u1")

B = image.read(url, pix_fmt="ya8")

print(B["shape"])

C = image.read(url, pix_fmt="rgb24")

D = image.read(url, pix_fmt="gray")

with tempfile.TemporaryDirectory() as tmpdirname:

out_url = path.join(tmpdirname, re.sub(r"\..*?$", outext, path.basename(url)))

print(out_url, C['shape'])

image.write(out_url, C)

print(probe.video_streams_basic(out_url))

C = image.read(out_url, pix_fmt="rgba", show_log=True)

# with open(path.join(tmpdirname, "progress.txt")) as f:

# print(f.read())

# display = os.read(read_pipe, 128).strip()

# plt.subplot(4, 1, 1)

# plt.imshow(A)

# plt.subplot(4, 1, 2)

# plt.imshow(B[..., 0], alpha=B[..., -1] / 255.0)

# plt.subplot(4, 1, 3)

# plt.imshow(C)

# plt.subplot(4, 1, 4)

# plt.imshow(D)

# plt.show()

def test_read_basic_filter():

url = "tests/assets/ffmpeg-logo.png"

B = image.read(

url,

show_log=True,

pix_fmt="rgb24",

fill_color="red",

crop=(300, 50),

flip="horizontal",

transpose="clock",

)

B = image.read(url, show_log=True, s=(100, -2))

print(B['shape'])

url = "tests/assets/ffmpeg-logo.png"

B = image.read(

url,

show_log=True,

fill_color="red",

)

B = image.read(url, show_log=True, fill_color="red", pix_fmt="rgb24")

def test_square_pixels():

url = "tests/assets/testvideo-1m.mp4"

with tempfile.TemporaryDirectory() as tmpdirname:

out_url = path.join(tmpdirname, path.basename(url))

transcode(url, out_url, show_log=True, vf="setsar=11/13", t=0.5)

B = image.read(out_url)

Bu = image.read(out_url, square_pixels="upscale")

Bd = image.read(out_url, square_pixels="downscale")

Bue = image.read(out_url, square_pixels="upscale_even")

Bde = image.read(out_url, square_pixels="downscale_even")

print(B['shape'])

print(Bu['shape'])

print(Bd['shape'])

print(Bue['shape'])

print(Bde['shape'])

if __name__ == "__main__":

from matplotlib import pyplot as plt

import logging

from ffmpegio import utils, ffmpegprocess

from ffmpegio.utils import filter as filter_utils, log as log_utils

# logging.basicConfig(level=logging.DEBUG)

url = "tests/assets/ffmpeg-logo.png"

B = image.read(

url,

show_log=True,

fill_color="red",

)

plt.imshow(B)

plt.show()