/* applet No. 0352
 *
 *  tea break - Morse particles in the 4D box
 *
 * Created by Ikeuchi Mitsuru on September 06 2003.
 * Copyright (c) 2006 Ikeuchi Mitsuru. All rights reserved.
 *
 * ver 0.0.1   2003.09.06  created
 * ver 0.0.2   2006.11.25  added projection view 
 *
 */

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

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

/*-------------------------------------- define gloval -----*/

  Thread th = null;

  Choice ch_mat, ch_tempMode, ch_view;
  Scrollbar sc_contTemp, sc_scale;

  Dimension dim;
  Image imgOff;
  Graphics gOff;

  int sleepTime = 50;
  int atomSizeX = 10; 
  int atomSizeY = 10; 
  int atomSizeZ = 10;
  int atomSizeW = 10; 
  int dispR = 20;
  int paintSizeX = atomSizeX*dispR; 
  int paintSizeY = atomSizeY*dispR;
  int wakuX = paintSizeX + dispR;
  int wakuY = paintSizeY + dispR;

  int Nmt = 81;          /* number of molecules    */
  int NN = Nmt + 1;
  double t = 0.0;              /* time (s)               */
  double dt = 5.0*1.0e-15;     /* time division (s)      */
  double sgm = 3.40e-10;       /* L-J sigma for Ar (m)   */

  double xMax = atomSizeX*sgm; /* x-Box size in (m)      */
  double yMax = atomSizeY*sgm; /* y-Box size in (m)      */
  double zMax = atomSizeZ*sgm; /* z-Box size in (m)      */
  double wMax = atomSizeW*sgm; 

  double contTemp = 300.0;
  double scale = 1.0;
  int tempMode = 0;
  int dispMode = 0;

  int dgX, dgY, dgXb,dgYb;
  double cx = 0.5*xMax;
  double cy = 0.5*yMax;
  double cz = 0.5*zMax;
  double theta = -20.0*3.14/180.0;
  double fai = 10.0*3.14/180.0;
  double dtheta = 0.0*3.14/180.0;
  double pai = 3.1415926536;

  double rc = 1.0e-9;
  double rc2 = rc*rc;

  double rreg = 1.5e-9;
  double rreg2 = rreg*rreg;

  double AU = 1.661e-27;
  double EE = 1.602e-19;
  double AA = 1.0e-10;
  double Morse[][] = {
  /*         mass         charge   D(eV)     A(A^-1)    r0(A)   */
  /* 0 W */ {183.85 *AU,  0.0*EE, 0.9906*EE, 1.4116/AA, 3.032*AA},
  /* 1 Mo*/ { 95.94 *AU,  0.0*EE, 0.8032*EE, 1.5079/AA, 2.976*AA},
  /* 2 Cr*/ { 51.996*AU,  0.0*EE, 0.4414*EE, 1.5721/AA, 2.754*AA},
  /* 3 Fe*/ { 55.847*AU,  0.0*EE, 0.4174*EE, 1.3885/AA, 2.845*AA},
  /* 4 Ni*/ { 58.71 *AU,  0.0*EE, 0.4205*EE, 1.4199/AA, 2.780*AA},
  /* 5 Al*/ { 26.98 *AU,  0.0*EE, 0.2703*EE, 1.1646/AA, 3.253*AA},
  /* 6 Pb*/ {207.19 *AU,  0.0*EE, 0.2348*EE, 1.1836/AA, 3.733*AA},
  /* 7 Cu*/ { 63.54 *AU,  0.0*EE, 0.3429*EE, 1.3588/AA, 2.866*AA},
  /* 8 Ag*/ {107.87 *AU,  0.0*EE, 0.3323*EE, 1.3690/AA, 3.115*AA},
  /* 9 Ca*/ { 40.08 *AU,  0.0*EE, 0.1623*EE, 0.8054/AA, 4.569*AA},
  /*10 Sr*/ { 87.62 *AU,  0.0*EE, 0.1513*EE, 0.7878/AA, 4.988*AA},
  /*11 Ba*/ {137.34 *AU,  0.0*EE, 0.1416*EE, 0.6570/AA, 5.373*AA},
  /*12 Na*/ { 22.99 *AU,  0.0*EE, 0.0633*EE, 0.5900/AA, 5.336*AA},
  /*13 K */ { 39.102*AU,  0.0*EE, 0.0542*EE, 0.4977/AA, 6.369*AA},
  /*14 Rb*/ { 85.47 *AU,  0.0*EE, 0.0464*EE, 0.4298/AA, 7.207*AA},
  /*15 Cs*/ {132.905*AU,  0.0*EE, 0.0449*EE, 0.4157/AA, 7.557*AA},
  /*16 Ne*/ { 20.183*AU,  0.0*EE, 0.0031*EE, 1.6500/AA, 3.076*AA},
  /*17 Ar*/ { 39.948*AU,  0.0*EE, 0.0104*EE, 1.3400/AA, 3.816*AA},
  /*18 Kr*/ { 83.80 *AU,  0.0*EE, 0.0141*EE, 1.2500/AA, 4.097*AA},
  /*19 Xe*/ {131.30 *AU,  0.0*EE, 0.0200*EE, 1.2400/AA, 4.467*AA},
  /*20 Hg*/ {200.59 *AU,  0.0*EE, 0.0734*EE, 1.4900/AA, 3.255*AA}
  };
  int NmatKind = 21;

  int MAT0 = 0;
  int MAT1 = 0;
  int MAT2 = 0;
  int MAT3 = 0;
  int MAT4 = 0;
  
  int kind[] = new int[NN];       /* i-th molec kind */
  double xx[] = new double[NN];   /* i-th x-position */
  double yy[] = new double[NN];   /* i-th y-position */
  double zz[] = new double[NN];   /* i-th z-position */
  double ww[] = new double[NN];   /* i-th w-position */
  double vx[] = new double[NN];   /* i-th x-velocity */
  double vy[] = new double[NN];   /* i-th y-velocity */
  double vz[] = new double[NN];   /* i-th z-velocity */
  double vw[] = new double[NN];   /* i-th w-velocity */
  double ffx[] = new double[NN];  /* i-th x-force    */
  double ffy[] = new double[NN];  /* i-th y-force    */
  double ffz[] = new double[NN];  /* i-th z-force    */
  double ffw[] = new double[NN];  /* i-th w-force    */
  double fbx[] = new double[NN];  /* i-th x-force before dt */
  double fby[] = new double[NN];  /* i-th y-force before dt */
  double fbz[] = new double[NN];  /* i-th z-force before dt */
  double fbw[] = new double[NN];  /* i-th w-force before dt */

  int reg[][] = new int[NN][860];   /* registration */

  int srtz[][] = new int[2][NN];   /* z-sort work space */

  int kindT[] = new int[NmatKind];
  double forceTable[][][] = new double[5][5][10200];

  double px[] = new double[NN]; 
  double py[] = new double[NN]; 
  double pz[] = new double[NN]; 

  double wkx[] = new double[8]; 
  double wky[] = new double[8]; 
  double wkz[] = new double[8]; 

  double pwkx[] = new double[8]; 
  double pwky[] = new double[8]; 
  double pwkz[] = new double[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}
  };

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

  public void init() {
    resize(630,420);
    setBackground(Color.white);

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

    ch_mat = new Choice();
    ch_mat.addItem("W");  ch_mat.addItem("Mo"); ch_mat.addItem("Cr");
    ch_mat.addItem("Fe"); ch_mat.addItem("Ni"); ch_mat.addItem("Al");
    ch_mat.addItem("Pb"); ch_mat.addItem("Cu"); ch_mat.addItem("Ag"); 
    ch_mat.addItem("Ca"); ch_mat.addItem("Sr"); ch_mat.addItem("Ba");
    ch_mat.addItem("Na"); ch_mat.addItem("K");  ch_mat.addItem("Rb");
    ch_mat.addItem("Cs"); ch_mat.addItem("Ne"); ch_mat.addItem("Ar");
    ch_mat.addItem("Kr"); ch_mat.addItem("Xe");
    ch_mat.addItemListener(this);

    ch_tempMode = new Choice();
    ch_tempMode.addItem("wall cont");
    ch_tempMode.addItem("adiabatic");
    ch_tempMode.addItem("scaling");
    ch_tempMode.addItemListener(this); 

    ch_view = new Choice();
    ch_view.addItem("ball");
    ch_view.addItem("line");
    ch_view.addItem("both");
    ch_view.addItem("real");
    ch_view.addItem("temp");
    ch_view.addItemListener(this); 

    sc_contTemp = new Scrollbar(Scrollbar.HORIZONTAL,30,10,10,210);
    sc_contTemp.addAdjustmentListener(this);

    sc_scale = new Scrollbar(Scrollbar.HORIZONTAL,100,10,50,210);
    sc_scale.addAdjustmentListener(this);

    addMouseListener(this);  
    addMouseMotionListener(this);

    setLayout(new BorderLayout());
    Panel pnl = new Panel();
    pnl.setLayout(new GridLayout(1,6,5,0));
    //pnl.add(new Label("  "));
    pnl.add(ch_mat);
    pnl.add(new Label("  "));
    pnl.add(ch_tempMode);
    pnl.add(sc_contTemp);
    pnl.add(sc_scale);
    pnl.add(ch_view);
    add(pnl,"North");

    setInitialPosition();
    setWaku();
  }

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

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

  public void itemStateChanged(ItemEvent ev){
    if (ev.getSource() == ch_mat){
      MAT1 = ch_mat.getSelectedIndex();
      t = 0.0;
      setInitialPosition();
    } else if (ev.getSource() == ch_tempMode){
      tempMode = ch_tempMode.getSelectedIndex();
    } else if (ev.getSource() == ch_view){
      dispMode = ch_view.getSelectedIndex();
    }
  }

  public void adjustmentValueChanged(AdjustmentEvent ev){
    if (ev.getSource() == sc_contTemp) { 
      contTemp = 10.0*(double)(sc_contTemp.getValue());

    } else if(ev.getSource() == sc_scale) { 
      scale = 0.01*(double)(sc_scale.getValue());

    }
  }

  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;
    }
    theta += 0.5*3.14/180.0*(dgX-dgXb);
    fai += 0.5*3.14/180.0*(dgY-dgYb);
  }

  public void run() {
    while (th != null) {
      try {
        calcposition();
        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 i,j,k,jj,farPoint;
    int gx,gy,gz,g2x,g2y,gbx,gby,gbz;
    double tmp,tm,clr;

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

    rotate(theta,fai);
    zSort();
    tmp = temperature();


    farPoint = findMin();
    for (i=0; i<12; i++) {
      gx = (int)(paintSizeX*scale*pwkx[boxp[i][0]]/xMax)+60;
      gy = (int)(paintSizeY*scale*pwky[boxp[i][0]]/yMax)+120;
      g2x = (int)(paintSizeX*scale*pwkx[boxp[i][1]]/xMax)+60;
      g2y = (int)(paintSizeY*scale*pwky[boxp[i][1]]/yMax)+120;
      if (boxp[i][0]==farPoint || boxp[i][1]==farPoint) { 
        gOff.setColor(Color.lightGray);
        gOff.drawLine(gx,gy, g2x, g2y);
      } 
    }


    if (dispMode==0) {

      for (i=0; i<Nmt; i++) {
        j = srtz[0][i];
        gx = (int)(paintSizeX*scale*px[j]/xMax)+60;
        gy = (int)(paintSizeY*scale*py[j]/yMax)+120;
        gz = (int)(5.0*scale*(0.68+0.3*(pz[j]/zMax))*Morse[kind[j]][4]/AA);
        clr = 0.66-0.66*(ww[j]/wMax);
        gOff.setColor(Color.getHSBColor(
          (float)clr,0.99f,(float)(0.5*pz[j]/zMax+0.35)));
        gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);
      }

    } else if (dispMode==1) {

      for (i=0; i<Nmt; i++) {
        j = srtz[0][i];
        gx = (int)(paintSizeX*scale*px[j]/xMax)+60;
        gy = (int)(paintSizeY*scale*py[j]/yMax)+120;
        for (k=1; k<reg[j][0]; k++) {
          jj = reg[j][k];
          if ( distance(j,jj)<0.6*(Morse[kind[j]][4]+Morse[kind[jj]][4]) ) {
            gbx = (int)(paintSizeX*scale*px[jj]/xMax)+60;
            gby = (int)(paintSizeY*scale*py[jj]/yMax)+120;
            //gOff.setColor(Color.gray);
            clr = 0.66-0.66*(ww[j]/wMax);
            gOff.setColor(Color.getHSBColor(
              (float)clr,0.99f,(float)(0.5*pz[j]/zMax+0.35)));
            gOff.drawLine(gx,gy, gbx, gby);
          }
        }
      }

    } else if (dispMode==2) {

      for (i=0; i<Nmt; i++) {
        j = srtz[0][i];
        gx = (int)(paintSizeX*scale*px[j]/xMax)+60;
        gy = (int)(paintSizeY*scale*py[j]/yMax)+120;
        gz = (int)(3.0*scale*(0.68+0.3*(pz[j]/zMax))*Morse[kind[j]][4]/AA);
        clr = 0.66-0.66*(ww[j]/wMax);
        gOff.setColor(Color.getHSBColor(
          (float)clr,0.99f,(float)(0.5*pz[j]/zMax+0.35)));
        gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);

        for (k=1; k<reg[j][0]; k++) {
          jj = reg[j][k];
          if ( distance(j,jj)<0.6*(Morse[kind[j]][4]+Morse[kind[jj]][4]) ) {
            gbx = (int)(paintSizeX*scale*px[jj]/xMax)+60;
            gby = (int)(paintSizeY*scale*py[jj]/yMax)+120;
            //gOff.setColor(Color.gray);
            clr = 0.66-0.66*(ww[j]/wMax);
            gOff.setColor(Color.getHSBColor(
              (float)clr,0.99f,(float)(0.5*pz[j]/zMax+0.35)));
            gOff.drawLine(gx,gy, gbx, gby);
          }
        }
      }

    } else if (dispMode==3) {

      for (i=0; i<Nmt; i++) {
        j = srtz[0][i];
        gx = (int)(paintSizeX*scale*px[j]/xMax)+60;
        gy = (int)(paintSizeY*scale*py[j]/yMax)+120;
        gz = (int)(7.5*scale*(0.68+0.3*(pz[j]/zMax))*Morse[kind[j]][4]/AA);
        clr = 0.66-0.66*(ww[j]/wMax);
        gOff.setColor(Color.getHSBColor(
          (float)clr,0.99f,(float)(0.5*pz[j]/zMax+0.35)));
        gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);
      }

    } else if (dispMode==4) {

      for (i=0; i<Nmt; i++) {
        j = srtz[0][i];
        gx = (int)(paintSizeX*scale*px[j]/xMax)+60;
        gy = (int)(paintSizeY*scale*py[j]/yMax)+120;
        gz = (int)(3.0*scale*(0.68+0.3*(pz[j]/zMax))*Morse[kind[j]][4]/AA);
        clr = 0.66-0.66*(ww[j]/wMax);
        tm = 0.5*Morse[kind[j]][0]*(vx[j]*vx[j]+vy[j]*vy[j]+vz[j]*vz[j])/(2.0*1.38e-23)/3000.0 + 0.3;
        if(tm>0.99) {
          tm = 0.99;
        }
        gOff.setColor(Color.getHSBColor( (float)clr, 0.99f, (float)tm ));
        gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);
      }

    }

    for (i=0; i<12; i++) {
      gx = (int)(paintSizeX*scale*pwkx[boxp[i][0]]/xMax)+60;
      gy = (int)(paintSizeY*scale*pwky[boxp[i][0]]/yMax)+120;
      g2x = (int)(paintSizeX*scale*pwkx[boxp[i][1]]/xMax)+60;
      g2y = (int)(paintSizeY*scale*pwky[boxp[i][1]]/yMax)+120;
      if (boxp[i][0]!=farPoint && boxp[i][1]!=farPoint) {
        if (tempMode==1 || tempMode==2) {
          gOff.setColor(Color.gray); 
        } else if (tempMode==0) {
          gOff.setColor(Color.black); 
          if (tmp<contTemp) { gOff.setColor(Color.red); };
        }
        gOff.drawLine(gx,gy, g2x, g2y);
      }
    }

    plotXY(400,80,0.75);
    plotZW(400,250,0.75);

    gOff.setColor(Color.black);
    gOff.drawString("t="+Math.floor(t*1.0e15+0.5)+"(fs)",630/6*0+10,40);
    gOff.drawString("temp mode",630/6*2+10,40);

    gOff.drawString("cont T="+contTemp+"K",630/6*3+10,40);
    gOff.drawString("T="+temperature()+" K",630/6*3+10,60);
    gOff.drawString("scale="+(int)(scale*100.0+0.5)+"%",630/6*4+10,40);
    gOff.drawString("view",630/6*5+10,40);

    gOff.drawString("P = P(x,y,z,w)",100,400);
  }

  double distance(int j, int jj) {
    double r2;

    r2 = (xx[j]-xx[jj])*(xx[j]-xx[jj])+(yy[j]-yy[jj])*(yy[j]-yy[jj])+(zz[j]-zz[jj])*(zz[j]-zz[jj])+(ww[j]-ww[jj])*(ww[j]-ww[jj]);
    return ( Math.sqrt(r2) );
  }

  void plotXY(int gbx, int gby, double sc) {
    int i,gx,gy,gz;
    double clr;

    for (i=0; i<Nmt; i++) {
      gx = (int)(sc*paintSizeX*xx[i]/xMax)+gbx;
      gy = (int)(sc*paintSizeX*yy[i]/yMax)+gby;
      gz = (int)(3.0*scale*Morse[kind[i]][4]/AA);
      clr = 0.66-0.66*(ww[i]/wMax);
      gOff.setColor(Color.getHSBColor(
          (float)clr,0.99f,(float)(0.5*zz[i]/zMax+0.35)));
      gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);
    }
    gOff.setColor(Color.black);
    gOff.drawRect(gbx-4,gby-4,(int)(sc*paintSizeX)+8,(int)(sc*paintSizeY)+8);
    gOff.drawString("X",gbx+(int)(sc*paintSizeY)+8,gby+(int)(sc*paintSizeY)+8);
    gOff.drawString("Y",gbx-20,gby+10);
  }

  void plotZW(int gbx, int gby, double sc) {
    int i,gx,gy,gz;
    double clr;

    for (i=0; i<Nmt; i++) {
      gx = (int)(sc*paintSizeX*zz[i]/zMax)+gbx;
      gy = (int)(sc*paintSizeX*ww[i]/wMax)+gby;
      gz = (int)(3.0*scale*Morse[kind[i]][4]/AA);
      clr = 0.66-0.66*(ww[i]/wMax);
      gOff.setColor(Color.getHSBColor(
          (float)clr,0.99f,(float)(0.5*zz[i]/zMax+0.35)));
      gOff.fillOval(gx-gz/2,gy-gz/2, gz, gz);
    }
    gOff.setColor(Color.black);
    gOff.drawRect(gbx-4,gby-4,(int)(sc*paintSizeX)+8,(int)(sc*paintSizeY)+8);
    gOff.drawString("Z",gbx+(int)(sc*paintSizeY)+8,gby+(int)(sc*paintSizeY)+8);
    gOff.drawString("W",gbx-20,gby+10);
  }

