mask = np.loadtxt(maskFile, np.int32)
else:
mask = np.arange(N)
NFilled = np.max(mask[mask < N]) + 1
# Read transfer operator spectrum from file and create a bi-orthonormal basis
# of eigenvectors and backward eigenvectors:
print('Readig spectrum for tauDim = {:.3f}...'.format(tauDim))
(eigValForward, eigValBackward, eigVecForward, eigVecBackward) \
= ergoPlot.readSpectrum(eigValForwardFile, eigValBackwardFile,
eigVecForwardFile, eigVecBackwardFile,
makeBiorthonormal=~cfg.spectrum.makeBiorthonormal,
fileFormat=fileFormat)
print('Getting conditionning of eigenvectors...')
eigenCondition = ergoPlot.getEigenCondition(eigVecForward, eigVecBackward)
# Get generator eigenvalues
eigValGen = (np.log(np.abs(eigValForward)) +
np.angle(eigValForward) * 1j) / tau
# Plot eigenvectors of transfer operator
alpha = 0.05
csfilter = 0.5
os.makedirs('{}/spectrum/eigvec'.format(cfg.general.plotDir), exist_ok=True)
os.makedirs('{}/spectrum/reconstruction'.format(cfg.general.plotDir),
exist_ok=True)
for ev in evPlot:
cmap = cm.hot_r if ev == 0 else cm.RdBu_r
if plotForward:
mask = np.fromfile(maskFile, np.int32)
else:
mask = np.loadtxt(maskFile, np.int32)
else:
mask = np.arange(N)
NFilled = np.max(mask[mask < N]) + 1
# Read transfer operator spectrum from file and create a bi-orthonormal basis
# of eigenvectors and backward eigenvectors:
print('Readig spectrum for tauDim = {:.3f}...'.format(tauDim))
(eigValForward, eigValBackward, eigVecForward, eigVecBackward) \
= ergoPlot.readSpectrum(eigValForwardFile, eigValBackwardFile,
eigVecForwardFile, eigVecBackwardFile,
makeBiorthonormal=makeBiorthonormal,
fileFormat=fileFormat)
eigenCondition = ergoPlot.getEigenCondition(eigVecForward, eigVecBackward,
statDist)
eigValGenOrig = ergoPlot.eig2Generator(eigValForward, tau)
# Read unfolding eigenvalues
# eigValForwardUnfold = None
st = statDist.copy()
st2 = np.concatenate((st, st))
alpha = 0.05
if unfold:
print('Readig spectrum for tauDimUnfold = {:.3f} to unfold...'.format(
tauDimUnfold))
(eigValForwardUnfold, eigValBackwardUnfold,
eigVecForwardUnfold, eigVecBackwardUnfold) \
= ergoPlot.readSpectrum(eigValForwardFileUnfold,
eigValBackwardFileUnfold,