import Foundation
import Metal

public class Simplex3DAuraRenderable: ConfiguredQuadRenderable<Simplex3DAuraRenderable.Config> {
    public struct Config: QuadRenderConfiguration {
        let display_size: ADVector2
        let fractal_noise_x: ADFloat
        let fractal_noise_y: ADFloat
        let fractal_noise_z: ADFloat
        let fractal_uniform_scale: ADFloat

        let grain_noise_z: ADFloat
        let grain_noise_scale: ADFloat

        let circle_mask_radius: ADFloat

        let background_color: ADVector3
        let color_a: ADVector3
        let color_b: ADVector3
        let color_c: ADVector3

        let mix_point_start: ADFloat
        let mix_point_background: ADFloat
        let mix_point_backgroundA: ADFloat
        let mix_point_AB: ADFloat
        let mix_point_BC: ADFloat
        let mix_point_end: ADFloat

        public init(
            displaySize: ADVector2 = .ones,
            fractalNoiseX: ADFloat = .zero,
            fractalNoiseY: ADFloat = .zero,
            fractalNoiseZ: ADFloat = .zero,
            fractalUniformScale: ADFloat = .zero,
            grainNoiseZ: ADFloat = .zero,
            grainNoiseScale: ADFloat = 1.0,
            circleMaskRadius: ADFloat = 0.25,
            backgroundColor: ADVector3 = .zero,
            colorA: ADVector3 = .zero,
            colorB: ADVector3 = .init(x: 0.5, y: 0.5, z: 0.5),
            colorC: ADVector3 = .ones,
            mixPointStart: ADFloat = -0.2,
            mixPointBackground: ADFloat = -0.1,
            mixPointBackgroundA: ADFloat = 0.0,
            mixPointAB: ADFloat = 0.2,
            mixPointBC: ADFloat = 0.4,
            mixPointEnd: ADFloat = 2.0
        ) {
            display_size = displaySize
            fractal_noise_x = fractalNoiseX
            fractal_noise_y = fractalNoiseY
            fractal_noise_z = fractalNoiseZ
            fractal_uniform_scale = fractalUniformScale
            grain_noise_z = grainNoiseZ
            grain_noise_scale = grainNoiseScale
            circle_mask_radius = circleMaskRadius
            background_color = backgroundColor
            color_a = colorA
            color_b = colorB
            color_c = colorC
            mix_point_start = mixPointStart
            mix_point_background = mixPointBackground
            mix_point_backgroundA = mixPointBackgroundA
            mix_point_AB = mixPointAB
            mix_point_BC = mixPointBC
            mix_point_end = mixPointEnd
        }

        public static func simpleConfig(
            colorSet: SunoTriColorSet,
            width: Int,
            height: Int,
            scale: Float = 1.0,
            seed: Float = .zero
        ) -> Simplex3DAuraRenderable.Config {
            return .init(
                displaySize: .init(x: ADFloat(width), y: ADFloat(height)),
                fractalNoiseX: .zero,
                fractalNoiseY: .zero,
                fractalNoiseZ: seed,
                fractalUniformScale: 0.5 * scale,
                grainNoiseZ: seed,
                grainNoiseScale: 5.0 * scale,
                circleMaskRadius: 100.0, // Just means will not mask
                backgroundColor: SunoColorPalette.backgroundDotCom.floatThree,
                colorA: colorSet.floatColorA,
                colorB: colorSet.floatColorB,
                colorC: colorSet.floatColorC,
                mixPointStart: -0.2, // Negative value here just means won't be used
                mixPointBackground: -0.1, // Negative value here just means won't be used
                mixPointBackgroundA: 0.0,
                mixPointAB: 0.2,
                mixPointBC: 0.4,
                mixPointEnd: 2.0
            )
        }

        public static var defaultConfiguration: Config {
            return Config()
        }
    }

    override public var configureAction: ConfigureAction {
        return { encoder in
            super.configureAction(encoder)
        }
    }

    override var fragmentShaderName: String {
        return "simplex_3d_aura_frag"
    }
}