/*----------------------------------  findMin()  ----------*/

  int findMin() {
    int i,im;
    double m;

    im = 0; m = pwkz[im];
    for (i=0; i<8; i++) {
      if (pwkz[i]<m) {
        m = pwkz[i];
        im = i;
      }
    }
    return(im);
  }

/*-------------------------------------  sort z  ----------*/

  void zSort() {
    int i,j,w;

    for (i=0; i<Nmt; i++) {
      j = srtz[0][i];
      srtz[1][i] = (int)(32767.0*pz[j]/zMax);
    } 
    for (i=0; i<100; i++) {
      for (j=0; j<Nmt-1; j++) {
        if (srtz[1][j]>srtz[1][j+1]) {
          w = srtz[0][j]; srtz[0][j]=srtz[0][j+1]; srtz[0][j+1]=w;
          w = srtz[1][j]; srtz[1][j]=srtz[1][j+1]; srtz[1][j+1]=w;
        }
      }
    }    
  }

  void zSortInit() {
    int i,j,w;

    for (i=0; i<Nmt; i++) {
      srtz[0][i] = i;
      srtz[1][i] = (int)(32767.0*pz[i]/zMax);
    } 
    for (i=0; i<Nmt; i++) {
      for (j=0; j<Nmt-1; j++) {
        if (srtz[1][j]>srtz[1][j+1]) {
          w = srtz[0][j]; srtz[0][j]=srtz[0][j+1]; srtz[0][j+1]=w;
          w = srtz[1][j]; srtz[1][j]=srtz[1][j+1]; srtz[1][j+1]=w;
        }
      }
    }    
  }

