Homec4science
Diffusion COSOlver (improved_sugama) Browse

cosolver/python_scripts/tools.py
cd1286e2d410improved_sugama

Last Change
Raw File

tools.py
View Options

import os
import sys
import time
import numpy as np
def mkdir(dir):
'''make a directory (dir) if it doesn't exist'''
if not os.path.exists(dir):
#print('Create ', dir)
os.mkdir(dir)
return True
return False
# Exec
exec_path = '/home/sernst/cosolver/bin/CO'
exec_name = 'CO'
T4_in_path = '/home/sernst/cosolver/T4.in'
def print_percent_bar(ratio, width=10, timestamp_start=time.time(), title=''):
elapsed = time.time()-timestamp_start
elem = round(ratio*width)
bar_str = ('|'*elem) + (' '*(10-elem))
print('\r- %s %s %.2f%% [%.2fs elapsed]' % (title, bar_str, 100*ratio, elapsed), end='')
# This functions block untile all jobs in `job_ids` list are done
# It displays stats about the process like percentage and elapsed time
def wait_untile_all_jobs_done(job_ids, timestamp_start = time.time()):
all_jobs_done = False
while not all_jobs_done:
cmd_stream = os.popen('squeue')
l = 0
matching_job_count = 0
for line in cmd_stream:
if l == 0:
l += 1
continue
job_id = int(line.strip(' ').split(' ')[0])
if job_id in job_ids:
matching_job_count += 1
l += 1
del l
print_percent_bar((len(job_ids)-matching_job_count)/len(job_ids),
timestamp_start=timestamp_start,
title='Cluster is working...')
if matching_job_count == 0:
all_jobs_done = True
time.sleep(0.1)
print()
# === Matrices manipulations ===
# Utils
def get_int(line):
lhs = line.split('=')[1].strip()
cidx = 0
for i in range(len(lhs)):
if lhs[i] not in '0123456789':
break
cidx = i+1
return int(lhs[:cidx])
def get_Pmax_Jmax(h5):
f90_lines = np.array(h5['/files/fort.f90'])[0].astype(str).split('\n')
for line in f90_lines:
if 'pmaxi' in line.lower():
pi = get_int(line)
if 'jmaxi' in line.lower():
ji = get_int(line)
if 'pmaxe' in line.lower():
pe = get_int(line)
if 'jmaxe' in line.lower():
je = get_int(line)
return pi,ji,pe,je
def resize(data, p,j, tp, tj):
p, j, tp, tj = p+1, j+1, tp+1, tj+1
assert p*j == data.shape[0]
assert p*j == data.shape[1]
assert tp <= p, "You can't resize a matrix with more Hermite polynomials than the orignal one: from %i to %i" % (p, tp)
assert tj <= j, "You can't resize a matrix with more Laguerre polynomials than the orignal one: from %i to %i" % (j, tj)
data = data.reshape([p,j,p,j])
data = data[:tp,:tj,:tp,:tj]
data = data.reshape([tp*tj,tp*tj])
return data
c4science · Help