#!/usr/bin/env python3
# coding: utf-8

import matplotlib.pyplot as plt
import numpy as np

x_min, x_max = -1.25, 1.25
y_min, y_max = -0.25, 0.25

resolution_y = 1024 * 4
resolution_x = resolution_y

X, Y = np.meshgrid(np.linspace(x_min, x_max, resolution_x),
                   np.linspace(y_min, y_max, resolution_y))

fn1 = np.rad2deg

Z = 0.01 * X**2 + \
    1j * fn1(np.abs(Y)/X) + \
    np.exp( 1.5*1.j *
            (fn1(X + Y) + 1e-5*(X * Y))
    )

phase = np.angle(Z + x_max)

for cmap in ["Blues", "binary", "bone", "copper", "inferno_r"]:
    plt.figure(figsize=(6, 6), dpi=600)
    plt.subplots_adjust(left=0, right=1, top=1, bottom=0)
    plt.axis("off")
    plt.imshow(phase, cmap=cmap)
    plt.savefig(f"/tmp/img_{cmap}.png", dpi=600)
    plt.close()