/*-------------------------------------  rotate  ----------*/

  void rotate(double th,double fi) {
    int i;
    double cosTh,sinTh,cosFi,sinFi;

    cosTh = Math.cos(th); sinTh = Math.sin(th);
    cosFi = Math.cos(fi); sinFi = Math.sin(fi);
    for (i=0; i<Nmt; i++) {
      py[i] = cosFi*(yy[i]-cy)+sinFi*(zz[i]-cz) + cy; 
      pz[i] = -sinFi*(yy[i]-cy)+cosFi*(zz[i]-cz) + cz;
    } 
    for (i=0; i<Nmt; i++) {
      px[i] = cosTh*(xx[i]-cx)+sinTh*(pz[i]-cz) + cx; 
      pz[i] = -sinTh*(xx[i]-cx)+cosTh*(pz[i]-cz) + cz;
    } 

    for (i=0; i<8; i++) {
      pwky[i] = cosFi*(wky[i]-cy)+sinFi*(wkz[i]-cz) + cy; 
      pwkz[i] = -sinFi*(wky[i]-cy)+cosFi*(wkz[i]-cz) + cz;
    } 
    for (i=0; i<8; i++) {
      pwkx[i] = cosTh*(wkx[i]-cx)+sinTh*(pwkz[i]-cz) + cx; 
      pwkz[i] = -sinTh*(wkx[i]-cx)+cosTh*(pwkz[i]-cz) + cz;
    }  
  }

