spacepaste

Paste details

reply | raw

  1.  
  2. # -*- coding : utf-8 -*-
  4. from abc import ABCMeta, abstractmethod
  5. import matplotlib.pyplot as plt
  6. import numpy as np
  7. from abc import ABCMeta, abstractmethod
  8. from cycler import cycler
  9. import matplotlib.font_manager as font_manager
  10. from matplotlib.ticker import MultipleLocator, FormatStrFormatter
  11. from matplotlib.axes import Axes
  12. import matplotlib
  13. from matplotlib import cm
  14. from collections import OrderedDict
  15. from matplotlib.ticker import AutoMinorLocator
  18. class BasePlot(metaclass=ABCMeta):
  20. @abstractmethod
  21. def __init__(self,**kwargs):
  22. self.parameters = kwargs
  23. self.nrows = None
  24. self.ncols = None
  25. self.figsize = None
  27. font_dirs = ['/home/marco/.fonts', ]
  28. font_files = font_manager.findSystemFonts(fontpaths=font_dirs)
  29. font_list = font_manager.createFontList(font_files)
  30. font_manager.fontManager.ttflist.extend(font_list)
  33. #----------------------------------------------------------------------------------------------
  34. def findLimits(self):
  35. '''
  36. found the vectors limits in order to set the figure limits x,y
  37. '''
  38. self._minX = 10E20
  39. self._minY = 10E20
  40. self._maxX = 0.000
  41. self._maxY = 0.000
  42. for i in range(0,len(self.variable)-1,3):
  43. if self._minX >= min(self.variable[i]):
  44. self._minX = min(self.variable[i])
  45. if self._maxX <= max(self.variable[i]):
  46. self._maxX = max(self.variable[i])
  47. if self._minY >= min(self.variable[i+1]):
  48. self._minY = min(self.variable[i+1])
  49. if self._maxY <= max(self.variable[i+1]):
  50. self._maxY = max(self.variable[i+1])
  51. return [round(self._minX,2), round(self._maxX,2) , self._minY, self._maxY]
  52. #return [round(self._minX,2), round(self._maxX,2) , round(self._minY,2) , round(self._maxY,2)]
  55. def schemes(self, style:str = 'nb'):
  57. if style == 'nb':
  58. return ['#8DA0CB', '#E58AC3', '#A6D853', '#FFD930', '#B3B3B3', '#5FC3A4', '#FC8D62', '#E5C494']
  59. elif style == 'nb2': ##7aa0c4","#ca82e1" ,"#8bcd50
  60. return ['#7aa0c4', '#ca82e1', '#8bcd50', '#FFD930', '#B3B3B3', '#5FC3A4', '#FC8D62', '#66C2A5']
  61. elif style == 'nbc': ##7aa0c4","#ca82e1" ,"#8bcd50
  62. return ['#7aa0c4', '#ca82e1', '#8bcd50','#B3B3B3', '#FFD930', '#5FC3A4', '#FC8D62', '#66C2A5']
  63. elif style == 'nbd':
  64. return ['#8DA0CB', '#E58AC3', '#A6D853', '#B3B3B3','#FFD930', '#5FC3A4', '#FC8D62', '#E5C494']
  65. elif style == 'mycolor': ##7aa0c4","#ca82e1" ,"#8bcd50
  66. return ['#7aa0c4', '#ca82e1', '#8bcd50',"#FFD92F", '#B3B3B3',"#64b9a1", "#745ea6", "#db7e76" ]
  67. elif style == 'migcolor':
  68. return ['#66C2A5', '#FC8D62', '#8DA0CB', '#E78AC3', '#A6D854', '#FFD92F', '#E5C494', '#B3B3B3']
  69. elif style == 'vega':
  70. return ['#1F77B4', '#FF7F0E', '#2CA02C', '#D62728', '#9467BD', '#8C564B', '#E377C2', '#7F7F7F', '#BCBD22', '#17BECF']
  71. elif style == 'gg':
  72. return ['#E24A33', '#348ABD', '#988ED5', '#777777', '#FBC15E', '#8EBA42', '#FFB5B8']
  73. elif style == 'brewers':
  74. return ['#66C2A5', '#FC8D62', '#8DA0CB', '#E78AC3', '#A6D854', '#FFD92F', '#E5C494', '#B3B3B3']
  75. elif style == 'nb2':
  76. return ['#8DA0CB', '#E78AC3', '#A6D854', '#FFD92F', '#E5C494', '#B3B3B3', '#66C2A5', '#FC8D62']
  77. # color['nb3'] = ['#8DA0CB', '#E78AC3', '#A6D854', '#FFD92F', '#E5C494', '#B3B3B3', '#66C2A5', '#FC8D62']
  78. elif style == 'mystyle':
  79. return ['#8DA0CB', '#E58AC3', '#A6D853', '#FFD930', '#B2B2B2', '#5FC3A4', '#FC8D62', '#66C2A5']
  80. elif style == 'tthmod':
  81. return ['#30a2da', '#fc4f30', '#e5ae38', '#6d904f', '#8b8b8b']
  82. elif style == 'grayscale':
  83. return ['#252525', '#525252', '#737373', '#969696', '#BDBDBD', '#D9D9D9', '#F0F0F0', '#F0F0FF' ]
  84. elif style == 'grayscale2':
  85. return ['#525252', '#969696', '#BDBDBD' ,'#D9D9D9', '#F0F0F0', '#F0F0FF' ]
  86. elif style == 'grayscale3':
  87. return ['#4B4B4B', '#686868', '#868686' ,'#B2B2B2', '#CFCFCF', '#F0F0FF' ]
  88. elif style =='accent':
  89. return ['#7fc97f','#beaed4','#fdc086','#ffff99','#386cb0','#f0027f','#bf5b17','#666666']
  90. elif style == 'palette2':
  91. return ['#1B9E77','#D95F02','#7570B3','#E7298A','#66A61E','#E6AB02','#A6761D','#666666']
  92. elif style == 'set1':
  93. return ['#E41A1C','#377EB8','#4DAF4A','#984EA3','#FF7F00','#FFFF33', '#F781BF','#A65628']
  94. elif style == 'set2':
  95. return ['#66C2A5','#FC8D62','#8DA0CB','#E78AC3','#A6D854','#FFD92F','#E5C494','#B3B3B3']
  96. elif style =='set3':
  97. return ['#8dd3c7','#ffd92f','#bebada','#fb8072','#80b1d3','#fdb462','#b3de69','#fccde5']
  98. elif style =='pastel1':
  99. return ['#b3e2cd','#fdcdac','#cbd5e8','#f4cae4','#e6f5c9','#fff2ae','#f1e2cc','#cccccc']
  100. elif style == 'pastel2':
  101. return ['#fbb4ae','#b3cde3','#ccebc5','#decbe4','#fed9a6','#ffffcc','#e5d8bd','#fddaec']
  102. elif style == 'paraview':
  103. return ['#3B4CC0','#7C9FF9','#C0D4F5','#F2CBB7','#EE8568','#B40426']
  104. elif style =='paraview2':
  105. return ['#3B4CC0','#6889EE','#9ABAFF','#C9D8EF','#EDD1C2','#F7A889','#E26A53','#B40426']
  106. elif style =='radiance1':
  107. return ['#EDED49','#E6B500','#E66300','#E61000','#A40000','#520000','#000000']
  108. elif style =='xray':
  109. return ['#000000','#333333','#666666','#999999','#CCCCCC']
  110. elif style == 'bluesea':
  111. return ['#22204D','#35426B','#3C778F','#4D9BA8','#5DBCBF','#70D4CD','#9DEADA']
  112. elif style == 'bluesky':
  113. return ['#5C5C5C','#59647F','#4D67A0','#4872C4','#5085E3','#62A1F7','#86C1FF','#BFE0FF']
  114. elif style =='bluenice':
  115. return ['#3B4CC0','#4F69D9','#6585EC','#7C9FF9','#93B5FF','#AAC7FD','#C0D4F5','#D4DBE6']
  116. elif style =='paired':
  117. return ['#A6CEE3','#1F78B4','#B2DF8A','#33A02C','#FB9A99','#E31A1C','#FDBF6F','#FF7F00','#CAB2D6','#6A3D9A']
  118. elif style == 'new':
  119. return ['#30A2DA','#FC4F30','#6D904F','#E5AE38','#8B8B8B','#FFB5B8','#988ED5','#30A2DA']
  120. elif style == 'new2':
  121. return ['#0072B2','#A60628','#467821','#F0E442','#7A68A6','#56B4E9','#CC79A7','#009E73']
  122. elif style == 'new3':
  123. return ['#0072B2','#A60628','#467821','#F0E442','#7A68A6','#56B4E9','#CC79A7','#009E73','#348ABD','#D55E00']
  124. elif style == 'new4':
  125. return ['#92C6FF','#FF9F9A','#97F0AA','#FFD92F','#B0E0E6','#D0BBFF','#B8BEAF','#FFB6FC']
  126. elif style == 'new5':
  127. return ['#7A8FB1','#EC9CAE','#B2E295','#FFDFA8','#405C89','#BE4F69','#72B54B','#CFA356']
  128. elif style == 'new6':
  129. return ['#B8CCED','#FBBDCB','#D2F8BB','#FFDA9A']
  130. elif style =='new6a':
  131. return ['#B8CCED','#FBBDCB']
  132. elif style == '3gray':
  133. return ['#000000' , '#000000', '#000000' , '#666666','#666666','#666666', '#999999', '#999999', '#999999']
  134. elif style == '2gray':
  135. return ['#000000' ,'#888A85', '#BABDB6', '#000000','#888A85', '#BABDB6']
  136. elif style == 'grayimproved':
  137. return ['#000000','#808080','#778899','#A9A9A9','#B0C4DE','#C0C0C0']
  138. elif style == 'monocrome':
  139. return ['#000000','#666666','#999999','#B3B3B3']
  141. #------------------------------------------------------------------------------------------
  143. def linestyles(self, style : str = 'ls1'):
  145. if style =='ls1':
  146. return ['-', '--', ':', '-.']
  147. elif style == 'ls2':
  148. return ['-', '--', ':', '-.','-','--',':','-.']
  149. elif style == 'ls3':
  150. return ['-','-','-','-.','-.','-.',':',':',':']
  151. elif style == 'ls3a':
  152. return ['-','-','-.','-.',':',':']
  153. elif style == 'ls10':
  154. return ['-', '--', ':', '-.','-','--',':','-.','-','--']
  155. elif style =='lsa8':
  156. return ['-','--','-','--','-','--','-','--']
  157. elif style == 'lp8':
  158. return ['-',':','-',':','-',':','-',':']
  159. elif style == 'llt10':
  160. return ['-','--','-','--','-','--','-','--','-','--']
  161. elif style == 'lp10':
  162. return ['-',':','-',':','-',':','-',':','-',':']
  163. elif style == 'ldash':
  164. return [(0, ()),(0, (1, 10)),(0, (1, 5)),(0, (1, 1))]
  165. elif style == 'ls8':
  166. return [(0, ()),(0, (1, 1)),(0, (5, 10)),(0, (5, 5)),(0, (5, 1)),(0, (3, 10, 1, 10)),(0, (3, 5, 1, 5)),(0, (3, 5, 1, 5, 1, 5)) ]
  167. elif style == 'ln':
  168. return ["--", (0,(5,2,5,5,1,4))]
  169. elif style == 'paper':
  170. return [(0, ()), (0, (3, 1, 1, 1, 1, 1)), (0, (5, 5)),(0, (5, 1)),(0, (3, 5, 1, 5)),(0, (3, 5, 1, 5, 1, 5)),(0, (1, 1)),(0, (5, 10)) ]
  173. # linestyle = OrderedDict(
  174. # [('solid', (0, ())),
  175. # ('loosely dotted', (0, (1, 10))),
  176. # ('dotted', (0, (1, 5))),
  177. # ('densely dotted', (0, (1, 1))),
  178. # ('loosely dashed', (0, (5, 10))),
  179. # ('dashed', (0, (5, 5))),
  180. # ('densely dashed', (0, (5, 1))),
  181. # ('loosely dashdotted', (0, (3, 10, 1, 10))),
  182. # ('dashdotted', (0, (3, 5, 1, 5))),
  183. # ('densely dashdotted', (0, (3, 1, 1, 1))),
  184. # ('loosely dashdotdotted', (0, (3, 10, 1, 10, 1, 10))),
  185. # ('dashdotdotted', (0, (3, 5, 1, 5, 1, 5))),
  186. # ('densely dashdotdotted', (0, (3, 1, 1, 1, 1, 1)))])
  187. #----------------------------------------------------------------------------------------------------
  188. def cycle(self,n : str):
  189. if n == '0':
  190. return plt.cycler("color", self.schemes(self.parameters['scheme']) ) #colors)
  191. elif n=='1':
  192. return plt.cycler("color", self.schemes(self.parameters['scheme']) ) + plt.cycler("linestyle", self.linestyles(self.parameters['linestyle']))
  193. elif n=='2':
  194. return plt.cycler("color", self.schemes(self.parameters['scheme']) ) * plt.cycler("linestyle", self.linestyles(self.parameters['linestyle']))
  195. elif n=='3':
  196. return plt.cycler("linestyle", self.linestyles(self.parameters['linestyle'])) * plt.cycler("color", self.schemes(self.parameters['scheme']))
  197. else :
  198. raise ValueError('Index out of bound')
  204. #------------------------------------------------------------------------------------------------
  205. def setparams(self):
  207. #self.parameters.update(kwargs)
  209. if 'box' not in self.parameters.keys() :
  210. self.parameters['box'] = '#AAAAAA'
  211. if 'axeslabel' not in self.parameters.keys() :
  212. self.parameters['axeslabel'] = 'gray'
  213. if 'xtickcolor' not in self.parameters.keys() :
  214. self.parameters['xtickcolor'] = 'gray'
  215. if 'ytickcolor' not in self.parameters.keys() :
  216. self.parameters['ytickcolor'] = 'gray'
  217. if 'gridcolor' not in self.parameters.keys():
  218. self.parameters['gridcolor'] = '#dddddd'
  219. if 'font' not in self.parameters.keys():
  220. self.parameters['font'] = 'serif'
  221. if 'fontstyle' not in self.parameters.keys():
  222. self.parameters['fontstyle'] = 'italic'
  223. if 'text' not in self.parameters.keys():
  224. self.parameters['text'] = False
  225. if 'fontsize' not in self.parameters.keys():
  226. self.parameters['fontsize'] = 10.0
  227. if 'legendfontsize' not in self.parameters.keys():
  228. self.parameters['legendfontsize'] =10.0
  229. if 'legendEdgeColor' not in self.parameters.keys():
  230. self.parameters['legendEdgeColor'] = '#dddddd'
  231. if 'labelsize' not in self.parameters.keys():
  232. self.parameters['labelsize'] = [12,12,12]
  233. if 'linestyle' not in self.parameters.keys():
  234. self.parameters['linestyle'] = self.linestyles('paper')
  235. if 'scheme' not in self.parameters.keys():
  236. self.parameters['scheme'] = 'nb'
  237. if 'cycle' not in self.parameters.keys():
  238. self.parameters['cycle'] = self.cycle('0')
  239. if 'legend.loc' not in self.parameters.keys():
  240. self.parameters['legend.loc'] = 'best'
  241. if 'lines.linewidth' not in self.parameters.keys():
  242. self.parameters['lines.linewidth'] = '2'
  243. if 'grid.alpha' not in self.parameters.keys():
  244. self.parameters['grid.alpha'] = '1'
  245. if 'text.color' not in self.parameters.keys():
  246. self.parameters['text.color'] = 'gray'
  247. if 'font.sans-serif' not in self.parameters.keys():
  248. self.parameters['font.sans-serif'] = 'serif'
  249. if 'figure.figsize' not in self.parameters.keys():
  250. self.parameters['figure.figsize'] = [9.5,4.5]
  251. if 'subplot.bottom' not in self.parameters.keys():
  252. self.parameters['subplot.bottom'] = 0.125
  253. if 'text' not in self.parameters.keys():
  254. self.parameters['text'] = False
  255. if 'package' not in self.parameters.keys():
  256. self.parameters['package'] = []
  257. if 'grid.linewidth' not in self.parameters.keys():
  258. self.parameters['grid.linewidth'] = 1
  259. if 'axes.linewidth' not in self.parameters.keys():
  260. self.parameters['axes.linewidth'] = 0.7
  261. if 'font.weight' not in self.parameters.keys():
  262. self.parameters['font.weight'] = 'normal'
  263. if 'legend.borderpad' not in self.parameters.keys():
  264. self.parameters['legend.borderpad'] = 0.5
  265. if 'legend.borderaxespad' not in self.parameters.keys():
  266. self.parameters['legend.borderaxespad'] =0.5
  267. if 'legend.framelinewidth' not in self.parameters.keys():
  268. self.parameters['legend.framelinewidth'] = 1.0
  269. if 'boxplot.boxprops.linestyle' not in self.parameters.keys():
  270. self.parameters['boxplot.boxprops.linestyle'] = '--'
  271. if 'tick.MajW' not in self.parameters.keys():
  272. self.parameters['tick.MajW'] = 0.7
  273. if 'tick.minW' not in self.parameters.keys():
  274. self.parameters['tick.minW'] = 0.7
  281. myparams = {
  282. 'patch.linewidth' : '0.5',
  283. 'patch.facecolor' : '#348ABD', # blue
  284. 'patch.edgecolor' : 'EEEEEE',
  285. 'patch.antialiased' : True,
  286. 'font.size': self.parameters['fontsize'],
  287. 'xtick.major.size' : 0.1,
  288. 'xtick.minor.size' : 0.025,
  289. 'axes.edgecolor': self.parameters['box'] ,
  290. 'axes.linewidth': self.parameters['axes.linewidth'], # BOX width
  291. 'axes.xmargin': 0,
  292. 'axes.ymargin': 0,
  293. 'axes.labelcolor': self.parameters['axeslabel'],
  294. 'axes.axisbelow': True,
  295. 'xtick.color': self.parameters['xtickcolor'],
  296. 'ytick.color': self.parameters['ytickcolor'],
  297. 'axes.prop_cycle': self.parameters['cycle'],
  298. 'grid.linestyle': '--',
  299. 'grid.alpha': self.parameters['grid.alpha'],
  300. 'grid.linewidth' : self.parameters['grid.linewidth'],
  301. 'grid.color' : self.parameters['gridcolor'],
  302. 'font.family': self.parameters['font'] ,
  303. #'labelsize' : self.parameters['labelsize'],
  304. 'font.size' : self.parameters['fontsize'],
  305. 'legend.edgecolor' : self.parameters['legendEdgeColor'],
  306. 'legend.fancybox' : False,
  307. 'legend.borderpad' : self.parameters['legend.borderpad'],
  308. 'legend.fontsize' : self.parameters['legendfontsize'],
  309. 'legend.loc' : self.parameters['legend.loc'] ,
  310. 'legend.framealpha': 1,
  311. 'legend.borderaxespad' : self.parameters['legend.borderaxespad'],
  312. 'lines.linewidth' : self.parameters['lines.linewidth'],
  313. 'font.style' : self.parameters['fontstyle'],
  314. 'figure.autolayout': False,
  315. 'font.sans-serif' : self.parameters['font.sans-serif'],
  316. 'text.usetex': self.parameters['text'] ,
  317. 'patch.linewidth' : self.parameters['legend.framelinewidth'],
  318. 'boxplot.boxprops.linestyle' : self.parameters['boxplot.boxprops.linestyle'],
  319. #'figure.dpi': 72.0,
  320. #'figure.edgecolor': (1, 1, 1, 0),
  321. #'figure.facecolor': (1, 1, 1, 0),
  322. 'figure.figsize': self.parameters['figure.figsize'], #[6.0, 4.0],
  323. #'figure.frameon': True,
  324. #'figure.max_open_warning': 20,
  325. 'figure.subplot.bottom': self.parameters['subplot.bottom'], #0.125,
  326. 'figure.subplot.hspace': 0.2,
  327. 'figure.subplot.left': 0.125,
  328. 'figure.subplot.right': 0.9,
  329. 'figure.subplot.top': 0.9,
  330. 'figure.subplot.wspace': 0.2,
  331. 'figure.titlesize': 'large',
  332. 'figure.titleweight': 'normal',
  333. 'text.color' : self.parameters['text.color'],
  334. #'axes.axisbelow': False,
  335. #'axes.edgecolor': 'k',
  336. #'axes.facecolor': 'w',
  337. #'axes.formatter.limits': [-7, 7],
  338. #'axes.formatter.use_locale': False,
  339. #'axes.formatter.use_mathtext': False,
  340. #'axes.formatter.useoffset': True,
  341. #'axes.grid': False,
  342. 'axes.grid.axis': 'both',
  343. #'axes.grid.which': 'major',
  344. #'axes.hold': True,
  345. #'axes.labelcolor': 'k',
  346. #'axes.labelpad': 5.0,
  347. 'axes.labelsize': 'large',
  348. #'axes.labelweight': 'normal',
  349. #'axes.linewidth': 1.0,
  350. #'axes.prop_cycle': cycler('color', ['b', 'g', 'r', 'c', 'm', 'y', 'k']),
  351. #'axes.spines.bottom': True,
  352. #'axes.spines.left': True,
  353. #'axes.spines.right': True,
  354. #'axes.spines.top': True,
  355. 'axes.titlesize': 'large',
  356. #'axes.titleweight': 'normal',
  357. #'axes.unicode_minus': True,
  358. #'axes.xmargin': 0.0,
  359. #'axes.ymargin': 0.0,
  360. #font.family : sans-serif
  361. #font.style : normal
  362. #font.variant : normal
  363. 'font.weight' : self.parameters['font.weight'],
  364. #font.stretch : normal
  365. # 'font.cursive': ['Apple Chancery',
  366. # 'Textile',
  367. # 'Zapf Chancery',
  368. # 'Sand',
  369. # 'Script MT',
  370. # 'Felipa',
  371. # 'cursive'],
  372. # 'font.family': ['sans-serif'],
  373. # 'font.fantasy': ['Comic Sans MS',
  374. # 'Chicago',
  375. # 'Charcoal',
  376. # 'ImpactWestern',
  377. # 'Humor Sans',
  378. # 'fantasy'],
  379. # 'font.monospace': ['Bitstream Vera Sans Mono',
  380. # 'DejaVu Sans Mono',
  381. # 'Andale Mono',
  382. # 'Nimbus Mono L',
  383. # 'Courier New',
  384. # 'Courier',
  385. # 'Fixed',
  386. # 'Terminal',
  387. # 'monospace'],
  388. # 'font.sans-serif': ['Bitstream Vera Sans',
  389. # 'DejaVu Sans',
  390. # 'Lucida Grande',
  391. # 'Verdana',
  392. # 'Geneva',
  393. # 'Lucid',
  394. # 'Arial',
  395. # 'Helvetica',
  396. # 'Avant Garde',
  397. # 'sans-serif'],
  398. # 'font.serif': ['Bitstream Vera Serif',
  399. # 'DejaVu Serif',
  400. # 'New Century Schoolbook',
  401. # 'Century Schoolbook L',
  402. # 'Utopia',
  403. # 'ITC Bookman',
  404. # 'Bookman',
  405. # 'Nimbus Roman No9 L',
  406. # 'Times New Roman',
  407. # 'Times',
  408. # 'Palatino',
  409. # 'Charter',
  410. # 'serif'],
  411. # 'font.size': 10.0,
  412. # 'text.antialiased': True,
  413. # 'text.color': 'k',
  414. # 'text.dvipnghack': None,
  415. # 'text.hinting': 'auto',
  416. # 'text.hinting_factor': 8,
  417. 'text.latex.preamble': self.parameters['package'],
  418. # 'text.latex.preview': False,
  419. # 'text.latex.unicode': False,
  420. #
  421. #
  422. # 'xtick.color': 'k',
  423. # 'xtick.direction': 'in',
  424. # 'xtick.labelsize': 'medium',
  425. # 'xtick.major.pad': 4.0,
  426. # 'xtick.major.size': 4.0,
  427. # 'xtick.major.width': 1.5,
  428. # 'xtick.minor.pad': 4.0,
  429. # 'xtick.minor.size': 2.0,
  430. # 'xtick.minor.visible': False,
  431. # 'xtick.minor.width': 0.5,
  432. # 'ytick.color': 'k',
  433. # 'ytick.direction': 'in',
  434. # 'ytick.labelsize': 'medium',
  435. # 'ytick.major.pad': 4.0,
  436. # 'ytick.major.size': 4.0,
  437. # 'ytick.major.width': 1.5
  438. # 'ytick.minor.pad': 4.0,
  439. # 'ytick.minor.size': 2.0,
  440. # 'ytick.minor.visible': False,
  441. # 'ytick.minor.width': 0.5
  442. }
  444. # if self.parameters['text']:
  445. # plt.rc('text', usetex=True )
  446. # plt.rcParams['text.latex.preamble']=[r"\usepackage{amsmath}"]
  447. # plt.rc('font',family='' ,size=14 )
  448. #
  449. # if 'package' in self.parameters.keys():
  450. # for i in self.parameters['package']:
  451. # string = r"\usepackage{" + i + "}"
  452. # plt.rcParams['text.latex.preamble']=[string]
  455. ###########################################
  456. plt.rcParams.update(myparams)
  457. ###########################################
  462. #----------------------------------------------------------------------------------------------------------
  463. def set(self, nrows, ncols, fig , axs, kwargs):
  465. self.config = kwargs
  466. #print(self.config)
  468. if 'grid.linestyle' not in self.config.keys():
  469. self.config['grid.linestyle'] = '--'
  470. if 'grid.dashes' not in self.config.keys():
  471. self.config['grid.dashes'] = (5,7)
  472. if 'title.colors' not in self.config.keys():
  473. self.config['title.colors'] = '#555555'
  474. if 'subplot.bottom' not in self.config.keys():
  475. self.config['subplot.bottom'] = 0.125
  476. if 'subplot.right' not in self.config.keys():
  477. self.config['right.bottom'] = 0.93
  478. if 'legend.bbox' not in self.config.keys():
  479. self.config['legend.bbox'] = (0,0)
  480. if 'legend.loc' not in self.config.keys():
  481. self.config['legend.loc'] = 'best'
  482. if 'figure.figuresize' not in self.config.keys():
  483. self.config['figure.figuresize'] = [8,5.5]
  484. if 'legend.axespad' not in self.config.keys():
  485. self.config['legend.axespad'] = 0.
  486. if 'legend.framelinewidth' not in self.config.keys():
  487. self.config['legend.framelinewidth'] = 0.2
  488. if 'legend.textcolor' not in self.config.keys():
  489. self.config['legend.textcolor'] = '#555555'
  491. #self.config['legend.axespad'] = 0.
  494. for i in range(nrows):
  495. for j in range(ncols):
  496. #self.axs[i,j].set_xlim( self.findLimits()[0] , self.findLimits()[1] )
  497. #self.axs[i,j].set_ylim( self.findLimits()[2] , self.findLimits()[3] + 0.02 )
  498. minorLocator = AutoMinorLocator(4)
  499. majorLocator = MultipleLocator(10)
  500. majorFormatter = FormatStrFormatter('%f')
  501. if nrows == ncols == 1:
  502. self.axs.grid(linestyle= self.config['grid.linestyle'] , dashes= self.config['grid.dashes'])
  503. self.axs.set_title('', color=self.config['title.colors'] )
  504. self.axs.xaxis.set_minor_locator(AutoMinorLocator(5))
  505. self.axs.yaxis.set_minor_locator(AutoMinorLocator(5))
  506. self.axs.yaxis.set_ticks_position('both')
  507. self.axs.xaxis.set_ticks_position('both')
  508. self.axs.tick_params(which='major', width=self.config['tick.MajW']) # 1
  509. self.axs.tick_params(which='major', length=5.5)
  510. self.axs.tick_params(which='minor', width=self.config['tick.minW'])
  511. self.axs.tick_params(which='minor', length=3)
  512. legend = leg = self.axs.legend()
  513. #self.axs.legend( bbox_to_anchor=self.config['legend.bbox'] , loc= self.config['legend.bbox'] )
  514. legend.get_frame().set_linewidth(self.config['legend.framelinewidth'])
  515. #legend.get_frame().set_linewidth(0.1)
  516. #legend.get_frame().set_ linewidth(0.1)
  517. plt.setp(legend.get_texts(), color=self.config['legend.textcolor'] )
  519. elif nrows == 1 or ncols==1 :
  520. self.axs[i].grid(linestyle= self.config['grid.linestyle'] , dashes= self.config['grid.dashes'])
  521. self.axs[i].set_title('', color='#555555')
  522. self.axs[i].xaxis.set_minor_locator(AutoMinorLocator(5))
  523. self.axs[i].yaxis.set_minor_locator(AutoMinorLocator(5))
  524. self.axs[i].yaxis.set_ticks_position('both')
  525. self.axs[i].xaxis.set_ticks_position('both')
  526. # self.axs[i].tick_params(which='major', width=0.5)
  527. self.axs[i].tick_params(which='major', width=self.config['tick.MajW']) # 1
  528. self.axs[i].tick_params(which='major', length=5)
  529. self.axs[i].tick_params(which='minor', width=self.config['tick.minW'])
  530. self.axs[i].tick_params(which='minor', length=2.5)
  531. legend = leg = self.axs[i].legend()
  532. legend.get_frame().set_linewidth(0.5)
  533. plt.setp(legend.get_texts(), color=self.config['legend.textcolor'])
  534. else :
  535. self.axs[i,j].grid(linestyle= self.config['grid.linestyle'] , dashes= self.config['grid.dashes'])
  536. self.axs[i,j].set_title('', color='#555555')
  537. self.axs[i,j].xaxis.set_minor_locator(AutoMinorLocator(5))
  538. self.axs[i,j].yaxis.set_minor_locator(AutoMinorLocator(5))
  539. self.axs[i,j].yaxis.set_ticks_position('both')
  540. self.axs[i,j].xaxis.set_ticks_position('both')
  541. # self.axs[i,j].tick_params(which='major', width=0.5)
  542. self.axs[i,j].tick_params(which='major', width=self.config['tick.MajW']) # 1
  543. self.axs[i,j].tick_params(which='major', length=5)
  544. self.axs[i,j].tick_params(which='minor', width=self.config['tick.minW'])
  545. self.axs[i,j].tick_params(which='minor', length=2.5)
  547. #linestyles = ["--","-.",":", (0,(5,2,1,4))]
  548. # linestyles = [(0,(7,7)),(0,(7,7)),(0,(7,7)), (0,(7,7))]
  549. # for ax, ls in zip(axs.flat, linestyles):
  550. # for spine in ax.spines.values():
  551. # spine.set_linestyle(ls)
  553. legend = leg = self.axs[i,j].legend()
  554. #legend.get_frame().set_linewidth(self.config['legend.framelinewidth'])
  555. legend.get_frame().set_linewidth(0.5)
  556. plt.setp(legend.get_texts(), color=self.config['legend.textcolor'])
  557. #axs.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0.)
  558. ##plt.title(y=1.25)
  559. #### plt.legend([l1, l2, l3],["HHZ 1", "HHN", "HHE"])
  560. if nrows == ncols ==1 :
  561. self.config['subplot.bottom'] = 0.175
  562. self.config['subplot.right'] = 0.93
  564. plt.subplots_adjust(left=0.125 , bottom=self.config['subplot.bottom'], right=self.config['subplot.right'], top=0.87, wspace=0, hspace=0)
  565. else:
  566. plt.subplots_adjust(left=0.06 , bottom=0.1, right=0.95, top=0.93, wspace=0.2, hspace=0.6) # 0.2 , 0.4
  567. return self.fig,self.axs
  572. def __call__ (self,nrows,ncols,figsize=(9.5,4.5)):
  574. self.nrows = nrows
  575. self.ncols = ncols
  576. self.figsize = figsize
  577. self.parameters['figure.figsize'] = self.figsize
  578. self.setparams()
  580. if self.nrows == self.ncols ==1:
  581. self.fig, self.axs = plt.subplots( self.nrows, self.ncols,figsize=self.figsize)
  582. else:
  583. self.fig, self.axs = plt.subplots( self.nrows, self.ncols,figsize=self.figsize)
  585. self.fig,self.axs = self.set( self.nrows, self.ncols,self.fig,self.axs, self.parameters )
  586. return self.fig,self.axs
  588. # left = 0.125 # the left side of the subplots of the figure
  589. # right = 0.9 # the right side of the subplots of the figure
  590. # bottom = 0.1 # the bottom of the subplots of the figure
  591. # top = 0.9 # the top of the subplots of the figure
  592. # wspace = 0.2 # the amount of width reserved for space between subplots,
  593. # # expressed as a fraction of the average axis width
  594. # hspace = 0.2 # the amount of height reserved for space between subplots,
  595. # # expressed as a fraction of the average axis height
  596. #
  600. #------------------------------------------------------------------------------------------------
  601.  
Powered by spacepaste. Paste removal.