[firedrake] defining an expression piecewise