/*-------------------------------------  setWaku  ---------*/

  void setWaku() {
    int i,ix,iy,iz;

    i=0;

    for (ix=0; ix<2; ix++) {
      for (iy=0; iy<2; iy++) {
        for (iz=0; iz<2; iz++) {
          wkx[i] = -0.5*sgm+(xMax+sgm)*(double)(ix);
          wky[i] = -0.5*sgm+(yMax+sgm)*(double)(iy);
          wkz[i] = -0.5*sgm+(zMax+sgm)*(double)(iz);
          i += 1;
        }
      }
    }
  }

/*-------------------------------------  statictics  ----------*/

  double rmsV() {
    int i;
    double s;

    s = 0.0; 
    for (i=0; i<Nmt; i++) {
      s += vx[i]*vx[i] + vy[i]*vy[i] + vz[i]*vz[i]+ vw[i]*vw[i];
    }
    return ((double)Math.floor(Math.sqrt(s/(double)Nmt)*10.0)/10.0);
  }

  double temperature() {
    int i;
    double s;

    s = 0.0; 
    for (i=0; i<Nmt; i++) {
      s += 0.5*(Morse[kind[i]][0])*(vx[i]*vx[i] + vy[i]*vy[i] + vz[i]*vz[i] + vw[i]*vw[i]);
    }
    return ((double)Math.floor(s/((double)Nmt*2.0*1.38e-23)*10.0)/10.0 );
  }


