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
180
181
182
183

use minterpolate::InterpolationPrimitive;
use serde::{Deserialize, Serialize};

use amethyst_core::{
    alga::general::{SubsetOf, SupersetOf},
    ecs::prelude::{Entity, WriteStorage},
    math::{convert, RealField, Vector2, Vector3, Vector4},
};

use crate::resources::{AnimationControlSet, AnimationSampling};

use self::SamplerPrimitive::*;

/// Get the animation set for an entity. If none exists, one will be added. If entity is invalid,
/// (eg. removed before) None will be returned.
///
/// ### Type parameters:
///
/// - `I`: identifier type for running animations, only one animation can be run at the same time
///        with the same id
/// - `T`: the component type that the animation applies to
pub fn get_animation_set<'a, I, T>(
    controls: &'a mut WriteStorage<'_, AnimationControlSet<I, T>>,
    entity: Entity,
) -> Option<&'a mut AnimationControlSet<I, T>>
where
    I: Send + Sync + 'static,
    T: AnimationSampling,
{
    controls
        .entry(entity)
        .ok()
        .map(|entry| entry.or_insert_with(AnimationControlSet::default))
}

/// Sampler primitive
#[derive(Debug, Clone, Copy, Serialize, Deserialize)]
pub enum SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    /// A single value
    Scalar(S),
    /// Two values
    Vec2([S; 2]),
    /// Three values
    Vec3([S; 3]),
    /// Four values
    Vec4([S; 4]),
}

impl<S> From<[S; 2]> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: [S; 2]) -> Self {
        SamplerPrimitive::Vec2(arr)
    }
}

impl<S> From<[S; 3]> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: [S; 3]) -> Self {
        SamplerPrimitive::Vec3(arr)
    }
}

impl<S> From<[S; 4]> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: [S; 4]) -> Self {
        SamplerPrimitive::Vec4(arr)
    }
}

impl<S> From<Vector2<S>> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: Vector2<S>) -> Self {
        SamplerPrimitive::Vec2(arr.into())
    }
}

impl<S> From<Vector3<S>> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: Vector3<S>) -> Self {
        SamplerPrimitive::Vec3(arr.into())
    }
}

impl<S> From<Vector4<S>> for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn from(arr: Vector4<S>) -> Self {
        SamplerPrimitive::Vec4(arr.into())
    }
}

impl<S> InterpolationPrimitive for SamplerPrimitive<S>
where
    S: RealField + SubsetOf<f32> + SupersetOf<f32>,
{
    fn add(&self, other: &Self) -> Self {
        match (*self, *other) {
            (Scalar(ref s), Scalar(ref o)) => Scalar(*s + *o),
            (Vec2(ref s), Vec2(ref o)) => Vec2([s[0] + o[0], s[1] + o[1]]),
            (Vec3(ref s), Vec3(ref o)) => Vec3([s[0] + o[0], s[1] + o[1], s[2] + o[2]]),
            (Vec4(ref s), Vec4(ref o)) => {
                Vec4([s[0] + o[0], s[1] + o[1], s[2] + o[2], s[3] + o[3]])
            }
            _ => panic!("Interpolation can not be done between primitives of different types"),
        }
    }

    fn sub(&self, other: &Self) -> Self {
        match (*self, *other) {
            (Scalar(ref s), Scalar(ref o)) => Scalar(*s - *o),
            (Vec2(ref s), Vec2(ref o)) => Vec2([s[0] - o[0], s[1] - o[1]]),
            (Vec3(ref s), Vec3(ref o)) => Vec3([s[0] - o[0], s[1] - o[1], s[2] - o[2]]),
            (Vec4(ref s), Vec4(ref o)) => {
                Vec4([s[0] - o[0], s[1] - o[1], s[2] - o[2], s[3] - o[3]])
            }
            _ => panic!("Interpolation can not be done between primitives of different types"),
        }
    }

    fn mul(&self, scalar: f32) -> Self {
        match *self {
            Scalar(ref s) => Scalar(mul_f32(*s, scalar)),
            Vec2(ref s) => Vec2([mul_f32(s[0], scalar), mul_f32(s[1], scalar)]),
            Vec3(ref s) => Vec3([
                mul_f32(s[0], scalar),
                mul_f32(s[1], scalar),
                mul_f32(s[2], scalar),
            ]),
            Vec4(ref s) => Vec4([
                mul_f32(s[0], scalar),
                mul_f32(s[1], scalar),
                mul_f32(s[2], scalar),
                mul_f32(s[3], scalar),
            ]),
        }
    }

    fn dot(&self, other: &Self) -> f32 {
        convert(match (*self, *other) {
            (Scalar(s), Scalar(o)) => (s * o),
            (Vec2(s), Vec2(o)) => (s[0] * o[0] + s[1] * o[1]),
            (Vec3(s), Vec3(o)) => (s[0] * o[0] + s[1] * o[1] + s[2] * o[2]),
            (Vec4(s), Vec4(o)) => (s[0] * o[0] + s[1] * o[1] + s[2] * o[2] + s[3] * o[3]),
            _ => panic!("Interpolation can not be done between primitives of different types"),
        })
    }

    fn magnitude2(&self) -> f32 {
        self.dot(self)
    }

    fn magnitude(&self) -> f32 {
        match *self {
            Scalar(s) => convert(s),
            Vec2(_) | Vec3(_) | Vec4(_) => self.magnitude2().sqrt(),
        }
    }

    fn normalize(&self) -> Self {
        match *self {
            Scalar(_) => *self,
            Vec2(_) | Vec3(_) | Vec4(_) => self.mul(1. / self.magnitude()),
        }
    }
}

fn mul_f32<T: RealField + SubsetOf<f32> + SupersetOf<f32>>(s: T, scalar: f32) -> T {
    convert::<f32, T>(scalar) * s
}