#
# RTEMS Pretty Printers
# Copyright 2010 Chris Johns (chrisj@rtems.org)
#
# $Id$
#

import gdb
import re

import objects
import threads
import classic

nesting = 0

def type_from_value(val):
    type = val.type;
    # If it points to a reference, get the reference.
    if type.code == gdb.TYPE_CODE_REF:
        type = type.target ()
    # Get the unqualified type
    return type.unqualified ()

def register_rtems_printers (obj):
    "Register RTEMS pretty-printers with objfile Obj."

    if obj == None:
        obj = gdb

    obj.pretty_printers.append (lookup_function)

def lookup_function (val):
    "Look-up and return a pretty-printer that can print val."

    global nesting

    typename = str(type_from_value(val))

    for function in pp_dict:
        if function.search (typename):
            nesting += 1
            result = pp_dict[function] (val)
            nesting -= 1
            if nesting == 0:
                objects.information.invalidate()
            return result

    # Cannot find a pretty printer.  Return None.
    return None

def build_rtems_dict():
    pp_dict[re.compile('^rtems_id$')]   = lambda val: objects.id_printer(val)
    pp_dict[re.compile('^Objects_Id$')] = lambda val: objects.id_printer(val)
    pp_dict[re.compile('^Objects_Name$')] = lambda val: objects.name_printer(val)
    pp_dict[re.compile('^Objects_Control$')] = lambda val: objects.control_printer(val)
    pp_dict[re.compile('^States_Control$')] = lambda val: threads.state_printer(val)
    pp_dict[re.compile('^rtems_attribute$')] = lambda val: classic.attribute_printer(val)
    pp_dict[re.compile('^Semaphore_Control$')] = lambda val: classic.semaphore_printer(val)

class rtems(gdb.Command):
    """Prefix command for RTEMS."""

    def __init__(self):
        super(rtems, self).__init__('rtems',
                                    gdb.COMMAND_STATUS,
                                    gdb.COMPLETE_NONE,
                                    True)

class rtems_object(gdb.Command):
    """Object sub-command for RTEMS"""

    objects = {
        'classic/semaphores': lambda id: classic.semaphore(id),
        'classic/tasks': lambda id: classic.task(id),
        'classic/message_queues': lambda id: classic.message_queue(id)
        }

    def __init__(self):
        self.__doc__ = 'Display the RTEMS object given a numeric ID.'
        super(rtems_object, self).__init__('rtems object',
                                           gdb.COMMAND_STATUS)

    def invoke(self, arg, from_tty):
        for num in arg.split():
            try:
                val = gdb.parse_and_eval(num)
                num = int(val)
            except:
                print 'error: "%s" is not a number' % (num)
                return
            id = objects.ident(num)
            if not id.valid():
                print 'Invalid object id'
            print 'API:%s Class:%s Node:%d Index:%d Id:%08X' % \
                (id.api(), id._class(), id.node(), id.index(), id.value())
            objectname = id.api() + '/' + id._class()
            if objectname in self.objects:
                object = self.objects[objectname](id)
                object.show(from_tty)
        objects.information.invalidate()

#
# Main
#
pp_dict = {}
build_rtems_dict()
gdb.pretty_printers = []
gdb.pretty_printers.append (lookup_function)
rtems()
rtems_object()