Python ccode.CCodePrinter類代碼示例

def test_ccode_Indexed():
    from sympy.tensor import IndexedBase, Idx
    from sympy import symbols
    s, n, m, o = symbols('s n m o', integer=True)
    i, j, k = Idx('i', n), Idx('j', m), Idx('k', o)

    x = IndexedBase('x')[j]
    A = IndexedBase('A')[i, j]
    B = IndexedBase('B')[i, j, k]

    with warnings.catch_warnings():
        warnings.filterwarnings("ignore", category=SymPyDeprecationWarning)
        p = CCodePrinter()
        p._not_c = set()

        assert p._print_Indexed(x) == 'x[j]'
        assert p._print_Indexed(A) == 'A[%s]' % (m*i+j)
        assert p._print_Indexed(B) == 'B[%s]' % (i*o*m+j*o+k)
        assert p._not_c == set()

        A = IndexedBase('A', shape=(5,3))[i, j]
        assert p._print_Indexed(A) == 'A[%s]' % (3*i + j)

        A = IndexedBase('A', shape=(5,3), strides='F')[i, j]
        assert ccode(A) == 'A[%s]' % (i + 5*j)

        A = IndexedBase('A', shape=(29,29), strides=(1, s), offset=o)[i, j]
        assert ccode(A) == 'A[o + s*j + i]'

        Abase = IndexedBase('A', strides=(s, m, n), offset=o)
        assert ccode(Abase[i, j, k]) == 'A[m*j + n*k + o + s*i]'
        assert ccode(Abase[2, 3, k]) == 'A[3*m + n*k + o + 2*s]'

def test_ccode_Indexed():
    from sympy.tensor import IndexedBase, Idx
    from sympy import symbols
    n, m, o = symbols('n m o', integer=True)
    i, j, k = Idx('i', n), Idx('j', m), Idx('k', o)
    p = CCodePrinter()
    p._not_c = set()

    x = IndexedBase('x')[j]
    assert p._print_Indexed(x) == 'x[j]'
    A = IndexedBase('A')[i, j]
    assert p._print_Indexed(A) == 'A[%s]' % (m*i+j)
    B = IndexedBase('B')[i, j, k]
    assert p._print_Indexed(B) == 'B[%s]' % (i*o*m+j*o+k)

    assert p._not_c == set()

def test_ccode_Indexed():
    from sympy.tensor import IndexedBase, Idx
    from sympy import symbols
    i, j, k, n, m, o = symbols('i j k n m o', integer=True)

    p = CCodePrinter()
    p._not_c = set()

    x = IndexedBase('x')[Idx(j, n)]
    assert p._print_Indexed(x) == 'x[j]'
    A = IndexedBase('A')[Idx(i, m), Idx(j, n)]
    assert p._print_Indexed(A) == 'A[%s]' % str(j + n*i)
    B = IndexedBase('B')[Idx(i, m), Idx(j, n), Idx(k, o)]
    assert p._print_Indexed(B) == 'B[%s]' % str(k + i*n*o + j*o)

    assert p._not_c == set()

 def _print_Mul(self, expr):
     if len(expr.args) == 2: # log2 and log10
         a, b = expr.args
         if is_inv(a) and is_log(a.base) and is_log(b):
             log_base = a.base.args[0]
             if log_base in [2, 10]:
                 return "log%d(%s)" % (log_base, self._print(b.args[0]))
     return CCodePrinter._print_Mul(self, expr)

def test_ccode_Indexed():
    from sympy.tensor import IndexedBase, Idx
    from sympy import symbols
    n, m, o = symbols('n m o', integer=True)
    i, j, k = Idx('i', n), Idx('j', m), Idx('k', o)

    x = IndexedBase('x')[j]
    A = IndexedBase('A')[i, j]
    B = IndexedBase('B')[i, j, k]

    with warnings.catch_warnings():
        warnings.filterwarnings("ignore", category=SymPyDeprecationWarning)
        p = CCodePrinter()
        p._not_c = set()

        assert p._print_Indexed(x) == 'x[j]'
        assert p._print_Indexed(A) == 'A[%s]' % (m*i+j)
        assert p._print_Indexed(B) == 'B[%s]' % (i*o*m+j*o+k)
        assert p._not_c == set()

 def _print_Pow(self, expr):
     if expr.base == 2:
         return "exp2(%s)" % self._print(expr.exp)
     if expr.base == 10:
         return "exp10(%s)" % self._print(expr.exp)
     if expr.exp == 0.5 and isinstance(expr.base, Add):
         args = expr.base.args
         if len(args) == 2 and all(map(is_square, args)):
             if self.order != 'none':
                 args = self._as_ordered_terms(expr.base, order=None)
                 # TODO: (above) is it fine to use `order=None`?
             return "hypot(%s, %s)" % (self._print(args[0].args[0]),
                                       self._print(args[1].args[0]))
     if expr.exp == Rational(1, 3):
         return "cbrt(%s)" % self._print(expr.base)
     if expr.exp.is_integer and expr.exp != -1:
         return "pown(%s, %s)" % (
             self._print(expr.base), self._print(expr.exp))
     if expr.exp.is_positive and expr.exp != 0.5:
         return "powr(%s, %s)" % (
             self._print(expr.base), self._print(expr.exp))
     return CCodePrinter._print_Pow(self, expr)

 def _indent_code(self, codelines):
     p = CCodePrinter()
     return p.indent_code(codelines)

 def __init__(self, settings={}):
     CCodePrinter.__init__(self, settings)

 def __init__(self, settings={}):
     settings = dict(settings)
     userfuncs = settings.setdefault('user_functions', {})
     for k, v in known_functions.items():
         userfuncs.setdefault(k, v)
     CCodePrinter.__init__(self, settings)