Plant genetics (#11407)

This commit is contained in:
Kevin Zheng
2022-10-15 23:25:41 -07:00
committed by GitHub
parent cb2f3a058b
commit 7fc357afd2
14 changed files with 519 additions and 25 deletions

View File

@@ -0,0 +1,142 @@
using System.Threading;
using Content.Server.Botany.Components;
using Content.Server.DoAfter;
using Content.Server.Hands.Components;
using Content.Server.Nutrition.EntitySystems;
using Content.Server.Popups;
using Content.Shared.Examine;
using Content.Shared.Interaction;
using Content.Shared.Inventory;
using Content.Shared.Tools.Components;
using Robust.Shared.Audio;
using Robust.Shared.Player;
using Robust.Shared.Random;
using Robust.Shared.Utility;
namespace Content.Server.Botany
{
public sealed class BotanySwabSystem : EntitySystem
{
[Dependency] private readonly DoAfterSystem _doAfterSystem = default!;
[Dependency] private readonly PopupSystem _popupSystem = default!;
[Dependency] private readonly MutationSystem _mutationSystem = default!;
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<BotanySwabComponent, AfterInteractEvent>(OnAfterInteract);
SubscribeLocalEvent<BotanySwabComponent, ExaminedEvent>(OnExamined);
// Private Events
SubscribeLocalEvent<TargetSwabSuccessfulEvent>(OnTargetSwabSuccessful);
SubscribeLocalEvent<SwabCancelledEvent>(OnSwabCancelled);
}
/// <summary>
/// Handles swabbing a plant.
/// </summary>
private void OnAfterInteract(EntityUid uid, BotanySwabComponent swab, AfterInteractEvent args)
{
if (swab.CancelToken != null)
{
swab.CancelToken.Cancel();
swab.CancelToken = null;
return;
}
if (args.Target == null || !args.CanReach)
return;
if (!TryComp<PlantHolderComponent>(args.Target, out var plant))
return;
swab.CancelToken = new CancellationTokenSource();
_doAfterSystem.DoAfter(new DoAfterEventArgs(args.User, swab.SwabDelay, swab.CancelToken.Token, target: args.Target)
{
BroadcastFinishedEvent = new TargetSwabSuccessfulEvent(args.User, args.Target, swab, plant),
BroadcastCancelledEvent = new SwabCancelledEvent(swab),
BreakOnTargetMove = true,
BreakOnUserMove = true,
BreakOnStun = true,
NeedHand = true
});
}
/// <summary>
/// This handles swab examination text
/// so you can tell if they are used or not.
/// </summary>
private void OnExamined(EntityUid uid, BotanySwabComponent swab, ExaminedEvent args)
{
if (args.IsInDetailsRange)
{
if (swab.SeedData != null)
args.PushMarkup(Loc.GetString("swab-used"));
else
args.PushMarkup(Loc.GetString("swab-unused"));
}
}
/// <summary>
/// Save seed data or cross-pollenate.
/// </summary>
private void OnTargetSwabSuccessful(TargetSwabSuccessfulEvent args)
{
if (args.Target == null)
return;
if (args.Swab.SeedData == null)
{
// Pick up pollen
args.Swab.SeedData = args.Plant.Seed;
_popupSystem.PopupEntity(Loc.GetString("botany-swab-from"), args.Target.Value, Filter.Entities(args.User));
}
else
{
var old = args.Plant.Seed; // Save old plant pollen
if (old == null)
return;
args.Plant.Seed = _mutationSystem.Cross(args.Swab.SeedData, old); // Cross-pollenate
args.Swab.SeedData = old; // Transfer old plant pollen to swab
_popupSystem.PopupEntity(Loc.GetString("botany-swab-to"), args.Target.Value, Filter.Entities(args.User));
}
if (args.Swab.CancelToken != null)
{
args.Swab.CancelToken.Cancel();
args.Swab.CancelToken = null;
}
}
private static void OnSwabCancelled(SwabCancelledEvent args)
{
args.Swab.CancelToken = null;
}
private sealed class SwabCancelledEvent : EntityEventArgs
{
public readonly BotanySwabComponent Swab;
public SwabCancelledEvent(BotanySwabComponent swab)
{
Swab = swab;
}
}
private sealed class TargetSwabSuccessfulEvent : EntityEventArgs
{
public EntityUid User { get; }
public EntityUid? Target { get; }
public BotanySwabComponent Swab { get; }
public PlantHolderComponent Plant { get; }
public TargetSwabSuccessfulEvent(EntityUid user, EntityUid? target, BotanySwabComponent swab, PlantHolderComponent plant)
{
User = user;
Target = target;
Swab = swab;
Plant = plant;
}
}
}
}

View File

