from math import exp
import os
import math
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import MoNeT_MGDrivE as monet
MKRS = ('o', '^', 's', 'p', 'd')
MCOL = ('#e0c3fc', '#bdb2ff', '#a0c4ff', '#ffd6a5')
RVB = monet.colorPaletteFromHexList(['#e0c3fc', '#00296b'])
KPARS = {
'Trap': {'A': 0.5, 'b': 1.5},
'Escape': {'A': 0, 'b': 100}
}
# KPARS = {
# 'Trap': {'A': 0.5, 'b': 0.15},
# # 'Trap': {'A': 0.5, 'b': 0.2},
# 'Escape': {'A': 0, 'b': 100}
# }
MKERN = [1, 1.0e-10, math.inf]
(DPI, PAD) = (300, 1)
def selectPaths(USR):
if USR=='lab':
(PT_DTA, PT_GA, PT_IMG) = (
'/Volumes/marshallShare/Mov/dta',
'/Volumes/marshallShare/Mov/GA',
'/Volumes/marshallShare/Mov/trp',
}
fltr.pop('i_ren')
keys = list(fltr.keys())
ks = [all(i) for i in zip(*[np.isclose(DATA[k], fltr[k]) for k in keys])]
resVals = list(DATA[ks]['i_res'].unique())
###############################################################################
# Plot
###############################################################################
# resVals = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
clist = ['#000000', '#03045e', '#6247aa', '#815ac0', '#c19ee0', '#d6e3f8']
clist = [
'#fe1d23', '#fe576f', '#fdcbff', '#d6e3f8', '#aacbff', '#00affe',
'#013af4', '#0000ee'
]
clist.reverse()
rvb = monet.colorPaletteFromHexList(clist)
colors = rvb(np.linspace(0, 1, len(resVals)))
(fig, ax) = plt.subplots(figsize=(10, 10))
fltr['i_res'] = 0.5
for (i, res) in enumerate(resVals):
fltr['i_res'] = res
keys = list(fltr.keys())
ks = [all(i) for i in zip(*[np.isclose(DATA[k], fltr[k]) for k in keys])]
dfSrf = DATA[ks]
# plt.scatter(dfSrf['i_ren'], dfSrf['WOP'])
plt.plot(list(dfSrf['i_ren']), list(dfSrf['WOP']), color=colors[i])
ax.xaxis.set_ticks(np.arange(0, 24, 4))
ax.yaxis.set_ticks(np.arange(0, 10 * 365, 365 / 2))
if not TICKS_HIDE:
leg = [
Patch(facecolor=colors[len(resVals) - (i + 1)],