Add VInt hint EnumIndex<T>

vshaxe · Apr 4, 2024 · f4921c5 · f4921c5
1 parent 519a3ff
commit f4921c5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/hld/Eval.hx b/hld/Eval.hx
@@ -777,6 +777,11 @@ class Eval {
 				if( eproto == null )
 					throw "Can't resolve enum " + t;
 				Value.intEnumFlags(i, eproto);
+			case HEnumIndex(t):
+				var eproto = module.resolveEnum(t);
+				if( eproto == null )
+					throw "Can't resolve enum " + t;
+				Value.intEnumIndex(i, eproto);
 			default: "" + i;
 			}
 		case VInt64(i):

diff --git a/hld/Value.hx b/hld/Value.hx
@@ -29,9 +29,10 @@ typedef InlinedField = { name : String, addr : Eval.VarAddress }
 
 enum Hint {
 	HNone;
-	HHex;
-	HBin;
-	HEnumFlags(t : String);
+	HHex; // v:h
+	HBin; // v:b
+	HEnumFlags(t : String); // v:EnumFlags<T>
+	HEnumIndex(t : String); // v:EnumIndex<T>
 }
 
 @:structInit class Value {
@@ -46,6 +47,8 @@ enum Hint {
 			return HBin;
 		if( StringTools.startsWith(s,"EnumFlags<") && StringTools.endsWith(s,">") )
 			return HEnumFlags(s.substr(10, s.length - 11));
+		if( StringTools.startsWith(s,"EnumIndex<") && StringTools.endsWith(s,">") )
+			return HEnumIndex(s.substr(10, s.length - 11));
 		return HNone;
 	}
 
@@ -88,4 +91,10 @@ enum Hint {
 		return f;
 	}
 
+	public static function intEnumIndex( value : Int, eproto : format.hl.Data.EnumPrototype ) : String {
+		if( value < 0 || value >= eproto.constructs.length )
+			throw "Out of range [0," + eproto.constructs.length + ")";
+		return eproto.constructs[value].name;
+	}
+
 }