Decouple ParticleSimulation into separate class
This commit is contained in:
33
ParticleSimulation/Particle.cs
Normal file
33
ParticleSimulation/Particle.cs
Normal file
@@ -0,0 +1,33 @@
|
||||
using Godot;
|
||||
|
||||
namespace Particles
|
||||
{
|
||||
public class Particle
|
||||
{
|
||||
|
||||
public Vector2 Position = new Vector2();
|
||||
public Vector2 Velocity = new Vector2();
|
||||
public ParticleType Type { get; }
|
||||
public float AverageSpeed { get; private set; } = 1f;
|
||||
|
||||
public int Id { get; }
|
||||
public float Health
|
||||
{
|
||||
get => _health;
|
||||
set => _health = Mathf.Clamp(value, 0f, 1f);
|
||||
}
|
||||
|
||||
private float _health = 1f;
|
||||
|
||||
public Particle(int id, ParticleType type)
|
||||
{
|
||||
Id = id;
|
||||
Type = type;
|
||||
}
|
||||
|
||||
public void AddAverageSpeedValue(float speed)
|
||||
{
|
||||
AverageSpeed = (0.99f * AverageSpeed) + (0.01f * speed);
|
||||
}
|
||||
}
|
||||
}
|
||||
225
ParticleSimulation/ParticleSimulation.cs
Normal file
225
ParticleSimulation/ParticleSimulation.cs
Normal file
@@ -0,0 +1,225 @@
|
||||
using Godot;
|
||||
using System.Collections.Generic;
|
||||
using System.ComponentModel;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Particles
|
||||
{
|
||||
public class ParticleSimulation
|
||||
{
|
||||
|
||||
public Vector2 ScreenSize;
|
||||
|
||||
private Dictionary<int, Particle> _particles = new Dictionary<int, Particle>();
|
||||
|
||||
private List<ParticleType> _particleTypes = new List<ParticleType>();
|
||||
private Dictionary<ParticleType, int> _particleTypeCounts = new Dictionary<ParticleType, int>();
|
||||
private List<Task> tasks = new List<Task>();
|
||||
|
||||
public List<int> LastParticlesAdded { get; private set; } = new List<int>();
|
||||
public List<int> LastParticlesRemoved { get; private set; } = new List<int>();
|
||||
|
||||
private int _idCount = 0;
|
||||
|
||||
private const int MaxParticles = 1200;
|
||||
private const int MaxParticleTypes = 10;
|
||||
private const float HealthDelta = 0.002f;
|
||||
private const float NegativeHealthMultiplier = 1.5f;
|
||||
|
||||
public void Initialize()
|
||||
{
|
||||
for (var i = 0; i < MaxParticleTypes; i++)
|
||||
CreateRandomParticleType();
|
||||
for (var i = 0; i < MaxParticles; i++)
|
||||
CreateRandomParticle();
|
||||
}
|
||||
|
||||
public void Update()
|
||||
{
|
||||
LastParticlesRemoved.Clear();
|
||||
LastParticlesAdded.Clear();
|
||||
tasks.Clear();
|
||||
foreach (var id in _particles.Keys)
|
||||
tasks.Add(Task.Factory.StartNew(UpdateParticle, id));
|
||||
Task.WaitAll(tasks.ToArray());
|
||||
var deletedParticle = false;
|
||||
foreach (var p in _particles)
|
||||
{
|
||||
var particle = p.Value;
|
||||
if (deletedParticle == false && particle.Health == 0f)
|
||||
{
|
||||
if (GD.Randf() < 0.2f)
|
||||
{
|
||||
LastParticlesRemoved.Add(p.Key);
|
||||
deletedParticle = true;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
var position = particle.Position;
|
||||
particle.Velocity = particle.Velocity.Clamped(3f * 4f);
|
||||
position += particle.Velocity;
|
||||
particle.Velocity *= 0.855f; // friction
|
||||
if (position.x > ScreenSize.x)
|
||||
position.x -= ScreenSize.x;
|
||||
else if (position.x < 0)
|
||||
position.x += ScreenSize.x;
|
||||
if (position.y > ScreenSize.y)
|
||||
position.y -= ScreenSize.y;
|
||||
else if (position.y < 0)
|
||||
position.y += ScreenSize.y;
|
||||
particle.AddAverageSpeedValue(particle.Position.DistanceTo(position));
|
||||
|
||||
if (particle.AverageSpeed < 0.3f)
|
||||
particle.Health -= HealthDelta * NegativeHealthMultiplier;
|
||||
else
|
||||
particle.Health += HealthDelta;
|
||||
|
||||
if (deletedParticle == false && particle.Health == 0f)
|
||||
{
|
||||
if (GD.Randf() < 0.2f)
|
||||
{
|
||||
LastParticlesRemoved.Add(p.Key);
|
||||
deletedParticle = true;
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
particle.Position = position;
|
||||
}
|
||||
|
||||
foreach (var id in LastParticlesRemoved)
|
||||
{
|
||||
_particles.Remove(id);
|
||||
}
|
||||
|
||||
if (_particles.Count < MaxParticles)
|
||||
{
|
||||
if (GD.Randf() < 0.2f)
|
||||
{
|
||||
CreateRandomParticle();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void RemoveParticleType(ParticleType type)
|
||||
{
|
||||
_particleTypes.Remove(type);
|
||||
foreach (var t in _particleTypes)
|
||||
{
|
||||
t.RemoveRelationship(type);
|
||||
}
|
||||
}
|
||||
|
||||
private void CreateRandomParticleType()
|
||||
{
|
||||
var type = new ParticleType()
|
||||
{
|
||||
Hue = (float) GD.RandRange(0, 1)
|
||||
};
|
||||
_particleTypes.Add(type);
|
||||
_particleTypeCounts[type] = 0;
|
||||
foreach (var type1 in _particleTypes)
|
||||
foreach(var type2 in _particleTypes)
|
||||
type1.AddRelationship(type2,
|
||||
new ParticleRelationshipProps(16, (float)GD.RandRange(20, 60),
|
||||
(float)GD.RandRange(-0.35 * 2, 0.35 * 2)));
|
||||
}
|
||||
|
||||
private void CreateRandomParticle()
|
||||
{
|
||||
var randomIndex = (int) (GD.Randi() % _particleTypes.Count);
|
||||
var type = _particleTypes[randomIndex];
|
||||
CreateParticle(type);
|
||||
}
|
||||
|
||||
private void CreateParticle(ParticleType type)
|
||||
{
|
||||
var particle = new Particle(_idCount, type)
|
||||
{
|
||||
Position = GetRandomParticlePosition(),
|
||||
Health = 1f
|
||||
};
|
||||
LastParticlesAdded.Add(_idCount);
|
||||
_particles.Add(_idCount, particle);
|
||||
_idCount++;
|
||||
_particleTypeCounts[type]++;
|
||||
}
|
||||
|
||||
private Vector2 GetRandomParticlePosition()
|
||||
{
|
||||
var position = new Vector2(
|
||||
(float) GD.RandRange(0, ScreenSize.x),
|
||||
(float) GD.RandRange(0, ScreenSize.y));
|
||||
return position;
|
||||
}
|
||||
|
||||
private Vector2 GetScreenWrapPosition(Vector2 p1, Vector2 p2)
|
||||
{
|
||||
var newPosition = p2;
|
||||
if (p2.x > (p1.x + (ScreenSize.x / 2f)))
|
||||
newPosition.x = p2.x - ScreenSize.x;
|
||||
else if (p2.x < (p1.x - (ScreenSize.x / 2f)))
|
||||
newPosition.x = p2.x + ScreenSize.x;
|
||||
if (p2.y > (p1.y + (ScreenSize.y / 2f)))
|
||||
newPosition.y = p2.y - ScreenSize.y;
|
||||
else if (p2.y < (p1.y - (ScreenSize.y / 2f)))
|
||||
newPosition.y = p2.y + ScreenSize.y;
|
||||
return newPosition;
|
||||
}
|
||||
|
||||
public Particle GetParticle(int id)
|
||||
{
|
||||
return _particles[id];
|
||||
}
|
||||
|
||||
private Task UpdateParticle(object i)
|
||||
{
|
||||
var id = (int)i;
|
||||
var particle1 = _particles[id];
|
||||
var closeCount = 0;
|
||||
foreach (var p2 in _particles)
|
||||
{
|
||||
var particle2 = p2.Value;
|
||||
if (particle1 == particle2)
|
||||
continue;
|
||||
var position = GetScreenWrapPosition(particle1.Position, particle2.Position);
|
||||
var distanceSquared = particle1.Position.DistanceSquaredTo(position);
|
||||
if (distanceSquared > (60f * 60f))
|
||||
continue;
|
||||
var direction = particle1.Position.DirectionTo(position);
|
||||
|
||||
if (distanceSquared < (70f * 70f))
|
||||
closeCount++;
|
||||
|
||||
// collision force
|
||||
float distance;
|
||||
if (distanceSquared < (16f * 16f))
|
||||
{
|
||||
distance = particle1.Position.DistanceTo(position);
|
||||
var collisionForce = 1f / (0.3f + Mathf.Pow(Mathf.E, -(distance - 11f))) - 1f / 0.3f;
|
||||
particle1.Velocity += direction * collisionForce;
|
||||
}
|
||||
|
||||
// particle relationship force
|
||||
var props = particle1.Type.GetRelationship(particle2.Type);
|
||||
if (props.Force != 0f && distanceSquared >= props.MinRadius * props.MinRadius &&
|
||||
distanceSquared <= props.MaxRadius * props.MaxRadius)
|
||||
{
|
||||
distance = particle1.Position.DistanceTo(position);
|
||||
var slope = props.Force / ((props.MaxRadius - props.MinRadius) / 2f);
|
||||
var particleForce = -slope * Mathf.Abs(distance - (props.MinRadius + props.MaxRadius) / 2f) +
|
||||
props.Force;
|
||||
particle1.Velocity += direction * particleForce;
|
||||
}
|
||||
}
|
||||
|
||||
if (closeCount < 4 || closeCount > 33)
|
||||
particle1.Health -= HealthDelta * NegativeHealthMultiplier;
|
||||
else
|
||||
particle1.Health += HealthDelta;
|
||||
|
||||
return Task.CompletedTask;
|
||||
}
|
||||
}
|
||||
}
|
||||
48
ParticleSimulation/ParticleType.cs
Normal file
48
ParticleSimulation/ParticleType.cs
Normal file
@@ -0,0 +1,48 @@
|
||||
using System.Collections.Generic;
|
||||
using Godot;
|
||||
|
||||
public struct ParticleRelationshipProps
|
||||
{
|
||||
public ParticleRelationshipProps(float minRadius, float maxRadius, float force)
|
||||
{
|
||||
MinRadius = minRadius;
|
||||
MaxRadius = maxRadius;
|
||||
Force = force;
|
||||
}
|
||||
|
||||
public float MinRadius { get; }
|
||||
public float MaxRadius { get; }
|
||||
public float Force { get; }
|
||||
}
|
||||
|
||||
public class ParticleType
|
||||
{
|
||||
private readonly Dictionary<ParticleType, ParticleRelationshipProps> _particleRelationships =
|
||||
new Dictionary<ParticleType, ParticleRelationshipProps>();
|
||||
|
||||
private float _hue;
|
||||
|
||||
public float Hue
|
||||
{
|
||||
get => _hue;
|
||||
set => _hue = Mathf.Clamp(value, 0, 1);
|
||||
}
|
||||
|
||||
public void AddRelationship(ParticleType type, ParticleRelationshipProps props)
|
||||
{
|
||||
if (_particleRelationships.ContainsKey(type))
|
||||
return;
|
||||
_particleRelationships[type] = props;
|
||||
}
|
||||
|
||||
public void RemoveRelationship(ParticleType type)
|
||||
{
|
||||
if (_particleRelationships.ContainsKey(type))
|
||||
_particleRelationships.Remove(type);
|
||||
}
|
||||
|
||||
public ParticleRelationshipProps GetRelationship(ParticleType type)
|
||||
{
|
||||
return _particleRelationships[type];
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user