Skip to content

GitLab

Commit 86ee2e8a authored by ph's avatar ph
Browse files
Options

up

parent 48bf31ee
Hide whitespace changes
Inline Side-by-side
Showing with 100 additions and 127 deletions
stvenant/stvenant_cl.py
View file @ 86ee2e8a
......@@ -34,7 +34,7 @@ vmax = np.sqrt(gpes * hmax) + umax
# time stepping
cfl = 0.8
_Tmax = 0.05
_Tmax = 1.0
def solve_ocl(m=_m, nx=_nx, ny=_ny, Tmax=_Tmax, Lx=_Lx, Ly=_Ly, animate=True, ivplot=0, precision="single", **kwargs):
......@@ -60,6 +60,7 @@ def solve_ocl(m=_m, nx=_nx, ny=_ny, Tmax=_Tmax, Lx=_Lx, Ly=_Ly, animate=True, iv
np_real, source = load_kernel("stvenant_kernels.cl", parameters, precision=precision, print_source=True,
module_file=__file__)
# OpenCL init
ctx = cl.create_some_context()
mf = cl.mem_flags
......
stvenant/stvenant_kernels.cl
View file @ 86ee2e8a
......@@ -11,66 +11,54 @@
#define _VOL (_DX * _DY)
__constant int dir[4][2] = { {1, 0}, {-1, 0},
{0, 1}, {0, -1}};
__constant int dir[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
__constant float ds[4] = { _DY, _DY, _DX, _DX };
__constant float ds[4] = {_DY, _DY, _DX, _DX};
#define double float
void riem_stvenant(double *wL,
double *wR,
double xi,
double *w);
void riem_stvenant(double *wL, double *wR, double xi, double *w);
double Z(double h1,
double h2);
double Z(double h1, double h2);
double dZ(double h1,
double h2);
double dZ(double h1, double h2);
void flux_riem_2d(double *wL, double *wR, double *vn, double *flux);
void fluxphy(float *w, float* n, float *flux){
void fluxphy(float *w, float *n, float *flux) {
float h = w[0];
float u = w[1] / h;
float v=w[2] / h;
float v = w[2] / h;
float un = u * n[0] + v * n[1];
flux[0] = h * un;
flux[1] = h * u * un + _G * h * h / 2 * n[0];
flux[2] = h * v * un + _G * h * h / 2 * n[1];
flux[1] = h * u * un + _G * h * h / 2 * n[0];
flux[2] = h * v * un + _G * h * h / 2 * n[1];
}
void fluxnum(float *wL, float *wR, float* vnorm, float* flux){
void fluxnum(float *wL, float *wR, float *vnorm, float *flux) {
float fL[_M];
float fR[_M];
fluxphy(wL, vnorm, fL);
fluxphy(wR, vnorm, fR);
for(int iv = 0; iv < _M; iv++){
for (int iv = 0; iv < _M; iv++) {
flux[iv] = 0.5f * (fL[iv] + fR[iv]) - 0.5f * _LAMBDA * (wR[iv] - wL[iv]);
}
}
__constant double g = 9.81;
void flux_riem_2d(double *wL, double *wR, double *vnorm, double *flux){
void flux_riem_2d(double *wL, double *wR, double *vnorm, double *flux) {
double qnL = wL[1] * vnorm[0] + wL[2] * vnorm[1];
double qnR = wR[1] * vnorm[0] + wR[2] * vnorm[1];
double qnL = wL[1] * vnorm[0] + wL[2] * vnorm[1];
double qnR = wR[1] * vnorm[0] + wR[2] * vnorm[1];
double qtL = -wL[1] * vnorm[1] + wL[2] * vnorm[0];
double qtR = -wR[1] * vnorm[1] + wR[2] * vnorm[0];
double qtL = -wL[1] * vnorm[1] + wL[2] * vnorm[0];
double qtR = -wR[1] * vnorm[1] + wR[2] * vnorm[0];
double vL[2] = {wL[0], qnL};
double vR[2] = {wR[0], qnR};
......@@ -100,38 +88,30 @@ void flux_riem_2d(double *wL, double *wR, double *vnorm, double *flux){
// puis calcul du flux.
fluxphy(w, vnorm, flux);
//if (wL[0] != wR[0])
//printf("flux=%f %f %f w=%f %f %f\n",flux[0], flux[1], flux[2],
// if (wL[0] != wR[0])
// printf("flux=%f %f %f w=%f %f %f\n",flux[0], flux[1], flux[2],
// w[0], w[1], w[2]);
}
void riem_stvenant(double *wL,
double *wR,
double xi,
double *w){
void riem_stvenant(double *wL, double *wR, double xi, double *w) {
double hL = wL[0];
double uL = wL[1]/wL[0];
double uL = wL[1] / wL[0];
double hR = wR[0];
double uR = wR[1]/wR[0];
double uR = wR[1] / wR[0];
double hs = 1e-10;
int itermax = 10;
for(int it = 0; it < itermax; it++){
double f = uL - (hs - hL) * Z(hs, hL) -
uR - (hs - hR) * Z(hs, hR);
double df = -(hs - hL) * dZ(hs, hL) -
Z(hs, hL) -
(hs - hR) * dZ(hs, hR) -
Z(hs, hR);
for (int it = 0; it < itermax; it++) {
double f = uL - (hs - hL) * Z(hs, hL) - uR - (hs - hR) * Z(hs, hR);
double df = -(hs - hL) * dZ(hs, hL) - Z(hs, hL) - (hs - hR) * dZ(hs, hR) -
Z(hs, hR);
double dhs = -f / df;
hs += dhs;
//printf("it=%d f=%e df=%e hs=%e dhs=%e\n",it,f,df,hs,dhs);
// printf("it=%d f=%e df=%e hs=%e dhs=%e\n",it,f,df,hs,dhs);
}
double us = uL - (hs - hL) * Z(hs, hL);
......@@ -139,7 +119,7 @@ void riem_stvenant(double *wL,
double v1m, v1p, v2m, v2p;
// 1-onde
if (hs < hL){
if (hs < hL) {
v1m = uL - sqrt(g * hL);
v1p = us - sqrt(g * hs);
} else {
......@@ -151,7 +131,7 @@ void riem_stvenant(double *wL,
}
// 2 onde
if (hs < hR){
if (hs < hR) {
v2m = us + sqrt(g * hs);
v2p = uR + sqrt(g * hR);
} else {
......@@ -162,22 +142,22 @@ void riem_stvenant(double *wL,
v2p = v2m;
}
//if (hL != hR)
//printf("v=%f %f %f %f\n hs=%f us=%f\n", v1m,v1p,v2m,v2p, hs,us);
// if (hL != hR)
// printf("v=%f %f %f %f\n hs=%f us=%f\n", v1m,v1p,v2m,v2p, hs,us);
if (xi < v1m) {
w[0] = wL[0];
w[1] = wL[1];
} else if (xi < v1p){
double u = (uL + 2 * xi + 2 *sqrt(g * hL)) / 3;
} else if (xi < v1p) {
double u = (uL + 2 * xi + 2 * sqrt(g * hL)) / 3;
double h = (u - xi) * (u - xi) / g;
w[0] = h;
w[1] = h * u;
} else if (xi < v2m){
} else if (xi < v2m) {
w[0] = hs;
w[1] = hs * us;
} else if (xi < v2p){
double u = (uR + 2 * xi - 2 *sqrt(g * hR)) / 3;
} else if (xi < v2p) {
double u = (uR + 2 * xi - 2 * sqrt(g * hR)) / 3;
double h = (u - xi) * (u - xi) / g;
w[0] = h;
w[1] = h * u;
......@@ -185,62 +165,61 @@ void riem_stvenant(double *wL,
w[0] = wR[0];
w[1] = wR[1];
}
}
double Heaviside(double x){
double Heaviside(double x) {
if (x > 0)
return 1;
else
return 0;
}
double Dirac(double x){
return 0;
}
double Dirac(double x) { return 0; }
double Z(double hs,
double h){
double Z(double hs, double h) {
double t0 = 2.0*sqrt(g)/(sqrt(hs)+sqrt(h))*Heaviside(h-hs)+sqrt(2.0)*sqrt(g)*sqrt(h+hs)/sqrt(h*hs)/2.0-sqrt(2.0)*sqrt(g)*sqrt(h+hs)/sqrt(h*hs)*Heaviside(h-hs)/2.0;
double t0 = 2.0 * sqrt(g) / (sqrt(hs) + sqrt(h)) * Heaviside(h - hs) +
sqrt(2.0) * sqrt(g) * sqrt(h + hs) / sqrt(h * hs) / 2.0 -
sqrt(2.0) * sqrt(g) * sqrt(h + hs) / sqrt(h * hs) *
Heaviside(h - hs) / 2.0;
return t0;
return t0;
}
double dZ(double hs,
double h){
double t0 = -sqrt(g)/pow(sqrt(hs)+sqrt(h),2.0f)*Heaviside(h-hs)/sqrt(hs)-2.0*sqrt
(g)/(sqrt(hs)+sqrt(h))*Dirac(-h+hs)+sqrt(2.0)*sqrt(g)/sqrt(h+hs)/sqrt(h*hs)/4.0
-sqrt(2.0)*sqrt(g)*sqrt(h+hs)/sqrt(h*h*h*hs*hs*hs)*h/4.0-sqrt(2.0)*sqrt(g)/sqrt
(h+hs)/sqrt(h*hs)*Heaviside(h-hs)/4.0+sqrt(2.0)*sqrt(g)*sqrt(h+hs)/sqrt(h*h*h*
hs*hs*hs)*Heaviside(h-hs)*h/4.0+sqrt(2.0)*sqrt(g)*sqrt(h+hs)/sqrt(h*hs)*Dirac(-
h+hs)/2.0;
return t0;
double dZ(double hs, double h) {
double t0 =
-sqrt(g) / pow(sqrt(hs) + sqrt(h), 2.0f) * Heaviside(h - hs) / sqrt(hs) -
2.0 * sqrt(g) / (sqrt(hs) + sqrt(h)) * Dirac(-h + hs) +
sqrt(2.0) * sqrt(g) / sqrt(h + hs) / sqrt(h * hs) / 4.0 -
sqrt(2.0) * sqrt(g) * sqrt(h + hs) / sqrt(h * h * h * hs * hs * hs) * h /
4.0 -
sqrt(2.0) * sqrt(g) / sqrt(h + hs) / sqrt(h * hs) * Heaviside(h - hs) /
4.0 +
sqrt(2.0) * sqrt(g) * sqrt(h + hs) / sqrt(h * h * h * hs * hs * hs) *
Heaviside(h - hs) * h / 4.0 +
sqrt(2.0) * sqrt(g) * sqrt(h + hs) / sqrt(h * hs) * Dirac(-h + hs) / 2.0;
return t0;
}
void exact_sol(float* xy, float t, float* w){
void exact_sol(float *xy, float t, float *w) {
float h = 1;
float x = xy[0];// - 0.5f;
float y = xy[1];// - 0.5f;
float x = xy[0]; // - 0.5f;
float y = xy[1]; // - 0.5f;
if (x < 0.75f && x > 0.25f && y < 0.75f && y > 0.25f) h = 2;
//if (x * x + y * y < 0.2f * 0.2f) h = 2;
if (x < 0.75f && x > 0.25f && y < 0.75f && y > 0.25f)
h = 2;
// if (x * x + y * y < 0.2f * 0.2f) h = 2;
w[0] = h;
w[1] = 0;
w[2] = 0;
}
__kernel void init_sol(__global float *wn){
__kernel void init_sol(__global float *wn) {
int id = get_global_id(0);
int i = id % _NX;
int j = id / _NX;
......@@ -250,45 +229,40 @@ __kernel void init_sol(__global float *wn){
float t = 0;
float xy[2] = {i * _DX + _DX / 2, j * _DY + _DY / 2};
exact_sol(xy, t, wnow);
//printf("x=%f, y=%f \n",xy[0],xy[1]);
// printf("x=%f, y=%f \n",xy[0],xy[1]);
// load middle value
for(int iv = 0; iv < _M; iv++){
for (int iv = 0; iv < _M; iv++) {
int imem = i + j * _NX + iv * ngrid;
wn[imem] = wnow[iv];
wn[imem] = wnow[iv];
// boundary values
//if (i == 0 || i == _NX - 1 || j == 0 || j == _NY - 1)
// if (i == 0 || i == _NX - 1 || j == 0 || j == _NY - 1)
// wn[imem] = _WBORD;
}
}
__kernel void time_step(__global float *wn, __global float *wnp1){
__kernel void time_step(__global float *wn, __global float *wnp1) {
int id = get_global_id(0);
int i = id % _NX;
int j = id / _NX;
int ngrid = _NX * _NY;
float wnow[_M];
float wnext[_M];
// load middle value
for(int iv = 0; iv < _M; iv++){
for (int iv = 0; iv < _M; iv++) {
int imem = i + j * _NX + iv * ngrid;
wnow[iv] = wn[imem];
wnext[iv] = wnow[iv];
}
}
// only compute internal cells
if (i > 0 && i < _NX - 1 && j > 0 && j < _NY - 1){
if (i > 0 && i < _NX - 1 && j > 0 && j < _NY - 1) {
float flux[_M];
// loop on all directions:
......@@ -296,7 +270,7 @@ __kernel void time_step(__global float *wn, __global float *wnp1){
// idir = 1 (west)
// idir = 2 (north)
// idir = 3 (south)
for(int idir = 0; idir < 4; idir++){
for (int idir = 0; idir < 4; idir++) {
float vn[2];
vn[0] = dir[idir][0];
vn[1] = dir[idir][1];
......@@ -304,33 +278,31 @@ __kernel void time_step(__global float *wn, __global float *wnp1){
int jR = j + dir[idir][1];
// load neighbour values
float wR[_M];
for(int iv = 0; iv < _M; iv++){
int imem = iv * ngrid + iR + jR * _NX;
wR[iv] = wn[imem];
for (int iv = 0; iv < _M; iv++) {
int imem = iv * ngrid + iR + jR * _NX;
wR[iv] = wn[imem];
}
if (iR == 0 || iR == _NX - 1 || jR == 0 || jR == _NY - 1){
wR[0] = wnow[0];
float qn = wnow[1] * vn[0] + wnow[2] * vn[1];
wR[1] = wnow[1] - 2 * qn * vn[0];
wR[2] = wnow[2] - 2 * qn * vn[1];
if (iR == 0 || iR == _NX - 1 || jR == 0 || jR == _NY - 1) {
wR[0] = wnow[0];
float qn = wnow[1] * vn[0] + wnow[2] * vn[1];
wR[1] = wnow[1] - 2 * qn * vn[0];
wR[2] = wnow[2] - 2 * qn * vn[1];
}
//flux_riem_2d(wnow, wR, vn, flux);
fluxnum(wnow, wR, vn, flux);
flux_riem_2d(wnow, wR, vn, flux);
//fluxnum(wnow, wR, vn, flux);
// time evolution
for(int iv = 0; iv < _M; iv++){
wnext[iv] -= _DT * ds[idir] / _VOL * flux[iv];
}
for (int iv = 0; iv < _M; iv++) {
wnext[iv] -= _DT * ds[idir] / _VOL * flux[iv];
}
}
} // end test for internal cells
} // end test for internal cells
// copy wnext into global memory
// (including boundary cells)
for(int iv = 0; iv < _M; iv++){
for (int iv = 0; iv < _M; iv++) {
int imem = iv * ngrid + i + j * _NX;
wnp1[imem] = wnext[iv];
}
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment