/* drop - smoothed particle hydrodynamics 2D */
/* coded by Ikeuchi Mitsuru   */
/* ver 0.0.1   2005.05.20     */
/* ver 0.0.2   2005.07.09 bug of dwwvdr(r,h) fixed */

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

public class dropSPH2D extends Applet
  implements ItemListener, ActionListener, AdjustmentListener, Runnable {

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

  Button bt_reset, bt_start, bt_stop, bt_go;
  Choice ch_view;
  Scrollbar sc_rate,sc_alpha;
  Thread th = null;

  Dimension dim;
  Image imgOff;
  Graphics gOff;

  int sleepTime = 30;

  int dispMode = 0;
  int incRate = 10;
  int resetSW = 0;
  int started = 1;
  double alpha = 5.0;
  double dens0 = 1.0;


  double xMax = 20.0;
  double yMax = 15.0;

  double t = 0.0;
  double dt = 0.005;
  double hh0 = 0.25;
  double rBond = 0.3;

  int NNp = 4000+100;
  int kind[] = new int [NNp];
  int bond[][] = new int[NNp][8];
  double xx[] = new double [NNp];
  double yy[] = new double [NNp];
  double vx[] = new double [NNp];
  double vy[] = new double [NNp];
  double ax[] = new double [NNp];
  double ay[] = new double [NNp];

  double hh[] = new double [NNp];
  double mass[] = new double [NNp];
  double density[] = new double [NNp];
  double pressure[] = new double [NNp];
  double energy[] = new double [NNp];
  double power[] = new double [NNp];

  int Nsx = 40;
  int Nsy = 30;
  int section[][][] = new int[Nsx][Nsy][100];

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

  public void init() {

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

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

    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_go = new Button("go");
    bt_go.addActionListener(this);

    ch_view = new Choice();
    ch_view.addItem("particle");
    ch_view.addItem("velocity");
    ch_view.addItem("density");
    ch_view.addItemListener(this);
    ch_view.select("particle");


    sc_rate= new Scrollbar(Scrollbar.HORIZONTAL,10,10,1,110);
    sc_rate.addAdjustmentListener(this);

    sc_alpha= new Scrollbar(Scrollbar.HORIZONTAL,50,10,20,210);
    sc_alpha.addAdjustmentListener(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(bt_go);
    pnl.add(sc_alpha);
	/*
    pnl.add(new Label("  "));*/
    pnl.add(ch_view);
    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() == ch_view) {
      dispMode = ch_view.getSelectedIndex();
    }
  }

  public void actionPerformed(ActionEvent ev){
    if(ev.getSource() == bt_reset){ 
      t = 0.0;
      resetSW = 1;
      started = 0;
    } else if(ev.getSource() == bt_start){ 
      started = 1;
    } else if(ev.getSource() == bt_stop){
      started = 0;
    } else if(ev.getSource() == bt_go){
      gogogo();
    }
}

  public void adjustmentValueChanged(AdjustmentEvent ev){
    if(ev.getSource() == sc_rate) { 
      incRate = sc_rate.getValue();
    } else if(ev.getSource() == sc_alpha) { 
      alpha = 0.1*sc_alpha.getValue();
    }
  }

  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;

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

    gOff.setColor(Color.black);
    gOff.drawRect(30,80,400,300);

    if (dispMode==0) {

      ovalPlot();

	} else if (dispMode==1) {

      linePlot(10.0);

	} else if (dispMode==2) {

      fieldPlot(density,0.02);

    }

    gOff.setColor(Color.black);
    gOff.drawString("t="+(int)(t*100.0+0.5)/100.0+" ",630/6*0+10,40);
    gOff.drawString("viscosity="+(int)(alpha*10.0)/10.0+" ",630/6*4+10,40);
    gOff.drawString("view",630/6*5+10,40);
    gOff.drawString("maxSec="+maxSection()+" ",150,40);

  }

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

  void ovalPlot() {
    int i,j,gx,gy;

    gx = 30; gy = 80;

    gOff.setColor(Color.blue);
    for (i=0; i<NNp; i++) {
	  if (kind[i]==0) {
	    gOff.setColor(Color.blue);
	  } else {
		gOff.setColor(Color.orange);
	  }
      gOff.drawOval(gx+(int)(xx[i]*20.0+0.5)-5,gy+(int)(yy[i]*20+0.5)-5, 10, 10);
    }
	
	gOff.setColor(Color.red);
    for (i=0; i<NNp; i++) {
	  if (kind[i]>0) {
	    for (j=0; j<NNp; j++) {
		  if (bondQ(i,j)==1) {
		    gOff.setColor(Color.red);
		    gOff.drawLine(gx+(int)(xx[i]*20.0+0.5),gy+(int)(yy[i]*20+0.5),gx+(int)(xx[j]*20.0+0.5),gy+(int)(yy[j]*20+0.5));
		  }
		}
	  }
    }
  }

  void linePlot(double mag) {
    int i,ix,iy,gx,gy,ir;

    gx = 30; gy = 80;ir = 10;

    for (i=0; i<NNp; i++) {
      ix = gx+(int)(xx[i]*20.0+0.5);
      iy = gy+(int)(yy[i]*20.0+0.5);
      if (vx[i]>0.0) {
        gOff.setColor(Color.getHSBColor(0.66f,0.8f,0.8f));
      } else {
        gOff.setColor(Color.getHSBColor(0.01f,0.8f,0.8f));
      }
      gOff.drawLine(ix,iy,ix+(int)(vx[i]*mag+0.5),iy+(int)(vy[i]*mag+0.5));
      if (kind[i]!=0) {
	    gOff.setColor(Color.orange);
		gOff.drawOval(ix-5,iy-5, 10, 10);
      }
    }
  }

  void fieldPlot(double field[], double mag) {
    int i,j,gx,gy;
	double c;

    gx = 30; gy = 80;

    for (i=0; i<NNp; i++) {
	  c = mag*field[i]; if (c<0.0) c = 0.0;
	  if (c>1.0) c = 1.0;
	  gOff.setColor(Color.getHSBColor((float)(0.66-0.66*c),0.8f,0.8f));
      gOff.drawOval(gx+(int)(xx[i]*20.0+0.5)-5,gy+(int)(yy[i]*20+0.5)-5, 10, 10);
    }
  }

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

  void setInitialCondition() {
    int i,j,ix,iy;
    double s;

    t = 0.0;

    i = 0;
    for (ix=0; ix<66; ix++) {
      for (iy=0; iy<20; iy++) {
        kind[i] = 0;
	    mass[i] = 1.0;
	    xx[i] = 0.10+0.3*(ix)+0.15*(iy%2);
	    yy[i] = yMax-0.15-0.3*(0.866*iy);
	    vx[i] = 0.5*(Math.random()-0.5);
	    vy[i] = 0.5*(Math.random()-0.5);
	    ax[i] = 0.0;
	    ay[i] = 0.0;
	    hh[i] = hh0;
        i += 1;
	  }
    }

    for (ix=0; ix<12; ix++) {
      for (iy=0; iy<8; iy++) {
        kind[i] = -1;
	    mass[i] = 0.6;
	    xx[i] = 5.0+0.866*rBond*ix;
	    yy[i] = 1.0+rBond*iy+0.5*rBond*(ix%2);
	    vx[i] = 0.0;
	    vy[i] = 0.0;
	    ax[i] = 0.0;
	    ay[i] = 0.0;
	    hh[i] = hh0;
	    i += 1;
	  }
    }

	NNp = i;
    bonding();

    for (i=0; i<NNp; i++) {
      energy[i] = 1.0;
	  power[i] = 0.0;
    }

    dividingSection();
    setDensity();
    dens0 = density[66*10+33]*0.85;

  }

  void gogogo() {
    int i;
	
    for (i=0; i<NNp; i++) {
      if (kind[i]==-1) {
	    kind[i] = 1;
	  }
    }
  }

