summaryrefslogtreecommitdiff
path: root/lib/correlation.h
diff options
context:
space:
mode:
Diffstat (limited to 'lib/correlation.h')
-rw-r--r--lib/correlation.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/correlation.h b/lib/correlation.h
index d08f713..29357a5 100644
--- a/lib/correlation.h
+++ b/lib/correlation.h
@@ -10,7 +10,11 @@ template <class R_t, class X_t>
double correlation_length(const state_t <R_t, X_t>& s) {
double total = 0;
+#ifdef DIMENSION
+ for (D_t j = 0; j < DIMENSION; j++) {
+#else
for (D_t j = 0; j < s.D; j++) {
+#endif
total += norm_squared(s.ReF[j]) + norm_squared(s.ImF[j]);
}