@@ -1,11 +1,14 @@
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using Content.Server.Botany.Components;
using Content.Server.Mind.Commands;
using Content.Server.Kitchen.Components;
using Content.Shared.Botany;
using Content.Shared.Examine;
using Content.Shared.Popups;
using Content.Shared.Random.Helpers;
using Content.Shared.Slippery;
using Content.Shared.StepTrigger.Components;
using Robust.Server.GameObjects;
using Robust.Shared.Map;
using Robust.Shared.Player;
@@ -158,6 +161,20 @@ public sealed partial class BotanySystem
metaData.EntityName += "?";
metaData.EntityDescription += " " + Loc.GetString("botany-mysterious-description-addon");
}
if (proto.Bioluminescent)
{
var light = EnsureComp<PointLightComponent>(entity);
light.Radius = proto.BioluminescentRadius;
light.Color = proto.BioluminescentColor;
}
if (proto.Slip)
{
var slippery = EnsureComp<SlipperyComponent>(entity);
EntityManager.Dirty(slippery);
EnsureComp<StepTriggerComponent>(entity);
}
}
return products;

View File

@@ -0,0 +1,211 @@
using Robust.Shared.Random;
namespace Content.Server.Botany;
public class MutationSystem : EntitySystem
{
[Dependency] private readonly IRobustRandom _robustRandom = default!;
/// <summary>
// Main idea: Simulate genetic mutation using random binary flips. Each
// seed attribute can be encoded with a variable number of bits, e.g.
// NutrientConsumption is represented by 5 bits randomly distributed in the
// plant's genome which thermometer code the floating value between 0.1 and
// 5. 1 unit of mutation flips one bit in the plant's genome, which changes
// NutrientConsumption if one of those 5 bits gets affected.
//
// You MUST clone() seed before mutating it!
/// </summary>
public void MutateSeed(SeedData seed, float severity)
{
// Add up everything in the bits column and put the number here.
const int totalbits = 215;
// Tolerances (55)
MutateFloat(ref seed.NutrientConsumption , 0.05f , 1.2f , 5 , totalbits , severity);
MutateFloat(ref seed.WaterConsumption , 3f , 9f , 5 , totalbits , severity);
MutateFloat(ref seed.IdealHeat , 263f , 323f , 5 , totalbits , severity);
MutateFloat(ref seed.HeatTolerance , 2f , 25f , 5 , totalbits , severity);
MutateFloat(ref seed.IdealLight , 0f , 14f , 5 , totalbits , severity);
MutateFloat(ref seed.LightTolerance , 1f , 5f , 5 , totalbits , severity);
MutateFloat(ref seed.ToxinsTolerance , 1f , 10f , 5 , totalbits , severity);
MutateFloat(ref seed.LowPressureTolerance , 60f , 100f , 5 , totalbits , severity);
MutateFloat(ref seed.HighPressureTolerance , 100f , 140f , 5 , totalbits , severity);
MutateFloat(ref seed.PestTolerance , 0f , 15f , 5 , totalbits , severity);
MutateFloat(ref seed.WeedTolerance , 0f , 15f , 5 , totalbits , severity);
// Stats (30*2 = 60)
MutateFloat(ref seed.Endurance , 50f , 150f , 5 , totalbits , 2*severity);
MutateInt(ref seed.Yield , 3 , 10 , 5 , totalbits , 2*severity);
MutateFloat(ref seed.Lifespan , 10f , 80f , 5 , totalbits , 2*severity);
MutateFloat(ref seed.Maturation , 3f , 8f , 5 , totalbits , 2*severity);
MutateFloat(ref seed.Production , 1f , 10f , 5 , totalbits , 2*severity);
MutateFloat(ref seed.Potency , 30f , 100f , 5 , totalbits , 2*severity);
// Kill the plant (30)
MutateBool(ref seed.Viable , false , 30 , totalbits , severity);
// Fun (70)
MutateBool(ref seed.Seedless , true , 10 , totalbits , severity);
MutateBool(ref seed.Slip , true , 10 , totalbits , severity);
MutateBool(ref seed.Sentient , true , 10 , totalbits , severity);
MutateBool(ref seed.Ligneous , true , 10 , totalbits , severity);
MutateBool(ref seed.Bioluminescent , true , 10 , totalbits , severity);
seed.BioluminescentColor = RandomColor(seed.BioluminescentColor, 10, totalbits, severity);
}
public SeedData Cross(SeedData a, SeedData b)
{
SeedData result = b.Clone();
result.Chemicals = random(0.5f) ? a.Chemicals : result.Chemicals;
CrossFloat(ref result.NutrientConsumption, a.NutrientConsumption);
CrossFloat(ref result.WaterConsumption, a.WaterConsumption);
CrossFloat(ref result.IdealHeat, a.IdealHeat);
CrossFloat(ref result.HeatTolerance, a.HeatTolerance);
CrossFloat(ref result.IdealLight, a.IdealLight);
CrossFloat(ref result.LightTolerance, a.LightTolerance);
CrossFloat(ref result.ToxinsTolerance, a.ToxinsTolerance);
CrossFloat(ref result.LowPressureTolerance, a.LowPressureTolerance);
CrossFloat(ref result.HighPressureTolerance, a.HighPressureTolerance);
CrossFloat(ref result.PestTolerance, a.PestTolerance);
CrossFloat(ref result.WeedTolerance, a.WeedTolerance);
CrossFloat(ref result.Endurance, a.Endurance);
CrossInt(ref result.Yield, a.Yield);
CrossFloat(ref result.Lifespan, a.Lifespan);
CrossFloat(ref result.Maturation, a.Maturation);
CrossFloat(ref result.Production, a.Production);
CrossFloat(ref result.Potency, a.Potency);
CrossBool(ref result.Seedless, a.Seedless);
CrossBool(ref result.Viable, a.Viable);
CrossBool(ref result.Slip, a.Slip);
CrossBool(ref result.Sentient, a.Sentient);
CrossBool(ref result.Ligneous, a.Ligneous);
CrossBool(ref result.Bioluminescent, a.Bioluminescent);
result.BioluminescentColor = random(0.5f) ? a.BioluminescentColor : result.BioluminescentColor;
// Hybrids have a high chance of being seedless. Balances very
// effective hybrid crossings.
if (a.Name == result.Name && random(0.7f))
{
result.Seedless = true;
}
return result;
}
// Mutate reference 'val' between 'min' and 'max' by pretending the value
// is representable by a thermometer code with 'bits' number of bits and
// randomly flipping some of them.
//
// 'totalbits' and 'mult' are used only to calculate the probability that
// one bit gets flipped.
private void MutateFloat(ref float val, float min, float max, int bits, int totalbits, float mult)
{
// Probability that a bit flip happens for this value.
float p = mult*bits/totalbits;
if (!random(p))
{
return;
}
// Starting number of bits that are high, between 0 and n.
int n = (int)Math.Round((val - min) / (max - min) * bits);
// Probability that the bit flip increases n.
float p_increase = 1-(float)n/bits;
int np;
if (random(p_increase))
{
np = n + 1;
}
else
{
np = n - 1;
}
// Set value based on mutated thermometer code.
float nval = MathF.Min(MathF.Max((float)np/bits * (max - min) + min, min), max);
val = nval;
}
private void MutateInt(ref int n, int min, int max, int bits, int totalbits, float mult)
{
// Probability that a bit flip happens for this value.
float p = mult*bits/totalbits;
if (!random(p))
{
return;
}
// Probability that the bit flip increases n.
float p_increase = 1-(float)n/bits;
int np;
if (random(p_increase))
{
np = n + 1;
}
else
{
np = n - 1;
}
np = Math.Min(Math.Max(np, min), max);
n = np;
}
private void MutateBool(ref bool val, bool polarity, int bits, int totalbits, float mult)
{
// Probability that a bit flip happens for this value.
float p = mult*bits/totalbits;
if (!random(p))
{
return;
}
val = polarity;
}
private Color RandomColor(Color color, int bits, int totalbits, float mult)
{
float p = mult*bits/totalbits;
if (random(p))
{
var colors = new List<Color>{
Color.White,
Color.Red,
Color.Yellow,
Color.Green,
Color.Blue,
Color.Purple,
Color.Pink
};
var rng = IoCManager.Resolve<IRobustRandom>();
return rng.Pick(colors);
}
return color;
}
private void CrossFloat(ref float val, float other)
{
val = random(0.5f) ? val : other;
}
private void CrossInt(ref int val, int other)
{
val = random(0.5f) ? val : other;
}
private void CrossBool(ref bool val, bool other)
{
val = random(0.5f) ? val : other;
}
private bool random(float p)
{
return _robustRandom.Prob(p);
}
}

View File

@@ -28,8 +28,12 @@ public sealed class SeedExtractorSystem : EntitySystem
return;
if (!TryComp(args.Used, out ProduceComponent? produce)) return;
if (!_botanySystem.TryGetSeed(produce, out var seed))
if (!_botanySystem.TryGetSeed(produce, out var seed) || seed.Seedless)
{
_popupSystem.PopupCursor(Loc.GetString("seed-extractor-component-no-seeds",("name", args.Used)),
Filter.Entities(args.User), PopupType.MediumCaution);
return;
}
_popupSystem.PopupCursor(Loc.GetString("seed-extractor-component-interact-message",("name", args.Used)),
Filter.Entities(args.User), PopupType.Medium);