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

export class AxisAngleGenerator implements RotationGenerator {
    type: 'rotation';
    constructor(
        public axis: Vector3,
        public angle: FunctionValueGenerator | ValueGenerator
    ) {
        this.type = 'rotation';
    }

    startGen(memory: GeneratorMemory): void {
        this.angle.startGen(memory);
    }

    genValue(memory: GeneratorMemory, quat: Quaternion, delta: number, t?: number): Quaternion {
        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
        return quat.setFromAxisAngle(this.axis, this.angle.genValue(memory, t!) * delta);
    }

    toJSON(): FunctionJSON {
        return {
            type: 'AxisAngle',
            axis: {x: this.axis.x, y: this.axis.y, z: this.axis.z},
            angle: this.angle.toJSON(),
        };
    }

    static fromJSON(json: FunctionJSON): AxisAngleGenerator {
        return new AxisAngleGenerator(
            new Vector3(json.axis.x, json.axis.y, json.axis.z),
            ValueGeneratorFromJSON(json.angle) as FunctionValueGenerator
        );
    }

    clone(): RotationGenerator {
        return new AxisAngleGenerator(this.axis.clone(), this.angle.clone());
    }
}