veloren/common/src/sys/animation.rs

use crate::{
    comp::{
        Animation, AnimationInfo, Attacking, ForceUpdate, Gliding, Jumping, OnGround, Ori, Pos,
        Rolling, Vel,
    },
    state::DeltaTime,
};
use specs::{Entities, Join, Read, ReadStorage, System, WriteStorage};

/// This system will apply the animation that fits best to the users actions
pub struct Sys;
impl<'a> System<'a> for Sys {
    type SystemData = (
        Entities<'a>,
        Read<'a, DeltaTime>,
        ReadStorage<'a, Vel>,
        ReadStorage<'a, OnGround>,
        ReadStorage<'a, Jumping>,
        ReadStorage<'a, Gliding>,
        ReadStorage<'a, Attacking>,
        ReadStorage<'a, Rolling>,
        WriteStorage<'a, AnimationInfo>,
    );

    fn run(
        &mut self,
        (
            entities,
            dt,
            velocities,
            on_grounds,
            jumpings,
            glidings,
            attackings,
            rollings,
            mut animation_infos,
        ): Self::SystemData,
    ) {
        for (entity, vel, on_ground, jumping, gliding, attacking, rolling, mut animation_info) in (
            &entities,
            &velocities,
            on_grounds.maybe(),
            jumpings.maybe(),
            glidings.maybe(),
            attackings.maybe(),
            rollings.maybe(),
            &mut animation_infos,
        )
            .join()
        {
            animation_info.time += dt.0 as f64;

            fn impossible_animation(message: &str) -> Animation {
                warn!("{}", message);
                Animation::Idle
            }

            let animation = match (
                on_ground.is_some(),
                vel.0.magnitude() > 3.0, // Moving
                attacking.is_some(),
                gliding.is_some(),
                rolling.is_some(),
            ) {
                (_, _, true, true, _) => impossible_animation("Attack while gliding"),
                (_, _, true, _, true) => impossible_animation("Roll while attacking"),
                (_, _, _, true, true) => impossible_animation("Roll while gliding"),
                (_, false, _, _, true) => impossible_animation("Roll without moving"),
                (_, true, false, false, true) => Animation::Roll,
                (true, false, false, false, false) => Animation::Idle,
                (true, true, false, false, false) => Animation::Run,
                (false, _, false, false, false) => Animation::Jump,
                (_, _, false, true, false) => Animation::Gliding,
                (_, _, true, false, false) => Animation::Attack,
            };

            let last = animation_info.clone();
            let changed = last.animation != animation;

            *animation_info = AnimationInfo {
                animation,
                time: if changed { 0.0 } else { last.time },
                changed,
            };
        }
    }
}