add Encounter::from_header_id

It makes sense to expose this logic as a function, as other programs
like raidgrep might want to use the same logic when dealing with partial
evtc files.

1 files changed, 16 insertions, 0 deletions

diff --git a/src/gamedata.rs b/src/gamedata.rs
index 64b6731..ac8d46a 100644
--- a/src/gamedata.rs
+++ b/src/gamedata.rs
@@ -1,5 +1,6 @@
 //! This module contains some low-level game data, such as different boss IDs.
 use num_derive::FromPrimitive;
+use num_traits::FromPrimitive;
 use std::{
     fmt::{self, Display, Formatter},
     str::FromStr,
@@ -118,6 +119,21 @@ impl Encounter {
             Encounter::WhisperOfJormag => &[Boss::WhisperOfJormag],
         }
     }
+
+    /// Converts a combat ID as given in the arcdps header into the correct encounter.
+    ///
+    /// This properly takes care of encounters with multiple bosses or which could be saved as
+    /// multiple bosses.
+    ///
+    /// ```
+    /// # use evtclib::gamedata::Encounter;
+    /// assert_eq!(Encounter::from_header_id(0x3C4E), Some(Encounter::ValeGuardian));
+    /// assert_eq!(Encounter::from_header_id(0x5261), Some(Encounter::TwinLargos));
+    /// ```
+    #[inline]
+    pub fn from_header_id(id: u16) -> Option<Encounter> {
+        Boss::from_u16(id).map(Boss::encounter)
+    }
 }
 
 /// Error for when converting a string to an encounter fails.