用rust实现黑客帝国代码雨

最新推荐文章于 2024-08-07 23:02:59 发布

dkddu

最新推荐文章于 2024-08-07 23:02:59 发布

阅读量191

点赞数 5

文章标签： rust restful 后端

本文链接：https://blog.csdn.net/weixin_72603661/article/details/137055957

版权

本文详细展示了如何利用Rust库firework-rs在终端中创建绚丽的烟花动画，代码涉及事件处理、粒子系统和实时渲染。

摘要由CSDN通过智能技术生成

代码

参考资料

效果图

代码

use crossterm::{
    cursor,
    event::{poll, read, Event, KeyCode},
    execute, terminal,
};
use firework_rs::{
    fireworks::{ExplosionForm, Firework, FireworkConfig, FireworkManager},
    particle::ParticleConfig,
    term::Terminal,
    utils::gen_points_circle,
};
use glam::Vec2;
use rand::{seq::IteratorRandom, thread_rng, Rng};
use std::{
    io::{stdout, Result},
    thread::sleep,
    time::{Duration, SystemTime},
};

fn main() -> Result<()> {
    let mut stdout = stdout();
    let (_width, _height) = terminal::size()?;
    let mut is_running = true;
    terminal::enable_raw_mode()?;
    execute!(stdout, terminal::EnterAlternateScreen, cursor::Hide)?;
    let mut time = SystemTime::now();
    let mut term = Terminal::default();
    let mut fm = FireworkManager::default().with_firework(gen_rain_firework(Vec2::new(
        _width as f32 / 4.,
        _height as f32 / 2.,
    )));

    while is_running {
        if poll(Duration::ZERO)? {
            match read()? {
                Event::Key(e) => {
                    if e.code == KeyCode::Esc {
                        is_running = false;
                    }
                }
                Event::Resize(_, _) => {
                    fm.reset();
                    term.reinit();
                }
                _ => {}
            };
        }

        let delta_time = SystemTime::now().duration_since(time).unwrap();
        fm.update(time, delta_time);
        time = SystemTime::now();
        term.render(&fm);
        term.print(&mut stdout);

        if delta_time < Duration::from_secs_f32(0.05) {
            let rem = Duration::from_secs_f32(0.05) - delta_time;
            sleep(rem);
        }
    }

    execute!(stdout, cursor::Show, terminal::LeaveAlternateScreen)?;
    terminal::disable_raw_mode()?;
    Ok(())
}

fn gen_rain_firework(center: Vec2) -> Firework {
    let colors = vec![(0, 150, 0), (50, 200, 50), (100, 255, 100)];
    let mut particles = Vec::new();

    for p in gen_points_circle(20, 50).iter() {
        particles.push(ParticleConfig::new(
            center + *p,
            Vec2::new(0., thread_rng().gen_range(10.0..15.0)),
            thread_rng().gen_range(50..70),
            Duration::from_secs_f32(thread_rng().gen_range(2.0..3.0)),
            *colors.iter().choose(&mut thread_rng()).unwrap(),
        ));
    }

    let mut config = FireworkConfig::default()
        .with_ar_scale(0.05)
        .with_gravity_scale(0.5)
        .with_gradient_scale(gradient);

    config.set_enable_gradient(true);

    Firework {
        init_time: SystemTime::now(),
        spawn_after: Duration::ZERO,
        center,
        particles,
        config,
        form: ExplosionForm::Sustained {
            lasts: Duration::from_secs(10),
            time_interval: Duration::from_secs_f32(0.01),
            timer: Duration::ZERO,
        },
        ..Default::default()
    }
}

fn gradient(x: f32) -> f32 {
    if x < 0.8125 {
        -0.4 * x + 1.1
    } else {
        -2. * x + 2.2
    }
}