WeaponOutSupport.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using Loot.Api.Graphics;
using Loot.Api.Graphics.Glowmask;
using Loot.Api.Graphics.Shader;
using Microsoft.Xna.Framework;
using Microsoft.Xna.Framework.Graphics;
using Terraria;
using Terraria.DataStructures;
using Terraria.ModLoader;

namespace Loot.ModSupport
{
    internal class WeaponOutSupport : ModSupport
    {
        public override string ModName => "WeaponOut";

        public override bool CheckValidity(Mod mod) => mod.Version >= new Version(1, 6, 4);

        public override void AddClientSupport(Mod mod)
        {
            On.Terraria.Main.DrawPlayer += MainOnDrawPlayer;
            On.Terraria.DataStructures.DrawData.Draw += DrawDataOnDraw;
            mod.Call("AddCustomPreDrawMethod", (Func<Player, Item, DrawData, bool>)CustomPreDraw);
        }

        private void MainOnDrawPlayer(On.Terraria.Main.orig_DrawPlayer orig, Main self, Player drawplayer, Vector2 position, float rotation, Vector2 rotationorigin, float shadow)
        {
            _lightColor = null;
            _cache.Clear();
            orig(self, drawplayer, position, rotation, rotationorigin, shadow);
        }

        private void DrawDataOnDraw(On.Terraria.DataStructures.DrawData.orig_Draw orig, ref DrawData self, SpriteBatch sb)
        {
            _lightColor = _lightColor ?? Lighting.GetColor((int)(Main.LocalPlayer.MountedCenter.X / 16), (int)(Main.LocalPlayer.MountedCenter.Y / 16));
            if (!DrawWeaponOutFists(self, sb) && !DrawWeaponOutFromCache(self, sb))
                orig(ref self, sb);
        }

        // Specific useStyle used for WO's fists
        private const int WEAPONOUT_FISTS_USESTYLE = 102115116;

        private bool DrawWeaponOutFists(DrawData self, SpriteBatch sb)
        {
            if (Main.LocalPlayer.HeldItem.useStyle != WEAPONOUT_FISTS_USESTYLE) return false;
            bool isHandOn = Main.LocalPlayer.handon > 0 && self.texture == Main.accHandsOnTexture[Main.LocalPlayer.handon];
            bool isHandOff = Main.LocalPlayer.handoff > 0 && self.texture == Main.accHandsOffTexture[Main.LocalPlayer.handoff];
            if (!isHandOn && !isHandOff) return false;

            var info = GraphicsGlobalItem.GetInfo(Main.LocalPlayer.HeldItem);
            if (info.HasShaders)
            {
                for (int i = 0; i < info.ShaderEntities.Count; i++)
                {
                    var entity = info.ShaderEntities[i];
                    if (entity == null) continue;
                    entity.Properties.SkipUpdatingDrawData = true;
                    entity.DrawData = self;
                    entity.DoDrawLayeredEntity(sb, _lightColor.Value, self.color, self.scale.LengthSquared(), self.rotation,
                        info.GlowmaskEntities[i], self.texture, self.texture, self.texture);
                }

                return true;
            }

            foreach (var entity in info.GlowmaskEntities.Where(x => x != null))
            {
                entity.Properties.SkipUpdatingDrawData = true;
                entity.DrawData = self;
                entity.DoDrawGlowmask(sb, _lightColor.Value, self.color, self.rotation, self.scale.LengthSquared(), entity.Entity.whoAmI, self.texture);
            }

            return info.HasGlowmasks;
        }

        private bool DrawWeaponOutFromCache(DrawData self, SpriteBatch sb)
        {
            var data = self;
            var cachedDatas = _cache.Where(x => x.drawData.texture == data.texture).ToList();
            if (cachedDatas.Any())
            {
                foreach (var cachedData in cachedDatas)
                {
                    var shaderEntity = cachedData.shaderEntity;
                    var glowmaskEntity = cachedData.glowmaskEntity;
                    var plr = cachedData.player;
                    _lightColor = _lightColor ?? Lighting.GetColor((int)(plr.MountedCenter.X / 16), (int)(plr.MountedCenter.Y / 16));

                    if (shaderEntity != null)
                    {
                        shaderEntity.Properties.SkipUpdatingDrawData = true;
                        shaderEntity.DrawData = self;
                        shaderEntity.DoDrawLayeredEntity(sb, _lightColor.Value, self.color, self.scale.LengthSquared(), self.rotation, glowmaskEntity);
                    }
                    else if (glowmaskEntity != null)
                    {
                        glowmaskEntity.Properties.SkipUpdatingDrawData = true;
                        glowmaskEntity.DrawData = self;
                        glowmaskEntity.DoDrawGlowmask(sb, _lightColor.Value, self.color, self.rotation, self.scale.LengthSquared(), glowmaskEntity.Entity.whoAmI);
                    }
                }

                return true;
            }

            return false;
        }

        private Color? _lightColor;
        private readonly List<(DrawData drawData, Player player, ShaderEntity shaderEntity, GlowmaskEntity glowmaskEntity)> _cache
            = new List<(DrawData, Player, ShaderEntity, GlowmaskEntity)>();

        private bool CustomPreDraw(Player player, Item item, DrawData drawDrata)
        {
            var info = GraphicsGlobalItem.GetInfo(item);
            if (info.HasShaders)
            {
                for (int i = 0; i < info.ShaderEntities.Count; i++)
                {
                    ShaderEntity entity = info.ShaderEntities[i];
                    if (entity != null)
                    {
                        _cache.Add((drawDrata, player, entity, info.GlowmaskEntities[i]));
                    }
                }
            }
            else if (info.HasGlowmasks)
            {
                foreach (var entity in info.GlowmaskEntities.Where(x => x != null))
                {
                    _cache.Add((drawDrata, player, null, entity));
                }
            }
            return true;
        }
    }
}