Add files via upload
This commit is contained in:
committed by
GitHub
parent
660e3d15fd
commit
dadfb985ed
467
mangrove.c
467
mangrove.c
@@ -49,8 +49,10 @@
|
||||
#define WINWIDTH 1280 /* window width */
|
||||
#define WINHEIGHT 720 /* window height */
|
||||
|
||||
#define NX 1280 /* number of grid points on x axis */
|
||||
#define NY 720 /* number of grid points on y axis */
|
||||
#define NX 640 /* number of grid points on x axis */
|
||||
#define NY 360 /* number of grid points on y axis */
|
||||
// #define NX 1280 /* number of grid points on x axis */
|
||||
// #define NY 720 /* number of grid points on y axis */
|
||||
|
||||
#define XMIN -2.0
|
||||
#define XMAX 2.0 /* x interval */
|
||||
@@ -63,6 +65,7 @@
|
||||
|
||||
#define B_DOMAIN 20 /* choice of domain shape, see list in global_pdes.c */
|
||||
|
||||
// #define CIRCLE_PATTERN 1 /* pattern of circles, see list in global_pdes.c */
|
||||
#define CIRCLE_PATTERN 8 /* pattern of circles, see list in global_pdes.c */
|
||||
|
||||
#define P_PERCOL 0.25 /* probability of having a circle in C_RAND_PERCOL arrangement */
|
||||
@@ -80,6 +83,17 @@
|
||||
#define NGRIDX 15 /* number of grid point for grid of disks */
|
||||
#define NGRIDY 20 /* number of grid point for grid of disks */
|
||||
|
||||
#define X_SHOOTER -0.2
|
||||
#define Y_SHOOTER -0.6
|
||||
#define X_TARGET 0.4
|
||||
#define Y_TARGET 0.7 /* shooter and target positions in laser fight */
|
||||
|
||||
#define ISO_XSHIFT_LEFT -1.65
|
||||
#define ISO_XSHIFT_RIGHT 0.4
|
||||
#define ISO_YSHIFT_LEFT -0.05
|
||||
#define ISO_YSHIFT_RIGHT -0.05
|
||||
#define ISO_SCALE 0.85 /* coordinates for isospectral billiards */
|
||||
|
||||
/* You can add more billiard tables by adapting the functions */
|
||||
/* xy_in_billiard and draw_billiard below */
|
||||
|
||||
@@ -90,18 +104,24 @@
|
||||
#define OSCILLATE_TOPBOT 1 /* set to 1 to enforce a planar wave on top and bottom boundary */
|
||||
#define X_SHIFT -0.9 /* x range on which to apply OSCILLATE_TOPBOT */
|
||||
|
||||
#define OMEGA 0.002 /* frequency of periodic excitation */
|
||||
#define K_BC 3.0 /* spatial period of periodic excitation in y direction */
|
||||
#define KX_BC 30.0 /* spatial period of periodic excitation in x direction */
|
||||
#define KY_BC 10.0 /* spatial period of periodic excitation in y direction */
|
||||
#define OMEGA 0.00133333333 /* frequency of periodic excitation */
|
||||
#define K_BC 3.0 /* spatial period of periodic excitation in y direction */
|
||||
#define KX_BC 10.0 /* spatial period of periodic excitation in x direction */
|
||||
#define KY_BC 3.3333 /* spatial period of periodic excitation in y direction */
|
||||
// #define KX_BC 20.0 /* spatial period of periodic excitation in x direction */
|
||||
// #define KY_BC 6.66666 /* spatial period of periodic excitation in y direction */
|
||||
// #define OMEGA 0.002 /* frequency of periodic excitation */
|
||||
// #define K_BC 3.0 /* spatial period of periodic excitation in y direction */
|
||||
// #define KX_BC 30.0 /* spatial period of periodic excitation in x direction */
|
||||
// #define KY_BC 10.0 /* spatial period of periodic excitation in y direction */
|
||||
#define AMPLITUDE 1.0 /* amplitude of periodic excitation */
|
||||
#define COURANT 0.02 /* Courant number */
|
||||
#define COURANTB 0.01 /* Courant number in medium B */
|
||||
#define COURANTB 0.015 /* Courant number in medium B */
|
||||
// #define COURANTB 0.00666 /* Courant number in medium B */
|
||||
#define GAMMA 2.0e-6 /* damping factor in wave equation */
|
||||
#define GAMMAB 2.5e-4 /* damping factor in wave equation */
|
||||
// #define GAMMAB 5.0e-4 /* damping factor in wave equation */
|
||||
// #define GAMMAB 1.0e-4 /* damping factor in wave equation */
|
||||
#define GAMMA 3.0e-6 /* damping factor in wave equation */
|
||||
#define GAMMAB 5.0e-4 /* damping factor in wave equation */
|
||||
// #define GAMMA 2.0e-6 /* damping factor in wave equation */
|
||||
// #define GAMMAB 2.5e-4 /* damping factor in wave equation */
|
||||
#define GAMMA_SIDES 1.0e-4 /* damping factor on boundary */
|
||||
#define GAMMA_TOPBOT 1.0e-6 /* damping factor on boundary */
|
||||
#define KAPPA 0.0 /* "elasticity" term enforcing oscillations */
|
||||
@@ -119,8 +139,8 @@
|
||||
|
||||
/* Parameters for length and speed of simulation */
|
||||
|
||||
// #define NSTEPS 1000 /* number of frames of movie */
|
||||
#define NSTEPS 4500 /* number of frames of movie */
|
||||
#define NSTEPS 1000 /* number of frames of movie */
|
||||
// #define NSTEPS 5500 /* number of frames of movie */
|
||||
#define NVID 60 /* number of iterations between images displayed on screen */
|
||||
#define NSEG 100 /* number of segments of boundary */
|
||||
#define INITIAL_TIME 100 /* time after which to start saving frames */
|
||||
@@ -167,42 +187,40 @@
|
||||
#define MANGROVE_HUE_MIN 180.0 /* color of original mangrove */
|
||||
#define MANGROVE_HUE_MAX -50.0 /* color of saturated mangrove */
|
||||
// #define MANGROVE_EMAX 5.0e-3 /* max energy for mangrove to survive */
|
||||
#define MANGROVE_EMAX 1.1e-3 /* max energy for mangrove to survive */
|
||||
#define MANGROVE_EMAX 1.5e-3 /* max energy for mangrove to survive */
|
||||
|
||||
#define RANDOM_RADIUS 1 /* set to 1 for random circle radius */
|
||||
#define ERODE_MANGROVES 0 /* set to 1 for mangroves to be eroded */
|
||||
#define ERODE_MANGROVES 1 /* set to 1 for mangroves to be eroded */
|
||||
#define RECOVER_MANGROVES 1 /* set to 1 to allow mangroves to recover */
|
||||
#define MOVE_MANGROVES 1 /* set to 1 for mobile mangroves */
|
||||
#define DETACH_MANGROVES 1 /* set to 1 for mangroves to be able to detach */
|
||||
#define INERTIA 1 /* set to 1 for taking inertia into account */
|
||||
#define REPELL_MANGROVES 1 /* set to 1 for mangroves to repell each other */
|
||||
#define DT_MANGROVE 0.1 /* time step for mangrove displacement */
|
||||
#define KSPRING 0.25 /* spring constant of mangroves */
|
||||
#define KWAVE 2.0 /* constant in force due to wave gradient */
|
||||
#define KSPRING 0.05 /* spring constant of mangroves */
|
||||
#define KWAVE 4.0 /* constant in force due to wave gradient */
|
||||
#define KREPEL 5.0 /* constant in repelling force between mangroves */
|
||||
#define REPEL_RADIUS 1.1 /* radius in which repelling force acts (in units of mangrove radius) */
|
||||
#define DXMAX 0.02 /* max displacement of mangrove in one time step */
|
||||
#define L_DETACH 0.2 /* spring length beyond which mangroves detach */
|
||||
#define DAMP_MANGROVE 0.2 /* damping coefficient of mangroves */
|
||||
#define L_DETACH 0.25 /* spring length beyond which mangroves detach */
|
||||
#define DAMP_MANGROVE 0.1 /* damping coefficient of mangroves */
|
||||
#define MANGROVE_MASS 1.5 /* mass of mangrove of radius MU */
|
||||
|
||||
#define HASHX 25 /* size of hashgrid in x direction */
|
||||
#define HASHY 15 /* size of hashgrid in y direction */
|
||||
#define HASHMAX 10 /* maximal number of mangroves per hashgrid cell */
|
||||
#define HASHGRID_PADDING 0.1 /* padding of hashgrid outside simulation window */
|
||||
|
||||
/* For debugging purposes only */
|
||||
#define FLOOR 1 /* set to 1 to limit wave amplitude to VMAX */
|
||||
#define VMAX 10.0 /* max value of wave amplitude */
|
||||
|
||||
|
||||
#include "hsluv.c"
|
||||
|
||||
#include "global_pdes.c"
|
||||
#include "sub_wave.c"
|
||||
#include "wave_common.c"
|
||||
|
||||
double courant2, courantb2; /* Courant parameters squared */
|
||||
double circle_energy[NMAXCIRCLES]; /* energy dissipated by the circles */
|
||||
double circley_wrapped[NMAXCIRCLES]; /* position of circle centers wrapped vertically */
|
||||
double anchor_x[NMAXCIRCLES]; /* points moving circles are attached to */
|
||||
double anchor_y[NMAXCIRCLES]; /* points moving circles are attached to */
|
||||
double vx[NMAXCIRCLES]; /* x velocity of circles */
|
||||
double vy[NMAXCIRCLES]; /* y velocity of circles */
|
||||
double circlerad_initial[NMAXCIRCLES]; /* initial circle radii */
|
||||
double mass_inverse[NMAXCIRCLES]; /* inverse of mangrove mass */
|
||||
short int circle_attached[NMAXCIRCLES]; /* has value 1 if the circle is attached to its anchor */
|
||||
|
||||
|
||||
/*********************/
|
||||
/* animation part */
|
||||
@@ -411,15 +429,109 @@ void evolve_wave(double *phi[NX], double *psi[NX], double *phi_tmp[NX], double *
|
||||
evolve_wave_half(phi_tmp, psi_tmp, phi, psi, xy_in);
|
||||
}
|
||||
|
||||
void hash_xy_to_ij(double x, double y, int ij[2])
|
||||
{
|
||||
static int first = 1;
|
||||
static double lx, ly;
|
||||
int i, j;
|
||||
|
||||
if (first)
|
||||
{
|
||||
lx = XMAX - XMIN + 2.0*HASHGRID_PADDING;
|
||||
ly = YMAX - YMIN + 2.0*HASHGRID_PADDING;
|
||||
first = 0;
|
||||
}
|
||||
|
||||
i = (int)((double)HASHX*(x - XMIN + HASHGRID_PADDING)/lx);
|
||||
j = (int)((double)HASHY*(y - YMIN + HASHGRID_PADDING)/ly);
|
||||
|
||||
if (i<0) i = 0;
|
||||
if (i>=HASHX) i = HASHX-1;
|
||||
if (j<0) j = 0;
|
||||
if (j>=HASHY) j = HASHY-1;
|
||||
|
||||
ij[0] = i;
|
||||
ij[1] = j;
|
||||
|
||||
// printf("Mapped (%.3f,%.3f) to (%i, %i)\n", x, y, ij[0], ij[1]);
|
||||
}
|
||||
|
||||
|
||||
void compute_repelling_force(int i, int j, double force[2])
|
||||
/* compute repelling force of mangrove j on mangrove i */
|
||||
{
|
||||
double x1, y1, x2, y2, distance, r, f;
|
||||
|
||||
x1 = circlex[i];
|
||||
y1 = circley[i];
|
||||
x2 = circlex[j];
|
||||
y2 = circley[j];
|
||||
|
||||
distance = module2(x2 - x1, y2 - y1);
|
||||
r = circlerad[i] + circlerad[j];
|
||||
if (r <= 0.0) r = 0.001*MU;
|
||||
f = KREPEL/(0.001 + distance*distance);
|
||||
|
||||
if ((distance > 0.0)&&(distance < REPEL_RADIUS*r))
|
||||
{
|
||||
force[0] = f*(x1 - x2)/distance;
|
||||
force[1] = f*(y1 - y2)/distance;
|
||||
}
|
||||
else
|
||||
{
|
||||
force[0] = 0.0;
|
||||
force[1] = 0.0;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void update_hashgrid(int* mangrove_hashx, int* mangrove_hashy, int* hashgrid_number, int* hashgrid_mangroves)
|
||||
{
|
||||
int i, j, k, n, m, ij[2], max = 0;
|
||||
|
||||
printf("Updating hashgrid_number\n");
|
||||
for (i=0; i<HASHX*HASHY; i++) hashgrid_number[i] = 0;
|
||||
printf("Updated hashgrid_number\n");
|
||||
|
||||
/* place each mangrove in hash grid */
|
||||
for (k=1; k<ncircles; k++)
|
||||
// if (circleactive[k])
|
||||
{
|
||||
// printf("placing circle %i\t", k);
|
||||
hash_xy_to_ij(circlex[k], circley[k], ij);
|
||||
i = ij[0]; j = ij[1];
|
||||
// printf("ij = (%i, %i)\t", i, j);
|
||||
n = hashgrid_number[i*HASHY + j];
|
||||
m = i*HASHY*HASHMAX + j*HASHMAX + n;
|
||||
// printf("n = %i, m = %i\n", n, m);
|
||||
if (m < HASHX*HASHY*HASHMAX) hashgrid_mangroves[m] = k;
|
||||
else printf("Too many mangroves in hash cell, try increasing HASHMAX\n");
|
||||
hashgrid_number[i*HASHY + j]++;
|
||||
mangrove_hashx[k] = i;
|
||||
mangrove_hashy[k] = j;
|
||||
|
||||
if (n > max) max = n;
|
||||
// printf("Placed mangrove %i at (%i,%i) in hashgrid\n", k, ij[0], ij[1]);
|
||||
// printf("%i mangroves at (%i,%i)\n", hashgrid_number[ij[0]][ij[1]], ij[0], ij[1]);
|
||||
}
|
||||
|
||||
printf("Maximal number of mangroves per hash cell: %i\n", max);
|
||||
}
|
||||
|
||||
|
||||
void animation()
|
||||
{
|
||||
double time, scale, diss, rgb[3], hue, y, dissip, ej, gradient[2], dx, dy, dt, xleft, xright, length;
|
||||
double time, scale, diss, rgb[3], hue, y, dissip, ej, gradient[2], dx, dy, dt, xleft, xright,
|
||||
length, fx, fy, force[2];
|
||||
double *phi[NX], *psi[NX], *phi_tmp[NX], *psi_tmp[NX];
|
||||
short int *xy_in[NX], redraw = 0;
|
||||
int i, j, s, ij[2];
|
||||
int i, j, k, n, s, ij[2], i0, iplus, iminus, j0, jplus, jminus, p, q;
|
||||
static int imin, imax;
|
||||
static short int first = 1;
|
||||
|
||||
double *circleenergy, *circley_wrapped, *anchor_x, *anchor_y, *vx, *vy, *circlerad_initial, *mass_inverse;
|
||||
short int *circle_attached;
|
||||
int *mangrove_hashx, *mangrove_hashy, *hashgrid_number, *hashgrid_mangroves;
|
||||
|
||||
/* Since NX and NY are big, it seemed wiser to use some memory allocation here */
|
||||
for (i=0; i<NX; i++)
|
||||
@@ -431,6 +543,21 @@ void animation()
|
||||
xy_in[i] = (short int *)malloc(NY*sizeof(short int));
|
||||
}
|
||||
|
||||
circleenergy = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* energy dissipated by the circles */
|
||||
circley_wrapped = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* position of circle centers wrapped vertically */
|
||||
anchor_x = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* points moving circles are attached to */
|
||||
anchor_y = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* points moving circles are attached to */
|
||||
vx = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* x velocity of circles */
|
||||
vy = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* y velocity of circles */
|
||||
circlerad_initial = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* initial circle radii */
|
||||
mass_inverse = (double *)malloc(NMAXCIRCLES*sizeof(double)); /* inverse of mangrove mass */
|
||||
circle_attached = (short int *)malloc(NMAXCIRCLES*sizeof(short int)); /* has value 1 if the circle is attached to its anchor */
|
||||
|
||||
mangrove_hashx = (int *)malloc(NMAXCIRCLES*sizeof(int)); /* hash grid positions of mangroves */
|
||||
mangrove_hashy = (int *)malloc(NMAXCIRCLES*sizeof(int)); /* hash grid positions of mangroves */
|
||||
hashgrid_number = (int *)malloc(HASHX*HASHY*sizeof(int)); /* total number of mangroves in each hash grid cell */
|
||||
hashgrid_mangroves = (int *)malloc(HASHX*HASHY*HASHMAX*sizeof(int)); /* numbers of mangoves in each hash grid cell */
|
||||
|
||||
/* initialise positions and radii of circles */
|
||||
if (B_DOMAIN == D_CIRCLES) init_circle_config();
|
||||
|
||||
@@ -455,13 +582,14 @@ void animation()
|
||||
/* initialise mangroves */
|
||||
for (i=0; i < ncircles; i++)
|
||||
{
|
||||
circle_energy[i] = 0.0;
|
||||
circleenergy[i] = 0.0;
|
||||
y = circley[i];
|
||||
if (y >= YMAX) y -= circlerad[i];
|
||||
if (y <= YMIN) y += circlerad[i];
|
||||
// if (y >= YMAX) y -= (YMAX - YMIN);
|
||||
// if (y <= YMIN) y += (YMAX - YMIN);
|
||||
circley_wrapped[i] = y;
|
||||
// circleactive[i] = 1;
|
||||
|
||||
if (RANDOM_RADIUS) circlerad[i] = circlerad[i]*(0.75 + 0.5*((double)rand()/RAND_MAX));
|
||||
|
||||
@@ -483,6 +611,9 @@ void animation()
|
||||
}
|
||||
}
|
||||
|
||||
/* initialise hash table for interacting mangroves */
|
||||
if (REPELL_MANGROVES) update_hashgrid(mangrove_hashx, mangrove_hashy, hashgrid_number, hashgrid_mangroves);
|
||||
|
||||
if (first) /* compute box limits where circles are reset */
|
||||
{
|
||||
/* find leftmost and rightmost circle */
|
||||
@@ -515,7 +646,7 @@ void animation()
|
||||
|
||||
for (i=0; i<=INITIAL_TIME + NSTEPS; i++)
|
||||
{
|
||||
//printf("%d\n",i);
|
||||
printf("Computing frame %d\n",i);
|
||||
/* compute the variance of the field to adjust color scheme */
|
||||
/* the color depends on the field divided by sqrt(1 + variance) */
|
||||
if (SCALE)
|
||||
@@ -525,83 +656,24 @@ void animation()
|
||||
}
|
||||
else scale = 1.0;
|
||||
|
||||
printf("Drawing wave\n");
|
||||
draw_wave(phi, psi, xy_in, scale, i, PLOT);
|
||||
|
||||
|
||||
printf("Evolving wave\n");
|
||||
for (j=0; j<NVID; j++)
|
||||
{
|
||||
// printf("%i ", j);
|
||||
evolve_wave(phi, psi, phi_tmp, psi_tmp, xy_in);
|
||||
// if (i % 10 == 9) oscillate_linear_wave(0.2*scale, 0.15*(double)(i*NVID + j), -1.5, YMIN, -1.5, YMAX, phi, psi);
|
||||
}
|
||||
|
||||
/* compute energy dissipated in obstacles */
|
||||
if (ERODE_MANGROVES) for (j=0; j<ncircles; j++)
|
||||
{
|
||||
dissip = compute_dissipation(phi, psi, xy_in, circlex[j], circley_wrapped[j]);
|
||||
|
||||
/* make sure the dissipation does not grow too fast because of round-off/blow-up */
|
||||
if (dissip > 0.1*MANGROVE_EMAX)
|
||||
{
|
||||
dissip = 0.1*MANGROVE_EMAX;
|
||||
printf("Flooring dissipation!\n");
|
||||
}
|
||||
|
||||
if (circleactive[j])
|
||||
{
|
||||
circle_energy[j] += dissip;
|
||||
ej = circle_energy[j];
|
||||
if (ej <= MANGROVE_EMAX)
|
||||
{
|
||||
if (ej > 0.0)
|
||||
{
|
||||
hue = MANGROVE_HUE_MIN + (MANGROVE_HUE_MAX - MANGROVE_HUE_MIN)*ej/MANGROVE_EMAX;
|
||||
if (hue < 0.0) hue += 360.0;
|
||||
}
|
||||
else hue = MANGROVE_HUE_MIN;
|
||||
hsl_to_rgb(hue, 0.9, 0.5, rgb);
|
||||
if (j%NGRIDY == 0) printf("Circle %i, energy %.5lg, hue %.5lg\n", j, ej, hue);
|
||||
draw_colored_circle(circlex[j], circley[j], circlerad[j], NSEG, rgb);
|
||||
|
||||
/* shrink mangrove */
|
||||
if (ej > 0.0)
|
||||
{
|
||||
// circlerad[j] -= MU*ej*ej/(MANGROVE_EMAX*MANGROVE_EMAX);
|
||||
// if (circlerad[j] < 0.0) circlerad[j] = 0.0;
|
||||
circlerad[j] = circlerad_initial[j]*(1.0 - ej*ej/(MANGROVE_EMAX*MANGROVE_EMAX));
|
||||
redraw = 1;
|
||||
}
|
||||
else circlerad[j] = circlerad_initial[j];
|
||||
}
|
||||
else /* remove mangrove */
|
||||
{
|
||||
circleactive[j] = 0;
|
||||
/* reinitialize table xy_in */
|
||||
redraw = 1;
|
||||
}
|
||||
}
|
||||
else /* allow disabled mangroves to recover */
|
||||
{
|
||||
circle_energy[j] -= 0.15*dissip;
|
||||
// circlerad[j] += 0.005*MU;
|
||||
// if (circlerad[j] > MU) circlerad[j] = MU;
|
||||
// if ((circle_energy[j] < 0.0)&&(circlerad[j] > 0.0))
|
||||
if (circle_energy[j] < 0.0)
|
||||
{
|
||||
circleactive[j] = 1;
|
||||
// circlerad[j] = circlerad[j]*(0.75 + 0.5*((double)rand()/RAND_MAX));
|
||||
circlerad[j] = circlerad_initial[j];
|
||||
circle_energy[j] = -MANGROVE_EMAX;
|
||||
/* reinitialize table xy_in */
|
||||
redraw = 1;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// printf("Circle %i, energy %.5lg\n", j, circle_energy[j]);
|
||||
}
|
||||
|
||||
/* move mangroves */
|
||||
if (MOVE_MANGROVES) for (j=0; j<ncircles; j++) if (circleactive[j])
|
||||
{
|
||||
compute_gradient(phi, psi, circlex[j], circley_wrapped[j], gradient);
|
||||
// printf("gradient = (%.3lg, %.3lg)\t", gradient[0], gradient[1]);
|
||||
|
||||
// if (j%NGRIDY == 0) printf("gradient (%.3lg, %.3lg)\n", gradient[0], gradient[1]);
|
||||
// if (j%NGRIDY == 0) printf("circle %i (%.3lg, %.3lg) -> ", j, circlex[j], circley[j]);
|
||||
@@ -617,11 +689,41 @@ void animation()
|
||||
dy += DT_MANGROVE*(-KSPRING*(circley_wrapped[j] - anchor_y[j]));
|
||||
}
|
||||
|
||||
/* compute repelling force from other mangroves */
|
||||
if (REPELL_MANGROVES)
|
||||
{
|
||||
/* determine neighboring grid points */
|
||||
i0 = mangrove_hashx[j];
|
||||
iminus = i0 - 1; if (iminus < 0) iminus = 0;
|
||||
iplus = i0 + 1; if (iplus >= HASHX) iplus = HASHX-1;
|
||||
|
||||
j0 = mangrove_hashy[j];
|
||||
jminus = j0 - 1; if (jminus < 0) jminus = 0;
|
||||
jplus = j0 + 1; if (jplus >= HASHY) jplus = HASHY-1;
|
||||
|
||||
fx = 0.0;
|
||||
fy = 0.0;
|
||||
for (p=iminus; p<= iplus; p++)
|
||||
for (q=jminus; q<= jplus; q++)
|
||||
for (k=0; k<hashgrid_number[p*HASHY+q]; k++)
|
||||
if (circleactive[hashgrid_mangroves[p*HASHY*HASHMAX + q*HASHMAX + k]])
|
||||
{
|
||||
compute_repelling_force(j, hashgrid_mangroves[p*HASHY*HASHMAX + q*HASHMAX + k], force);
|
||||
fx += force[0];
|
||||
fy += force[1];
|
||||
}
|
||||
|
||||
// if (fx*fx + fy*fy > 0.001) printf("Force on mangrove %i: (%.3f, %.3f)\n", j, fx, fy);
|
||||
|
||||
dx += DT_MANGROVE*fx;
|
||||
dy += DT_MANGROVE*fy;
|
||||
}
|
||||
|
||||
/* detach mangrove if spring is too long */
|
||||
if (DETACH_MANGROVES)
|
||||
{
|
||||
length = module2(circlex[j] - anchor_x[j], circley_wrapped[j] - anchor_y[j]);
|
||||
if (j%NGRIDY == 0) printf("spring length %.i: %.3lg\n", j, length);
|
||||
// if (j%NGRIDY == 0) printf("spring length %.i: %.3lg\n", j, length);
|
||||
// if (length > L_DETACH) circle_attached[j] = 0;
|
||||
if (length*mass_inverse[j] > L_DETACH) circle_attached[j] = 0;
|
||||
}
|
||||
@@ -638,9 +740,9 @@ void animation()
|
||||
circlex[j] += vx[j]*DT_MANGROVE;
|
||||
circley[j] += vy[j]*DT_MANGROVE;
|
||||
circley_wrapped[j] += vy[j]*DT_MANGROVE;
|
||||
if (j%NGRIDY == 0)
|
||||
printf("circle %.i: (dx,dy) = (%.3lg,%.3lg), (vx,vy) = (%.3lg,%.3lg)\n",
|
||||
j, circlex[j]-anchor_x[j], circley[j]-anchor_y[j], vx[j], vy[j]);
|
||||
// if (j%NGRIDY == 0)
|
||||
// printf("circle %.i: (dx,dy) = (%.3lg,%.3lg), (vx,vy) = (%.3lg,%.3lg)\n",
|
||||
// j, circlex[j]-anchor_x[j], circley[j]-anchor_y[j], vx[j], vy[j]);
|
||||
}
|
||||
else
|
||||
{
|
||||
@@ -655,10 +757,148 @@ void animation()
|
||||
if (circley_wrapped[j] >= YMAX) circley_wrapped[j] = YMAX;
|
||||
|
||||
// if (j%NGRIDY == 0) printf("(%.3lg, %.3lg)\n", circlex[j], circley[j]);
|
||||
|
||||
|
||||
redraw = 1;
|
||||
}
|
||||
|
||||
/* test for debugging */
|
||||
if (1) for (j=0; j<ncircles; j++)
|
||||
{
|
||||
dissip = compute_dissipation(phi, psi, xy_in, circlex[j], circley_wrapped[j]);
|
||||
|
||||
/* make sure the dissipation does not grow too fast because of round-off/blow-up */
|
||||
if (dissip > 0.1*MANGROVE_EMAX)
|
||||
{
|
||||
dissip = 0.1*MANGROVE_EMAX;
|
||||
printf("Flooring dissipation!\n");
|
||||
}
|
||||
|
||||
if (circleactive[j])
|
||||
{
|
||||
circleenergy[j] += dissip;
|
||||
ej = circleenergy[j];
|
||||
// printf("ej = %.3f\n", ej);
|
||||
if (ej <= MANGROVE_EMAX)
|
||||
{
|
||||
if (ej > 0.0)
|
||||
{
|
||||
hue = MANGROVE_HUE_MIN + (MANGROVE_HUE_MAX - MANGROVE_HUE_MIN)*ej/MANGROVE_EMAX;
|
||||
if (hue < 0.0) hue += 360.0;
|
||||
}
|
||||
else hue = MANGROVE_HUE_MIN;
|
||||
hsl_to_rgb(hue, 0.9, 0.5, rgb);
|
||||
// if (j%NGRIDY == 0) printf("Circle %i, energy %.5lg, hue %.5lg\n", j, ej, hue);
|
||||
draw_colored_circle(circlex[j], circley[j], circlerad[j], NSEG, rgb);
|
||||
|
||||
/* shrink mangrove */
|
||||
if ((ERODE_MANGROVES)&&(ej > 0.0))
|
||||
{
|
||||
circlerad[j] = circlerad_initial[j]*(1.0 - ej*ej/(MANGROVE_EMAX*MANGROVE_EMAX));
|
||||
redraw = 1;
|
||||
}
|
||||
else circlerad[j] = circlerad_initial[j];
|
||||
|
||||
}
|
||||
else /* remove mangrove */
|
||||
{
|
||||
circleactive[j] = 0;
|
||||
/* reinitialize table xy_in */
|
||||
redraw = 1;
|
||||
}
|
||||
}
|
||||
else if (RECOVER_MANGROVES) /* allow disabled mangroves to recover */
|
||||
{
|
||||
circleenergy[j] -= 0.15*dissip;
|
||||
printf("Circle %i energy %.3lg\n", j, circleenergy[j]);
|
||||
if (circleenergy[j] < 0.0)
|
||||
{
|
||||
printf("Reactivating circle %i?\n", j);
|
||||
/* THE PROBLEM occurs when circleactive[0] is set to 1 again */
|
||||
if (j>0) circleactive[j] = 1;
|
||||
circlerad[j] = circlerad_initial[j];
|
||||
circleenergy[j] = -MANGROVE_EMAX;
|
||||
/* reinitialize table xy_in */
|
||||
redraw = 1;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
/* compute energy dissipated in obstacles */
|
||||
/* if (ERODE_MANGROVES) for (j=0; j<ncircles; j++)
|
||||
{
|
||||
// printf("j = %i\t", j);
|
||||
dissip = compute_dissipation(phi, psi, xy_in, circlex[j], circley_wrapped[j]);
|
||||
printf("dissip = %.3f\t", dissip);
|
||||
|
||||
/* make sure the dissipation does not grow too fast because of round-off/blow-up */
|
||||
// if (dissip > 0.1*MANGROVE_EMAX)
|
||||
// {
|
||||
// dissip = 0.1*MANGROVE_EMAX;
|
||||
// printf("Flooring dissipation!\n");
|
||||
// }
|
||||
//
|
||||
// if (circleactive[j])
|
||||
// {
|
||||
// circleenergy[j] += dissip;
|
||||
// ej = circleenergy[j];
|
||||
// printf("ej = %.3f\n", ej);
|
||||
// if (ej <= MANGROVE_EMAX)
|
||||
// {
|
||||
// if (ej > 0.0)
|
||||
// {
|
||||
// hue = MANGROVE_HUE_MIN + (MANGROVE_HUE_MAX - MANGROVE_HUE_MIN)*ej/MANGROVE_EMAX;
|
||||
// if (hue < 0.0) hue += 360.0;
|
||||
// }
|
||||
// else hue = MANGROVE_HUE_MIN;
|
||||
// hsl_to_rgb(hue, 0.9, 0.5, rgb);
|
||||
// if (j%NGRIDY == 0) printf("Circle %i, energy %.5lg, hue %.5lg\n", j, ej, hue);
|
||||
// draw_colored_circle(circlex[j], circley[j], circlerad[j], NSEG, rgb);
|
||||
//
|
||||
// /* shrink mangrove */
|
||||
// if (ej > 0.0)
|
||||
// {
|
||||
// circlerad[j] -= MU*ej*ej/(MANGROVE_EMAX*MANGROVE_EMAX);
|
||||
// if (circlerad[j] < 0.0) circlerad[j] = 0.0;
|
||||
// circlerad[j] = circlerad_initial[j]*(1.0 - ej*ej/(MANGROVE_EMAX*MANGROVE_EMAX));
|
||||
// redraw = 1;
|
||||
// }
|
||||
// else circlerad[j] = circlerad_initial[j];
|
||||
// }
|
||||
// else /* remove mangrove */
|
||||
// {
|
||||
// circleactive[j] = 0;
|
||||
/* reinitialize table xy_in */
|
||||
// redraw = 1;
|
||||
// }
|
||||
// }
|
||||
// else /* allow disabled mangroves to recover */
|
||||
// {
|
||||
// circleenergy[j] -= 0.15*dissip;
|
||||
// printf("ej = %.3f\n", circleenergy[j]);
|
||||
// circlerad[j] += 0.005*MU;
|
||||
// if (circlerad[j] > MU) circlerad[j] = MU;
|
||||
// if ((circleenergy[j] < 0.0)&&(circlerad[j] > 0.0))
|
||||
// if (circleenergy[j] < 0.0)
|
||||
// {
|
||||
// circleactive[j] = 1;
|
||||
// circlerad[j] = circlerad[j]*(0.75 + 0.5*((double)rand()/RAND_MAX));
|
||||
// circlerad[j] = circlerad_initial[j];
|
||||
// circleenergy[j] = -MANGROVE_EMAX;
|
||||
/* reinitialize table xy_in */
|
||||
// redraw = 1;
|
||||
// }
|
||||
|
||||
// }
|
||||
|
||||
// printf("Circle %i, energy %.5lg\n", j, circleenergy[j]);
|
||||
// }
|
||||
|
||||
printf("Updating hashgrid\n");
|
||||
if (REPELL_MANGROVES) update_hashgrid(mangrove_hashx, mangrove_hashy, hashgrid_number, hashgrid_mangroves);
|
||||
|
||||
|
||||
printf("Drawing billiard\n");
|
||||
draw_billiard();
|
||||
|
||||
glutSwapBuffers();
|
||||
@@ -666,7 +906,7 @@ void animation()
|
||||
if (redraw)
|
||||
{
|
||||
printf("Reinitializing xy_in\n");
|
||||
init_xyin_xrange(xy_in, imin, NX-1);
|
||||
init_xyin_xrange(xy_in, imin, NX);
|
||||
// init_xyin_xrange(xy_in, imin, imax);
|
||||
}
|
||||
redraw = 0;
|
||||
@@ -701,6 +941,21 @@ void animation()
|
||||
free(psi_tmp[i]);
|
||||
free(xy_in[i]);
|
||||
}
|
||||
|
||||
free(circleenergy);
|
||||
free(circley_wrapped);
|
||||
free(anchor_x);
|
||||
free(anchor_y);
|
||||
free(vx);
|
||||
free(vy);
|
||||
free(circlerad_initial);
|
||||
free(mass_inverse);
|
||||
free(circle_attached);
|
||||
|
||||
free(mangrove_hashx);
|
||||
free(mangrove_hashy);
|
||||
free(hashgrid_number);
|
||||
free(hashgrid_mangroves);
|
||||
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user