Graph functions of complex numbers in Processing and/or P5

Let's say you have a function

f(x) = x^2 + 1

which has no root in the domain of the Real numbers, but still has two roots in the complex numbers.

It would be nice to be able to show the function graph in Processing/p5 in 3D, where the x and z axis represent the complex input (x+zi) and the y axis represents the length of the resulting complex "vector". The function value can then be colored with the vector's angle (0 to 360 or -180 to 180 or -pi to pi etc.)

Can you do this, maybe also showing the axes?