/* --------------------------- bonding manager ---------- */

  void bonding() {
    int i,j;
    double r2,rb,rb2;

    rb = rBond*1.2;
	rb2=rb*rb;

    bondInit();

    for (i=0; i<NNp; i++) {
	  if (kind[i]!=0) {
        for (j=i+1; j<NNp; j++) {
		  if (kind[j]!=0) {
            r2 = (xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j]);
            if (r2<=rb2) {
              bondReg(i, j); bondReg(j, i);
			}
		  }
        }
      }
    }
  }

  void bondInit() {
    int i,j;

    for (i=0; i<NNp; i++) {
      for (j=0; j<8; j++) {
        bond[i][j]=-1;
      }
    }
  }

  void bondReg(int ii, int jj) {
    int j;

    for (j=0; j<8; j++) {
      if (bond[ii][j]==-1) {
	    bond[ii][j] = jj;
		break;
	  }
	}
  }

  void bondUnreg(int ii, int jj) {
    int j;

    for (j=0; j<8; j++) {
      if (bond[ii][j]==jj) {
	    bond[ii][j] = -1;
		break;
	  }
	}
  }

  int bondQ(int ii, int jj) {
    int j,q;
    double r2,rb,rb2;

    rb = rBond*1.2;
    rb2=rb*rb;

    q = 0;
    for (j=0; j<8; j++) {
      if (bond[ii][j]==jj) {
        q = 1;
        r2 = (xx[ii]-xx[jj])*(xx[ii]-xx[jj])+(yy[ii]-yy[jj])*(yy[ii]-yy[jj]);
        if (r2>rb2) {
          bondReg(ii,jj); bondReg(jj,ii);
          q = 0;
        }
        break;
      }
    }
    return (q);
  }

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

  void timeEvolution() {
    int i;

    if (resetSW==1) {
      setInitialCondition();
	  resetSW = 0;
    }
    if (started==1) {
      for (i=0; i<incRate; i++) {
        timeStep();
      }
    }
  }

