run_control.py

import os
import sys
import subprocess
import random
import numpy
from time import sleep

from calc_sign import calc_sign

nruns = 20
nnodes = 1

NT = 64
NX = 64
nupdates = 10000
nmeasure = 1
U = 0
m = 0.1
mu= 0
step = 0.01
init_nstep = 10000
tolerance = -40

accuracy = 1e-7
max_iterations = 1000

seed = random.randint(0,99999999)


if len(sys.argv) == 2 and sys.argv[1] in ["local", "-l"]:
  use_srun = False
else :
  use_srun = True


def new_parameter_file( run ):
  global seed
  seed = seed+1
  parameter = f'''{NT}
{NX}
{seed}
.config_{run}
{nupdates}
{nmeasure}
{int(nupdates/nmeasure)}
{m}
{U}
{mu}
{init_nstep}
{step}
{tolerance}
WL_F_{run}'''
  
  text_file = open(f".parameter_{run}", "w")
  text_file.write(parameter)
  text_file.close()


error = 1
for i in range(max_iterations):
  for run in range(nruns):
    new_parameter_file( run )

  processes = []
  for run in range(nruns):
    new_parameter_file( run )
    with open(f".parameter_{run}", "r") as inputfile:
      with open(f"output_{run}", "a+") as outputfile:
        if(use_srun):
          p = subprocess.Popen(["srun","-n1","-N1","./wanglandau"], stdin=inputfile, stdout=outputfile)
        else:
          p = subprocess.Popen(["./wanglandau"], stdin=inputfile, stdout=outputfile)
        processes.append(p)
    while len(processes) == nnodes:
      sleep(0.5)
      for p in processes:
        if p.poll() is not None:
          processes.remove(p)
  
  for p in processes:
    p.wait()
  
  sign_mean, sign_std, weights = calc_sign(nruns, weights=True)
      
  print(sign_mean,sign_std)
  sys.stdout.flush()
  
  error = sign_std/abs(sign_mean)
  if( error < accuracy ):
    break