/*-------------------------------------  v ajustment  ---------*/

  void vAjustment() {
    int i;
    double tmp, r;

    tmp = temperature();
    r = Math.sqrt(contTemp/tmp);
    for (i=0; i<Nmt; i++) {
      vx[i] = r*vx[i];
      vy[i] = r*vy[i];
      vz[i] = r*vz[i];
      vw[i] = r*vw[i];
    }
  }


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

  void setForceTable(){
    int i,j,ir,k0,k1; 
    double d,a,r0,y,r,sc;
    int kkk[] = {MAT0,MAT1,MAT2,MAT3,MAT4};

    for (i=0; i<5; i++) {
      k0=kkk[i];
      for (j=0; j<5; j++) {
        k1=kkk[j];

        d = Math.sqrt(Morse[k0][2]*Morse[k1][2]);
        a = 0.5*(Morse[k0][3]+Morse[k1][3]);
        r0 = 0.5*(Morse[k0][4]+Morse[k1][4]);

        for (ir=100; ir<10000; ir++) {
          r = (double)ir*1.0e-13;
          y = Math.exp(-a*(r-r0));
          forceTable[i][j][ir]=2.0*d*a*y*(y-1.0);
        }
        for (ir=0; ir<100; ir++) {
          forceTable[i][j][ir]=forceTable[i][j][100];
        }

        for (ir=6000; ir<10000; ir++) {
          forceTable[i][j][ir] = cutoff(ir)*forceTable[i][j][ir];
        }
      }
    }
  }

  double cutoff(int ir) {
    double rt;

    rt = 0.0;
    if (ir<6000) {
      rt = 1.0;
    } else if (ir>=6000 && ir<10000) {
      rt = 0.5+0.5*Math.cos(3.1415926536*(((double)ir)/10000.0-0.6)/0.4);
    } else {
      rt = 0.0;
    }
    return( rt );
  }

  void setInitialPosition(){
    int i; 
    double s;

    kindT[MAT0] = 0; kindT[MAT1] = 1; kindT[MAT2] = 2;
    kindT[MAT3] = 3; kindT[MAT4] = 4;
    setForceTable();

    s = 3.4e-10;
    for (i=0; i<Nmt; i++) {
      kind[i] = MAT1;
      xx[i] = s*2.5*(double)(i/27)+2.5*s;
      yy[i] = s*2.5*(double)((i%27)/9)+2.5*s;
      zz[i] = s*2.5*(double)((i%9)/3)+2.5*s;
      ww[i] = s*2.5*(double)(i%3)+2.5*s;

      vx[i] = 800.0*Math.random()-400.0;
      vy[i] = 800.0*Math.random()-400.0;
      vz[i] = 800.0*Math.random()-400.0;
      vw[i] = 800.0*Math.random()-400.0;

      ffx[i] = 0.0;
      ffy[i] = 0.0;
      ffz[i] = 0.0;
      ffw[i] = 0.0;

      fbx[i] = 0.0;
      fby[i] = 0.0;
      fbz[i] = 0.0;
      fbw[i] = 0.0;
    }

    zSortInit();
  }

