| 586 |  | atomColData.electricField.end(), V3Zero); | 
| 587 |  | } | 
| 588 |  |  | 
| 589 | + | if (storageLayout_ & DataStorage::dslSitePotential) { | 
| 590 | + | fill(atomRowData.sitePotential.begin(), | 
| 591 | + | atomRowData.sitePotential.end(), 0.0); | 
| 592 | + | fill(atomColData.sitePotential.begin(), | 
| 593 | + | atomColData.sitePotential.end(), 0.0); | 
| 594 | + | } | 
| 595 | + |  | 
| 596 |  | #endif | 
| 597 |  | // even in parallel, we need to zero out the local arrays: | 
| 598 |  |  | 
| 624 |  | if (storageLayout_ & DataStorage::dslElectricField) { | 
| 625 |  | fill(snap_->atomData.electricField.begin(), | 
| 626 |  | snap_->atomData.electricField.end(), V3Zero); | 
| 627 | + | } | 
| 628 | + | if (storageLayout_ & DataStorage::dslSitePotential) { | 
| 629 | + | fill(snap_->atomData.sitePotential.begin(), | 
| 630 | + | snap_->atomData.sitePotential.end(), 0.0); | 
| 631 |  | } | 
| 632 |  | } | 
| 633 |  |  | 
| 843 |  | snap_->atomData.electricField[i] += efield_tmp[i]; | 
| 844 |  | } | 
| 845 |  |  | 
| 846 | + | if (storageLayout_ & DataStorage::dslSitePotential) { | 
| 847 | + |  | 
| 848 | + | int nsp = snap_->atomData.sitePotential.size(); | 
| 849 | + | vector<RealType> sp_tmp(nsp, 0.0); | 
| 850 | + |  | 
| 851 | + | AtomPlanRealRow->scatter(atomRowData.sitePotential, sp_tmp); | 
| 852 | + | for (int i = 0; i < nsp; i++) { | 
| 853 | + | snap_->atomData.sitePotential[i] += sp_tmp[i]; | 
| 854 | + | sp_tmp[i] = 0.0; | 
| 855 | + | } | 
| 856 | + |  | 
| 857 | + | AtomPlanRealColumn->scatter(atomColData.sitePotential, sp_tmp); | 
| 858 | + | for (int i = 0; i < nsp; i++) | 
| 859 | + | snap_->atomData.sitePotential[i] += sp_tmp[i]; | 
| 860 | + | } | 
| 861 |  |  | 
| 862 |  | nLocal_ = snap_->getNumberOfAtoms(); | 
| 863 |  |  | 
| 1347 |  | atomColData.electricField[atom2] += *(idat.eField2); | 
| 1348 |  | } | 
| 1349 |  |  | 
| 1350 | + | if (storageLayout_ & DataStorage::dslSitePotential) { | 
| 1351 | + | atomRowData.sitePotential[atom1] += *(idat.sPot1); | 
| 1352 | + | atomColData.sitePotential[atom2] += *(idat.sPot2); | 
| 1353 | + | } | 
| 1354 | + |  | 
| 1355 |  | #else | 
| 1356 |  | pairwisePot += *(idat.pot); | 
| 1357 |  | excludedPot += *(idat.excludedPot); | 
| 1376 |  | if (storageLayout_ & DataStorage::dslElectricField) { | 
| 1377 |  | snap_->atomData.electricField[atom1] += *(idat.eField1); | 
| 1378 |  | snap_->atomData.electricField[atom2] += *(idat.eField2); | 
| 1379 | + | } | 
| 1380 | + |  | 
| 1381 | + | if (storageLayout_ & DataStorage::dslSitePotential) { | 
| 1382 | + | snap_->atomData.sitePotential[atom1] += *(idat.sPot1); | 
| 1383 | + | snap_->atomData.sitePotential[atom2] += *(idat.sPot2); | 
| 1384 |  | } | 
| 1385 |  |  | 
| 1386 |  | #endif |