60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# last edit: 11.04.2018
|
|
|
|
from collections import Iterable
|
|
|
|
import numpy as np
|
|
|
|
INF = 1e9
|
|
EPSILON = 1e-7
|
|
TWO_PI = 2 * np.pi
|
|
|
|
|
|
def xy_to_z(*xy):
|
|
if len(xy) == 1:
|
|
z = xy[0]
|
|
if isinstance(z, Iterable):
|
|
z = np.array(z)
|
|
if len(z.shape) == 2:
|
|
z = z[0] + 1j * z[1]
|
|
elif len(z.shape) > 2:
|
|
raise ValueError("Something went wrong!")
|
|
elif len(xy) == 2:
|
|
x, y = xy
|
|
if isinstance(x, Iterable):
|
|
if isinstance(y, Iterable) and len(x) == len(y):
|
|
z = np.array(x) + 1j * np.array(y)
|
|
else:
|
|
raise ValueError("x and y vectors dont match in type and/or size")
|
|
else:
|
|
z = x + 1j * y
|
|
else:
|
|
raise ValueError("Arguments are not valid - specify either complex number/vector z or real and imaginary number/vector x, y")
|
|
|
|
return z
|
|
|
|
|
|
def z_to_xy(z):
|
|
return z.real, z.imag
|
|
|
|
|
|
def moebius_z(*args, norm):
|
|
z = xy_to_z(*args)
|
|
return 1 - 2 * norm / (z + norm)
|
|
|
|
|
|
def moebius_inv_z(*args, norm):
|
|
z = xy_to_z(*args)
|
|
return norm * (1 + z) / (1 - z)
|
|
|
|
|
|
def ang_to_c(ang, radius=1):
|
|
return radius * (np.cos(ang) + np.sin(ang) * 1j)
|
|
|
|
|
|
def lambda_to_rad(lmb):
|
|
return lmb * 4 * np.pi
|
|
|
|
|
|
def rad_to_lambda(rad):
|
|
return rad * 0.25 / np.pi
|