/*-------------------------------  particle motion   ---------*/

  void timeStep() {
    int i,j;
	double dtv2,s,rr;

    t = t + dt;
    dtv2 = dt/2.0;

    dividingSection();
    setDensity();

    for (i=0; i<NNp; i++) {
	  if (kind[i]==0) {
	    pressure[i] = 20.0*(Math.pow(density[i]/dens0,7.0)-1);
	  } else {
	    pressure[i] = 0.5*density[i];
	  }
    }

    for (i=0; i<NNp; i++) {
	  if (kind[i]>=0) {
        vx[i] += dtv2*ax[i];
        vy[i] += dtv2*ay[i];
        xx[i] += vx[i]*dt;
        yy[i] += vy[i]*dt;
	  }
    }
    changeCalc();
    for (i=0; i<NNp; i++) {
      if (kind[i]>=0) {
        vx[i] += dtv2*ax[i];
        vy[i] += dtv2*ay[i];
        energy[i] += power[i]*dt;
	  }
    }

    rr = 1.0;
    for (i=0; i<NNp; i++) {
      if (xx[i] < 0.0) { 
        xx[i] = 0.0; vx[i] = -rr*vx[i]; vy[i] = rr*vy[i]; 
      }
      if (xx[i] > xMax) {
        xx[i] = xMax; vx[i] = -rr*vx[i]; vy[i] = rr*vy[i]; 
      }
      if (yy[i] < 0.0) { 
        yy[i] = 0.0; vx[i] = rr*vx[i]; vy[i] = -rr*vy[i]; 
      }
      if (yy[i] > yMax) {
        yy[i] = yMax; vx[i] = rr*vx[i]; vy[i] = -rr*vy[i];
      }
    }
  }

  void changeCalc() {
    int i,j,i0,i1,j0,j1,ii,jj,iq;
    double ai,aj,b,rij,gradW,gradWx,gradWy,a;
    double rcc,rcc2,r2,vr,mu,pij;

    rcc = hh0*2.0;
    rcc2 = rcc*rcc;

    for (i=0; i<NNp; i++) {
      ax[i] = 0.0; ay[i] = 1.0;
      boundaryForce(i, rcc, 100.0);
      power[i] = 0.0;
      ai = pressure[i]/(density[i]*density[i]);

      i0 = (int)(Nsx*(xx[i]-rcc)/xMax); if (i0<0) i0 = 0;
      i1 = (int)(Nsx*(xx[i]+rcc)/xMax); if (i1>=Nsx) i1 = Nsx-1;
      j0 = (int)(Nsy*(yy[i]-rcc)/yMax); if (j0<0) j0 = 0;
      j1 = (int)(Nsy*(yy[i]+rcc)/yMax); if (j1>=Nsy) j1 = Nsy-1;
      for(ii=i0;ii<=i1;ii++) {
        for(jj=j0;jj<=j1;jj++) {
          for(iq=1;iq<=section[ii][jj][0];iq++) {
            j = section[ii][jj][iq];
            r2=(xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j]);
            if (r2<rcc2) {
              rij = Math.sqrt(r2);
              if (rij>0) {
                aj = pressure[j]/(density[j]*density[j]);
                vr = (xx[i]-xx[j])*(vx[i]-vx[j])+(yy[i]-yy[j])*(vy[i]-vy[j]);
                pij = 0;
                if (vr<0.0) {
                  mu = 0.5*vr/(r2+0.001*0.5*0.5);
                  pij = (-alpha*0.5*mu+0.0*mu*mu)/((density[i]+density[j])/2.0);
                }
                b = mass[j]*(ai+aj+pij);
                gradW = dwwvdr(rij, (hh[i]+hh[j])/2.0);
                gradWx = gradW*(xx[i]-xx[j])/rij;
                gradWy = gradW*(yy[i]-yy[j])/rij;
                ax[i] += -b*gradWx;
                ay[i] += -b*gradWy;
                power[i] += 0.5*b*((vx[i]-vx[j])*gradWx+(vy[i]-vy[j])*gradWy);
                if (kind[j]!=0 || kind[i]!=0) {
                  a = force(rij,i,j)/mass[i];
                  ax[i] += a*(xx[i]-xx[j])/rij;
                  ay[i] += a*(yy[i]-yy[j])/rij;
                }
              }
            }
          }
        }
      }
    }
  }

  double force(double r, int ii, int jj) {
    int i,j;
    double ret,rc,ra,ra1,ra2,ram,fa,f,r1,r3,kc;


    if (kind[ii]==0 || kind[jj]==0) {

      rc = hh0*1.2;
      ret = 0.0;
      if (r<rc) {
	    ra = (rc-r)/rc;
	    ret = 2000.0*ra*ra;
		return ( ret );
	  }

	}

    if( kind[ii]>0 && kind[jj]>0 ) {

      kc = 2000.0;
      ra = rBond;
      ra1 = 1.2*ra;
      ra2 = 1.5*ra;
      ram = (ra1+ra2)/2.0;
      fa = 2.0/((ra2-ra1)*(ra2-ra1));

      if (ii<jj) {
        i= ii; j=jj;
      } else {
        i= jj; j=ii;
      }
      if (r>ra2) {
        bondUnreg(i,j); bondUnreg(j,i);
        return ( 0.0 );
      }  

      f = 0.0;
      if (bondQ(i,j)==0) {
        if (r<ra) {
          f = kc*(r-ra);
        }
      } else {
        if (r<=ra1) {
          f = kc*(r-ra);
        } else if (r>=ra2) {
          f = 0.0;
          bondUnreg(i,j); bondUnreg(j,i);
        } else if (ra1<=r && r<ram) {
          f = kc*(r-ra)*(1.0-fa*(r-ra1)*(r-ra1));
        } else if (ram<=r && r<ra2) {
          f = kc*(r-ra)*fa*(r-ra2)*(r-ra2);
        }
      }
      return ( -f );

	}

    return (0.0);
  }

  void boundaryForce(int i, double rc, double k) {
    double r;

    if (xx[i] < rc) {
	  r = rc-xx[i];
      ax[i] += k*r*r; 
    }
    if (xx[i] > xMax-rc) {
	  r = xx[i]-(xMax-rc);
      ax[i] += -k*r*r; 
    }
    if (yy[i] < rc) {
	  r = rc-yy[i];
      ay[i] += k*r*r; 
    }
    if (yy[i] > yMax-rc) {
	  r = yy[i]-(yMax-rc);
      ay[i] += -k*r*r; 
    }
  }


  void setDensity() {
    int i,j,i0,i1,j0,j1,ii,jj,iq;
    double s,rij;
    double rcc,rcc2,r2;

    rcc = hh0*2.0;
    rcc2 = rcc*rcc;

    for (i=0; i<NNp; i++) {
	  s = 0.0;
	  i0 = (int)(Nsx*(xx[i]-rcc)/xMax); if (i0<0) i0 = 0;
      i1 = (int)(Nsx*(xx[i]+rcc)/xMax); if (i1>=Nsx) i1 = Nsx-1;
      j0 = (int)(Nsy*(yy[i]-rcc)/yMax); if (j0<0) j0 = 0;
      j1 = (int)(Nsy*(yy[i]+rcc)/yMax); if (j1>=Nsy) j1 = Nsy-1;
      for(ii=i0;ii<=i1;ii++) {
        for(jj=j0;jj<=j1;jj++) {
          for(iq=1;iq<=section[ii][jj][0];iq++) {
            j = section[ii][jj][iq];
            r2=(xx[i]-xx[j])*(xx[i]-xx[j])+(yy[i]-yy[j])*(yy[i]-yy[j]);
            if (r2<rcc2) {
              rij = Math.sqrt(r2);
              s += mass[j]*kernel(rij, (hh[i]+hh[j])/2.0);
            }
          }
        }
      }
      density[i] = s;
    }
  }

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

  void dividingSection() {
    int i,j,ip,iq;

    for(i=0;i<Nsx;i++) {
      for(j=0;j<Nsy;j++) {
        section[i][j][0] = 0;
      }
    }

    for(ip=0;ip<NNp;ip++) {
      i = (int)(Nsx*xx[ip]/xMax); if (i>=Nsx) i = Nsx-1;
      j = (int)(Nsy*yy[ip]/yMax); if (j>=Nsy) j = Nsy-1;
      iq = section[i][j][0]+1;
      section[i][j][0] = iq;
      section[i][j][iq] = ip;
    }
  }

  int maxSection() {
    int i,j,m;

    m = 0;
    for(i=0;i<Nsx;i++) {
      for(j=0;j<Nsy;j++) {
        if (section[i][j][0]>m) m = section[i][j][0];
      }
    }
    return ( m );
  }

/*----------------------------  smoothed particle ---------*/

  double kernel(double r, double h) {
    double a,q,ret;

    ret = 0.0;
	q = r/h;
	a = 10.0/(7.0*3.141592)/(h*h);
	if (q<1.0) {
      ret = a*(1.0-1.5*q*q+0.75*q*q*q);
    } else if (q<2.0) {
      ret = a*0.25*(2.0-q)*(2.0-q)*(2.0-q);
	}
    return ( ret );
  }

  double dwwvdr(double r, double h) {
    double a,q,ret;

    ret = 0.0;
	q = r/h;
	a = 10.0/(7.0*3.141592)/(h*h*h);
	if (q<1.0) {
      ret = a*(-3.0*q+2.25*q*q);
    } else if (q<=2.0) {
      ret = -a*0.75*(2.0-q)*(2.0-q);
	}
    return ( ret );
  }

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

}