def pytest6(verbose=1):
""" Check generation of OA(32, t, 2^a) """
N = 32
k = 8
t = 3
l = [2] * 8
rr = []
gtrr = [3, 5, 10, 17, 33]
s = oalib.intVector(l)
adata = oalib.arraydata_t(s, N, t, k)
if verbose:
print('pytest6: run different algorithms on the same case')
algs = [oalib.MODE_ORIGINAL, oalib.MODE_J4]
for ii, alg in enumerate(algs):
algname = oalib.algnames(alg)
if verbose >= 2:
print('pytest6: running %s, alg %s' % (adata.fullidstr(), algname))
rr = []
tmp = oahelper.runExtend(N,
k,
t,
l,
verbose=verbose,
nums=rr,
algorithm=alg)
if not rr == gtrr:
print('pytest6: case %s' % adata.fullidstr())
print(
' algorithm %s: error: incorrect number of arrays! %s -> %s'
% (algname, rr, gtrr))
def selectArraysInFile(infile,
outfile,
idx,
afmode=oalib.ATEXT,
verbose=1,
cache=1):
""" Select arrays in a file by indices
Args:
infile (str): file with designs
outfile (str): output file with designs
inx (list of indices)
Returns:
None
"""
if not checkFiles(outfile, cache=cache):
gidxint = oalib.intVector([int(x) for x in idx])
sols = oalib.arraylist_t()
oalib.selectArrays(infile, gidxint, sols, 0)
af = oalib.arrayfile_t(infile, 1)
_ = oalib.writearrayfile(outfile, sols, afmode, af.nrows, af.ncols)
if verbose >= 2:
print('selectArrays: write array file %s in mode %d' %
(outfile, afmode))
else:
if verbose >= 2:
print('output file %s already exists' % outfile)
def pytest2level(verbose=1):
""" Check generation of OA(16,2, 4^2 2^a) """
N = 32
k = 10
t = 3
l = [2]
rr = []
t0 = time.time()
oahelper.runExtend(N,
k,
t,
l,
verbose=1,
nums=rr,
algorithm=oalib.MODE_ORIGINAL)
dt = time.time() - t0
t0 = time.time()
oahelper.runExtend(N,
k,
t,
l,
verbose=1,
nums=rr,
algorithm=oalib.MODE_LMC_2LEVEL)
dt2level = time.time() - t0
ll = l * k
s = oalib.intVector(ll)
adata = oalib.arraydata_t(s, N, t, k)
if verbose:
print('case %s: 2-level method %.2f [s] -> %.2f [s]' %
(adata.idstr(), dt, dt2level))
def selectArrays(infile, outfile, idx, afmode=oalib.ATEXT, verbose=1, cache=1):
""" Select arrays in a file by indices
Input:
- infile (string)
- outfile (string)
- inx (list of indices)
Output:
- None
"""
if not checkFiles(outfile, cache=cache):
gidxint = oalib.intVector([int(x) for x in idx])
# print(' xx select: %s' % str(gidxint))
# print(' xx select: %s' % str([v for v in gidxint]))
sols = oalib.arraylist_t()
oalib.selectArrays(infile, gidxint, sols, 0)
af = oalib.arrayfile_t(infile, 1)
af.nrows
r = oalib.writearrayfile(outfile, sols, afmode, af.nrows, af.ncols)
if verbose >= 2:
print('selectArrays: write array file %s in mode %d' %
(outfile, afmode))
else:
if verbose >= 2:
print('output file %s already exists' % outfile)
def runExtend(N,
k,
t=3,
l=2,
verbose=1,
initsols=None,
nums=[],
algorithm=None):
""" Run extension algorithm and return arrays
Args:
N (int): number of rows
k (int): number of columns to extend to
t (int): strength of the arrays
l (int): factors of the designs
initsols (None or list): list of arrays to extend, None to start with root
Returns:
list: list of generated designs
Example:
>>> import oapackage
>>> designs = oapackage.oahelper.runExtend(16, 5, 3, verbose=0)
"""
if verbose:
print('runExtend: N=%d, k=%d, t=%d' % (N, k, t))
if isinstance(l, list):
ll = l
else:
ll = [l]
ll = ll + [ll[-1]] * (k - len(ll))
s = oalib.intVector(ll)
adata = oalib.arraydata_t(s, N, t, k)
al = oalib.array_link(adata.N, adata.strength, 1)
al.create_root(adata)
if initsols is None:
sols0 = oalib.arraylist_t()
sols0.append(al)
tstart = t
else:
sols0 = initsols
tstart = sols0[0].n_columns
oaoptions = oalib.OAextend()
if algorithm is None:
oaoptions.setAlgorithmAuto(adata)
else:
oaoptions.setAlgorithm(algorithm, adata)
solsx = sols0
for ii in range(tstart, k):
solsx = oalib.arraylist_t()
oalib.extend_arraylist(sols0, adata, oaoptions, ii, solsx)
if verbose >= 2:
print(' ii %d: %d' % (ii, solsx.size()))
sols0 = solsx
nums.append(solsx.size())
sys.stdout.flush()
return solsx
def runExtend(N,
k,
t=3,
l=2,
verbose=1,
initsols=None,
nums=[],
algorithm=None):
""" Run extension algorithm and return arrays
Arguments
---------
N : integer
number of rows
k: integer
number of columns
t: integer
strength of the arrays
>>> r = runExtend(16, 5, 3, verbose=0)
"""
if verbose:
print('runExtend: N=%d, k=%d, t=%d' % (N, k, t))
if isinstance(l, list): # types.ListType):
ll = l
else:
ll = [l]
ll = ll + [ll[-1]] * (k - len(ll))
s = oalib.intVector(ll)
adata = oalib.arraydata_t(s, N, t, k)
al = oalib.array_link(adata.N, adata.strength, 1)
al.create_root(adata)
if initsols is None:
sols0 = oalib.arraylist_t()
sols0.append(al)
tstart = t
else:
sols0 = initsols
tstart = sols0[0].n_columns
oaoptions = oalib.OAextend()
if algorithm is None:
oaoptions.setAlgorithmAuto(adata)
else:
oaoptions.setAlgorithm(algorithm, adata)
solsx = sols0
for ii in range(tstart, k):
solsx = oalib.arraylist_t()
oalib.extend_arraylist(sols0, adata, oaoptions, ii, solsx)
if verbose >= 2:
print(' ii %d: %d' % (ii, solsx.size()))
sols0 = solsx
nums.append(solsx.size())
sys.stdout.flush()
return solsx
def runExtend(N, k, t=3, l=2, verbose=1, initsols=None, nums=[], algorithm=None):
""" Run extension algorithm and return arrays
Args:
N (int): number of rows
k (int): number of columns to extend to
t (int): strength of the arrays
l (int): factors of the designs
initsols (None or list): list of arrays to extend, None to start with root
Returns:
list: list of generated designs
Example:
>>> import oapackage
>>> designs = oapackage.oahelper.runExtend(16, 5, 3, verbose=0)
"""
if verbose:
print('runExtend: N=%d, k=%d, t=%d' % (N, k, t))
if isinstance(l, list):
ll = l
else:
ll = [l]
ll = ll + [ll[-1]] * (k - len(ll))
s = oalib.intVector(ll)
adata = oalib.arraydata_t(s, N, t, k)
al = oalib.array_link(adata.N, adata.strength, 1)
al.create_root(adata)
if initsols is None:
sols0 = oalib.arraylist_t()
sols0.append(al)
tstart = t
else:
sols0 = initsols
tstart = sols0[0].n_columns
oaoptions = oalib.OAextend()
if algorithm is None:
oaoptions.setAlgorithmAuto(adata)
else:
oaoptions.setAlgorithm(algorithm, adata)
solsx = sols0
for ii in range(tstart, k):
solsx = oalib.arraylist_t()
oalib.extend_arraylist(sols0, adata, oaoptions, ii, solsx)
if verbose >= 2:
print(' ii %d: %d' % (ii, solsx.size()))
sols0 = solsx
nums.append(solsx.size())
sys.stdout.flush()
return solsx
def selectArraysInFile(infile, outfile, idx, afmode=oalib.ATEXT, verbose=1, cache=1):
""" Select arrays in a file by indices
Args:
infile (str): file with designs
outfile (str): output file with designs
inx (list of indices)
Returns:
None
"""
if not checkFiles(outfile, cache=cache):
gidxint = oalib.intVector([int(x) for x in idx])
sols = oalib.arraylist_t()
oalib.selectArrays(infile, gidxint, sols, 0)
af = oalib.arrayfile_t(infile, 1)
_ = oalib.writearrayfile(outfile, sols, afmode, af.nrows, af.ncols)
if verbose >= 2:
print('selectArrays: write array file %s in mode %d' %
(outfile, afmode))
else:
if verbose >= 2:
print('output file %s already exists' % outfile)