Source code for mcgdb.model.task.environment.p2012.npm.user.overlay.matrix

import gdb

import mcgdb
from mcgdb.model.task import representation, capture
from mcgdb.model.task.environment.p2012 import npm

[docs]class HostDMAPush(npm.representation.dma_link.P2012HostDMAPushEndpoint):
    count = 0
[docs]    def do_produce_message(self):
        
        # payload = str(gdb.parse_and_eval("*((int *)%d)@%d" % 
        #                                  (self.baseAddr, self.increment /
        #                                   gdb.lookup_type("int").sizeof)))
        
        payload = "Push>%s" % self.id
        HostDMAPush.count += 1
        return model.representation.Message(self, payload, 
                                      "Message #%d" % HostDMAPush.count)

[docs]class HostDMAPull(npm.representation.dma_link.P2012HostDMAPullEndpoint):
[docs]    def get_message(self, msg):
        
        # payload = str(gdb.parse_and_eval("*((int *)%d)@%d" % 
        #                                  (self.baseAddr, self.increment /
        #                                   gdb.lookup_type("int").sizeof)))
        
        
        payload = "Pull<%s" % self.id
        msg.checkpoint(self, payload)
        return msg

[docs]class MyComponent(representation.CommComponent):
[docs]    def init(self):
        self.mailbox = []
		
[docs]    def do_consume_message(self, endpoint, msg):
        self.mailbox.insert(0, msg)
    
[docs]    def do_produce_message(self, endpoint):
        try:
            msg = self.mailbox.pop()
            msg.checkpoint("MatrixProcessor/%d" % self.id, 
                           "%sx%s" % (msg.last_payload(), self.id))
            return msg
        except IndexError:
            return model.representation.Message(self, "EMPTY")
        
[docs]    def get_messages(self):
        return self.mailbox
    
[docs]class MyHost: #(representation.CommHost):
[docs]    def init(self):
        self.mailbox = []

[docs]    def do_consume_message(self, endpoint, msg):
        self.mailbox.append(msg)
        
[docs]    def get_messages(self):
        return self.mailbox
        
definitions = {}
[docs]def initialize():
    definitions[model.representation.process.CommHost] = MyHost
    definitions[representation.CommComponent] = MyComponent
    definitions[npm.representation.dma_link.P2012HostDMAPushEndpoint] = HostDMAPush
    definitions[npm.representation.dma_link.P2012HostDMAPullEndpoint] = HostDMAPull