.

src/pipelines/agents/agent.wgsl

@@ -7,6 +7,15 @@ struct Settings {
   individualTrailWeight: f32,
   agentCount: f32,
   introProgress: f32,
+  color1ToColor1: f32,
+  color1ToColor2: f32,
+  color1ToColor3: f32,
+  color2ToColor1: f32,
+  color2ToColor2: f32,
+  color2ToColor3: f32,
+  color3ToColor1: f32,
+  color3ToColor2: f32,
+  color3ToColor3: f32,
 };
 
 @group(1) @binding(0) var<uniform> settings: Settings;
@@ -58,17 +67,19 @@ fn main(
   let sourceRightSample = textureLoad(sourceMap, rightSensor, 0);
 
   let channelMask = get_channel_mask(agent.colorIndex);
-  let friendForward = dot(trailForward.rgb, channelMask);
-  let friendLeft = dot(trailLeft.rgb, channelMask);
-  let friendRight = dot(trailRight.rgb, channelMask);
+  let reactionMask = get_reaction_mask(agent.colorIndex);
 
-  let sourceForward = dot(sourceForwardSample.rgb, channelMask);
-  let sourceLeft = dot(sourceLeftSample.rgb, channelMask);
-  let sourceRight = dot(sourceRightSample.rgb, channelMask);
+  let trailForwardWeight = dot(trailForward.rgb, reactionMask);
+  let trailLeftWeight = dot(trailLeft.rgb, reactionMask);
+  let trailRightWeight = dot(trailRight.rgb, reactionMask);
 
-  let weightForward = friendForward + sourceForward * 24.0;
-  let weightLeft = friendLeft + sourceLeft * 24.0;
-  let weightRight = friendRight + sourceRight * 24.0;
+  let sourceForwardWeight = dot(sourceForwardSample.rgb, reactionMask);
+  let sourceLeftWeight = dot(sourceLeftSample.rgb, reactionMask);
+  let sourceRightWeight = dot(sourceRightSample.rgb, reactionMask);
+
+  let weightForward = trailForwardWeight + sourceForwardWeight * 24.0;
+  let weightLeft = trailLeftWeight + sourceLeftWeight * 24.0;
+  let weightRight = trailRightWeight + sourceRightWeight * 24.0;
 
   var rotation = (random.r - 0.5) * settings.turnWhenLost;
   if weightForward >= weightLeft && weightForward >= weightRight {
@@ -78,7 +89,8 @@ fn main(
   }
 
   let sourceAtAgent = textureLoad(sourceMap, vec2<i32>(agent.position), 0);
-  let sourceAtAgentStrength = clamp(dot(sourceAtAgent.rgb, channelMask), 0.0, 1.0);
+  let positiveReactionMask = max(reactionMask, vec3<f32>(0.0));
+  let sourceAtAgentStrength = clamp(dot(sourceAtAgent.rgb, positiveReactionMask), 0.0, 1.0);
   var moveRate = settings.moveRate * mix(1.0, 0.08, sourceAtAgentStrength);
   var introTargetOffset = vec2<f32>(0.0, 0.0);
   var introTargetDistance = 0.0;
@@ -111,7 +123,7 @@ fn main(
   }
 
   let sourceBelow = textureLoad(sourceMap, vec2<i32>(nextPosition), 0);
-  let sourceBelowStrength = dot(sourceBelow.rgb, channelMask);
+  let sourceBelowStrength = clamp(dot(sourceBelow.rgb, positiveReactionMask), 0.0, 1.0);
   let trailWeight = settings.individualTrailWeight * (1.0 + sourceBelowStrength * 16.0);
   var trailBelow = textureLoad(trailMapIn, vec2<i32>(nextPosition), 0);
   trailBelow = vec4<f32>(
@@ -145,6 +157,28 @@ fn get_channel_mask(colorIndex: f32) -> vec3<f32> {
   return vec3<f32>(0, 0, 1);
 }
 
+fn get_reaction_mask(colorIndex: f32) -> vec3<f32> {
+  if colorIndex < 0.5 {
+    return vec3<f32>(
+      settings.color1ToColor1,
+      settings.color1ToColor2,
+      settings.color1ToColor3
+    );
+  }
+  if colorIndex < 1.5 {
+    return vec3<f32>(
+      settings.color2ToColor1,
+      settings.color2ToColor2,
+      settings.color2ToColor3
+    );
+  }
+  return vec3<f32>(
+    settings.color3ToColor1,
+    settings.color3ToColor2,
+    settings.color3ToColor3
+  );
+}
+
 fn angle_delta(sourceAngle: f32, targetAngle: f32) -> f32 {
   return atan2(sin(targetAngle - sourceAngle), cos(targetAngle - sourceAngle));
 }