python/ctrump/format.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.

"""Format internal information"""

from ctrump import *

def format_loop_subscripts(subscripts):
    ret = ""
    varname = "<loop_counter>"

    for i in subscripts:
        indices_str = ""

        for j in i.indices:
            is_sign = ""
            if j.flags & ctrump.LOOP_INDEX_NEGATIVE:
                is_sign = "-"

            if j.code == ctrump.LOOP_INDEX_INDUCTIVE:
                varname = j.var.name
                indices_str = indices_str + '%s%s(incr=%d)+'%(is_sign,varname, j.incr)
            elif j.code == ctrump.LOOP_INDEX_POINTER_INC:
                indices_str = indices_str + '(ptrincr=%d)+'%(j.incr)
            elif j.code == ctrump.LOOP_INDEX_INVARIANT:
                indices_str = indices_str + '<%s%s>+'%(is_sign,j.invariant.name)
            else:
                raise("unknown index code '%s'"%ctrump.loop_index_code_string_table[j.code])


        if i.code == ctrump.LOOP_SUBSCRIPT_COEF_TERMINAL:
            ret = ret + ("{%s%d}"%(indices_str, i.offset))
        elif i.code == ctrump.LOOP_SUBSCRIPT_COEF_ARRAYSIZE:
            ret = ret + ("[%s%d(arraysize=%d)]"%(indices_str, i.offset, i.array_size_coef))
        elif i.code == ctrump.LOOP_SUBSCRIPT_COEF_SCALE:
            ret = ret + ("[%s%d(scale=%s)]"%(indices_str, i.offset, i.scale_array_coef.name))
        elif i.code == ctrump.LOOP_SUBSCRIPT_RECORD_MEMBER:
            ret = ret + ("[%s.%s]"%(indices_str, i.member_name))
        elif i.code == ctrump.LOOP_SUBSCRIPT_RECORD_MEMBER_TERMINAL:
            ret = ret + ("{%s.%s}"%(indices_str, i.member_name))
        elif i.code == ctrump.LOOP_SUBSCRIPT_LOAD_RECORD:
            ret = ret + ("[%s%d(load_type=%s)]"%(indices_str, i.offset, i.load_type.name))
        else:
            raise("unknown subscript code '%s'"%ctrump.loop_subscript_code_string_table[i.code])
    return ret

def format_tree_subscripts(subscripts):
    ret = ""
    for i in subscripts:
        for j in i.nd_subscripts:
            if (j.code == ctrump.SUBSCRIPT_TERMINAL or
                j.code == ctrump.SUBSCRIPT_COEF_ARRAY or
                j.code == ctrump.SUBSCRIPT_COEF_SIZEOF_RECORD):

                if j.expr:
                    ret = ("[%s]"%(ctrump.pprint_expr(j.expr))) + ret
                else:
                    ret = "->" + ret
            elif (j.code == ctrupm.SUBSCRIPT_RECORD_MEMBER):
                ret = (".%s"%j.member_name) + ret
    return ret

def format_random_access(a):
    (path,lineno) = ctrump.get_stmt_loc(a.access_at_stmt)
    pp = ctrump.pprint_expr
    if a.code == ctrump.RANDOM_ACCESS_POINTER_NOT_INDUCTIVE:
        return "%s:%d:'%s' is not inductive."%(path,lineno, pp(a.not_inductive_pointer))
    elif a.code == ctrump.RANDOM_ACCESS_COMPLICATED_POINTER:
        return "%s:%d:Can't analyze '%s'. too complicated."%(path,lineno, pp(a.complicated_pointer))
    elif a.code == ctrump.RANDOM_ACCESS_ARRAY_NOT_INVARIANT:
        return "%s:%d:'%s' is not invariant."%(path,lineno, pp(a.array))
    elif a.code == ctrump.RANDOM_ACCESS_SUBSCRIPT_NOT_INDUCTIVE:
        return "%s:%d:'%s' is not inductive."%(path,lineno, ctrump.format_tree_subscripts(a.subscripts))
    elif a.code == ctrump.RANDOM_ACCESS_COMPLICATED_ARRAY:
        if a.subscripts:
            return "%s:%d:'%s%s' is too complicated."%(path,lineno, ctrump.pprint_expr(a.array),
                                                         ctrump.format_tree_subscripts(a.subscripts))
        else:
            return "%s:%d:'%s' is too complicated."%(path,lineno, ctrump.pprint_expr(a.array))
    elif a.code == ctrump.RANDOM_ACCESS_MULTIPLE_SCALED_INDEX:
        return "%s:%d:multiple scaled index '%s' and '%s'."(path, lineno,
                                                            ctrump.pprint_expr(a.index0),
                                                            ctrump.pprint_expr(a.index1))
    elif a.code == ctrump.RANDOM_ACCESS_COMPLICATED_SUBSCRIPT:
        return "%s:%d:too complicated subscript '%s'."%(path, lineno, ctrump.pprint_expr(a.complicated_subscript))
    elif a.code == ctrump.RANDOM_ACCESS_INDEX_VAR_NOT_INDUCTIVE:
        return "%s:%d:'%s' is not inductive or invariant."%(path, lineno, a.index_var_not_inductive.name)
    else:
        return '%s:%d: error.'%(path,lineno)
    

def format_pdg(pdg):
    code = pdg.code
    if code == ctrump.PDG_VALUE_REDUCTION:
        return 'reduction : += ...' % (pdg.value)
    elif code == ctrump.PDG_VALUE_INCREMENTAL:
        return 'reduction : ++'
    elif code == ctrump.PDG_VALUE_EXPR:
        return 'value : %s'%pdg.val
    elif code == ctrump.PDG_VALUE_PHI:
        return 'phi : (...)'
    elif code == ctrump.PDG_VALUE_INITIALIZER:
        return 'initializer : ...'
    else:
        raise Exception('unknown pdg code %d'%code)

---