cyber_rider/src/glamor.rs

89 lines
2.7 KiB
Rust
Raw Normal View History

2022-01-14 06:05:51 +00:00
use bevy::{
pbr::wireframe::{Wireframe, WireframeConfig, WireframePlugin},
prelude::*,
render::{
mesh::{Indices, VertexAttributeValues},
options::WgpuOptions,
render_resource::WgpuFeatures,
},
2022-01-14 06:05:51 +00:00
};
use bevy_polyline::{Polyline, PolylineBundle, PolylineMaterial, PolylinePlugin};
use rand::{thread_rng, Rng};
2022-01-14 06:05:51 +00:00
use crate::{geometry::CyberSphere, lights::AnimateCyberLightWireframe};
2022-01-14 06:05:51 +00:00
2022-01-29 21:31:15 +00:00
pub const BISEXY_COLOR: Color = Color::hsla(292.0, 0.9, 0.60, 1.1);
fn wireframe_planet(
2022-01-14 06:05:51 +00:00
mut commands: Commands,
meshes: Res<Assets<Mesh>>,
mut polylines: ResMut<Assets<Polyline>>,
mut polymats: ResMut<Assets<PolylineMaterial>>,
query: Query<&Handle<Mesh>, With<CyberSphere>>,
2022-01-14 06:05:51 +00:00
) {
let handle = query.single();
let mesh = meshes.get(handle).unwrap();
let vertices = mesh.attribute(Mesh::ATTRIBUTE_POSITION).unwrap();
let mut pts = Vec::with_capacity(vertices.len());
if let VertexAttributeValues::Float32x3(verts) = vertices {
let indices = mesh.indices().unwrap();
if let Indices::U32(indices) = indices {
for i in indices.iter() {
let v = verts[*i as usize];
let v = Vec3::from_slice(&v);
pts.push(v);
}
}
}
if !pts.is_empty() {
commands.spawn_bundle(PolylineBundle {
polyline: polylines.add(Polyline { vertices: pts }),
material: polymats.add(PolylineMaterial {
2022-01-29 21:31:15 +00:00
width: 101.0,
color: BISEXY_COLOR,
perspective: true,
}),
..Default::default()
});
}
2022-01-14 06:05:51 +00:00
}
fn wireframify_lights(
mut commands: Commands,
no_wires: Query<Entity, (With<AnimateCyberLightWireframe>, Without<Wireframe>)>,
wires: Query<Entity, (With<AnimateCyberLightWireframe>, With<Wireframe>)>,
) {
let chance = 0.005;
let rng = &mut thread_rng();
for e in no_wires.iter() {
if rng.gen::<f32>() < chance {
commands.entity(e).insert(Wireframe);
}
}
for e in wires.iter() {
if rng.gen::<f32>() < chance {
commands.entity(e).remove::<Wireframe>();
}
}
}
2022-01-14 06:05:51 +00:00
// public plugin
pub struct CyberGlamorPlugin;
impl Plugin for CyberGlamorPlugin {
fn build(&self, app: &mut App) {
app.insert_resource(WgpuOptions {
features: WgpuFeatures::POLYGON_MODE_LINE,
..Default::default()
})
.insert_resource(WireframeConfig { global: false })
.add_startup_system_to_stage(StartupStage::PostStartup, wireframe_planet)
.add_system(wireframify_lights)
.add_plugin(PolylinePlugin)
2022-01-14 06:05:51 +00:00
.add_plugin(WireframePlugin);
}
}