diff options
Diffstat (limited to 'lib/finite_states.h')
| -rw-r--r-- | lib/finite_states.h | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/lib/finite_states.h b/lib/finite_states.h index 08eff30..426edad 100644 --- a/lib/finite_states.h +++ b/lib/finite_states.h @@ -11,20 +11,26 @@ // invoking header std::array<std::array<std::array<double, N_STATES>, N_STATES>, N_STATES> J_probs; +#ifndef NOFIELD std::array<std::array<double, N_STATES>, N_STATES> H_probs; +#endif template <class X_t> +#ifndef NOFIELD void initialize_probs(std::function <double(X_t, X_t)> J, std::function <double(X_t)> H, double T) { for (q_t i = 0; i < N_STATES; i++) { for (q_t j = 0; j < N_STATES; j++) { - for (q_t k = 0; k < N_STATES; k++) { - J_probs[i][j][k] = 1.0 - exp(-(J(states[i], states[k]) - J(states[j], states[k])) / T); - } + H_probs[i][j] = 1.0 - exp(-(H(states[i]) - H(states[j])) / T); } } +#else +void initialize_probs(std::function <double(X_t, X_t)> J, double T) { +#endif for (q_t i = 0; i < N_STATES; i++) { for (q_t j = 0; j < N_STATES; j++) { - H_probs[i][j] = 1.0 - exp(-(H(states[i]) - H(states[j])) / T); + for (q_t k = 0; k < N_STATES; k++) { + J_probs[i][j][k] = 1.0 - exp(-(J(states[i], states[k]) - J(states[j], states[k])) / T); + } } } } |