Added forward comm for atomscale values

This commit is contained in:
Aidan Thompson
2025-01-15 14:11:00 -07:00
parent 2a5e20c1c1
commit b8fcb927ee
2 changed files with 35 additions and 0 deletions

View File

@ -35,6 +35,10 @@ PairHybridScaled::PairHybridScaled(LAMMPS *lmp) :
PairHybrid(lmp), fsum(nullptr), tsum(nullptr), scaleval(nullptr), scaleidx(nullptr), atomvar(nullptr), atomscale(nullptr) PairHybrid(lmp), fsum(nullptr), tsum(nullptr), scaleval(nullptr), scaleidx(nullptr), atomvar(nullptr), atomscale(nullptr)
{ {
nmaxfsum = -1; nmaxfsum = -1;
// set comm size needed by this Pair (if atomscaleflag)
comm_forward = 1;
} }
/* ---------------------------------------------------------------------- */ /* ---------------------------------------------------------------------- */
@ -195,8 +199,11 @@ void PairHybridScaled::compute(int eflag, int vflag)
} else { } else {
int igroupall = 0; int igroupall = 0;
input->variable->compute_atom(atomvar[m],igroupall,atomscale,1,0); input->variable->compute_atom(atomvar[m],igroupall,atomscale,1,0);
comm->forward_comm(this);
for (i = 0; i < nall; ++i) { for (i = 0; i < nall; ++i) {
const double ascale = atomscale[i]; const double ascale = atomscale[i];
// if (i >= atom->nlocal)
// printf("id = %lld scale = %g\n", atom->tag[i], ascale);
fsum[i][0] += ascale * f[i][0]; fsum[i][0] += ascale * f[i][0];
fsum[i][1] += ascale * f[i][1]; fsum[i][1] += ascale * f[i][1];
fsum[i][2] += ascale * f[i][2]; fsum[i][2] += ascale * f[i][2];
@ -710,3 +717,28 @@ void PairHybridScaled::copy_svector(int itype, int jtype)
} }
} }
} }
/* ---------------------------------------------------------------------- */
int PairHybridScaled::pack_forward_comm(int n, int *list, double *buf, int /*pbc_flag*/, int * /*pbc*/)
{
int i,j,m;
m = 0;
for (i = 0; i < n; i++) {
j = list[i];
buf[m++] = atomscale[j];
}
return m;
}
/* ---------------------------------------------------------------------- */
void PairHybridScaled::unpack_forward_comm(int n, int first, double *buf)
{
int i,m,last;
m = 0;
last = first + n;
for (i = first; i < last; i++) atomscale[i] = buf[m++];
}

View File

@ -44,6 +44,9 @@ class PairHybridScaled : public PairHybrid {
void init_svector() override; void init_svector() override;
void copy_svector(int, int) override; void copy_svector(int, int) override;
int pack_forward_comm(int, int *, double *, int, int *) override;
void unpack_forward_comm(int, int, double *) override;
protected: protected:
double **fsum, **tsum; double **fsum, **tsum;
double *scaleval; double *scaleval;