vertex color example

examples/flag_example.rs

@@ -1,4 +1,4 @@
-use bevy::{prelude::*, render::mesh::VertexAttributeValues};
+use bevy::prelude::*;
 use bevy_inspector_egui::quick::{ResourceInspectorPlugin, WorldInspectorPlugin};
 use bevy_silk::prelude::*;
 
@@ -112,17 +112,16 @@ fn spawn_cloth(
 
     // Color flag
     let mut mesh = mesh;
-    if let Some(VertexAttributeValues::Float32x3(positions)) =
-        mesh.attribute(Mesh::ATTRIBUTE_POSITION)
-    {
-        let colors: Vec<[f32; 4]> = positions
-            .iter()
-            .map(|[r, g, b]| [(1. - *r) / 2., (1. - *g) / 2., (1. - *b) / 2., 1.])
-            .collect();
-        println!("{}", colors.len());
-        mesh.insert_attribute(Mesh::ATTRIBUTE_COLOR, colors);
-    }
-    let cloth = ClothBuilder::new().with_pinned_vertex_ids((0..size_y).map(|i| i * size_x));
+    let colors: Vec<[f32; 4]> = (0..size_y)
+        .flat_map(|_| {
+            (0..size_x).map(|v| {
+                let v = v as f32 / size_x as f32;
+                [1.0, v, v, 1.0]
+            })
+        })
+        .collect();
+    mesh.insert_attribute(Mesh::ATTRIBUTE_COLOR, colors);
+    let cloth = ClothBuilder::new().with_pinned_vertex_color(Color::RED);
     commands.spawn((
         PbrBundle {
             mesh: meshes.add(mesh),