draw: unify linear and elts draw jit functions