LAB/rust-otus
0
0
Fork 0
Code Pull Requests Activity

Проектная работа - WIP

Browse Source
This commit is contained in:
Баранов Александр Сергеевич
2026-06-07 11:30:11 +03:00
parent a24c1259a3
commit 5cc62bc430
7 changed files with 85 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files
snake-game/src/domain.rs
+1 -1
View File
@@ -18,7 +18,7 @@ pub const SNAKE_VELOCITY: u8 = (GRID_SIZE as f32 / 1.0) as u8;
/// Голова змеи
#[derive(Component)]
pub struct Head;
pub struct Head(pub Direction);
/// Сегмент змеиного хвоста
#[derive(Component)]
snake-game/src/domain/snake_path.rs
+2 -48
View File
@@ -4,51 +4,5 @@ use std::collections::HashMap;
use crate::domain::{Direction, Position};
/// Путь змеи
#[derive(Resource)]
pub struct SnakePath {
positions: HashMap<Position, Direction>,
fresh: Option<Position>,
}
impl SnakePath {
pub fn new() -> Self {
SnakePath {
positions: HashMap::with_capacity(16),
fresh: None,
}
}
pub fn push(&mut self, position: Position, direction: Direction) -> bool {
if let Some(_) = self.positions.get(&position) {
if let Some(fresh) = self.fresh {
if position == fresh {
self.positions.insert(position, direction);
return true;
}
}
return false;
}
self.positions.insert(position, direction);
self.fresh = Some(position);
return true;
}
pub fn peek(&mut self, position: &Position) -> Option<Direction> {
if let Some(fresh) = self.fresh {
if fresh == *position {
self.fresh = None;
}
}
self.positions.get(position).cloned()
}
pub fn drop(&mut self, position: &Position) {
self.positions.remove(position);
}
}
impl Default for SnakePath {
fn default() -> Self {
Self::new()
}
}
#[derive(Resource, Default, Deref, DerefMut)]
pub struct SnakePath(HashMap<Position, Direction>);
snake-game/src/gameplay.rs
+47 -38
View File
@@ -1,63 +1,46 @@
use bevy::prelude::*;
use crate::{
domain::{Direction, End, GRID_SIZE, GridBarier, Head, Position, SnakePath, Tail, Velocity},
domain::{Direction, GRID_SIZE, GridBarier, Head, Position, SnakePath, Tail, Velocity},
tools::split_by_grid,
};
/// Система поворота головы змеи
pub fn head_rotation(
input: Res<ButtonInput<KeyCode>>, //
query: Single<(&mut Transform, &Direction), With<Head>>,
mut snake_path: ResMut<SnakePath>,
query: Single<(&mut Transform, &mut Head, &Direction)>,
) {
let _span = trace_span!("head_rotation").entered();
if input.just_pressed(KeyCode::ArrowLeft) {
let (mut transform, direction) = query.into_inner();
let (mut transform, mut head, direction) = query.into_inner();
let translation = transform.translation.clone();
let cur_pos = Position::from(&*transform);
let mut dst_pos = direction.dst_pos(&*transform);
let mut new_direction = direction.clone();
new_direction.rotate_left();
if !snake_path.push(dst_pos, new_direction) {
trace!("try 1 failed: cannot push [{dst_pos} => {new_direction}] into SnakePath");
dst_pos.shift(&direction);
if !snake_path.push(dst_pos, new_direction) {
trace!("try 2 failed: cannot push [{dst_pos} => {new_direction}] into SnakePath");
return;
}
}
head.0 = new_direction;
transform.rotation = new_direction.orientation();
trace!("running {direction} at {cur_pos} {translation}: go {new_direction} on {dst_pos}");
trace!("running {direction} at {cur_pos} {translation}: go {new_direction}");
} else if input.just_pressed(KeyCode::ArrowRight) {
let (mut transform, direction) = query.into_inner();
let (mut transform, mut head, direction) = query.into_inner();
let translation = transform.translation.clone();
let cur_pos = Position::from(&*transform);
let mut dst_pos = direction.dst_pos(&*transform);
let mut new_direction = direction.clone();
new_direction.rotate_right();
if !snake_path.push(dst_pos, new_direction) {
trace!("try 1 failed: cannot push [{dst_pos} => {new_direction}] into SnakePath");
dst_pos.shift(&direction);
if !snake_path.push(dst_pos, new_direction) {
trace!("try 2 failed: cannot push [{dst_pos} => {new_direction}] into SnakePath");
return;
}
}
head.0 = new_direction;
transform.rotation = new_direction.orientation();
trace!("running {direction} at {cur_pos} {translation}: go {new_direction} on {dst_pos}");
trace!("running {direction} at {cur_pos} {translation}: go {new_direction}");
}
}
/// Система движения змеи
pub fn snake_moving(
time: Res<Time<Fixed>>,
mut snake_path: ResMut<SnakePath>,
query: Query<(&mut Transform, &mut Direction, &Velocity, Entity, Has<End>), Or<(With<Head>, With<Tail>, With<End>)>>,
mut commands: Commands,
fn moving(
velocity: &f32,
time: &Time<Fixed>,
direction: &mut Direction,
mut transform: &mut Transform,
entity: Entity,
commands: &mut Commands,
switch_direction: impl FnOnce(Position) -> Option<Direction>,
) {
let _span = trace_span!("snake_moving").entered();
for (mut transform, mut direction, Velocity(velocity), entity, has_end) in query.into_iter() {
// первая часть движения - до границы сетки
let delta = velocity * time.delta_secs();
let src = direction.get_coord(&transform);
@@ -71,9 +54,9 @@ pub fn snake_moving(
if let Some(second) = splited.next() {
let delta = second.1.abs();
let pos = Position::from(&*transform);
if let Some(dir) = snake_path.peek(&pos) {
if let Some(dir) = switch_direction(pos) {
trace!("set new direction for {entity} in {pos} --> {dir}");
*direction = dir.clone();
*direction = dir;
}
let src = direction.get_coord(&*transform);
let sign = direction.get_change_sign();
@@ -82,11 +65,37 @@ pub fn snake_moving(
// создать событие прохождения границы сетки
commands.entity(entity).trigger(|e| GridBarier::new(e, pos, (&*direction).clone()));
}
}
// если это кончик хвоста, то нужно очистить [SnakePath] за собой
if has_end {
snake_path.drop(&pos);
pub fn head_moving(
query: Query<(&mut Transform, &mut Direction, &Velocity, &Head, Entity)>,
time: Res<Time<Fixed>>,
mut snake_path: ResMut<SnakePath>,
mut commands: Commands,
) {
let _span = trace_span!("head_moving").entered();
for (mut transform, mut direction, Velocity(velocity), head, entity) in query.into_iter() {
let next_dir = head.0.clone();
let snake_path = &mut snake_path;
let switch_direction = move |pos: Position| {
snake_path.insert(pos.clone(), next_dir.clone());
Some(next_dir)
};
moving(velocity, &time, &mut direction, &mut transform, entity, &mut commands, switch_direction);
}
}
pub fn snake_moving(
query: Query<(&mut Transform, &mut Direction, &Velocity, Entity), (With<Tail>, Without<Head>)>,
time: Res<Time<Fixed>>,
snake_path: Res<SnakePath>,
mut commands: Commands,
) {
let _span = trace_span!("off_head_snake_moving").entered();
for (mut transform, mut direction, Velocity(velocity), entity) in query.into_iter() {
let snake_path = &snake_path;
let switch_direction = move |pos: Position| snake_path.get(&pos).cloned();
moving(velocity, &time, &mut direction, &mut transform, entity, &mut commands, switch_direction);
}
}
snake-game/src/lib.rs
+1 -1
View File
@@ -5,6 +5,6 @@ mod observers;
mod startup;
mod tools;
pub use gameplay::{head_rotation, snake_moving};
pub use gameplay::{head_moving, head_rotation, snake_moving};
pub use startup::startup;
pub use tools::define_log_layer;
snake-game/src/main.rs
+2 -2
View File
@@ -1,5 +1,5 @@
use bevy::prelude::*;
use snake_game::{head_rotation, snake_moving, startup};
use snake_game::{head_moving, head_rotation, snake_moving, startup};
fn main() {
App::new()
@@ -33,7 +33,7 @@ fn main() {
),
)
.add_systems(Update, (head_rotation,))
.add_systems(FixedUpdate, (snake_moving,))
.add_systems(FixedUpdate, (head_moving, snake_moving))
.add_systems(
PostUpdate,
(
snake-game/src/observers.rs
+7
View File
@@ -14,3 +14,10 @@ pub fn schedule_set_velocity(tail: Entity) -> impl Fn(On<GridBarier>, Commands,
}
}
}
/// Очистка [SnakePath] от старых значений
pub fn clean_snake_path(event: On<GridBarier>, query: Query<&End>, mut snake_path: ResMut<SnakePath>) {
if query.contains(event.entity) {
snake_path.remove(&event.position);
}
}
snake-game/src/startup.rs
+3 -1
View File
@@ -19,11 +19,13 @@ fn spawn_snake(commands: &mut Commands) {
commands.insert_resource(SnakePath::default());
commands.spawn(Observer::new(crate::observers::clean_snake_path));
commands.spawn(Observer::new(crate::debug::trace_snake_moving));
let head = commands
.spawn((
Head,
Head(Direction::default()),
Direction::default(),
Velocity(SNAKE_VELOCITY as f32),
Sprite::from_color(Color::srgb(1., 1., 1.), vec2(HEAD_SIZE as f32, HEAD_SIZE as f32)),