feature: Add more APIs to L0 Sysman python binding

Related-To: NEO-18602

Add following apis to python:
1. zesFrequencyGetProperties
2. zesPowerGetProperties
3. zesPowerGetUsage

Signed-off-by: Aviral Nigam <aviral.nigam@intel.com>

Author: aviralni

bindings/sysman/python/source/examples/pyzes_black_box_test.py

@@ -149,6 +149,19 @@ def get_frequency_domain_string(freq_domain):
     return domain_map.get(freq_domain, f"UNKNOWN_FREQ_DOMAIN_{freq_domain}")
 
 
+def get_power_domain_string(power_domain):
+    """Convert power domain enum to string"""
+    domain_map = {
+        pz.ZES_POWER_DOMAIN_UNKNOWN: "ZES_POWER_DOMAIN_UNKNOWN",
+        pz.ZES_POWER_DOMAIN_CARD: "ZES_POWER_DOMAIN_CARD",
+        pz.ZES_POWER_DOMAIN_PACKAGE: "ZES_POWER_DOMAIN_PACKAGE",
+        pz.ZES_POWER_DOMAIN_STACK: "ZES_POWER_DOMAIN_STACK",
+        pz.ZES_POWER_DOMAIN_MEMORY: "ZES_POWER_DOMAIN_MEMORY",
+        pz.ZES_POWER_DOMAIN_GPU: "ZES_POWER_DOMAIN_GPU",
+    }
+    return domain_map.get(power_domain, f"UNKNOWN_POWER_DOMAIN_{power_domain}")
+
+
 def get_throttle_reasons_string(throttle_reasons):
     """Convert throttle reason flags to human-readable string"""
     if throttle_reasons == 0:
@@ -666,8 +679,13 @@ def test_power_module(device_handle, device_index):
         print_verbose(f"\n  Power Domain {i}:")
 
         power_props = pz.zes_power_properties_t()
+        power_ext_props = pz.zes_power_ext_properties_t()
+        default_limit = pz.zes_power_limit_ext_desc_t()
         power_props.stype = pz.ZES_STRUCTURE_TYPE_POWER_PROPERTIES
-        power_props.pNext = None
+        power_props.pNext = cast(byref(power_ext_props), c_void_p)
+        power_ext_props.stype = pz.ZES_STRUCTURE_TYPE_POWER_EXT_PROPERTIES
+        power_ext_props.pNext = None
+        power_ext_props.defaultLimit = pointer(default_limit)
 
         rc = pz.zesPowerGetProperties(power_handles[i], byref(power_props))
         if check_rc(f"zesPowerGetProperties(power {i})", rc):
@@ -682,6 +700,11 @@ def test_power_module(device_handle, device_index):
             print_verbose(f"      Default Limit: {power_props.defaultLimit}")
             print_verbose(f"      Min Limit: {power_props.minLimit}")
             print_verbose(f"      Max Limit: {power_props.maxLimit}")
+            print_verbose(
+                "      Extended Domain: "
+                f"{get_power_domain_string(power_ext_props.domain)}"
+            )
+            print_verbose(f"      Extended Default Limit: {default_limit.limit}")
 
         # Test power usage
         instant_power = c_uint32(0)

bindings/sysman/python/source/pyzes.py

@@ -153,7 +153,7 @@ class zes_engine_handle_t(c_void_p):
 
 ##
 
-ze_bool_t = c_uint32
+ze_bool_t = c_uint8
 ze_device_property_flags_t = c_uint32
 zes_engine_type_flags_t = c_uint32
 zes_device_property_flags_t = c_uint32
@@ -224,6 +224,29 @@ class zes_engine_handle_t(c_void_p):
 ZES_POWER_DOMAIN_GPU = 5
 ZES_POWER_DOMAIN_FORCE_UINT32 = 0x7FFFFFFF
 
+# Power limit level enumeration
+zes_power_level_t = c_int32
+ZES_POWER_LEVEL_UNKNOWN = 0
+ZES_POWER_LEVEL_SUSTAINED = 1
+ZES_POWER_LEVEL_BURST = 2
+ZES_POWER_LEVEL_PEAK = 3
+ZES_POWER_LEVEL_INSTANTANEOUS = 4
+ZES_POWER_LEVEL_FORCE_UINT32 = 0x7FFFFFFF
+
+# Power source enumeration
+zes_power_source_t = c_int32
+ZES_POWER_SOURCE_ANY = 0
+ZES_POWER_SOURCE_MAINS = 1
+ZES_POWER_SOURCE_BATTERY = 2
+ZES_POWER_SOURCE_FORCE_UINT32 = 0x7FFFFFFF
+
+# Power limit unit enumeration
+zes_limit_unit_t = c_int32
+ZES_LIMIT_UNIT_UNKNOWN = 0
+ZES_LIMIT_UNIT_CURRENT = 1
+ZES_LIMIT_UNIT_POWER = 2
+ZES_LIMIT_UNIT_FORCE_UINT32 = 0x7FFFFFFF
+
 ## Frequency domain enums ##
 zes_freq_domain_t = c_int32
 ZES_FREQ_DOMAIN_GPU = 0
@@ -340,6 +363,9 @@ class zes_engine_handle_t(c_void_p):
 ZES_STRUCTURE_TYPE_MEM_STATE = 0x1E
 ZES_STRUCTURE_TYPE_FREQ_PROPERTIES = 0x9
 ZES_STRUCTURE_TYPE_FREQ_STATE = 0x1B
