#pragma once

#include <stddef.h>

#include "maths/matrix.h"
#include "maths/vector.h"

struct PsiVector psi_new_state_0(void);
struct PsiVector psi_new_state_1(void);

struct PsiQuantumGate
{
    const char* name;
    struct PsiMatrix matrix;
    size_t num_qubits;
};

struct PsiQuantumGate psi_new_quantum_gate(const char* name, struct PsiMatrix matrix,
                                           size_t num_qubits);
struct PsiQuantumGate psi_new_quantum_gate_from_matrix(const char* name, struct PsiMatrix matrix);
void psi_free_quantum_gate(struct PsiQuantumGate* gate);

struct PsiQuantumBit
{
    const char* name;
    struct PsiVector state;
};

struct PsiQuantumBit psi_new_quantum_bit(const char* name, struct PsiVector state);
void psi_free_quantum_bit(struct PsiQuantumBit* bit);

struct PsiQuantumRegister
{
    const char* name;
    struct PsiVector state_vector;
    struct PsiQuantumBit* qubits;
    size_t num_qubits;
};

struct PsiQuantumRegister psi_new_quantum_register(const char* name, const char** names,
                                                   size_t count);
struct PsiQuantumRegister
psi_new_quantum_register_from(const char* name, const struct PsiQuantumBit* bits, size_t count);
void psi_free_quantum_register(struct PsiQuantumRegister* reg);

void psi_apply_gate(struct PsiQuantumRegister* reg, struct PsiQuantumGate gate,
                    const size_t* targets, size_t target_count);