niklas/tracetools_analysis
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add addr_to_int util function

Browse source
This commit is contained in:
Christophe Bedard 2019-08-08 11:29:14 +02:00
parent e845a3a668
commit 4a41c2bd61
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tracetools_analysis/tracetools_analysis/processor/profile.py
View file

@ -43,13 +43,13 @@ class ProfileHandler(EventHandler):
def __init__( def __init__(
self, self,
address_to_func: Dict[int, str] = {}, address_to_func: Dict[Union[int, str], str] = {},
**kwargs, **kwargs,
) -> None: ) -> None:
""" """
Constructor. Constructor.
:param address_to_func: the mapping from function address to name :param address_to_func: the mapping from function address (`int` or hex `str`) to name
""" """
handler_map = { handler_map = {
'lttng_ust_cyg_profile_fast:func_entry': 'lttng_ust_cyg_profile_fast:func_entry':
@ -65,7 +65,9 @@ class ProfileHandler(EventHandler):
) )
self._data_model = ProfileDataModel() self._data_model = ProfileDataModel()
self._address_to_func = address_to_func self._address_to_func = {
self._addr_to_int(addr): name for addr, name in address_to_func.items()
}
# Temporary buffers # Temporary buffers
# tid -> # tid ->
@ -80,6 +82,11 @@ class ProfileHandler(EventHandler):
# ] # ]
self._current_funcs: Dict[int, List[List[Union[str, int]]]] = defaultdict(list) self._current_funcs: Dict[int, List[List[Union[str, int]]]] = defaultdict(list)
@staticmethod
def _addr_to_int(addr: Union[int, str]) -> int:
"""Transform an address into an `int` if it's a hex `str`."""
return int(addr, 16) if isinstance(addr, str) else addr
@property @property
def data(self) -> ProfileDataModel: def data(self) -> ProfileDataModel:
return self._data_model return self._data_model