deuxfleurs 0.1.0

use crate::attachment::internal::AttachmentPosition;

use super::{Context, DataUniformBuilder, GraphicalContext};

pub struct DataMut<'a, T, Ctxt: Context> {
    pub(crate) inner: T,
    pub(crate) context: &'a mut Ctxt,
    pub(crate) uniform: Ctxt::DataUniform<'a>,
}

pub type UninitedData<'a, 'b, T> = DataMut<'b, T, &'a mut crate::Settings>;
pub type DisplayData<'a, 'b, T> = DataMut<'b, T, GraphicalContext<'a>>;

impl<'a, T, Ctxt: Context> DataMut<'a, T, Ctxt> {
    pub(crate) fn convert<U, F: FnOnce(T) -> U>(self, f: F) -> DataMut<'a, U, Ctxt> {
        DataMut {
            inner: f(self.inner),
            uniform: self.uniform,
            context: self.context,
        }
    }
}

pub trait DataMutTrait {
    fn update_data_settings(&mut self);
}

impl<'a, 'b, T> DataMutTrait for UninitedData<'a, 'b, T> {
    fn update_data_settings(&mut self) {}
}

impl<'a, 'b, T> DataMutTrait for DisplayData<'a, 'b, T>
where
    T: DataUniformBuilder,
{
    fn update_data_settings(&mut self) {
        self.uniform
            .as_ref()
            .map(|uniform| self.inner.refresh_buffer(self.context.queue, uniform));
    }
}

pub trait NewAttachedGeometry {
    type UpgradedAttachedGeometry;

    fn init(
        self,
        device: &wgpu::Device,
        camera_light_bind_group_layout: &wgpu::BindGroupLayout,
        transform_bind_group_layout: &wgpu::BindGroupLayout,
        color_format: wgpu::TextureFormat,
    ) -> Self::UpgradedAttachedGeometry;

    fn downgrade(upgraded: &Self::UpgradedAttachedGeometry) -> Self;
}

impl<Ctxt: Context> AttachedGeometry<Ctxt> for () {
    type Args = ();
    type Settings<'a> = &'a mut ();

    fn new(
        _name: String,
        _args: Self::Args,
        _position: AttachmentPosition,
        _characteristic_l: f32,
        _context: &mut Ctxt,
        _transform_layout: &Ctxt::TransformLayout,
    ) -> Self {
        ()
    }

    fn get_settings(&mut self) -> Self::Settings<'_> {
        self
    }

    fn get_attached_position(&self) -> &AttachmentPosition {
        &AttachmentPosition::Vertex
    }

    fn move_elements(&mut self, _queue: &wgpu::Queue, _indices: &[u32], _pos: &[[f32; 3]]) {}
}

pub struct EmptyAttached(());

impl NewAttachedGeometry for () {
    type UpgradedAttachedGeometry = EmptyAttached;

    fn init(
        self,
        _device: &wgpu::Device,
        _camera_light_bind_group_layout: &wgpu::BindGroupLayout,
        _transform_bind_group_layout: &wgpu::BindGroupLayout,
        _color_format: wgpu::TextureFormat,
    ) -> Self::UpgradedAttachedGeometry {
        EmptyAttached(())
    }

    fn downgrade(_upgraded: &Self::UpgradedAttachedGeometry) -> Self {
        ()
    }
}

impl<Ctxt: Context> AttachedGeometry<Ctxt> for EmptyAttached {
    type Args = ();
    type Settings<'a> = &'a mut ();

    fn new(
        _name: String,
        _args: Self::Args,
        _position: AttachmentPosition,
        _characteristic_l: f32,
        _context: &mut Ctxt,
        _transform_layout: &Ctxt::TransformLayout,
    ) -> Self {
        EmptyAttached(())
    }

    fn get_settings(&mut self) -> Self::Settings<'_> {
        &mut self.0
    }

    fn get_attached_position(&self) -> &AttachmentPosition {
        &AttachmentPosition::Vertex
    }

    fn move_elements(&mut self, _queue: &wgpu::Queue, _indices: &[u32], _pos: &[[f32; 3]]) {}
}

pub trait ShapeSettings: DataUniformBuilder {
    fn new(name: &str, characteristic_length: f32) -> Self;

    fn draw_ui(&mut self, ui: &mut egui::Ui, rebuild_pipeline: &mut bool) -> bool;
}

pub trait ShapeGeometry {
    type Args;

    fn new(args: Self::Args) -> Self;

    fn can_be_replaced_by(&self, _other: &Self) -> bool;

    fn get_positions(&self) -> &[[f32; 3]];

    fn get_total_elements(&self) -> u32;

    fn get_vertex_pos(&self, vertex: u32) -> [f32; 3];

    fn move_vertex(
        &mut self,
        vertex: u32,
        pos: [f32; 3],
    ) -> ((Vec<u32>, Vec<[f32; 3]>), (Vec<u32>, Vec<[f32; 3]>));

    fn get_characteristic_length(&self) -> f32;
}

pub trait AttachedGeometry<Ctxt: Context> {
    type Args;
    type Settings<'a>
    where
        Self: 'a;

    fn new(
        name: String,
        args: Self::Args,
        _position: AttachmentPosition,
        characteristic_l: f32,
        context: &mut Ctxt,
        transform_layout: &Ctxt::TransformLayout,
    ) -> Self;

    fn shown(&self) -> bool {
        false
    }

    fn show(&mut self, _show: bool, _refresh_screen: &mut bool) {}

    fn draw_ui(
        &mut self,
        _ui: &mut egui::Ui,
        _device: &wgpu::Device,
        _queue: &wgpu::Queue,
        _camera_light_bind_group_layout: &wgpu::BindGroupLayout,
        _color_format: wgpu::TextureFormat,
        _refresh_screen: &mut bool,
    ) {
    }

    fn render<'c, 'd>(&'c self, _render_pass: &mut wgpu::RenderPass<'d>)
    where
        'c: 'd,
    {
    }

    fn get_settings(&mut self) -> Self::Settings<'_>;

    fn get_attached_position(&self) -> &AttachmentPosition;

    fn move_elements(&mut self, queue: &wgpu::Queue, indices: &[u32], pos: &[[f32; 3]]);
}