+ZES_STRUCTURE_TYPE_POWER_PROPERTIES = 0xD
+ZES_STRUCTURE_TYPE_POWER_LIMIT_EXT_DESC = 0x27
+ZES_STRUCTURE_TYPE_POWER_EXT_PROPERTIES = 0x28
 ZES_STRUCTURE_TYPE_TEMP_PROPERTIES = 0xA
 ZES_STRUCTURE_TYPE_TEMP_CONFIG = 0x1C
 ZES_STRUCTURE_TYPE_ENGINE_PROPERTIES = 0x5
@@ -500,6 +526,32 @@ class zes_power_properties_t(_PrintableStructure):
     }
 
 
+class zes_power_limit_ext_desc_t(_PrintableStructure):
+    _fields_ = [
+        ("stype", c_int32),  # ZES_STRUCTURE_TYPE_POWER_LIMIT_EXT_DESC
+        ("pNext", c_void_p),
+        ("level", zes_power_level_t),
+        ("source", zes_power_source_t),
+        ("limitUnit", zes_limit_unit_t),
+        ("enabledStateLocked", ze_bool_t),
+        ("enabled", ze_bool_t),
+        ("intervalValueLocked", ze_bool_t),
+        ("interval", c_int32),
+        ("limitValueLocked", ze_bool_t),
+        ("limit", c_int32),
+    ]
+    _fmt_ = {"interval": "%d ms", "limit": "%d"}
+
+
+class zes_power_ext_properties_t(_PrintableStructure):
+    _fields_ = [
+        ("stype", c_int32),  # ZES_STRUCTURE_TYPE_POWER_EXT_PROPERTIES
+        ("pNext", c_void_p),
+        ("domain", zes_power_domain_t),
+        ("defaultLimit", POINTER(zes_power_limit_ext_desc_t)),
+    ]
+
+
 class zes_power_energy_counter_t(_PrintableStructure):
     _fields_ = [
         ("energy", c_uint64),  # monotonic energy counter in microjoules

bindings/sysman/python/test/unit_tests/test_frequency.py

@@ -130,6 +130,5 @@ def mock_get_properties(frequency_handle, properties_ptr):
         mock_get_func.assert_called_with("zesFrequencyGetProperties")
         mock_func.assert_called_once()
 
-
 if __name__ == "__main__":
     unittest.main()

bindings/sysman/python/test/unit_tests/test_power.py

@@ -108,6 +108,8 @@ def test_GivenValidPowerHandleWhenCallingZesPowerGetPropertiesThenCallSucceedsWi
         mock_default_limit = 150000
         mock_min_limit = 100000
         mock_max_limit = 225000
+        mock_domain = self.pyzes.ZES_POWER_DOMAIN_PACKAGE
+        mock_ext_default_limit = 175000
 
         def mock_get_properties(power_handle, properties_ptr):
             properties_ptr._obj.onSubdevice = True
@@ -117,13 +119,29 @@ def mock_get_properties(power_handle, properties_ptr):
             properties_ptr._obj.defaultLimit = mock_default_limit
             properties_ptr._obj.minLimit = mock_min_limit
             properties_ptr._obj.maxLimit = mock_max_limit
+
+            ext_properties_ptr = cast(
+                properties_ptr._obj.pNext,
+                POINTER(self.pyzes.zes_power_ext_properties_t),
+            )
+            ext_properties_ptr.contents.domain = mock_domain
+            ext_properties_ptr.contents.defaultLimit.contents.limit = (
+                mock_ext_default_limit
+            )
             return self.pyzes.ZE_RESULT_SUCCESS
 
         mock_func = MagicMock(side_effect=mock_get_properties)
         mock_get_func.return_value = mock_func
 
         power_handle = self.pyzes.zes_pwr_handle_t()
         properties = self.pyzes.zes_power_properties_t()
+        ext_properties = self.pyzes.zes_power_ext_properties_t()
+        ext_default_limit = self.pyzes.zes_power_limit_ext_desc_t()
+
+        properties.stype = self.pyzes.ZES_STRUCTURE_TYPE_POWER_PROPERTIES
+        properties.pNext = cast(byref(ext_properties), c_void_p)
+        ext_properties.stype = self.pyzes.ZES_STRUCTURE_TYPE_POWER_EXT_PROPERTIES
+        ext_properties.defaultLimit = pointer(ext_default_limit)
 
         result = self.pyzes.zesPowerGetProperties(power_handle, byref(properties))
 
@@ -135,9 +153,11 @@ def mock_get_properties(power_handle, properties_ptr):
         self.assertEqual(properties.defaultLimit, mock_default_limit)
         self.assertEqual(properties.minLimit, mock_min_limit)
         self.assertEqual(properties.maxLimit, mock_max_limit)
+        self.assertEqual(ext_properties.stype, self.pyzes.ZES_STRUCTURE_TYPE_POWER_EXT_PROPERTIES)
+        self.assertEqual(ext_properties.domain, mock_domain)
+        self.assertEqual(ext_properties.defaultLimit.contents.limit, mock_ext_default_limit)
         mock_get_func.assert_called_with("zesPowerGetProperties")
         mock_func.assert_called_once()
 
-
 if __name__ == "__main__":
     unittest.main()