Merge pull request #4060 from myk002/myk_readable

[Units] add "agitated" and trained levels to getReadableName

docs/changelog.txt

@@ -59,6 +59,7 @@ Template for new versions:
 ## Fixes
 
 ## Misc Improvements
+- wherever units are listed in DFHack tools, properties like "agitated" or (-trained-) are now shown
 
 ## Documentation
 

docs/dev/Lua API.rst

@@ -1614,8 +1614,9 @@ Units module
 * ``dfhack.units.getReadableName(unit)``
 
   Returns a string that includes the language name of the unit (if any), the
-  race of the unit, whether it is trained for war or hunting, and any
-  syndrome-given descriptions (such as "necromancer").
+  race of the unit, whether it is trained for war or hunting, any
+  syndrome-given descriptions (such as "necromancer"), and the training level
+  (if tame).
 
 * ``dfhack.units.getStressCategory(unit)``
 

library/modules/Units.cpp

@@ -1273,6 +1273,36 @@ static string get_caste_name(df::unit* unit) {
     return raw->caste[caste]->caste_name[0];
 }
 
+// must subtract 1 from animal_training_level value to index this array
+static const char * training_quality_table[] = {
+    "trained",     // Trained
+    "-trained-",    // WellTrained
+    "+trained+",    // SkilfullyTrained
+    "*trained*",    // ExpertlyTrained
+    "\xF0trained\xF0", // (≡) ExceptionallyTrained
+    "\x0Ftrained\x0F"  // (☼) MasterfullyTrained
+};
+
+static const char * getTameTag(df::unit *unit) {
+    int32_t level = unit->training_level;
+    switch (level) {
+        case df::animal_training_level::SemiWild:
+            return "semi-wild";
+        case df::animal_training_level::Trained:
+        case df::animal_training_level::WellTrained:
+        case df::animal_training_level::SkilfullyTrained:
+        case df::animal_training_level::ExpertlyTrained:
+        case df::animal_training_level::ExceptionallyTrained:
+        case df::animal_training_level::MasterfullyTrained:
+            return training_quality_table[level-1];
+        case df::animal_training_level::Domesticated:
+            return "tame";
+        case df::animal_training_level::WildUntamed:
+        default:
+            return "wild";
+    }
+}
+
 string Units::getReadableName(df::unit* unit) {
     string race_name = isBaby(unit) ? getRaceBabyName(unit) :
         (isChild(unit) ? getRaceChildName(unit) : get_caste_name(unit));
@@ -1282,6 +1312,8 @@ string Units::getReadableName(df::unit* unit) {
         race_name = "hunter " + race_name;
     if (isWar(unit))
         race_name = "war " + race_name;
+    if (unit->flags4.bits.agitated_wilderness_creature)
+        race_name = "agitated " + race_name;
     string name = Translation::TranslateName(getVisibleName(unit), false);
     if (name.empty()) {
         name = race_name;
@@ -1302,6 +1334,11 @@ string Units::getReadableName(df::unit* unit) {
             break;
         }
     }
+    if (isTame(unit)) {
+        name += " (";
+        name += getTameTag(unit);
+        name += ")";
+    }
     return name;
 }