write code to model acoustic wave propagation underwater in javascript

index.tsx
const c = 1500; // speed of sound in water
const dt = 0.001; // time step
const dx = 1; // spatial step
const N = 100; // number of grid points in one dimension
const f0 = 500; // frequency of source
const source = [N/2, N/2]; // location of source
const tEnd = 0.5; // end time

// initialize velocity and pressure fields
let u = new Array(N*N).fill(0);
let u1 = new Array(N*N).fill(0);
let u2 = new Array(N*N).fill(0);
let p = new Array(N*N).fill(0);

// initialize source function in time domain
const tSrc = Array.from({length: Math.ceil(tEnd/dt)}, (_, i) => i*dt);
const src = tSrc.map(t => Math.sin(2*Math.PI*f0*t));

// main loop for wave propagation
for (let n = 0; n < tSrc.length; n++) {
  // update velocity field
  for (let i = 1; i < N-1; i++) {
    for (let j = 1; j < N-1; j++) {
      u1[i*N+j] = u[i*N+j] - c*c*dt/dx*(p[i*N+j+1] - p[i*N+j] + p[i*N+j+N] - p[i*N+j]);
    }
  }
  // update pressure field
  for (let i = 1; i < N-1; i++) {
    for (let j = 1; j < N-1; j++) {
      p[i*N+j] = p[i*N+j] - c*c*dt/dx*(u1[i*N+j] - u1[i*N+j-1] + u1[i*N+j-N] - u1[i*N+j]) + src[n]*dt;
    }
  }
  // update velocity field
  for (let i = 1; i < N-1; i++) {
    for (let j = 1; j < N-1; j++) {
      u2[i*N+j] = u1[i*N+j] - c*c*dt/dx*(p[i*N+j] - p[i*N+j-1] + p[i*N+j-N] - p[i*N+j]);
    }
  }
  // update velocity field for next time step
  for (let i = 0; i < N*N; i++) {
    u[i] = u2[i];
  }
}
1403 chars
44 lines