"""
SLM Demo
========
Example on how different geometries and patches work for an SLM. Currently, uses SLM number 0, which is the left
upper corner of the primary monitor.

EPILEPSY WARNING: YOUR PRIMARY SCREEN MAY QUICKLY FLASH DURING RUNNING THIS FILE
"""

import time

import astropy.units as u
import numpy as np
from skimage import data

import set_path  # noqa - needed for setting the module search path to find openwfs
from openwfs.devices.slm import SLM, Patch, geometry
from openwfs.utilities import patterns

# construct a new SLM object and add a patch to it
numerical_aperture = 0.8
s1 = SLM(0, shape=(300, 200))
s2 = SLM(0, pos=(0, 300))
s3 = s1.clone(0, pos=(300, 0), shape=(400, 500))
g = geometry.rectangle(numerical_aperture)

# p1 = Patch(s1, g)
s1.primary_patch.geometry = g
pf = Patch(s1)
p2 = Patch(s2)
p3 = Patch(s2, geometry.rectangle(0.2))
p4 = Patch(s2, geometry.rectangle(0.1))
p3.phases = 0.25
p4.phases = 1
p4.additive_blend = False

pf.phases = patterns.lens(100, f=1 * u.m, wavelength=0.8 * u.um, extent=(10 * u.mm, 10 * u.mm))
rng = np.random.default_rng()
for n in range(200):
    random_data = rng.random([10, 10], np.float32) * 2.0 * np.pi
    s1.set_phases(random_data)
    s2.set_phases(n / 4.0)
    if n == 100:
        del s3

p1 = None  # test deletion. After deleting the two windowed SLMs, we can create a new full screen one
s1.patches.clear()
s1 = 0
s2 = 0
s3 = SLM(1)  # full-screen window
s3.update()
s3.monitor_id = 0  # test switching monitor id
time.sleep(0.5)

s = SLM(0)
s.set_phases(data.camera() / 255 * 2 * np.pi)

s.update()