add legend control in rc_gracify

minyez · Jul 2, 2024 · 794bd56 · 794bd56
1 parent ded9dc7
commit 794bd56
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mushroom/aims/gw.py b/mushroom/aims/gw.py
@@ -149,7 +149,7 @@ def __process_single_self_energy_data(fpath):
 def read_aims_self_energy_dir(sedir: str = "self_energy",
                               filter_isbk_file: Callable[[int, int, int, int], bool] = None,
                               merge_band_kpoints: bool = False,
-                              filethres_mp: int = 100):
+                              filethres_mp: int = 10000):
     """read all sigc data in self energy directory `sedir`
 
     Args:

diff --git a/mushroom/visual/pyplot.py b/mushroom/visual/pyplot.py
@@ -19,7 +19,7 @@
 
 
 def rc_gracify(transparent: bool = False, dpi: int = 300, cmap: str = "RdBu", fontsize: int = 16,
-               tick_scale: float = 1.5):
+               tick_scale: float = 1.5, legend_frameon: bool = False, legend_fancybox: bool = False):
     """setup rcParams to mimic the style of XmGrace"""
     raise_no_module(plt, "matplotlib")
     plt.rcParams["font.family"] = ["serif",] + plt.rcParams["font.family"]
@@ -29,8 +29,8 @@ def rc_gracify(transparent: bool = False, dpi: int = 300, cmap: str = "RdBu", fo
     plt.rcParams['mathtext.fontset'] = 'dejavuserif'
     plt.rcParams["xtick.direction"] = "in"
     plt.rcParams["ytick.direction"] = "in"
-    plt.rcParams["legend.frameon"] = False
-    plt.rcParams["legend.fancybox"] = False
+    plt.rcParams["legend.frameon"] = legend_frameon
+    plt.rcParams["legend.fancybox"] = legend_fancybox
     # disable legend transparency
     plt.rcParams["legend.framealpha"] = 1.0
     plt.rcParams["savefig.dpi"] = dpi