Skip to content

Commit 0c914f2

Browse files
committed
symbolics: cleaner repr of Cast
1 parent f1a082d commit 0c914f2

2 files changed

Lines changed: 10 additions & 6 deletions

File tree

devito/symbolics/extended_sympy.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010
from devito.finite_differences.elementary import Min, Max
1111
from devito.tools import (Pickable, Bunch, as_tuple, is_integer, float2, # noqa
1212
float3, float4, double2, double3, double4, int2, int3,
13-
int4, dtype_to_ctype, ctypes_to_cstr, ctypes_vector_mapper)
13+
int4, dtype_to_ctype, ctypes_to_cstr, ctypes_vector_mapper,
14+
ctypes_to_cstr)
1415
from devito.types import Symbol
1516
from devito.types.basic import Basic
1617

@@ -422,7 +423,10 @@ def _C_ctype(self):
422423

423424
@property
424425
def _op(self):
425-
return '(%s)' % self._C_ctype
426+
return '(%s)' % ctypes_to_cstr(self._C_ctype)
427+
428+
def __str__(self):
429+
return "%s%s" % (self._op, self.base)
426430

427431

428432
class IndexedPointer(sympy.Expr, Pickable, BasicWrapperMixin):

examples/userapi/06_sparse_operations.ipynb

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -277,8 +277,8 @@
277277
"name": "stdout",
278278
"output_type": "stream",
279279
"text": [
280-
"Eq(posx, (int)(floor((-o_x + s_coords(p_s, 0))/h_x)))\n",
281-
"Eq(posy, (int)(floor((-o_y + s_coords(p_s, 1))/h_y)))\n",
280+
"Eq(posx, (int)floor((-o_x + s_coords(p_s, 0))/h_x))\n",
281+
"Eq(posy, (int)floor((-o_y + s_coords(p_s, 1))/h_y))\n",
282282
"Eq(px, -floor((-o_x + s_coords(p_s, 0))/h_x) + (-o_x + s_coords(p_s, 0))/h_x)\n",
283283
"Eq(py, -floor((-o_y + s_coords(p_s, 1))/h_y) + (-o_y + s_coords(p_s, 1))/h_y)\n",
284284
"Eq(sum, 0.0)\n",
@@ -484,8 +484,8 @@
484484
"name": "stdout",
485485
"output_type": "stream",
486486
"text": [
487-
"Eq(posx, (int)(floor((-o_x + s_coords(p_s, 0))/h_x)))\n",
488-
"Eq(posy, (int)(floor((-o_y + s_coords(p_s, 1))/h_y)))\n",
487+
"Eq(posx, (int)floor((-o_x + s_coords(p_s, 0))/h_x))\n",
488+
"Eq(posy, (int)floor((-o_y + s_coords(p_s, 1))/h_y))\n",
489489
"Eq(sum, 0.0)\n",
490490
"Inc(sum, wsincrsx(p_s, rsx + 3)*wsincrsy(p_s, rsy + 3)*f(t, rsx + posx, rsy + posy))\n",
491491
"Eq(s(time, p_s), sum)\n"

0 commit comments

Comments
 (0)