fixed identation
This commit is contained in:
alphataubio
@ -53,168 +53,167 @@ class RegSphereKokkos : public RegSphere, public KokkosBase {
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int surface_kokkos(double x, double y, double z, double cutoff)
|
||||
{
|
||||
int ncontact;
|
||||
double xs, ys, zs;
|
||||
double xnear[3], xorig[3];
|
||||
{
|
||||
int ncontact;
|
||||
double xs, ys, zs;
|
||||
double xnear[3], xorig[3];
|
||||
|
||||
if (dynamic) {
|
||||
xorig[0] = x; xorig[1] = y; xorig[2] = z;
|
||||
inverse_transform(x, y, z);
|
||||
}
|
||||
if (dynamic) {
|
||||
xorig[0] = x; xorig[1] = y; xorig[2] = z;
|
||||
inverse_transform(x, y, z);
|
||||
}
|
||||
|
||||
xnear[0] = x; xnear[1] = y; xnear[2] = z;
|
||||
xnear[0] = x; xnear[1] = y; xnear[2] = z;
|
||||
|
||||
if (!openflag) {
|
||||
if (interior)
|
||||
ncontact = surface_interior_kokkos(xnear, cutoff);
|
||||
if (!openflag) {
|
||||
if (interior) ncontact = surface_interior_kokkos(xnear, cutoff);
|
||||
else
|
||||
ncontact = surface_exterior_kokkos(xnear, cutoff);
|
||||
} else {
|
||||
// one of surface_int/ext() will return 0
|
||||
// so no need to worry about offset of contact indices
|
||||
ncontact = surface_exterior_kokkos(xnear, cutoff) + surface_interior_kokkos(xnear, cutoff);
|
||||
}
|
||||
|
||||
if (rotateflag && ncontact) {
|
||||
for (int i = 0; i < ncontact; i++) {
|
||||
xs = xnear[0] - d_contact[i].delx;
|
||||
ys = xnear[1] - d_contact[i].dely;
|
||||
zs = xnear[2] - d_contact[i].delz;
|
||||
forward_transform(xs, ys, zs);
|
||||
d_contact[i].delx = xorig[0] - xs;
|
||||
d_contact[i].dely = xorig[1] - ys;
|
||||
d_contact[i].delz = xorig[2] - zs;
|
||||
} else {
|
||||
// one of surface_int/ext() will return 0
|
||||
// so no need to worry about offset of contact indices
|
||||
ncontact = surface_exterior_kokkos(xnear, cutoff) + surface_interior_kokkos(xnear, cutoff);
|
||||
}
|
||||
}
|
||||
|
||||
return ncontact;
|
||||
}
|
||||
if (rotateflag && ncontact) {
|
||||
for (int i = 0; i < ncontact; i++) {
|
||||
xs = xnear[0] - d_contact[i].delx;
|
||||
ys = xnear[1] - d_contact[i].dely;
|
||||
zs = xnear[2] - d_contact[i].delz;
|
||||
forward_transform(xs, ys, zs);
|
||||
d_contact[i].delx = xorig[0] - xs;
|
||||
d_contact[i].dely = xorig[1] - ys;
|
||||
d_contact[i].delz = xorig[2] - zs;
|
||||
}
|
||||
}
|
||||
|
||||
return ncontact;
|
||||
}
|
||||
|
||||
Kokkos::View<Contact*, DeviceType> d_contact;
|
||||
|
||||
private:
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int surface_interior_kokkos(double *x, double cutoff)
|
||||
{
|
||||
double delx = x[0] - xc;
|
||||
double dely = x[1] - yc;
|
||||
double delz = x[2] - zc;
|
||||
double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
if (r > radius || r == 0.0) return 0;
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int surface_interior_kokkos(double *x, double cutoff)
|
||||
{
|
||||
double delx = x[0] - xc;
|
||||
double dely = x[1] - yc;
|
||||
double delz = x[2] - zc;
|
||||
double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
if (r > radius || r == 0.0) return 0;
|
||||
|
||||
double delta = radius - r;
|
||||
if (delta < cutoff) {
|
||||
d_contact[0].r = delta;
|
||||
d_contact[0].delx = delx * (1.0 - radius / r);
|
||||
d_contact[0].dely = dely * (1.0 - radius / r);
|
||||
d_contact[0].delz = delz * (1.0 - radius / r);
|
||||
d_contact[0].radius = -radius;
|
||||
d_contact[0].iwall = 0;
|
||||
d_contact[0].varflag = 1;
|
||||
return 1;
|
||||
double delta = radius - r;
|
||||
if (delta < cutoff) {
|
||||
d_contact[0].r = delta;
|
||||
d_contact[0].delx = delx * (1.0 - radius / r);
|
||||
d_contact[0].dely = dely * (1.0 - radius / r);
|
||||
d_contact[0].delz = delz * (1.0 - radius / r);
|
||||
d_contact[0].radius = -radius;
|
||||
d_contact[0].iwall = 0;
|
||||
d_contact[0].varflag = 1;
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int surface_exterior_kokkos(double *x, double cutoff)
|
||||
{
|
||||
double delx = x[0] - xc;
|
||||
double dely = x[1] - yc;
|
||||
double delz = x[2] - zc;
|
||||
double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
if (r < radius) return 0;
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int surface_exterior_kokkos(double *x, double cutoff)
|
||||
{
|
||||
double delx = x[0] - xc;
|
||||
double dely = x[1] - yc;
|
||||
double delz = x[2] - zc;
|
||||
double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
if (r < radius) return 0;
|
||||
|
||||
double delta = r - radius;
|
||||
if (delta < cutoff) {
|
||||
d_contact[0].r = delta;
|
||||
d_contact[0].delx = delx * (1.0 - radius / r);
|
||||
d_contact[0].dely = dely * (1.0 - radius / r);
|
||||
d_contact[0].delz = delz * (1.0 - radius / r);
|
||||
d_contact[0].radius = radius;
|
||||
d_contact[0].iwall = 0;
|
||||
d_contact[0].varflag = 1;
|
||||
return 1;
|
||||
double delta = r - radius;
|
||||
if (delta < cutoff) {
|
||||
d_contact[0].r = delta;
|
||||
d_contact[0].delx = delx * (1.0 - radius / r);
|
||||
d_contact[0].dely = dely * (1.0 - radius / r);
|
||||
d_contact[0].delz = delz * (1.0 - radius / r);
|
||||
d_contact[0].radius = radius;
|
||||
d_contact[0].iwall = 0;
|
||||
d_contact[0].varflag = 1;
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void add_contact(int n, double *x, double xp, double yp, double zp)
|
||||
{
|
||||
double delx = x[0] - xp;
|
||||
double dely = x[1] - yp;
|
||||
double delz = x[2] - zp;
|
||||
d_contact[n].r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
d_contact[n].radius = 0;
|
||||
d_contact[n].delx = delx;
|
||||
d_contact[n].dely = dely;
|
||||
d_contact[n].delz = delz;
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int k_inside(double x, double y, double z) const
|
||||
{
|
||||
const double delx = x - xc;
|
||||
const double dely = y - yc;
|
||||
const double delz = z - zc;
|
||||
const double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
|
||||
if (r <= radius) return 1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void forward_transform(double &x, double &y, double &z) const
|
||||
{
|
||||
if (rotateflag) rotate(x, y, z, theta);
|
||||
if (moveflag) {
|
||||
x += dx;
|
||||
y += dy;
|
||||
z += dz;
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void add_contact(int n, double *x, double xp, double yp, double zp)
|
||||
{
|
||||
double delx = x[0] - xp;
|
||||
double dely = x[1] - yp;
|
||||
double delz = x[2] - zp;
|
||||
d_contact[n].r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
d_contact[n].radius = 0;
|
||||
d_contact[n].delx = delx;
|
||||
d_contact[n].dely = dely;
|
||||
d_contact[n].delz = delz;
|
||||
}
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void inverse_transform(double &x, double &y, double &z) const
|
||||
{
|
||||
if (moveflag) {
|
||||
x -= dx;
|
||||
y -= dy;
|
||||
z -= dz;
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
int k_inside(double x, double y, double z) const
|
||||
{
|
||||
const double delx = x - xc;
|
||||
const double dely = y - yc;
|
||||
const double delz = z - zc;
|
||||
const double r = sqrt(delx * delx + dely * dely + delz * delz);
|
||||
|
||||
if (r <= radius) return 1;
|
||||
return 0;
|
||||
}
|
||||
if (rotateflag) rotate(x,y,z,-theta);
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void rotate(double &x, double &y, double &z, double angle) const
|
||||
{
|
||||
double a[3],b[3],c[3],d[3],disp[3];
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void forward_transform(double &x, double &y, double &z) const
|
||||
{
|
||||
if (rotateflag) rotate(x, y, z, theta);
|
||||
if (moveflag) {
|
||||
x += dx;
|
||||
y += dy;
|
||||
z += dz;
|
||||
}
|
||||
}
|
||||
|
||||
double sine = sin(angle);
|
||||
double cosine = cos(angle);
|
||||
d[0] = x - point[0];
|
||||
d[1] = y - point[1];
|
||||
d[2] = z - point[2];
|
||||
double x0dotr = d[0]*runit[0] + d[1]*runit[1] + d[2]*runit[2];
|
||||
c[0] = x0dotr * runit[0];
|
||||
c[1] = x0dotr * runit[1];
|
||||
c[2] = x0dotr * runit[2];
|
||||
a[0] = d[0] - c[0];
|
||||
a[1] = d[1] - c[1];
|
||||
a[2] = d[2] - c[2];
|
||||
b[0] = runit[1]*a[2] - runit[2]*a[1];
|
||||
b[1] = runit[2]*a[0] - runit[0]*a[2];
|
||||
b[2] = runit[0]*a[1] - runit[1]*a[0];
|
||||
disp[0] = a[0]*cosine + b[0]*sine;
|
||||
disp[1] = a[1]*cosine + b[1]*sine;
|
||||
disp[2] = a[2]*cosine + b[2]*sine;
|
||||
x = point[0] + c[0] + disp[0];
|
||||
y = point[1] + c[1] + disp[1];
|
||||
z = point[2] + c[2] + disp[2];
|
||||
}
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void inverse_transform(double &x, double &y, double &z) const
|
||||
{
|
||||
if (moveflag) {
|
||||
x -= dx;
|
||||
y -= dy;
|
||||
z -= dz;
|
||||
}
|
||||
if (rotateflag) rotate(x,y,z,-theta);
|
||||
}
|
||||
|
||||
KOKKOS_INLINE_FUNCTION
|
||||
void rotate(double &x, double &y, double &z, double angle) const
|
||||
{
|
||||
double a[3],b[3],c[3],d[3],disp[3];
|
||||
|
||||
double sine = sin(angle);
|
||||
double cosine = cos(angle);
|
||||
d[0] = x - point[0];
|
||||
d[1] = y - point[1];
|
||||
d[2] = z - point[2];
|
||||
double x0dotr = d[0]*runit[0] + d[1]*runit[1] + d[2]*runit[2];
|
||||
c[0] = x0dotr * runit[0];
|
||||
c[1] = x0dotr * runit[1];
|
||||
c[2] = x0dotr * runit[2];
|
||||
a[0] = d[0] - c[0];
|
||||
a[1] = d[1] - c[1];
|
||||
a[2] = d[2] - c[2];
|
||||
b[0] = runit[1]*a[2] - runit[2]*a[1];
|
||||
b[1] = runit[2]*a[0] - runit[0]*a[2];
|
||||
b[2] = runit[0]*a[1] - runit[1]*a[0];
|
||||
disp[0] = a[0]*cosine + b[0]*sine;
|
||||
disp[1] = a[1]*cosine + b[1]*sine;
|
||||
disp[2] = a[2]*cosine + b[2]*sine;
|
||||
x = point[0] + c[0] + disp[0];
|
||||
y = point[1] + c[1] + disp[1];
|
||||
z = point[2] + c[2] + disp[2];
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user