Extract monsters and NPCs into a spawns XML file (jo3bingham#42)

jo3bingham · web-flow · commit bb938951ad36 · 2020-08-28T12:21:54.000-04:00
diff --git a/Apps/Extract/Program.cs b/Apps/Extract/Program.cs
@@ -20,6 +20,7 @@ class Program
         private static readonly HashSet<ulong> _knownPositions = new HashSet<ulong>();
         private static readonly HashSet<uint> _knownMonsterIds = new HashSet<uint>();
         private static readonly HashSet<uint> _knownNpcIds = new HashSet<uint>();
+        private static readonly HashSet<uint> _knownSpawnIds = new HashSet<uint>();
 
         private static readonly HashSet<uint> _ignoreIds = new HashSet<uint>();
 
@@ -37,6 +38,8 @@ class Program
         private static StreamWriter _monsterFile;
         private static StreamWriter _npcFile;
 
+        private static XmlWriter _spawnsXml;
+
         private static Logger.LogLevel _logLevel = Logger.LogLevel.Error;
 
         private static Logger.LogOutput _logOutput = Logger.LogOutput.Console;
@@ -54,6 +57,7 @@ class Program
         private static bool _extractMapData = false;
         private static bool _extractMonsterData = false;
         private static bool _extractNpcData = false;
+        private static bool _extractSpawns = false;
 
         static bool ParseArgs(string[] args)
         {
@@ -89,6 +93,11 @@ static bool ParseArgs(string[] args)
                                 _extractNpcData = true;
                             }
                             break;
+                        case "--spawns":
+                            {
+                                _extractSpawns = true;
+                            }
+                            break;
                         case "-h":
                         case "--help":
                             {
@@ -124,6 +133,7 @@ static bool ParseArgs(string[] args)
                                 Console.WriteLine("--map: Used for extracting map data to the OTBM format.\n");
                                 Console.WriteLine("--monsters: Used for extracting monster information to monsters.txt.\n");
                                 Console.WriteLine("--npcs: Used for extracting npc information to npcs.txt.\n");
+                                Console.WriteLine("--spawns: Used for extracting spawns to spawns.xml.\n");
                             }
                             return false;
                         default:
@@ -208,7 +218,8 @@ static void Main(string[] args)
                     return;
                 }
 
-                if (!_extractItemData && !_extractLookItemText && !_extractMapData && !_extractMonsterData && !_extractNpcData)
+                if (!_extractItemData && !_extractLookItemText && !_extractMapData &&
+                    !_extractMonsterData && !_extractNpcData && !_extractSpawns)
                 {
                     Console.WriteLine("You must specificy at least one extraction option.");
                     Console.WriteLine("Use -h, or --help, for help.");
@@ -288,6 +299,18 @@ static void Main(string[] args)
                     _npcFile = new StreamWriter("npcs.txt", true);
                 }
 
+                if (_extractSpawns)
+                {
+                    Console.WriteLine("Extracting spawns...");
+                    var settings = new XmlWriterSettings
+                    {
+                        Indent = true
+                    };
+                    _spawnsXml = XmlWriter.Create("spawns.xml", settings);
+                    _spawnsXml.WriteStartDocument();
+                    _spawnsXml.WriteStartElement("spawns");
+                }
+
                 ExtractRecordings(filenames);
 
                 if (_itemFile != null)
@@ -310,6 +333,12 @@ static void Main(string[] args)
                     _npcFile.Close();
                 }
 
+                if (_spawnsXml != null)
+                {
+                    _spawnsXml.WriteEndDocument();
+                    _spawnsXml.Close();
+                }
+
                 Console.WriteLine("Extraction complete");
             }
             catch (Exception ex)
@@ -430,6 +459,32 @@ static void LoadXML(string filename)
             }
         }
 
