/* polarizability of hydrogen-like atom QED3D */
/* coded by Ikeuchi Mitsuru     */
/* ver 0.0.1   2004.05.21       */
/* ver 0.0.2   2004.05.24  improved reset code */

import java.awt.*;
import java.awt.event.*;
import java.applet.*;

public class polarizabilityQED3D extends Applet
  implements MouseListener, MouseMotionListener,
    ItemListener, ActionListener, AdjustmentListener, Runnable {

/* -------------------------------------- set global ------ */

  Choice cvw;
  Button bt_reset, bt_start, bt_stop,bt_loss;
  Scrollbar spx,spy,spz,sef;
  Thread th = null;

  Dimension dim;
  Image imgOff;
  Graphics gOff;

  int sleepTime = 30;

  int dispMode = 9;
  double px0 = 0.0;
  double py0 = 0.0;
  double pz0 = 0.0;
  double xField = 0.01;
  int lossFlag = 0;
  int fieldSetFlag = 0;
  int started = 1;


  int dgX, dgY, dgXb,dgYb;
  double viewTheta = -15.0*3.14/180.0;
  double viewFai = -72.0*3.14/180.0;
  double dtheta = 0.0*3.14/180.0;
  double pai = 3.1415926536;

  double cosTh = Math.cos(viewTheta); 
  double sinTh = Math.sin(viewTheta);
  double cosFi = Math.cos(viewFai);
  double sinFi = Math.sin(viewFai);


  double t = 0.0;

  double dx = 8.0/64.0;
  double dy = 8.0/64.0;
  double dz = 8.0/64.0;
  double dt = 1.0*(dx*dx);

  int NNx = 64+1;
  int NNy = 64+1;
  int NNz = 64+1;
  int NN = NNx+1; /* max(NNx,NNy,NNz)+1 */
  int NNr = 500;

  double phRe[][][] = new double[NNx][NNy][NNz];
  double phIm[][][] = new double[NNx][NNy][NNz];  

  double vv[][][] = new double[NNx][NNy][NNz];

  double aaRe[] = new double[NN];
  double aaIm[] = new double[NN];
  double bbRe[] = new double[NN];
  double bbIm[] = new double[NN];
  double bRe[] = new double[NN];
  double bIm[] = new double[NN];
  double uRe[] = new double[NN];
  double uIm[] = new double[NN];

  int xpts[] = new int[NN];
  int ypts[] = new int[NN];
  double icol[] = new double[NN];

  int pxBox[] = new int[8];
  int pyBox[] = new int[8];
  int boxp[][] = {
   {0,1},{0,2},{0,4},{1,3},{1,5},{2,3},{2,6},{4,5},{4,6},{3,7},{5,7},{6,7}
  };

  double srh[] = new double[NNr+1];
  int xptr[] = new int[NNr+1];
  int yptr[] = new int[NNr+1];
  int zptr[] = new int[NNr+1];


/* ----------------------------- applet control ------ */

  public void init() {

    resize(630,320);
    setBackground(Color.white);

    dim = getSize();
    imgOff = createImage(dim.width,dim.height);
    gOff = imgOff.getGraphics();

    cvw = new Choice();
    cvw.addItem("3D"); cvw.addItem("phase"); cvw.addItem("cloud");
    cvw.addItem("p flow"); cvw.addItem("p+dens");
    cvw.addItem("v flow"); cvw.addItem("v+dens");
    cvw.addItem("grid ph"); cvw.addItem("grid V");cvw.addItem("grid");
    cvw.addItem("dens"); cvw.addItem("ph 2D");
    cvw.addItemListener(this);
    cvw.select("grid");

    bt_reset= new Button("reset");
    bt_reset.addActionListener(this);

    bt_start= new Button("start");
    bt_start.addActionListener(this); 

    bt_stop = new Button("stop");
    bt_stop.addActionListener(this);

    bt_loss = new Button("loss");
    bt_loss.addActionListener(this);
    
    spx= new Scrollbar(Scrollbar.HORIZONTAL,50,10,0,110);
    spx.addAdjustmentListener(this);

    spy= new Scrollbar(Scrollbar.HORIZONTAL,0,10,0,110);
    spy.addAdjustmentListener(this);

    spz= new Scrollbar(Scrollbar.HORIZONTAL,0,10,0,110);
    spz.addAdjustmentListener(this);

    sef= new Scrollbar(Scrollbar.HORIZONTAL,10,10,0,110);
    sef.addAdjustmentListener(this);

    addMouseListener(this);  
    addMouseMotionListener(this);

    setLayout(new BorderLayout());
    Panel pnl = new Panel();
    pnl.setLayout(new GridLayout(1,6,5,0));
    pnl.add(bt_reset);
    pnl.add(bt_start);
    pnl.add(bt_stop);
    pnl.add(cvw);
    pnl.add(sef);
    pnl.add(bt_loss);
    add(pnl,"North");

    setInitialCondition();
  }

  public void start() {
    if (th == null) {
      th = new Thread(this);
      th.start();
    }
  }

  public void stop() {
    if (th != null) {
      th.stop();
      th = null;
    }
  }

  public void itemStateChanged(ItemEvent ev){
    if (ev.getSource() == cvw){
      dispMode = cvw.getSelectedIndex();
    }
  }

  public void actionPerformed(ActionEvent ev){
    if(ev.getSource() == bt_reset){ 
      started = -1;
    } else if(ev.getSource() == bt_start){ 
      started = 1;
    } else if(ev.getSource() == bt_stop){
      started = 0;
    } else if(ev.getSource() == bt_loss){
      lossFlag = 1;
    }
		}

  public void adjustmentValueChanged(AdjustmentEvent ev){
    if (ev.getSource() == spx) { 
      px0 = 0.1*(double)(spx.getValue());
    } else if (ev.getSource() == spy) { 
      py0 = 0.1*(double)(spy.getValue());
    } else if (ev.getSource() == spz) { 
      pz0 = 0.1*(double)(spz.getValue());
    } else if (ev.getSource() == sef) { 
      xField = 0.001*(double)(sef.getValue());
      fieldSetFlag = 1;
    }
  }

  public void mousePressed(MouseEvent ev){ }
  public void mouseReleased(MouseEvent ev){
    dgXb = 0; dgYb = 0;
    dgX = 0; dgY = 0;
  }
  public void mouseClicked(MouseEvent ev){ }
  public void mouseEntered(MouseEvent ev){ } 
  public void mouseExited (MouseEvent ev){ }

  public void mouseMoved  (MouseEvent ev){ }

  public void mouseDragged(MouseEvent ev){
    dgXb = dgX; dgYb = dgY;
    dgX=ev.getX();
    dgY=ev.getY();
    if (dgXb==0 && dgYb==0) {
      dgXb = dgX; dgYb = dgY;
    }
    viewTheta += 0.5*3.14/180.0*(dgX-dgXb);
    viewFai += 0.5*3.14/180.0*(dgY-dgYb);
    cosTh = Math.cos(viewTheta); 
    sinTh = Math.sin(viewTheta);
    cosFi = Math.cos(viewFai);
    sinFi = Math.sin(viewFai);
  }

  public void run() {
    while (th != null) {
      try {
        timeEvolution();
        offPaint();
        repaint();
        Thread.sleep(sleepTime);
      } catch (InterruptedException e) { }
    }
  }

  public void update(Graphics g) {
    paint(g);
  }

  public void paint(Graphics g) {
    g.drawImage(imgOff,0,0,this);
  }

/* ----------------------------- offPaint -------------------- */

  void offPaint() {
    int gb;
    double ev,ek,mx;
    double dens;

    gOff.setColor(Color.white);
    gOff.fillRect(0,0,dim.width,dim.height);

    dens = 200.0;
    ev = meanPotential();
    ek = meanKinetic();

    if (dispMode==0) {

      boxPlot();
      dens3DPlot(dens,0);
      boxPlot2();

    } else if (dispMode==1) {

      boxPlot();
      dens3DPlot(dens,1);
      boxPlot2();

    } else if (dispMode==2) {

      boxPlot();
      cloud3DPlot();
      boxPlot2();

    } else if (dispMode==3) {

      boxPlot();
      momentum3DPlot(dens,0);
      boxPlot2();

    } else if (dispMode==4) {

      boxPlot();
      momentum3DPlot(dens,1);
      boxPlot2();

    } else if (dispMode==5) {

      boxPlot();
      momentum3DPlot(dens,2);
      boxPlot2();

    } else if (dispMode==6) {

      boxPlot();
      momentum3DPlot(dens,3);
      boxPlot2();

    } else if (dispMode==7) {

      gridPlot(NNz/2, dens, 0.0);

    } else if (dispMode==8) {

      gridPlot(NNz/2, 0.0, 5.0);

    } else if (dispMode==9) {

      gridPlot(NNz/2, dens, 5.0);

    } else if (dispMode==10) {

      densPlot(25, 0);

    } else if (dispMode==11) {

      densPlot(25, 1);

    }


    gOff.setColor(Color.blue);
    gOff.drawString("t="+(int)(t*2418.9+0.5)/100.0+" as",10,40);
    gOff.drawString("norm="+norm()+" ",630/6*1+10,40);

    gOff.setColor(Color.black);
    gOff.drawString("view",630/6*3+10,40);
    gOff.drawString("Ex="+(int)(xField*1000.0+0.5)+"/1000 au",630/6*4+10,40);

    gb = 360;
    gOff.setColor(Color.black);
    gOff.drawString("box="+(int)(NNx*dx)+"x"+ +(int)(NNy*dy)+"x"+(int)(NNz*dz)+" au(1 au = 52.918 pm)",gb,60);

    gOff.setColor(Color.blue);
    gOff.drawString("<V>="+ev+" au",gb,100);
    gOff.drawString("<K>="+ek+" au",gb,120);
    gOff.drawString("<K>+<V>="+(ek+ev)+" au",gb,140);
    mx = meanX()-4.0;
    gOff.drawString("<x-x0>="+(mx)+" au",gb,180);
    if (xField>0.0) {
      gOff.drawString("<x-x0>/Ex="+(mx/xField)+" au",gb,200);
    }
    gOff.setColor(Color.black);
    gOff.drawString("V(r)= -1.0/r (r>=0.1)",gb,240);
    gOff.drawString("    = -10.0 (r<0.1)",gb,260);
    gOff.drawString("phi(r @t=0) = A exp(-r)",gb,280);

  }

/*------------------------------- plot methods  ------------*/

  void densPlot(double weight, int mode) {
    int i,j,k;
    int gbx,gby,mag;
    double ph2,col;

    gbx = 30;
    gby = 70;
    mag = 3;
    gOff.setColor(Color.black);
    gOff.fillRect(gbx-1,gby-1,mag*NNx+1,mag*NNy+1);

    k = NNz/2;
    for (i=1; i<NNx-1; i++) {
      for (j=1; j<NNy-1; j++) {
        if (vv[i][j][k]>0.0) {
          gOff.setColor(Color.getHSBColor(0.33f, 0.95f, 0.6f));
          gOff.fillRect(gbx+mag*i,gby+mag*j, mag, mag);
        }
        ph2 = weight*(phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k]);
        if (ph2>0.1) {
          if (ph2>1.0) {
            ph2 = 1.0;
		        }
          if (mode==0) {
            gOff.setColor(Color.getHSBColor(0.15f, 0.95f, (float)(ph2)));
          } else if (mode==1) {
            col = 0.5+0.5*phRe[i][j][k]/Math.sqrt(ph2/weight);
            gOff.setColor(Color.getHSBColor((float)col, 0.9f, (float)(ph2)));
          }
          gOff.fillRect(gbx+mag*i,gby+mag*j, mag, mag);
        }
      }
    }
  }

  void gridPlot(int zPos,double phWeight, double vvWeight) {
    int i,j,k,gbx,gby;
    int cx,cy,cz;
    double z,px,py,pz,sc;
    double ph2;

    cx = NNx/2;
    cy = NNy/2;
    cz = 0;
    sc = 4.0;

    gbx = 30+(int)(cx/sc);
    gby = 20+(int)(cy/sc);
    k = zPos;

    for(j=0; j<NNy; j += 2) {
      for (i=0; i<NNx; i++) { 
        ph2 = phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k];
        z = phWeight*ph2 + vvWeight*vv[i][j][k] + cz;
        py = cosFi*(j-cy)+sinFi*(z-cz) + cy; 
        pz = -sinFi*(j-cy)+cosFi*(z-cz) + cz;
        px = cosTh*(i-cx)+sinTh*(pz-cz) + cx; 
        pz = -sinTh*(i-cx)+cosTh*(pz-cz) + cz;
        xpts[i] = (int)(sc*px)+gbx;
        ypts[i] = (int)(sc*py)+gby;
        if (ph2>0.004) {
          icol[i] = 0.15;
        } else if (vv[i][j][k]>0.0) {
          icol[i] = 0.33;
        } else {
          icol[i] = 0.66;
        }
      }
      for (i=0; i<NNx-1; i++) {
        gOff.setColor(Color.getHSBColor((float)icol[i],0.9f,0.9f));
        gOff.drawLine(xpts[i],ypts[i],xpts[i+1],ypts[i+1]);
      }
    }

    for(i=0; i<NNx; i += 2) {
      for (j=0; j<NNy; j++) {
        ph2 = phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k];
        z = phWeight*ph2 + vvWeight*vv[i][j][k] + cz;
        py = cosFi*(j-cy)+sinFi*(z-cz) + cy; 
        pz = -sinFi*(j-cy)+cosFi*(z-cz) + cz;
        px = cosTh*(i-cx)+sinTh*(pz-cz) + cx; 
        pz = -sinTh*(i-cx)+cosTh*(pz-cz) + cz;
        xpts[j] = (int)(sc*px)+gbx;
        ypts[j] = (int)(sc*py)+gby;
        if (ph2>0.004) {
          icol[j] = 0.15;
        } else if (vv[i][j][k]>0.0) {
          icol[j] = 0.33;
        } else {
          icol[j] = 0.66;
        }
      }
      for (j=0; j<NNy-1; j++) { 
        gOff.setColor(Color.getHSBColor((float)icol[j],0.9f,0.9f));
        gOff.drawLine(xpts[j],ypts[j],xpts[j+1],ypts[j+1]);
      }
    }
  }


  void dens3DPlot(double weight,int mode) {
    int i,j,k,ix,iy,ir;
    int gbx,gby;
    double cxx,cyy,czz;
    double z,ppx,ppy,ppz;
    double mag,dens,col;

    gbx = 60;
    gby = 80;
    mag = 3.0;
    cxx = mag*(NNx/2.0); 
    cyy = mag*(NNy/2.0);
    czz = mag*(NNz/2.0);

    for (i=0; i<NNx; i+=2) {
      for(j=0; j<NNy; j +=2) {
        for (k=0; k<NNz; k+=2) {
          if (vv[i][j][k]>0.0) {
            z = k*mag;
            ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;
            ir = 2;
            ix = (int)(ppx)+gbx;
            iy = (int)(ppy)+gby;
            gOff.setColor(Color.getHSBColor(0.33f, 0.9f,0.9f));
            gOff.fillOval(ix-ir/2,iy-ir/2,ir,ir);
          }
          dens =  phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k];
          ir = (int)(weight*dens);
          if (ir>0) {
            if (ir>10) {
              ir = 10;
            }
            z = k*mag;
            ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;

            ix = (int)(ppx)+gbx;
            iy = (int)(ppy)+gby;
            col = 0.0;
            if (mode==0) {
              col = 0.66 -0.05*dens;
              if (col<0.0) {
                col = 0.0;
              }
            } else if (mode==1) {
              col = 0.5+0.5*phRe[i][j][k]/Math.sqrt(dens);
            }
            gOff.setColor(Color.getHSBColor((float)col, 0.9f,0.9f));
            gOff.fillOval(ix-ir/2,iy-ir/2,ir,ir);
          }
        }
      }
    }

  }


  void cloud3DPlot() {
    int i,j,k,ix,iy,ir,m;
    int gbx,gby;
    double cxx,cyy,czz;
    double z,ppx,ppy,ppz;
    double mag;

    setsrh();

    gbx = 60;
    gby = 80;
    mag = 3.0;
    cxx = mag*(NNx/2.0); 
    cyy = mag*(NNy/2.0);
    czz = mag*(NNz/2.0);
    ir = 4;

    for (i=1; i<NNx-1; i+=2) {
      for (j=1; j<NNy-1; j +=2) {
        for (k=1; k<NNz-1; k+=2) {  
          if (vv[i][j][k]>0.0) {
            z = k*mag;
            ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;

            ix = (int)(ppx)+gbx;
            iy = (int)(ppy)+gby;
            gOff.setColor(Color.getHSBColor(0.33f, 0.95f,0.95f));
            gOff.fillOval(ix-2,iy-2,4,4);
          }
        }
      }
    }

    for (m=0; m<NNr-1; m++) {
      i=xptr[m]; j=yptr[m]; k=zptr[m]; 
      z = k*mag;
      ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
      ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
      ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
      ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;

      ix = (int)(ppx)+gbx;
      iy = (int)(ppy)+gby;
      gOff.setColor(Color.getHSBColor(0.66f, 0.9f,0.9f));
      gOff.fillOval(ix-ir/2,iy-ir/2,ir,ir);
    }
  }


  void momentum3DPlot(double weight,int mode) {
    int i,j,k,ix,iy,ix2,iy2,ir;
    int gbx,gby;
    double cxx,cyy,czz;
    double z,ppx,ppy,ppz;
    double mag,dens,col,amp;
    double px,pxphRe,pxphIm,py,pyphRe,pyphIm,pz,pzphRe,pzphIm,pAbs;

    gbx = 60;
    gby = 80;
    mag = 3.0;
    amp = weight;
    cxx = mag*(NNx/2.0); 
    cyy = mag*(NNy/2.0);
    czz = mag*(NNz/2.0);

    for (i=2; i<NNx-1; i+=4) {
      for(j=2; j<NNy-1; j +=4) {
        for (k=2; k<NNz-1; k+=4) {
          if (vv[i][j][k]>0.0) {
            z = k*mag;
            ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;
            ir = 2;
            ix = (int)(ppx)+gbx;
            iy = (int)(ppy)+gby;
            gOff.setColor(Color.getHSBColor(0.33f, 0.9f,0.9f));
            gOff.fillOval(ix-ir/2,iy-ir/2,ir,ir);
          }
          dens =  phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k];
          if (dens>0.0025) {
            pxphRe = (phIm[i+1][j][k]-phIm[i-1][j][k])/(2*dx);
            pxphIm = (-phRe[i+1][j][k]+phRe[i-1][j][k])/(2*dx);
            px = (phRe[i][j][k]*pxphRe + phIm[i][j][k]*pxphIm);
            pyphRe = (phIm[i][j+1][k]-phIm[i][j-1][k])/(2*dy);
            pyphIm = (-phRe[i][j+1][k]+phRe[i][j-1][k])/(2*dy);
            py = (phRe[i][j][k]*pyphRe + phIm[i][j][k]*pyphIm);
            pzphRe = (phIm[i][j][k+1]-phIm[i][j][k-1])/(2*dz);
            pzphIm = (-phRe[i][j][k+1]+phRe[i][j][k-1])/(2*dz);
            pz = (phRe[i][j][k]*pzphRe + phIm[i][j][k]*pzphIm);

            z = k*mag;
            ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;
            ix = (int)(ppx)+gbx;
            iy = (int)(ppy)+gby;
            if (mode==2 || mode==3) {
              px = 0.02*px/dens; py = 0.02*py/dens; pz = 0.02*pz/dens;
            }
            z = k*mag+pz*amp;
            ppy = cosFi*(j*mag+py*amp-cyy)+sinFi*(z-czz) + cyy; 
            ppz = -sinFi*(j*mag+py*amp-cyy)+cosFi*(z-czz) + czz;
            ppx = cosTh*(i*mag+px*amp-cxx)+sinTh*(ppz-czz) + cxx; 
            ppz = -sinTh*(i*mag+px*amp-cxx)+cosTh*(ppz-czz) + czz;
            ix2 = (int)(ppx)+gbx;
            iy2 = (int)(ppy)+gby;
            gOff.setColor(Color.getHSBColor(0.01f, 0.9f,0.9f));
            gOff.drawLine(ix,iy,ix2,iy2);
            if (mode==1 || mode==3) {
              dens =  phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k];
              ir = (int)(weight*dens); 
              if (ir>5) ir = 5;
			  gOff.setColor(Color.getHSBColor(0.66f, 0.9f,0.9f));
              gOff.fillOval(ix-ir/2,iy-ir/2,ir,ir);
            }
          }
        }
      }
    }

  }


  void boxPlot() {
    int i,j,k,ip,i1,i2;
    int gbx,gby;
    double cxx,cyy,czz;
    double z,ppx,ppy,ppz;
    double mag,dens;

    gbx = 60;
    gby = 80;
    mag = 3.0;
    cxx = mag*(NNx/2.0); 
    cyy = mag*(NNy/2.0);
    czz = mag*(NNz/2.0);

    ip = 0;
    for (i=0; i<NNx; i+=NNx-1) {
      for(j=0; j<NNy; j+=NNy-1) {
        for (k=0; k<NNz; k+=NNz-1) {
          z = k*mag;
          ppy = cosFi*(j*mag-cyy)+sinFi*(z-czz) + cyy; 
          ppz = -sinFi*(j*mag-cyy)+cosFi*(z-czz) + czz;
          ppx = cosTh*(i*mag-cxx)+sinTh*(ppz-czz) + cxx; 
          ppz = -sinTh*(i*mag-cxx)+cosTh*(ppz-czz) + czz;

          pxBox[ip] = (int)(ppx)+gbx;
          pyBox[ip] = (int)(ppy)+gby;
          ip += 1;
        }
      }
    }
    gOff.setColor(Color.gray);
    for (i=0; i<12; i++) {
      i1= boxp[i][0]; i2= boxp[i][1];
      if (i1==0) {
        gOff.setColor(Color.gray);
        gOff.drawLine(pxBox[i1],pyBox[i1],pxBox[i2],pyBox[i2]);
      }
    }
  }

  void boxPlot2() {
    int i,i1,i2;

    for (i=0; i<12; i++) {
      i1= boxp[i][0]; i2= boxp[i][1];
      if (i1 != 0) {
        gOff.setColor(Color.black);
        gOff.drawLine(pxBox[i1],pyBox[i1],pxBox[i2],pyBox[i2]);
      }
    }
  }

