#include "animation.hpp"

int main(int argc, char* argv[]) {
  const unsigned n = 12;

  unsigned N = 1200;
  double R = 0.024;
  double ε = 0.01;
  unsigned steps = 1e6;
  double β = 1;

  initializeAnimation(argc, argv);

  Rng r;

  Model<2, SoftSphere<2, n>> m(N, 2 * R * 1.25);

  for (unsigned i = 0; i < N; i++) {
    Position<2> x = {r.uniform(0.0, 1.0), r.uniform(0.0, 1.0)};
    double rad = pow(r.uniform(pow(2.219, -2), 1.0), -0.5) * R / 2.219;
    m.insert(SoftSphere<2, n>(x, rad));
  }

  for (unsigned i = 0; i < steps * N; i++) {
    SoftSphere<2, n>& s = r.pick(m.particles);
    Vector<2> Δx;
    for (double& Δxi : Δx) {
      Δxi = r.variate<double, std::normal_distribution>(0, ε);
    }

    m.metropolis(β, s, Δx, r);

    SoftSphere<2, n>& s1 = r.pick(m.particles);
    SoftSphere<2, n>& s2 = r.pick(m.particles);
    
    m.swap(β, s1, s2, r);

    if (i % (N * 3) == 0) {
      draw(m);
    }
  }

  return 0;
}