from FFTMA import gen
from refine import refine as ref
import numpy as np
from random import randint as rdi

N = 128

for i in range(1):
    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 = 1

    k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)

    np.save("out" + str(i) + ".npy", ref(k, 4, 4, 4))