import * as neataptic from "neataptic"
import { Group } from "./group"
import { Node } from "./node"
import { Connection } from "./connection"

export interface Layer extends Group {
    nodes: Node[]
    connections: {
        in: Connection[],
        out: Connection[],
        self: Connection[],
    }
}

export function createDense(size: number): Layer {
    return new neataptic.Layer.Dense(size)
}

export function createLSTM(size: number): Layer {
    return new neataptic.Layer.LSTM(size)
}

export function createGRU(size: number): Layer {
    return new neataptic.Layer.GRU(size)
}

export function createMemory(size: number): Layer {
    return new neataptic.Layer.Memory(size)
}

export function create(): Layer {
    return new neataptic.Layer()
}

export const classConstructor: () => void = neataptic.Layer