1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179

//! Uniform interface for shaders.

pub use valora_derive::UniformSet;

use glium::{
    texture::{texture2d::Texture2d, texture2d_multisample::Texture2dMultisample},
    uniforms::{SamplerBehavior, UniformValue},
};

/// A trait proxying `glium::Uniforms` for types which own their uniforms.
pub trait OwnedUniforms {
    fn visit_owned_values(&self, f: &mut dyn FnMut(&str, &dyn IntoUniformValue));
}

/// A trait for types which can represent themselves as `glium::UniformValue`.
pub trait IntoUniformValue {
    fn into_uniform_value<'a>(&'a self) -> UniformValue<'a>;
}

macro_rules! primitive_uniform_value {
    ($primitive:ty, $wrapper:expr) => {
        impl IntoUniformValue for $primitive {
            fn into_uniform_value<'a>(&'a self) -> UniformValue<'a> { $wrapper(*self) }
        }
    };
}

macro_rules! referenced_uniform_value {
    ($base:ty, $wrapper:expr) => {
        impl IntoUniformValue for $base {
            fn into_uniform_value<'a>(&'a self) -> UniformValue<'a> { $wrapper(self) }
        }
    };
}

primitive_uniform_value!(f32, |v| UniformValue::Float(v));
primitive_uniform_value!((f32, f32), |v: (f32, f32)| UniformValue::Vec2([v.0, v.1]));
primitive_uniform_value!((f32, f32, f32), |v: (f32, f32, f32)| UniformValue::Vec3([
    v.0, v.1, v.2
]));
primitive_uniform_value!((f32, f32, f32, f32), |v: (f32, f32, f32, f32)| {
    UniformValue::Vec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([f32; 2], |v| UniformValue::Vec2(v));
primitive_uniform_value!([f32; 3], |v| UniformValue::Vec3(v));
primitive_uniform_value!([f32; 4], |v| UniformValue::Vec4(v));
primitive_uniform_value!([[f32; 2]; 2], |v| UniformValue::Mat2(v));
primitive_uniform_value!([[f32; 3]; 3], |v| UniformValue::Mat3(v));
primitive_uniform_value!([[f32; 4]; 4], |v| UniformValue::Mat4(v));

primitive_uniform_value!(f64, |v| UniformValue::Double(v));
primitive_uniform_value!((f64, f64), |v: (f64, f64)| UniformValue::DoubleVec2([
    v.0, v.1
]));
primitive_uniform_value!((f64, f64, f64), |v: (f64, f64, f64)| {
    UniformValue::DoubleVec3([v.0, v.1, v.2])
});
primitive_uniform_value!((f64, f64, f64, f64), |v: (f64, f64, f64, f64)| {
    UniformValue::DoubleVec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([f64; 2], |v| UniformValue::DoubleVec2(v));
primitive_uniform_value!([f64; 3], |v| UniformValue::DoubleVec3(v));
primitive_uniform_value!([f64; 4], |v| UniformValue::DoubleVec4(v));
primitive_uniform_value!([[f64; 2]; 2], |v| UniformValue::DoubleMat2(v));
primitive_uniform_value!([[f64; 3]; 3], |v| UniformValue::DoubleMat3(v));
primitive_uniform_value!([[f64; 4]; 4], |v| UniformValue::DoubleMat4(v));

primitive_uniform_value!(i32, |v| UniformValue::SignedInt(v));
primitive_uniform_value!((i32, i32), |v: (i32, i32)| UniformValue::IntVec2([
    v.0, v.1
]));
primitive_uniform_value!((i32, i32, i32), |v: (i32, i32, i32)| UniformValue::IntVec3(
    [v.0, v.1, v.2]
));
primitive_uniform_value!((i32, i32, i32, i32), |v: (i32, i32, i32, i32)| {
    UniformValue::IntVec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([i32; 2], |v| UniformValue::IntVec2(v));
primitive_uniform_value!([i32; 3], |v| UniformValue::IntVec3(v));
primitive_uniform_value!([i32; 4], |v| UniformValue::IntVec4(v));

primitive_uniform_value!(i64, |v| UniformValue::Int64(v));
primitive_uniform_value!((i64, i64), |v: (i64, i64)| UniformValue::Int64Vec2([
    v.0, v.1
]));
primitive_uniform_value!((i64, i64, i64), |v: (i64, i64, i64)| {
    UniformValue::Int64Vec3([v.0, v.1, v.2])
});
primitive_uniform_value!((i64, i64, i64, i64), |v: (i64, i64, i64, i64)| {
    UniformValue::Int64Vec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([i64; 2], |v| UniformValue::Int64Vec2(v));
primitive_uniform_value!([i64; 3], |v| UniformValue::Int64Vec3(v));
primitive_uniform_value!([i64; 4], |v| UniformValue::Int64Vec4(v));

primitive_uniform_value!(u64, |v| UniformValue::UnsignedInt64(v));
primitive_uniform_value!((u64, u64), |v: (u64, u64)| UniformValue::UnsignedInt64Vec2(
    [v.0, v.1]
));
primitive_uniform_value!((u64, u64, u64), |v: (u64, u64, u64)| {
    UniformValue::UnsignedInt64Vec3([v.0, v.1, v.2])
});
primitive_uniform_value!((u64, u64, u64, u64), |v: (u64, u64, u64, u64)| {
    UniformValue::UnsignedInt64Vec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([u64; 2], |v| UniformValue::UnsignedInt64Vec2(v));
primitive_uniform_value!([u64; 3], |v| UniformValue::UnsignedInt64Vec3(v));
primitive_uniform_value!([u64; 4], |v| UniformValue::UnsignedInt64Vec4(v));

primitive_uniform_value!(u32, |v| UniformValue::UnsignedInt(v));
primitive_uniform_value!((u32, u32), |v: (u32, u32)| UniformValue::UnsignedIntVec2([
    v.0, v.1
]));
primitive_uniform_value!((u32, u32, u32), |v: (u32, u32, u32)| {
    UniformValue::UnsignedIntVec3([v.0, v.1, v.2])
});
primitive_uniform_value!((u32, u32, u32, u32), |v: (u32, u32, u32, u32)| {
    UniformValue::UnsignedIntVec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([u32; 2], |v| UniformValue::UnsignedIntVec2(v));
primitive_uniform_value!([u32; 3], |v| UniformValue::UnsignedIntVec3(v));
primitive_uniform_value!([u32; 4], |v| UniformValue::UnsignedIntVec4(v));

primitive_uniform_value!(bool, |v| UniformValue::Bool(v));
primitive_uniform_value!((bool, bool), |v: (bool, bool)| UniformValue::BoolVec2([
    v.0, v.1
]));
primitive_uniform_value!((bool, bool, bool), |v: (bool, bool, bool)| {
    UniformValue::BoolVec3([v.0, v.1, v.2])
});
primitive_uniform_value!((bool, bool, bool, bool), |v: (bool, bool, bool, bool)| {
    UniformValue::BoolVec4([v.0, v.1, v.2, v.3])
});
primitive_uniform_value!([bool; 2], |v| UniformValue::BoolVec2(v));
primitive_uniform_value!([bool; 3], |v| UniformValue::BoolVec3(v));
primitive_uniform_value!([bool; 4], |v| UniformValue::BoolVec4(v));

referenced_uniform_value!(Texture2d, |t| UniformValue::Texture2d(t, None));
referenced_uniform_value!(
    Texture2dMultisample,
    |t| UniformValue::Texture2dMultisample(t, Some(SamplerBehavior::default()))
);

#[cfg(test)]
mod test {
    use super::*;
    use valora_derive::UniformSet;

    #[derive(UniformSet)]
    pub struct MyUniforms {
        camera: [f32; 3],
        flipped: bool,
    }

    #[test]
    fn test() {
        let uniforms = MyUniforms {
            camera: [0., 0., 0.],
            flipped: true,
        };

        let mut i = 0;
        OwnedUniforms::visit_owned_values(&uniforms, &mut |name: &str, value| {
            let value = value.into_uniform_value();
            match i {
                0 => match (name, value) {
                    ("camera", UniformValue::Vec3(pos)) if pos == [0., 0., 0.] => {}
                    _ => panic!("Wrong camera values"),
                },
                1 => match (name, value) {
                    ("flipped", UniformValue::Bool(true)) => {}
                    _ => panic!("Wrong flipped values"),
                },
                _ => panic!("unexpected uniform: {:?}", name),
            }
            i += 1;
        });
    }
}