# This is a code for 8.08 - 8.8S308 recitation 5
# You will simulate two particles of different temperatures and check time-reversal asymmetry

# for installing required packages
# using Pkg
# Pkg.add("Plots")
# Pkg.add("Random")

seed = 1234

# for plotting
using Plots
using Random

cd(@__DIR__)
include("completed_Rec5_module.jl")

#random number generator
rng = MersenneTwister(seed)

# parameters
dt = 1e-2       # time step
k = 10          # interaction strength
lam = 1         # external potential strength
kT1 = 1         # thermal energy of reservoir 1
kT2 = 0.01      # thermal energy of reservoir 2

#set parameters
param = Rec.setParam(dt, k, lam, kT1, kT2)

#set state
state = Rec.setState(param)

# parameters for generating density and current heatmap
n_sample  = 10000
dt_sample = 1
Nbinx     = 20
NbinxJ    = 15
xmax      = 1.5

P = zeros(Float64, Nbinx, Nbinx)      # density in (x1,x2) space
J = zeros(Float64, NbinxJ, NbinxJ, 2) # current in (x1,x2) space
get_densities!(state, param, rng, n_sample, dt_sample, xmax, P, J)