/*-------------------------------  registration   -------------*/

  void registration() {
    int i,j,k;
    double r2;

    for(i=0;i<Nmt;i++) {
      k = 1;
      for(j=i+1;j<Nmt;j++) {
        r2=(xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j])+(zz[i]-zz[j])*(zz[i]-zz[j])+(ww[i]-ww[j])*(ww[i]-ww[j]);
        if (r2<rreg2) {
          reg[i][k]=j;
          k += 1;
        }
      }
      reg[i][0]=k;
    }

  }

/*-------------------------------  molecules motion   ---------*/

  void calcposition(){
    int i;

    if (tempMode==2) {
      vAjustment();
    }

    registration();
    for (i=0; i<10; i++) {

      timeEvolution();

    }
  }


  void timeEvolution(){
    int i; 
    double rr, tmp;

    t = t + dt;

    rr=1.0;
    if (tempMode==0) {
      tmp = temperature();
      rr = Math.sqrt(contTemp/tmp);
      if (rr<0.5) { rr = 0.5; } else if (rr>1.2){ rr=1.2; };
    }

    for (i=0; i<Nmt; i++) {
      xx[i] += vx[i]*dt+0.5*dt*dt*ffx[i]/Morse[kind[i]][0];
      yy[i] += vy[i]*dt+0.5*dt*dt*ffy[i]/Morse[kind[i]][0];
      zz[i] += vz[i]*dt+0.5*dt*dt*ffz[i]/Morse[kind[i]][0];
      ww[i] += vw[i]*dt+0.5*dt*dt*ffw[i]/Morse[kind[i]][0];
      fbx[i] = ffx[i];
      fby[i] = ffy[i];
      fbz[i] = ffz[i];
      fbw[i] = ffw[i];
    }
    forceCalc();
    for (i=0; i<Nmt; i++) {
      vx[i] += 0.5*dt*(ffx[i]+fbx[i])/Morse[kind[i]][0];
      vy[i] += 0.5*dt*(ffy[i]+fby[i])/Morse[kind[i]][0];
      vz[i] += 0.5*dt*(ffz[i]+fbz[i])/Morse[kind[i]][0];
      vw[i] += 0.5*dt*(ffw[i]+fbw[i])/Morse[kind[i]][0];
    }

    for (i=0; i<Nmt; i++) {
      if (xx[i] < 0.0) { 
        xx[i] = 0.0; vx[i] = -rr*vx[i]; vy[i] = rr*vy[i];  
        vz[i] = rr*vz[i]; vw[i] = rr*vw[i]; 
      }
      if (xx[i] > xMax) {
        xx[i] = xMax; vx[i] = -rr*vx[i]; vy[i] = rr*vy[i]; 
        vz[i] = rr*vz[i]; vw[i] = rr*vw[i];  
      }
      if (yy[i] < 0.0) { 
        yy[i] = 0.0; vx[i] = rr*vx[i]; vy[i] = -rr*vy[i]; 
        vz[i] = rr*vz[i]; vw[i] = rr*vw[i];
      }
      if (yy[i] > yMax) {
        yy[i] = yMax; vx[i] = rr*vx[i]; vy[i] = -rr*vy[i]; 
        vz[i] = rr*vz[i]; vw[i] = rr*vw[i];
      }
      if (zz[i] < 0.0) { 
        zz[i] = 0.0; vx[i] = rr*vx[i]; vy[i] = rr*vy[i]; 
        vz[i] = -rr*vz[i]; vw[i] = rr*vw[i];
      }
      if (zz[i] > zMax) {
        zz[i] = zMax; vx[i] = rr*vx[i]; vy[i] = rr*vy[i]; 
        vz[i] = -rr*vz[i]; vw[i] = rr*vw[i];
      }
      if (ww[i] < 0.0) { 
        ww[i] = 0.0; vx[i] = rr*vx[i]; vy[i] = rr*vy[i]; 
        vz[i] = rr*vz[i]; vw[i] = -rr*vw[i]; 
      }
      if (ww[i] > wMax) {
        ww[i] = zMax; vx[i] = rr*vx[i]; vy[i] = rr*vy[i]; 
        vz[i] = rr*vz[i]; vw[i] = -rr*vw[i];
      }
    }
  }

  void forceCalc() {
    int i,j,k;
    double r2,rij,f,fxij,fyij,fzij,fwij;

    for(i=0;i<Nmt;i++) {
      ffx[i] = 0.0;
      ffy[i] = 0.0;
      ffz[i] = 0.0;
      ffw[i] = 0.0;
    }

    for(i=0;i<Nmt;i++) {
      for(k=1;k<reg[i][0];k++) {
        j = reg[i][k];
        r2=(xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j])+(zz[i]-zz[j])*(zz[i]-zz[j])+(ww[i]-ww[j])*(ww[i]-ww[j]);
        if (r2<rc2) {
          rij = Math.sqrt(r2);
          f = force(rij,kind[i],kind[j]);
          fxij = f*(xx[i]-xx[j])/rij;
          fyij = f*(yy[i]-yy[j])/rij;
          fzij = f*(zz[i]-zz[j])/rij;
          fwij = f*(ww[i]-ww[j])/rij;

          ffx[i] +=  fxij;
          ffy[i] +=  fyij;
          ffz[i] +=  fzij;
          ffw[i] +=  fwij;

          ffx[j] += -fxij;
          ffy[j] += -fyij;
          ffz[j] += -fzij;
          ffw[j] += -fwij;
        }
      }
    }
  }


  double force(double r, int knd0, int knd1) {
    return ( forceTable[kindT[knd0]][kindT[knd1]][(int)(r*1.0e13)] );
  }

/*----- end of molecules motion -----*/

}  

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