import random
import math
import torch

def random_circle_init(min_r : float = 0.5, on_edge : bool = False):
    theta = random.uniform(0, 2 * math.pi)
    if on_edge:
        r = 1.0
    else:
        r = random.uniform(min_r, 1.0)
    x = r * math.cos(theta)
    y = r * math.sin(theta)

    return x, y

def recursive_find_dtype(x):
    """
    Assuming x is some list/tuple of things that could be tensors, searches for any tensors and returns dtype
    """
    for i in x:
        if isinstance(i, list):
            res = recursive_find_dtype(i)
            if res is None:
                continue
            else:
                return res
        elif isinstance(i, torch.Tensor):
            return i.dtype

def recursive_find_device(x):
    """
    Assuming x is some list/tuple of things that could be tensors, searches for any tensors and returns device
    """
    for i in x:
        if isinstance(i, list):
            res = recursive_find_device(i)
            if res is None:
                continue
            return res
        elif isinstance(i, torch.Tensor):
            return i.device