import {RotationGenerator} from './RotationGenerator';
import {Euler, Quaternion} from '../math';
import {FunctionValueGenerator, ValueGenerator, ValueGeneratorFromJSON} from './ValueGenerator';
import {FunctionJSON} from './FunctionJSON';
import {EulerOrder} from '../math';
import {GeneratorMemory} from './GeneratorMemory';

export class EulerGenerator implements RotationGenerator {
    type: 'rotation';
    eular: Euler;

    constructor(
        public angleX: FunctionValueGenerator | ValueGenerator,
        public angleY: FunctionValueGenerator | ValueGenerator,
        public angleZ: FunctionValueGenerator | ValueGenerator,
        eulerOrder?: EulerOrder
    ) {
        this.type = 'rotation';
        this.eular = new Euler(0, 0, 0, eulerOrder);
    }

    startGen(memory: GeneratorMemory): void {
        this.angleX.startGen(memory);
        this.angleY.startGen(memory);
        this.angleZ.startGen(memory);
    }

    genValue(memory: GeneratorMemory, quat: Quaternion, delta: number, t?: number): Quaternion {
        this.eular.set(
            this.angleX.genValue(memory, t!) * delta,
            this.angleY.genValue(memory, t!) * delta,
            this.angleZ.genValue(memory, t!) * delta
        );
        return quat.setFromEuler(this.eular);
    }

    toJSON(): FunctionJSON {
        return {
            type: 'Euler',
            angleX: this.angleX.toJSON(),
            angleY: this.angleY.toJSON(),
            angleZ: this.angleZ.toJSON(),
            eulerOrder: this.eular.order,
        };
    }

    static fromJSON(json: FunctionJSON): EulerGenerator {
        return new EulerGenerator(
            ValueGeneratorFromJSON(json.angleX) as FunctionValueGenerator,
            ValueGeneratorFromJSON(json.angleY) as FunctionValueGenerator,
            ValueGeneratorFromJSON(json.angleZ) as FunctionValueGenerator,
            json.eulerOrder as EulerOrder
        );
    }

    clone(): RotationGenerator {
        return new EulerGenerator(this.angleX, this.angleY, this.angleZ, this.eular.order);
    }
}