mirror of
https://gitgud.io/AbstractConcept/rimworld-animations-patch.git
synced 2024-08-15 00:43:27 +00:00
207 lines
9.1 KiB
C#
207 lines
9.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Verse;
|
|
using RimWorld;
|
|
using AlienRace;
|
|
using Rimworld_Animations;
|
|
using UnityEngine;
|
|
using HarmonyLib;
|
|
|
|
namespace Rimworld_Animations_Patch
|
|
{
|
|
public class BodyAddonData
|
|
{
|
|
public AlienPartGenerator.BodyAddon bodyAddon;
|
|
public BodyPartRecord bodyPartRecord;
|
|
public List<Vector3> bodyAddonOffsets = new List<Vector3>();
|
|
public bool alignsWithHead = false;
|
|
|
|
private Pawn pawn;
|
|
private string bodyType;
|
|
private PawnRenderFlags renderFlags;
|
|
private bool canDraw = false;
|
|
private bool bodyPartMissing = false;
|
|
|
|
public BodyAddonData(Pawn pawn, AlienPartGenerator.BodyAddon bodyAddon, bool isPortrait = false)
|
|
{
|
|
this.pawn = pawn;
|
|
this.bodyAddon = bodyAddon;
|
|
|
|
if (isPortrait)
|
|
{ renderFlags |= PawnRenderFlags.Portrait; }
|
|
|
|
bodyPartRecord = pawn.def?.race?.body?.AllParts?.FirstOrDefault(x => x.def == bodyAddon?.bodyPart);
|
|
alignsWithHead = bodyAddon.alignWithHead || (bodyPartRecord != null && bodyPartRecord.IsInGroup(BodyPartGroupDefOf.FullHead));
|
|
|
|
GenerateOffsets();
|
|
UpdateVisibility();
|
|
}
|
|
|
|
public void GenerateOffsets()
|
|
{
|
|
bodyType = pawn.story.bodyType.defName;
|
|
bodyAddonOffsets.Clear();
|
|
|
|
int bodyAddonIndex = (pawn.def as ThingDef_AlienRace).alienRace.generalSettings.alienPartGenerator.bodyAddons.IndexOf(bodyAddon);
|
|
AlienPartGenerator.AlienComp alienComp = pawn.GetComp<AlienPartGenerator.AlienComp>();
|
|
Graphic addonGraphic = alienComp.addonGraphics[bodyAddonIndex];
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
{
|
|
Rot4 apparentRotation = new Rot4(i);
|
|
|
|
// Get basic offset for body addon
|
|
AlienPartGenerator.RotationOffset defaultOffsets = bodyAddon.defaultOffsets.GetOffset(apparentRotation);
|
|
Vector3 bodyTypeOffset = (defaultOffsets != null) ? defaultOffsets.GetOffset(renderFlags.FlagSet(PawnRenderFlags.Portrait), pawn.story.bodyType, pawn.story.headType) : Vector3.zero;
|
|
|
|
AlienPartGenerator.RotationOffset rotationOffsets = bodyAddon.offsets.GetOffset(apparentRotation);
|
|
Vector3 bodyAddonOffset = bodyTypeOffset + ((rotationOffsets != null) ? rotationOffsets.GetOffset(renderFlags.FlagSet(PawnRenderFlags.Portrait), pawn.story.bodyType, pawn.story.headType) : Vector3.zero);
|
|
|
|
// Offset private parts so that they render over tattoos but under apparel (rendering under tatoos looks weird)
|
|
if (bodyAddon.bodyPart == PatchBodyPartDefOf.Genitals || bodyAddon.bodyPart == PatchBodyPartDefOf.Chest || bodyAddon.bodyPart == PatchBodyPartDefOf.Anus || addonGraphic.path.Contains("belly") || addonGraphic.path.Contains("Belly"))
|
|
{
|
|
bodyAddonOffset.y = (bodyAddonOffset.y + 0.40f) / 1000f + 0.012f;
|
|
|
|
// Erected penises should be drawn over apparel
|
|
if (pawn.RaceProps.Humanlike &&
|
|
(addonGraphic.path.Contains("penis") || addonGraphic.path.Contains("Penis")) &&
|
|
(addonGraphic.path.Contains("flaccid") == false && addonGraphic.path.Contains("Flaccid") == false) &&
|
|
apparentRotation == Rot4.South)
|
|
{ bodyAddonOffset.y += 0.010f; }
|
|
}
|
|
|
|
// Otherwise use the standard offsets
|
|
else
|
|
{ bodyAddonOffset.y = 0.3f + bodyAddonOffset.y; }
|
|
|
|
// Draw addons infront of body
|
|
if (!bodyAddon.inFrontOfBody)
|
|
{ bodyAddonOffset.y *= -1f; }
|
|
|
|
// Adjust for facing
|
|
if (apparentRotation == Rot4.North)
|
|
{
|
|
if (bodyAddon.layerInvert)
|
|
{ bodyAddonOffset.y = -bodyAddonOffset.y; }
|
|
}
|
|
|
|
if (apparentRotation == Rot4.East)
|
|
{ bodyAddonOffset.x = -bodyAddonOffset.x; }
|
|
|
|
// Adjustment for body addons attached to the head that are not marked as such
|
|
if (alignsWithHead && bodyAddon.alignWithHead == false)
|
|
{ bodyAddonOffset -= pawn.Drawer.renderer.BaseHeadOffsetAt(apparentRotation); }
|
|
|
|
// Done
|
|
bodyAddonOffsets.Add(bodyAddonOffset);
|
|
}
|
|
}
|
|
|
|
public Vector3 GetOffset(Rot4 facing)
|
|
{
|
|
if (pawn.story.bodyType.defName != bodyType)
|
|
{ GenerateOffsets(); }
|
|
|
|
return bodyAddonOffsets[facing.AsInt];
|
|
}
|
|
|
|
public bool CanDraw()
|
|
{
|
|
return bodyPartMissing == false && SitutationalVisibiltyCheck() && (pawn.Drawer.renderer.graphics.apparelGraphics.Any() == false || canDraw);
|
|
}
|
|
|
|
public bool SitutationalVisibiltyCheck()
|
|
{
|
|
if (pawn == null || bodyAddon == null) return false;
|
|
|
|
if (renderFlags.FlagSet(PawnRenderFlags.Portrait) == false && pawn.CurrentBed()?.def.building.bed_showSleeperBody == false && bodyAddon.drawnInBed == false)
|
|
{ return false; }
|
|
|
|
if (renderFlags.FlagSet(PawnRenderFlags.Portrait) == false && (pawn.GetPosture() == PawnPosture.LayingOnGroundNormal || pawn.GetPosture() == PawnPosture.LayingOnGroundFaceUp) && bodyAddon.drawnOnGround == false)
|
|
{ return false; }
|
|
|
|
if (bodyAddon.backstoryRequirement != null && pawn.story?.AllBackstories?.Any((BackstoryDef x) => x == bodyAddon.backstoryRequirement) == false)
|
|
{ return false; }
|
|
|
|
if (bodyAddon.drawnDesiccated == false && pawn.Corpse?.GetRotStage() == RotStage.Dessicated)
|
|
{ return false; }
|
|
|
|
if (pawn.gender == Gender.Female && bodyAddon.drawForFemale == false || pawn.gender == Gender.Male && bodyAddon.drawForMale == false)
|
|
{ return false; }
|
|
|
|
if (bodyAddon.bodyTypeRequirement != null && pawn.story?.bodyType != bodyAddon.bodyTypeRequirement)
|
|
{ return false; }
|
|
|
|
return true;
|
|
}
|
|
|
|
public void UpdateVisibility()
|
|
{
|
|
if (pawn == null || bodyAddon == null || bodyPartRecord == null) return;
|
|
canDraw = true;
|
|
|
|
if (pawn.health?.hediffSet?.GetNotMissingParts()?.Contains(bodyPartRecord) == false)
|
|
{ bodyPartMissing = true; return; }
|
|
|
|
if (pawn?.apparel?.WornApparel == null || pawn.apparel.WornApparel.NullOrEmpty())
|
|
{ return; }
|
|
|
|
foreach (Apparel apparel in pawn.apparel.WornApparel)
|
|
{
|
|
CompApparelVisibility comp = apparel?.TryGetComp<CompApparelVisibility>();
|
|
if (comp == null) continue;
|
|
LoadRimNudeData(comp);
|
|
|
|
if (comp.isBeingWorn == false) continue;
|
|
|
|
if (bodyAddon.bodyPart == PatchBodyPartDefOf.Genitals ||
|
|
bodyAddon.bodyPart == PatchBodyPartDefOf.Anus ||
|
|
bodyAddon.bodyPart == PatchBodyPartDefOf.Chest ||
|
|
bodyAddon.hediffGraphics?.Any(x => x.path.NullOrEmpty() == false && (x.path.Contains("belly") || x.path.Contains("Belly"))) == true)
|
|
{
|
|
if ((bodyAddon.bodyPart == PatchBodyPartDefOf.Genitals || bodyAddon.bodyPart == PatchBodyPartDefOf.Anus) && comp.coversGroin)
|
|
{ canDraw = false; return; };
|
|
|
|
if (bodyAddon.bodyPart == PatchBodyPartDefOf.Chest && comp.coversChest)
|
|
{ canDraw = false; return; };
|
|
|
|
if (bodyAddon.hediffGraphics?.Any(x => x.path.NullOrEmpty() == false && (x.path.Contains("belly") || x.path.Contains("Belly"))) == true && comp.coversBelly)
|
|
{ canDraw = false; return; }
|
|
}
|
|
|
|
else
|
|
{
|
|
if (bodyAddon.hiddenUnderApparelFor?.Any(x => apparel?.def.apparel?.hatRenderedFrontOfFace == false && apparel.def.apparel?.bodyPartGroups?.Contains(x) == true) == true)
|
|
{ canDraw = false; return; };
|
|
|
|
if (bodyAddon.hiddenUnderApparelTag?.Any(x => apparel?.def.apparel?.hatRenderedFrontOfFace == false && apparel.def.apparel?.tags?.Contains(x) == true) == true)
|
|
{ canDraw = false; return; };
|
|
}
|
|
}
|
|
}
|
|
|
|
public void LoadRimNudeData(CompApparelVisibility comp)
|
|
{
|
|
if (comp?.rimNudeDataStatus == RimNudeDataStatus.Unavailable)
|
|
{ return; }
|
|
|
|
if (comp?.rimNudeDataStatus == RimNudeDataStatus.NotLoaded)
|
|
{
|
|
RimNudeData rimNudeData = ApparelSettings.GetRimNudeData(comp?.apparel);
|
|
|
|
if (rimNudeData == null)
|
|
{
|
|
comp.rimNudeDataStatus = RimNudeDataStatus.Unavailable;
|
|
return;
|
|
}
|
|
|
|
comp.coversBelly = rimNudeData.coversBelly;
|
|
comp.coversChest = rimNudeData.coversChest;
|
|
comp.coversGroin = rimNudeData.coversGroin;
|
|
|
|
comp.rimNudeDataStatus = RimNudeDataStatus.Loaded;
|
|
}
|
|
}
|
|
}
|
|
}
|