qudarap/tests/rng_sequence_with_skipahead.py

0001 #!/usr/bin/env python
0002
0003 import os, numpy as np, logging
0004 log = logging.getLogger(__name__)
0005
0006
0007 def rng_sequence_with_skipahead(base):
0008     fold = os.path.expandvars(base)
0009     symbols = "abcdefghijklmnopqrstuvwxyz"
0010     names = sorted(os.listdir(fold))
0011
0012     arrs = []
0013     for i, name in enumerate(names):
0014         sym = symbols[i]
0015         path = os.path.join(fold, name)
0016
0017         print("fold:[%s]" % fold)
0018         print("name:[%s]" % name)
0019         print("path:[%s]" % path)
0020
0021         arr = np.load(path)
0022         arrs.append(arr)
0023         msg = "%3s %20s %s" % (sym, str(arr.shape), path)
0024         print(msg)
0025         globals()[sym] = arr
0026     pass
0027     seq = np.concatenate(arrs)
0028     globals()["seq"] = seq
0029 pass
0030
0031
0032 if __name__ == '__main__':
0033     rng_sequence_with_skipahead("$FOLD/rng_sequence_f_ni1000000_nj16_nk16_tranche100000")
0034     print("seq.shape %s " % str(seq.shape))
0035 pass
0036
0037