Homec4science
Diffusion akantu (features/Data-Driven_Implementation) Browse

rAKA/cmake/libakantu/v3/printers.py
895e766f2bc6features/Data…

Last Change
Raw File

printers.py
View Options

#!/usr/bin/env python3
__copyright__ = (
"Copyright (©) 2016-2023 EPFL (Ecole Polytechnique Fédérale de Lausanne)"
"Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)"
)
__license__ = "LGPLv3"
#
# Inspired from boost's pretty printers from
# Rüdiger Sonderfeld <ruediger@c-plusplus.de>
# and from Pretty-printers for libstc++ from Free Software Foundation, Inc.
#
import gdb
import re
import itertools
try:
import gdb.printing
except ImportError:
raise RuntimeError("Your GDB is too old")
class AkantuPrinter(object):
regex = None
@classmethod
def supports(cls, typename):
# print('{0} ~= {1}'.format(typename, cls.regex), file=sys.stderr)
return cls.regex.search(typename)
@classmethod
def get_basic_type(cls, value):
"""Determines the type associated to a value"""
_type = value.type
# If it points to a reference, get the reference.
if _type.code == gdb.TYPE_CODE_REF:
_type = _type.target()
# Get the unqualified type, stripped of typedefs.
_type = _type.unqualified().strip_typedefs()
return _type.tag
__akantu_pretty_printers__ = \
gdb.printing.RegexpCollectionPrettyPrinter("libakantu-v3")
def register_pretty_printer(pretty_printer):
"""Registers a Pretty Printer"""
__akantu_pretty_printers__.add_printer(
pretty_printer.name, pretty_printer.regex, pretty_printer
)
return pretty_printer
@register_pretty_printer
class AkaArrayPrinter(AkantuPrinter):
"""Pretty printer for akantu::Array<T>"""
regex = re.compile(
"^akantu::Array<(.*?), (true|false)>$|^akantu::ArrayDataLayer<(.*?), \(akantu::ArrayAllocationType\)1>$" # noqa
)
name = "akantu::Array"
def __init__(self, value):
self.typename = self.get_basic_type(value)
self.value = value
self.ptr = self.value["values"]
self.size = int(self.value["size_"])
self.nb_component = int(self.value["nb_component"])
def to_string(self):
m = self.regex.search(self.typename)
return "Array<{0}>({1}, {2}) stored at {3}".format(
m.group(1), self.size, self.nb_component, self.ptr
)
def display_hint(self):
return "array"
def children(self):
_ptr = self.ptr
for i in range(self.size):
for j in range(self.nb_component):
yield ("[{0}, {1}]".format(i, j), (_ptr + j).dereference())
_ptr = _ptr + self.nb_component
class RbtreeIterator(object):
"""
Turn an RB-tree-based container (std::map, std::set etc.) into
a Python iterable object.
"""
def __init__(self, rbtree):
self.size = rbtree["_M_t"]["_M_impl"]["_M_node_count"]
self.node = rbtree["_M_t"]["_M_impl"]["_M_header"]["_M_left"]
self.count = 0
def __iter__(self):
return self
def __len__(self):
return int(self.size)
def __next__(self):
if self.count == self.size:
raise StopIteration
result = self.node
self.count = self.count + 1
if self.count < self.size:
# Compute the next node.
node = self.node
if node.dereference()["_M_right"]:
node = node.dereference()["_M_right"]
while node.dereference()["_M_left"]:
node = node.dereference()["_M_left"]
else:
parent = node.dereference()["_M_parent"]
while node == parent.dereference()["_M_right"]:
node = parent
parent = parent.dereference()["_M_parent"]
if node.dereference()["_M_right"] != parent:
node = parent
self.node = node
return result
def get_value_from_aligned_membuf(buf, valtype):
"""Returns the value held in a __gnu_cxx::__aligned_membuf."""
return buf["_M_storage"].address.cast(valtype.pointer()).dereference()
def get_value_from_Rb_tree_node(node):
"""Returns the value held in an _Rb_tree_node<_Val>"""
try:
member = node.type.fields()[1].name
if member == "_M_value_field":
# C++03 implementation, node contains the value as a member
return node["_M_value_field"]
elif member == "_M_storage":
# C++11 implementation, node stores value in __aligned_membuf
valtype = node.type.template_argument(0)
return get_value_from_aligned_membuf(node["_M_storage"], valtype)
except: # noqa: E722
pass
raise ValueError("Unsupported implementation for %s" % str(node.type))
def find_type(orig, name):
typ = orig.strip_typedefs()
while True:
# Strip cv-qualifiers. PR 67440.
search = "%s::%s" % (typ.unqualified(), name)
try:
return gdb.lookup_type(search)
except RuntimeError:
pass
# The type was not found, so try the superclass. We only need
# to check the first superclass, so we don't bother with
# anything fancier here.
field = typ.fields()[0]
if not field.is_base_class:
raise ValueError("Cannot find type %s::%s" % (str(orig), name))
typ = field.type
@register_pretty_printer
class AkaElementTypeMapArrayPrinter(AkantuPrinter):
"""Pretty printer for akantu::ElementTypeMap<Array<T>>"""
regex = re.compile(
"^akantu::ElementTypeMap<akantu::Array<(.*?), (true|false)>\*, akantu::(.*?)>$" # noqa
) # noqa: E501,W605
name = "akantu::ElementTypeMapArray"
# Turn an RbtreeIterator into a pretty-print iterator.
class _rb_iter(object):
def __init__(self, rbiter, type, ghost_type):
self.rbiter = rbiter
self.count = 0
self.type = type
self.ghost_type = ghost_type
def __iter__(self):
return self
def __next__(self):
if self.count % 2 == 0:
n = next(self.rbiter)
n = n.cast(self.type).dereference()
n = get_value_from_Rb_tree_node(n)
self.pair = n
item = "{0}:{1}".format(n["first"], self.ghost_type)
else:
item = self.pair["second"].dereference()
result = ("[{0}]".format(self.count), item)
self.count = self.count + 1
return result
def _iter(self, not_ghost, ghost, type):
iter_size = (len(not_ghost), len(ghost))
it = self._rb_iter(not_ghost, type, "_not_ghost")
for _ in range(iter_size[0] * 2):
yield next(it)
it = self._rb_iter(ghost, type, "_ghost")
for _ in range(iter_size[1] * 2):
yield next(it)
raise StopIteration
def __init__(self, value):
self.typename = self.get_basic_type(value)
self.value = value
self.data = self.value["data"]
self.ghost_data = self.value["ghost_data"]
def to_string(self):
m = self.regex.search(self.typename)
return "ElementTypMapArray<{0}> with " "{1} _not_ghost and {2} _ghost".format( # noqa
m.group(1),
len(RbtreeIterator(self.data)),
len(RbtreeIterator(self.ghost_data)))
def children(self):
# m = self.regex.search(self.typename)
rep_type = find_type(self.data.type, "_Rep_type")
node = find_type(rep_type, "_Link_type")
node = node.strip_typedefs()
return itertools.chain(
self._rb_iter(RbtreeIterator(self.data), node, "_not_ghost"),
self._rb_iter(RbtreeIterator(self.ghost_data), node, "_ghost"),
)
def display_hint(self):
return "map"
# @register_pretty_printer
class AkaTensorPrinter(AkantuPrinter):
"""Pretty printer for akantu::Tensor<T>"""
regex = re.compile("^akantu::Tensor<(.*), +(.*), +(.*)>$")
name = "akantu::Tensor"
value = None
typename = ""
ptr = None
dims = []
ndims = 0
def pretty_print(self):
def ij2str(i, j, m):
return "{0}".format((self.ptr + m * j + i).dereference())
def line(i, m, n):
return "[{0}]".format(
", ".join((ij2str(i, j, m) for j in range(n))))
m = int(self.dims[0])
if self.ndims == 1:
n = 1
else:
n = int(self.dims[1])
return "[{0}]".format(", ".join(line(i, m, n) for i in range(m)))
def __init__(self, value):
self.typename = self.get_basic_type(value)
self.value = value
self.ptr = self.value["values"]
self.dims = self.value["n"]
def children(self):
yield ("values", self.pretty_print())
yield ("wrapped", self.value["wrapped"])
@register_pretty_printer
class AkaVectorPrinter(AkaTensorPrinter):
"""Pretty printer for akantu::Vector<T>"""
regex = re.compile("^akantu::Vector<(.*)>$")
name = "akantu::Vector"
n = 0
ptr = 0x0
def __init__(self, value):
super(AkaVectorPrinter, self).__init__(value)
self.ndims = 1
def to_string(self):
m = self.regex.search(self.typename)
return "Vector<{0}>({1}) [{2}]".format(
m.group(1), int(self.dims[0]), str(self.ptr)
)
@register_pretty_printer
class AkaMatrixPrinter(AkaTensorPrinter):
"""Pretty printer for akantu::Matrix<T>."""
regex = re.compile("^akantu::Matrix<(.*)>$")
name = "akantu::Matrix"
def __init__(self, value):
super(AkaMatrixPrinter, self).__init__(value)
self.ndims = 2
def to_string(self):
m = self.regex.search(self.typename)
return "Matrix<%s>(%d, %d) [%s]" % (
m.group(1),
int(self.dims[0]),
int(self.dims[1]),
str(self.ptr),
)
@register_pretty_printer
class AkaElementPrinter(AkantuPrinter):
"""Pretty printer for akantu::Element."""
regex = re.compile("^akantu::Element$")
name = "akantu::Element"
def __init__(self, value):
self.typename = self.get_basic_type(value)
self.value = value
self.element = self.value["element"]
self.eltype = self.value["type"]
self.ghost_type = self.value["ghost_type"]
self._ek_not_defined = gdb.parse_and_eval("akantu::_not_defined")
self._casper = gdb.parse_and_eval("akantu::_casper")
self._max_uint = gdb.parse_and_eval("(akantu::Int) -1")
def to_string(self):
if (
(self.element == self._max_uint)
and (self.eltype == self._ek_not_defined)
and (self.ghost_type == self._casper)
):
return "ElementNull"
return "Element({0}, {1}, {2})".format(
self.element, self.eltype, self.ghost_type
)
def register_akantu_printers(obj):
"""Register Akantu Pretty Printers."""
gdb.printing.register_pretty_printer(obj, __akantu_pretty_printers__)
c4science · Help