yohai/yebisu
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

first commit

Browse source
This commit is contained in:
Yohai 2019-08-25 16:12:39 +08:00
commit 604f1cb18a
9 changed files with 1683 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

173
grape6.cpp Normal file
View file

@ -0,0 +1,173 @@
#include <algorithm>
#include <cassert>
#include "yebisu_g6.h"
#include "g6util.h"
#define NIMAX 2048
#define MAXDEV 4
#ifndef NB_FLAG
#define NB_FLAG 0
#endif
extern "C"
{
static int sort_mode = 0;
static double eps2_buf[MAXDEV][NIMAX];
static int nnb_buf[MAXDEV][NIMAX];
void g6_open(int clusterid){
assert(clusterid < MAXDEV);
assert(NIMAX >= g6_npipes());
yebisu_g6_open(clusterid);
}
void g6_close(int clusterid){
yebisu_g6_close(clusterid);
}
void g6_set_tunit(int newtunit){}
void g6_set_xunit(int newxunit){}
int g6_set_j_particle(
int clusterid,
int address,
int index,
double tj,
double dtj,
double mass,
double a2by18[3],
double a1by6[3],
double aby2[3],
double v[3],
double x[3])
{
yebisu_g6_push_jp(clusterid, x, v, aby2, a1by6, mass, tj, index, address);
return 0;
}
void g6_set_ti(int clusterid, double ti){
yebisu_g6_set_ti(clusterid, ti);
}
void g6calc_firsthalf0(
int clusterid,
int nj,
int ni,
int index[],
double xi[][3],
double vi[][3],
double fold[][3],
double jold[][3],
double phiold[],
double *eps2,
double h2[],
int mode)
{
double *eps2_ptr = eps2;
if(mode){ // constand eps2
eps2_ptr = eps2_buf[clusterid];
for(int i=0; i<ni; i++){
eps2_ptr[i] = *eps2;
}
}
yebisu_g6_set_ip(clusterid, ni, xi, vi, eps2_ptr, h2, index);
yebisu_g6_launch_gravity(clusterid, ni, nj, NB_FLAG);
}
void g6calc_firsthalf(
int clusterid,
int nj,
int ni,
int index[],
double xi[][3],
double vi[][3],
double fold[][3],
double jold[][3],
double phiold[],
double eps2,
double h2[])
{
g6calc_firsthalf0(clusterid, nj, ni, index, xi, vi, fold, jold, phiold, &eps2, h2, 1);
}
int g6calc_lasthalf(
int clusterid,
int nj,
int ni,
int index[],
double xi[][3],
double vi[][3],
double eps2,
double h2[],
double acc[][3],
double jerk[][3],
double pot[])
{
yebisu_g6_get_force(clusterid, ni, acc, jerk, pot, nnb_buf[clusterid]);
return 0;
}
int g6calc_lasthalf0(
int clusterid,
int nj,
int ni,
int index[],
double xi[][3],
double vi[][3],
double *eps2,
double h2[],
double acc[][3],
double jerk[][3],
double pot[],
int mode)
{
yebisu_g6_get_force(clusterid, ni, acc, jerk, pot, nnb_buf[clusterid]);
return 0;
}
int g6calc_lasthalf2(
int clusterid,
int nj,
int ni,
int index[],
double xi[][3],
double vi[][3],
double eps2,
double h2[],
double acc[][3],
double jerk[][3],
double pot[],
int nnbindex[])
{
assert(NB_FLAG);
yebisu_g6_get_force(clusterid, ni, acc, jerk, pot, nnbindex);
return 0;
}
int g6_read_neighbour_list(int clusterid){
assert(NB_FLAG);
yebisu_g6_receive_neighbor_list(clusterid);
return 0;
}
int g6_get_neighbour_list(
int clusterid,
int ipipe,
int maxlength,
int *nblen,
int nbl[])
{
assert(NB_FLAG);
yebisu_g6_get_neighbor_list(clusterid, ipipe, maxlength, nblen, nbl);
const int nnb = *nblen;
if(nnb < 0) return -1;
if(sort_mode){
std::sort(nbl, nbl+nnb);
}
return 0;
}
void g6_set_neighbour_list_sort_mode(int mode){
sort_mode = mode;
}
int g6_get_neighbour_list_sort_mode(void){
return sort_mode;
}
int g6_npipes(void){
return yebisu_g6_get_nimax();
}
int g6_getnjmax(int clusterid){
return yebisu_g6_get_njmax();
}
}