import random
import winkelumrechnungen as wu

def line2example(line):
    split = line.split()
    example = [float(value) for value in split[:7]]
    for i, value in enumerate(example):
        if i < 6:
            example[i] = wu.deg2rad(value)
            # example[i] = value
    return example

def get_random_examples(num):
    """
    beta0, lamb0, alpha0, beta1, lamb1, alpha1, s12
    :param num:
    :return:
    """
    # random.seed(42)
    with open("Karney_2024_Testset.txt") as datei:
        lines = datei.readlines()
    examples = []
    for i in range(num):
        example = line2example(lines[random.randint(0, len(lines) - 1)])
        examples.append(example)
    return examples

def get_examples(l_i):
    """
    beta0, lamb0, alpha0, beta1, lamb1, alpha1, s12
    :param num:
    :return:
    """
    with open("Karney_2024_Testset.txt") as datei:
        lines = datei.readlines()
    examples = []
    for i in l_i:
        example = line2example(lines[i])
        examples.append(example)
    return examples

if __name__ == "__main__":
    get_random_examples(10)