Add operator[]

dotnet · May 30, 2024 · 687ed43 · 687ed43
1 parent b2d9b1c
commit 687ed43
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/coreclr/jit/regMaskTPOps.cpp b/src/coreclr/jit/regMaskTPOps.cpp
@@ -263,3 +263,14 @@ void regMaskTP::operator^=(const regNumber reg)
     low ^= value;
 #endif
 }
+
+SingleTypeRegSet regMaskTP::operator[](int index) const
+{
+#ifdef HAS_MORE_THAN_64_REGISTERS
+    assert(index <= 2);
+    RegSet32 value = _registers[index];
+    return decodeForRegisterIndex(index, value);
+#else
+    return low;
+#endif
+}
diff --git a/src/coreclr/jit/target.h b/src/coreclr/jit/target.h
@@ -419,6 +419,7 @@ struct regMaskTP
 
     void operator|=(const regNumber reg);
     void operator^=(const regNumber reg);
+    SingleTypeRegSet operator[](int index) const;
 };
 
 static regMaskTP operator^(const regMaskTP& first, const regMaskTP& second)