/*-------------------------- statistics ----------*/

  double norm() {
    int i,j,k;
    double p;

    p = 0.0;
    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          p += (phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k]);
        }
      }
    }
    return ( p*dx*dy*dz );
  }

  double meanPotential() {
    int i,j,k;
    double p;

    p = 0.0;
    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          p += vv[i][j][k]*(phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k]);
        }
      }
    }
    return ( p*dx*dy*dz );
  }

  double meanKinetic() {
    int i,j,k;
    double p,d2phRe,d2phIm,h2;

    h2 = dx*dx;
    p = 0.0;
    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          d2phRe = (phRe[i+1][j][k]+phRe[i-1][j][k]+phRe[i][j+1][k]+phRe[i][j-1][k]+phRe[i][j][k+1]+phRe[i][j][k-1]-6.0*phRe[i][j][k])/h2;
          d2phIm = (phIm[i+1][j][k]+phIm[i-1][j][k]+phIm[i][j+1][k]+phIm[i][j-1][k]+phIm[i][j][k+1]+phIm[i][j][k-1]-6.0*phIm[i][j][k])/h2;
          p += (phRe[i][j][k]*d2phRe+phIm[i][j][k]*d2phIm);
        }
      }
    }
    return ( -0.5*p*dx*dy*dz );
  }

  double meanX() {
    int i,j,k;
    double x,p;

    p = 0.0;
    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          x = i*dx;
          p += x*(phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k]);
        }
      }
    }
    return ( p*dx*dy*dz );
  }

  void setsrh() {
    int i,j,k,m;
    double s;

    for (i=0;i<NNr;i++) {
      srh[i] = Math.random();
    }

    for (i=0;i<NNr;i++) {
      for (j=0;j<NNr-1;j++) {
        if (srh[j]>srh[j+1]) {
          s = srh[j+1]; srh[j+1] = srh[j]; srh[j] = s;
        }
      }
    }

    m = 0;
    s = 0.0;
    for (i=0;i<NNx;i++) {
      for (j=0;j<NNy;j++) {
        for (k=0;k<NNz;k++) {
          s += (phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k])*dx*dy*dz;
          while (s>srh[m] && m<NNr) {
            xptr[m] = i; yptr[m] = j; zptr[m] = k;
            m +=1;
          }
        }
      }
    }
  }


