Skip to content

Commit e9f5266

Browse files
committed
Adding default densities to ArbLatticeAccess
1 parent 9e8cf4f commit e9f5266

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/ArbLatticeAccess.hpp.Rt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,10 +115,14 @@ for (s in rows(all_stages)) { ?>
115115
template<class Node>
116116
CudaDeviceFunction void pop<?%s s$suffix ?>(Node& node) const {
117117
<?R
118-
for (d in rows(Density)[s$load.densities]) {
118+
dens = Density;
119+
dens$load = s$load.densities;
120+
for (d in rows(dens)) if (d$load) {
119121
f = rows(Fields)[[match(d$field, Fields$name)]]
120122
dp = c(-d$dx, -d$dy, -d$dz)
121123
cat(paste0(" node.", d$name, " = load_", f$nicename, "<"), paste(dp, collapse=", "), " >();\n")
124+
} else if (!is.na(d$default)) { ?>
125+
<?%s paste("node",d$name,sep=".") ?> = <?%f d$default ?>; <?R
122126
} ?> }
123127
template<class Node>
124128
CudaDeviceFunction void push<?%s s$suffix ?>(const Node& node) const {

0 commit comments

Comments
 (0)