Include NPC models in Shared Models list.

Author: Lunaretic

xivModdingFramework/Items/Categories/Gear.cs

@@ -736,6 +736,46 @@ public async Task<List<IItemModel>> GetSameModelList(IItemModel item)
                     && it.ModelInfo.SecondaryID == item.ModelInfo.SecondaryID
                     && it.SecondaryCategory == item.SecondaryCategory).Select(it => it as IItemModel).ToList()
                 );
+
+                // We don't really care which sub-model gets returned here, we just need a path to pull the data file off of.
+                var imc = new Imc(_gameDirectory, item.DataFile);
+
+                // Language doesn't actually matter for the tests we're doing here.
+                var info = await imc.GetFullImcInfo(item);
+                var slot = item.GetItemSlotAbbreviation();
+                var imcEntries = info.GetAllEntries(slot);
+
+                // Need to verify all of our IMC sets are properly represented in the item list.
+                for (int i = 0; i < imcEntries.Count; i++)
+                {
+                    // Already in it.  All set.
+                    if (sameModelItems.Any(x => x.ModelInfo.ImcSubsetID == i)) continue;
+
+                    // Need to create a new item for it.
+                    var npcItem = new XivGear()
+                    {
+                        Name = "NPC Equipment e" + item.ModelInfo.PrimaryID + " " + slot + " IMC Variant " + i,
+                        ModelInfo = new XivGearModelInfo(),
+                        PrimaryCategory = item.PrimaryCategory,
+                        SecondaryCategory = item.SecondaryCategory,
+                        TertiaryCategory = item.TertiaryCategory,
+                        DataFile = item.DataFile,
+                    };
+                    npcItem.ModelInfo.PrimaryID = item.ModelInfo.PrimaryID;
+                    npcItem.ModelInfo.SecondaryID = item.ModelInfo.SecondaryID;
+                    npcItem.ModelInfo.ImcSubsetID = i;
+                    try
+                    {
+                        ((XivGearModelInfo)npcItem.ModelInfo).IsWeapon = ((XivGearModelInfo)item.ModelInfo).IsWeapon;
+                    }
+                    catch
+                    {
+                        // No-op.  Should never actually get here, but safety check on the cast.
+                    }
+
+                    sameModelItems.Add(npcItem);
+                }
+
             }
             else
             {

xivModdingFramework/Models/FileTypes/Mdl.cs

@@ -372,42 +372,6 @@ private async Task ExportMaterialsForModel(TTModel model, string outputFilePath,
             }
         }
 
-        /// <summary>
-        /// Retrieves all items that share the same model.
-        /// </summary>
-        /// <param name="item"></param>
-        /// <param name="language"></param>
-        /// <returns></returns>
-        public async Task<List<IItemModel>> GetSameModelList(IItemModel item, XivLanguage language = XivLanguage.English)
-        {
-            var sameModelItems = new List<IItemModel>();
-            var gear = new Gear(_gameDirectory, language);
-            var character = new Character(_gameDirectory, language);
-            var companions = new Companions(_gameDirectory, language);
-            var ui = new UI(_gameDirectory, language);
-            var housing = new Housing(_gameDirectory, language);
-
-            if (item.PrimaryCategory.Equals(XivStrings.Gear))
-            {
-
-                // Scan the gear list for anything using the same model ID and slot.
-                sameModelItems.AddRange(
-                    (await gear.GetGearList())
-                    .Where(it =>
-                    it.ModelInfo.PrimaryID == item.ModelInfo.PrimaryID
-                    && it.SecondaryCategory == item.SecondaryCategory).Select(it => it as IItemModel).ToList()
-                );
-            }
-            else if (item.PrimaryCategory.Equals(XivStrings.Character))
-            {
-
-                // Character models are assumed to have no shared models,
-                // So return a copy of the original item.
-                sameModelItems.Add((IItemModel)item.Clone());
-            }
-            return sameModelItems;
-        }
-
 
         /// <summary>
         /// Retreives the high level TTModel representation of an underlying MDL file.

xivModdingFramework/Variants/FileTypes/Imc.cs

@@ -392,6 +392,16 @@ public int SubsetSize
             /// </summary>
             public List<XivImc> DefaultSubset { get; set; }
 
+            // Gets all (non-default) IMC entries for a given slot.
+            public List<XivImc> GetAllEntries(string slot = "")
+            {
+                var ret = new List<XivImc>(SubsetList.Count);
+                for(int i = 0; i < SubsetList.Count; i++)
+                {
+                    ret.Add(GetEntry(i, slot));
+                }
+                return ret;
+            }
 
             /// <summary>
             /// Retrieve a given IMC info. Negative values retrieve the default set.