/*-------------------------- set loss energy ----------*/

  void setLoss() {
    int i,j,k;
    double a;

    for (i=0;i<NNx;i++) {
      for (j=0;j<NNy;j++) {
        for (k=0;k<NNz;k++) {
          a = Math.sqrt(phRe[i][j][k]*phRe[i][j][k]+phIm[i][j][k]*phIm[i][j][k]);
          phRe[i][j][k] = a;
          phIm[i][j][k] = 0.0;
        }
      }
    }
  }


/*-------------------------- set initial condition ----------*/

  void setInitialCondition() {
    t = 0.0;
    setWave(4.0,4.0,4.0, 1.0, px0, py0,pz0);

    setPotential(4.0,4.0,4.0);

  }

/* ----------------------------- set potential -------------- */

  void setPotential(double x0,double y0,double z0) {
    int i,j,k;
    double x,y,z,r;

    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          x = i*dx;
          y = j*dy;
          z = k*dz;
          r = Math.sqrt((x-x0)*(x-x0)+(y-y0)*(y-y0)+(z-z0)*(z-z0));
          if (r<0.1) {
            r = 0.1;
          }
          vv[i][j][k] = -1.0/r+xField*(x-x0);

        }
      }
    }
  }

/* ----------------------------- set wave packet -------------- */

  void setWave(double xPos, double yPos, double zPos, double waveWidth, double kx, double ky, double kz) {
    int i,j,k;
    double a,x,y,z,r,phAb,phPh;

    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          x = i*dx;
          y = j*dy;
          z = k*dz;
          r = Math.sqrt((x-xPos)*(x-xPos)+(y-yPos)*(y-yPos)+(z-zPos)*(z-zPos));
          phAb = Math.exp( -r );
          phPh = kx*x+ky*y+kz*z;
          phRe[i][j][k] = phAb*Math.cos(phPh);
          phIm[i][j][k] = phAb*Math.sin(phPh);
        }
      }
    }

    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        phRe[i][j][0] = 0.0;
        phIm[i][j][0] = 0.0;
        phRe[i][j][NNz-1] = 0.0;
        phIm[i][j][NNz-1] = 0.0;
      }
    }
    for (i=1;i<NNx-1;i++) {
      for (k=1;k<NNz-1;k++) {
        phRe[i][0][k] = 0.0;
        phIm[i][0][k] = 0.0;
        phRe[i][NNy-1][k] = 0.0;
        phIm[i][NNy-1][k] = 0.0;
      }
    }
    for (j=1;j<NNy-1;j++) {
      for (k=1;k<NNz-1;k++) {
        phRe[0][j][k] = 0.0;
        phIm[0][j][k] = 0.0;
        phRe[NNx-1][j][k] = 0.0;
        phIm[NNx-1][j][k] = 0.0;
      }
    }

    a = Math.sqrt(norm());
	
    for (i=1;i<NNx-1;i++) {
      for (j=1;j<NNy-1;j++) {
        for (k=1;k<NNz-1;k++) {
          phRe[i][j][k] = phRe[i][j][k]/a;
          phIm[i][j][k] = phIm[i][j][k]/a;
        }
      }
    }

  }

