from FFTMA import gen
import numpy as np
import sys

def generate(n):
    nx, ny, nz = n,n,n
    dx, dy, dz = 1.0, 1.0, 1.0
    seed= 1548762 #rdi(10000,99999)
    var=1
    vario=2
    alpha=1
    lcx=2
    lcy=4
    lcz=16
    ap1x=1
    ap1y=0
    ap1z=0
    ap2x=0
    ap2y=1
    ap2z=0

    v1 = (var, vario, alpha, lcx, lcy, lcz, ap1x, ap1y, ap1z, ap2x, ap2y, ap2z)
    variograms = [v1]

    mean=15.3245987
    variance=3.5682389
    typ=3

    k=gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
    np.save(f"out_{n}.npy",k)

if __name__ == '__main__':
    N=int(sys.argv[1])
    generate(N)