+        private static void AddSpawnEntry(OXGaming.TibiaAPI.Creatures.Creature creature)
+        {
+            if ((creature.Type != OXGaming.TibiaAPI.Constants.CreatureType.Monster &&
+                creature.Type != OXGaming.TibiaAPI.Constants.CreatureType.Npc) ||
+                _knownSpawnIds.Contains(creature.Id))
+            {
+                return;
+            }
+
+            _knownSpawnIds.Add(creature.Id);
+
+            _spawnsXml.WriteStartElement("spawn");
+            _spawnsXml.WriteAttributeString("centerx", creature.Position.X.ToString());
+            _spawnsXml.WriteAttributeString("centery", creature.Position.Y.ToString());
+            _spawnsXml.WriteAttributeString("centerz", creature.Position.Z.ToString());
+            _spawnsXml.WriteAttributeString("radius", "5");
+            _spawnsXml.WriteStartElement(creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster ? "monster" : "npc");
+            _spawnsXml.WriteAttributeString("name", creature.Name);
+            _spawnsXml.WriteAttributeString("x", "0");
+            _spawnsXml.WriteAttributeString("y", "0");
+            _spawnsXml.WriteAttributeString("z", "0");
+            _spawnsXml.WriteAttributeString("spawntime", "60");
+            _spawnsXml.WriteEndElement();
+            _spawnsXml.WriteEndElement();
+        }
+
         private static void ExtractRecordings(List<string> filenames)
         {
             Console.WriteLine($"Extracting data from {filenames.Count} recording(s)...");
@@ -685,6 +740,11 @@ private static bool Connection_OnReceivedServerFullMapPacket(Packet packet)
                                 continue;
                             }
 
+                            if (_extractSpawns)
+                            {
+                                AddSpawnEntry(creature);
+                            }
+
                             if (_extractMonsterData && creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster && !_knownMonsterIds.Contains(creature.Id))
                             {
                                 _monsterFile.WriteLine($"{creature.Name} {creature.Position}");
@@ -707,6 +767,11 @@ private static bool Connection_OnReceivedServerCreatureDataPacket(Packet packet)
             var p = (CreatureData)packet;
             if (p.Creature != null)
             {
+                if (_extractSpawns)
+                {
+                    AddSpawnEntry(p.Creature);
+                }
+
                 if (_extractMonsterData && p.Creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster && !_knownMonsterIds.Contains(p.Creature.Id))
                 {
                     _monsterFile.WriteLine($"{p.Creature.Name} {p.Creature.Position}");
@@ -730,6 +795,11 @@ private static bool Connection_OnReceivedServerChangeOnMapPacket(Packet packet)
             }
             else if (p.Creature != null)
             {
+                if (_extractSpawns)
+                {
+                    AddSpawnEntry(p.Creature);
+                }
+
                 if (_extractMonsterData && p.Creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster && !_knownMonsterIds.Contains(p.Creature.Id))
                 {
                     _monsterFile.WriteLine($"{p.Creature.Name} {p.Creature.Position}");
@@ -753,6 +823,11 @@ private static bool Connection_OnReceivedServerCreateOnMapPacket(Packet packet)
             }
             else if (p.Creature != null)
             {
+                if (_extractSpawns)
+                {
+                    AddSpawnEntry(p.Creature);
+                }
+
                 if (_extractMonsterData && p.Creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster && !_knownMonsterIds.Contains(p.Creature.Id))
                 {
                     _monsterFile.WriteLine($"{p.Creature.Name} {p.Creature.Position}");
@@ -800,6 +875,11 @@ static bool Connection_OnReceivedMapPacket(Packet packet)
                                 continue;
                             }
 
+                            if (_extractSpawns)
+                            {
+                                AddSpawnEntry(creature);
+                            }
+
                             if (_extractMonsterData && creature.Type == OXGaming.TibiaAPI.Constants.CreatureType.Monster && !_knownMonsterIds.Contains(creature.Id))
                             {
                                 _monsterFile.WriteLine($"{creature.Name} {creature.Position}");
