import java.awt.Color;

public class Plot2Dez
{ 
   public static void show(ComplexFunction f, int sz)
   {
      StdDraw.setCanvasSize(sz, sz);
      StdDraw.setXscale(0, sz);
      StdDraw.setYscale(0, sz);
      double scale = 2.5;
      for (int i = 0; i < sz; i++)
         for (int j = 0; j < sz; j++)
         {
	    double x = ((1.0*i)/sz - .5)*scale;
            double y = ((1.0*j)/sz - .5)*scale;
            Complex z = new Complex(x, y);
            double val = f.eval(z).abs()*100;
            int t;
            if      (val <   0) t = 0;
            else if (val > 255) t = 255;
            else                t = (int) val;
            Color c = new Color(255-t, 255-t, 255-t);
            StdDraw.setPenColor(c);
            StdDraw.pixel(i, j);
         }

      Color c = new Color(0, 0, 0);
      StdDraw.setPenColor(c);
      StdDraw.line(sz/2, 0, sz/2, sz);
      StdDraw.line(0, sz/2, sz, sz/2);

      StdDraw.show();
   }

}