python/ctrump/libspe2.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
# Copyright (c) 2009, Fixstars Corporation
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without modification, 
# are permitted provided that the following conditions are met:
# 
#   * Redistributions of source code must retain the above copyright notice,
#     this list of conditions and the following disclaimer.
# 
#   * Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions and the following disclaimer in the documentation
#     and/or other materials provided with the distribution.
# 
#   * Neither the name of Fixstars Corporation nor the names of its contributors
#     may be used to endorse or promote products derived from this software
#     without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
"""@package libspe2 interface"""

import ctrump

def spe_in_param_member(have_spe_output, param, name):
    if have_spe_output:
        return param.member('in').member(name)
    else:
        return param.member(name)

class LibSPE2:
    def __init__(self, builder):
        self.builder = builder
        self.spe_context_ptr_t = builder.incomplete_typedefed_name('spe_context_ptr_t')

        self.spe_program_handle_t = builder.pointer_to(builder.incomplete_typedefed_name('spe_program_handle_t'))
        self.load_func_typespec = builder.func_type(builder.int_, [(self.spe_context_ptr_t,'spe'), 
                                                                   (self.spe_program_handle_t,'program')])

        uint_ptr_t = builder.pointer_to(builder.uint)
        void_ptr_t = builder.pointer_to(builder.void)

        spe_stop_info_t = builder.incomplete_typedefed_name('spe_stop_info_t')
        stop_info_ptr_t = builder.pointer_to(spe_stop_info_t)

        self.run_func_typespec = builder.func_type(builder.int_, [(self.spe_context_ptr_t, 'spe'),
                                                                  (void_ptr_t, 'argp')])
        self.run_func = builder.func('ctrump_runtime_spe_start', self.run_func_typespec)

        self.wait_func_typespec = builder.func_type(builder.int_, [(self.spe_context_ptr_t, 'spe')])

        self.fini = builder.func('ctrump_spe_fini')
        self.include = builder.include("ctrump-libspe2.h")

        self.load_func = builder.func('ctrump_runtime_spe_program_load', self.load_func_typespec)
        self.wait_func = builder.func('ctrump_runtime_spe_wait', self.wait_func_typespec)


    def program_typespec(self):
        builder = self.builder
        typespec = builder.incomplete_typedefed_name('spe_program_handle_t')
        return typespec

    def emit_load_func(self, spe_obj, program_obj):
        b = self.builder
        return self.load_func(spe_obj, b.addr(program_obj))

    def emit_run_func(self, spe_obj, param):
        return self.run_func(spe_obj, param)

    def emit_wait_func(self, spe_obj):
        return self.wait_func(spe_obj)

    def emit_spe_end(self, have_spe_output, spe_id, is_last_spe, sync_buffer_ea):
        fini = self.builder.func('ctrump_spe_fini')
        return self.builder.return_(fini(spe_id, is_last_spe, sync_buffer_ea))

    def emit_include(self):
        return self.include

    def get_program_load_func(self):
        return self.emit_load_func

    def get_run_func(self):
        return self.emit_run_func

    def get_wait_func(self):
        return self.emit_wait_func

    def get_spe_end_func(self):
        return self.emit_spe_end

    def speobj_typespec(self):
        return self.spe_context_ptr_t

def create(builder):
    return LibSPE2(builder)

---