/* ----------------------------- timeEvolution -------------- */

  void timeEvolution() {
    int i;

    if (started==-1) {
      t = 0.0;
      setInitialCondition();
      started = 0;
    } else if (started==1) {
      for (i=0; i<2; i++) {
        t = t + dt;
        phaseStep();
        kxStep();
        kyStep();
        kzStep();
        phaseStep();
      }
    }
    if (lossFlag==1) {
      setLoss();
      lossFlag = 0;
    } else if (fieldSetFlag==1) {
      setPotential(4.0,4.0,4.0);
      fieldSetFlag = 0;
    }
  }

/* ----------------------------- time step -------------- */

  void kxStep() {
    int i,j,k;
    double a,aaAb,auAb;

    a = 4.0*dy*dy/dt;
    for (k=1; k<NNz-1; k++) {
      for (j=1; j<NNy-1; j++) {
        for (i=1; i<NNx-1; i++) {
          bbRe[i] = 2.0;
          bbIm[i] = a;
          aaRe[i] = -2.0;
          aaIm[i] = a;
          bRe[i] = bbRe[i]*phRe[i][j][k]-bbIm[i]*phIm[i][j][k] - phRe[i+1][j][k] - phRe[i-1][j][k];
          bIm[i] = bbRe[i]*phIm[i][j][k]+bbIm[i]*phRe[i][j][k] - phIm[i+1][j][k] - phIm[i-1][j][k];
        }

        aaAb = aaRe[1]*aaRe[1]+aaIm[1]*aaIm[1];
        uRe[1] = aaRe[1]/aaAb;
        uIm[1] = -aaIm[1]/aaAb;
        phRe[1][j][k] = bRe[1]*uRe[1] - bIm[1]*uIm[1];
        phIm[1][j][k] = bIm[1]*uRe[1] + bRe[1]*uIm[1];

        for (i=2; i<NNx-1; i++) {
          auAb = (aaRe[i]-uRe[i-1])*(aaRe[i]-uRe[i-1])+(aaIm[i]-uIm[i-1])*(aaIm[i]-uIm[i-1]);
          uRe[i] = (aaRe[i]-uRe[i-1])/auAb;
          uIm[i] = -(aaIm[i]-uIm[i-1])/auAb;
          phRe[i][j][k] = (bRe[i]-phRe[i-1][j][k])*uRe[i] - (bIm[i]-phIm[i-1][j][k])*uIm[i];
          phIm[i][j][k] = (bRe[i]-phRe[i-1][j][k])*uIm[i] + (bIm[i]-phIm[i-1][j][k])*uRe[i];
        }

        for (i=NNx-3; i>=1; i--) {
          phRe[i][j][k] -= phRe[i+1][j][k]*uRe[i] - phIm[i+1][j][k]*uIm[i];
          phIm[i][j][k] -= phRe[i+1][j][k]*uIm[i] + phIm[i+1][j][k]*uRe[i];
        }
      }
    }
  }

  void kyStep() {
    int i,j,k;
    double a,aaAb,auAb;

    a = 4.0*dy*dy/dt;
    for (i=1; i<NNx-1; i++) {
      for (k=1; k<NNz-1; k++) {
        for (j=1; j<NNy-1; j++) {
          bbRe[j] = 2.0;
          bbIm[j] = a;
          aaRe[j] = -2.0;
          aaIm[j] = a;
          bRe[j] = bbRe[j]*phRe[i][j][k]-bbIm[j]*phIm[i][j][k] - phRe[i][j+1][k] - phRe[i][j-1][k];
          bIm[j] = bbRe[j]*phIm[i][j][k]+bbIm[j]*phRe[i][j][k] - phIm[i][j+1][k] - phIm[i][j-1][k];
        }

        aaAb = aaRe[1]*aaRe[1]+aaIm[1]*aaIm[1];
        uRe[1] = aaRe[1]/aaAb;
        uIm[1] = -aaIm[1]/aaAb;
        phRe[i][1][k] = bRe[1]*uRe[1] - bIm[1]*uIm[1];
        phIm[i][1][k] = bIm[1]*uRe[1] + bRe[1]*uIm[1];

        for (j=2; j<NNy-1; j++) {
          auAb = (aaRe[j]-uRe[j-1])*(aaRe[j]-uRe[j-1])+(aaIm[j]-uIm[j-1])*(aaIm[j]-uIm[j-1]);
          uRe[j] = (aaRe[j]-uRe[j-1])/auAb;
          uIm[j] = -(aaIm[j]-uIm[j-1])/auAb;
          phRe[i][j][k] = (bRe[j]-phRe[i][j-1][k])*uRe[j] - (bIm[j]-phIm[i][j-1][k])*uIm[j];
          phIm[i][j][k] = (bRe[j]-phRe[i][j-1][k])*uIm[j] + (bIm[j]-phIm[i][j-1][k])*uRe[j];
        }

        for (j=NNy-3; j>=1; j--) {
          phRe[i][j][k] -= phRe[i][j+1][k]*uRe[j] - phIm[i][j+1][k]*uIm[j];
          phIm[i][j][k] -= phRe[i][j+1][k]*uIm[j] + phIm[i][j+1][k]*uRe[j];
        }
      }
    }
  }

  void kzStep() {
    int i,j,k;
    double a,aaAb,auAb;

    a = 4.0*dy*dy/dt;
    for (i=1; i<NNx-1; i++) {
      for (j=1; j<NNy-1; j++) {
        for (k=1; k<NNz-1; k++) {
          bbRe[k] = 2.0;
          bbIm[k] = a;
          aaRe[k] = -2.0;
          aaIm[k] = a;
          bRe[k] = bbRe[k]*phRe[i][j][k]-bbIm[k]*phIm[i][j][k] - phRe[i][j][k+1] - phRe[i][j][k-1];
          bIm[k] = bbRe[k]*phIm[i][j][k]+bbIm[k]*phRe[i][j][k] - phIm[i][j][k+1] - phIm[i][j][k-1];
        }

        aaAb = aaRe[1]*aaRe[1]+aaIm[1]*aaIm[1];
        uRe[1] = aaRe[1]/aaAb;
        uIm[1] = -aaIm[1]/aaAb;
        phRe[i][j][1] = bRe[1]*uRe[1] - bIm[1]*uIm[1];
        phIm[i][j][1] = bIm[1]*uRe[1] + bRe[1]*uIm[1];

        for (k=2; k<NNz-1; k++) {
          auAb = (aaRe[k]-uRe[k-1])*(aaRe[k]-uRe[k-1])+(aaIm[k]-uIm[k-1])*(aaIm[k]-uIm[k-1]);
          uRe[k] = (aaRe[k]-uRe[k-1])/auAb;
          uIm[k] = -(aaIm[k]-uIm[k-1])/auAb;
          phRe[i][j][k] = (bRe[k]-phRe[i][j][k-1])*uRe[k] - (bIm[k]-phIm[i][j][k-1])*uIm[k];
          phIm[i][j][k] = (bRe[k]-phRe[i][j][k-1])*uIm[k] + (bIm[k]-phIm[i][j][k-1])*uRe[k];
        }

        for (k=NNz-3; k>=1; k--) {
          phRe[i][j][k] -= phRe[i][j][k+1]*uRe[k] - phIm[i][j][k+1]*uIm[k];
          phIm[i][j][k] -= phRe[i][j][k+1]*uIm[k] + phIm[i][j][k+1]*uRe[k];
        }
      }
    }

  }

  void phaseStep() {
    int i,j,k;
    double th,cs,sn,phr,phi;

    for (i=1; i<NNx-1; i++) {
      for (j=1; j<NNy-1; j++) {
        for (k=1; k<NNz-1; k++) {
          th = 0.5*dt*vv[i][j][k];
          cs = Math.cos(th); sn = Math.sin(th);
          phr = phRe[i][j][k];
          phi = phIm[i][j][k];
          phRe[i][j][k] = cs*phr+sn*phi;
          phIm[i][j][k] = cs*phi-sn*phr;
        }
      }
    }
  }

/* ----------------------------- end of applet -------------- */

}