temet 0.9.0__tar.gz → 0.9.2__tar.gz

{temet-0.9.0 → temet-0.9.2}/.github/workflows/pypi.yaml

@@ -20,7 +20,7 @@ jobs:
       - name: Build release sdist
         run: |
           python -m pip install build
-          python -m build
+          python -m build --sdist
 
       - name: Upload sdist
         uses: actions/upload-artifact@v4

{temet-0.9.0 → temet-0.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: temet
-Version: 0.9.0
+Version: 0.9.2
 Summary: temet simulation analysis framework.
 Keywords: astronomy,simulation,cosmology,astrophysics
 Author-Email: Dylan Nelson <dnelson@uni-heidelberg.de>
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering :: Astronomy
 Classifier: Topic :: Scientific/Engineering :: Physics
 Project-URL: Homepage, https://www.github.com/dnelson/temet/

{temet-0.9.0 → temet-0.9.2}/docs/index.rst

@@ -2,7 +2,8 @@
 temet toolkit: documentation
 ============================
 
-Welcome to the documentation for **temet**, a python toolkit for the execution, analysis, and 
+Welcome to the documentation for **temet** (`github repo <https://github.com/dnelson/temet>`_), 
+a python toolkit for the execution, analysis, and 
 visualization of numerical simulations. Specifically, hydrodynamical simulations run with the 
 `AREPO <https://wwwmpa.mpa-garching.mpg.de/~volker/arepo/>`_ moving mesh code, as well as codes 
 producing similarly structured outputs including 

{temet-0.9.0 → temet-0.9.2}/docs/installation.rst

@@ -45,7 +45,7 @@ Or, if you would like to install the most up-to-date version directly from the r
 
 .. code-block:: bash
 
-    pip install git+ssh://git@github.com/dnelson/temet.git
+    pip install git+https://git@github.com/dnelson/temet.git
 
 If you plan on editing, making changes, and adding functionality to this package (**recommended choice**), then first 
 clone the repository, then install the package in 'editable' mode. This means that the files in this directory are not copied 
@@ -60,6 +60,11 @@ anywhere, but are used as is. Any changes you make are reflected (i.e. immediate
 Download Data Files
 -------------------
 
+.. note::
+
+   When you ``import temet`` for the first time, you will be reminded to download these data files.
+
+
 Several large tabulated data files are used to compute e.g. stellar luminosities (from FSPS), ion abundances and emissivities (from CLOUDY), and X-ray emission (from APEC/XPSEC). For convenience these can be downloaded as
 
 .. code-block:: bash
@@ -76,7 +81,11 @@ The default plotting style uses the Roboto font. To install this font locally, d
     wget https://github.com/google/fonts/raw/main/apache/roboto/static/Roboto-Light.ttf
     wget https://github.com/google/fonts/raw/main/apache/roboto/static/Roboto-LightItalic.ttf
 
-Note: on MPCDF machines, suggested organization of simulation directories is as follows
+
+Organizing Simulation Folders
+-----------------------------
+
+On MPCDF machines (and the TNG Lab), we suggest organization of simulation directories as follows
 
 .. code-block:: bash
 
@@ -91,6 +100,8 @@ the last two lines create symlinks to the actual output directory where the simu
 (``groupcat_*`` and ``snapdir_*``) reside, as well as to the postprocessing directory (containing ``trees``, etc).
 Replace as needed with the actual path on your machine.
 
+This allows many "known" simulations to be loaded by name only.
+
 
 External Package Installation
 -----------------------------

{temet-0.9.0 → temet-0.9.2}/pyproject.toml

@@ -10,11 +10,11 @@ packages = [
 ]
 
 [tool.setuptools.package-data]
-temet = ["temet/plot/mpl.style", "temet/util/*.cpp"]
+temet = ["temet/plot/mpl.style", "temet/util/*.cpp", "temet/tables/fonts/*.ttf"]
 
 [project]
 name = "temet"
-version = "0.9.0"
+version = "0.9.2"
 authors = [
     { name="Dylan Nelson", email="dnelson@uni-heidelberg.de" },
 ]
@@ -46,6 +46,7 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
     "Topic :: Scientific/Engineering :: Astronomy",
     "Topic :: Scientific/Engineering :: Physics",
 ]

{temet-0.9.0 → temet-0.9.2}/temet/__init__.py

@@ -3,7 +3,10 @@
 __all__ = ["ICs", "catalog", "cosmo", "load", "obs", "plot", "spectra", "tracer", "util", "vis"]
 # currently exclude "ML" to avoid torch* dependencies
 # exclude "projects" as this is example/user code
-
 from temet import ICs, catalog, cosmo, load, obs, plot, spectra, tracer, util, vis
 
 from .util.simParams import simParams as sim
+
+
+# check for data tables/ download
+util.extern.check_data_tables()

{temet-0.9.0 → temet-0.9.2}/temet/catalog/box.py

@@ -7,7 +7,7 @@ import numpy as np
 from ..cosmo import hydrogen
 from ..cosmo.cloudy import cloudyIon
 from ..cosmo.hydrogen import calculateCDDF
-from ..util.helper import numPartToChunkLoadSize, reportMemory
+from ..util.helper import num_cpus, numPartToChunkLoadSize, reportMemory
 
 
 def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSFR=False):
@@ -176,6 +176,14 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
             # SFR>0 gas has a fraction=1 of the given species
             assert species in preCompSpecies  # otherwise handle
 
+        # determine parallelism (high memory load)
+        nThreads = min(num_cpus() // 2, 8)
+        if boxGridDim > 60000:
+            nThreads = nThreads // 2
+        if boxGridDim > 100000:
+            nThreads = nThreads // 2
+        nThreads = max(nThreads, 1)
+
         if species in hDensSpecies:
             # calculate atomic hydrogen mass (HI) or total neutral hydrogen mass (HI+H2) [10^10 Msun/h]
             atomic = species == "HI" or species == "HI2" or species == "HI3"
@@ -196,6 +204,7 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
                 nPixels=boxGridDim,
                 sP=sP,
                 colDens=True,
+                nThreads=nThreads,
                 sliceFac=boxWidthFrac,
             )
 
@@ -235,6 +244,7 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
                 nPixels=boxGridDim,
                 sP=sP,
                 colDens=True,
+                nThreads=nThreads,
                 sliceFac=boxWidthFrac,
             )
 
@@ -242,12 +252,6 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
 
         if species in preCompSpecies:
             # anything directly loaded from the snapshots, return in units of [10^10 Msun * h / ckpc^2]
-            nThreads = 8
-            if boxGridDim > 60000:
-                nThreads = 4
-            if boxGridDim > 100000:
-                nThreads = 2
-
             if allSFR:
                 w = np.where(gas["StarFormationRate"] > 0)
                 gas[species][w] = gas["Masses"][w]
@@ -278,6 +282,7 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
                 nPixels=boxGridDim,
                 sP=sP,
                 colDens=False,
+                nThreads=nThreads,
                 sliceFac=boxWidthFrac,
             )
 
@@ -293,6 +298,7 @@ def wholeBoxColDensGrid(sP, pSplit, species, gridSize=None, onlySFR=False, allSF
                 nPixels=boxGridDim,
                 sP=sP,
                 colDens=False,
+                nThreads=nThreads,
                 sliceFac=boxWidthFrac,
             )
 

{temet-0.9.0 → temet-0.9.2}/temet/cosmo/cloudy.py

@@ -9,7 +9,7 @@ import numpy as np
 from scipy.ndimage import map_coordinates
 
 from ..cosmo.cloudyGrid import basePath, getEmissionLines
-from ..util.helper import closest, iterable, pSplitRange
+from ..util.helper import closest, iterable, num_cpus, pSplitRange
 
 
 class cloudyIon:
@@ -160,7 +160,7 @@ class cloudyIon:
               "XXIX": 29, "XXX": 30, "XXXI": 31}  # fmt: skip
     _romanInv = {v: k for k, v in _roman.items()}
 
-    def __init__(self, sP, el=None, res="lg_c17", redshiftInterp=False, order=3):
+    def __init__(self, sP, el=None, res="lg_c23", redshiftInterp=False, order=3):
         """Load the table, optionally only for a given element(s)."""
         if sP is None:
             return  # instantiate only for misc methods
@@ -370,7 +370,7 @@ class cloudyIon:
         if temp is None:
             return self.grid["temp"], locData[i0, i1, i2, :]
 
-    def frac(self, element, ionNum, dens, metal, temp, redshift=None, nThreads=16):
+    def frac(self, element, ionNum, dens, metal, temp, redshift=None, nThreads=None):
         """Interpolate the ion abundance table, return log(ionization fraction).
 
         Input gas properties can be scalar or np.array(), in which case they must have the same size.
@@ -386,7 +386,7 @@ class cloudyIon:
           temp (ndarrray): temperature [log K]
           metal (ndarray): metallicity [log solar]
           redshift (float or None): redshift, if we are interpolating in redshift space.
-          nThreads (int): number of threads to use for interpolation (1=serial).
+          nThreads (int): number of threads to use for interpolation (1=serial). if None, automatically determine.
 
         Return:
           ndarray: ionization fraction per cell [log].
@@ -394,6 +394,10 @@ class cloudyIon:
         element = self._resolveElementNames(element)
         ionNum = self._resolveIonNumbers(ionNum)
 
+        if nThreads is None:
+            # determine automatically
+            nThreads = min(num_cpus() // 4, 16)  # half of available, at most 36
+
         if redshift is not None and not self.redshiftInterp:
             raise Exception("Redshift input for interpolation, but we have selected nearest hyperslice.")
         if redshift is None and self.redshiftInterp:

{temet-0.9.0 → temet-0.9.2}/temet/cosmo/cloudyGrid.py

@@ -12,11 +12,12 @@ import h5py
 import numpy as np
 
 from ..cosmo import hydrogen
-from ..util.helper import closest, logZeroNaN, rootPath
+from ..util.extern import tables_path
+from ..util.helper import closest, logZeroNaN, num_cpus, rootPath
 from ..util.simParams import simParams
 
 
-basePath = rootPath + "tables/cloudy/"
+basePath = tables_path + "cloudy/"
 basePathTemp = expanduser("~") + "/data/cloudy_tables/"
 
 # emission lines recorded (must redo Cloudy grid to add lines)
@@ -386,7 +387,7 @@ def _loadExternalUVB(redshifts=None, hm12=False, puchwein19=False):
     if hm12:
         filePath = rootPath + "data/haardt.madau/hm2012.uvb.txt"
     if puchwein19:
-        filePath = rootPath + "/data/puchwein/p19.uvb.txt"
+        filePath = rootPath + "data/puchwein/p19.uvb.txt"
 
     sP = simParams(res=1820, run="tng")  # for units
 
@@ -714,10 +715,14 @@ def _getRhoTZzGrid(res, uvb):
     return densities, temps, metals, redshifts
 
 
-def runGrid(redshiftInd, nThreads=71, res="lg", uvb="FG11"):
+def runGrid(redshiftInd, nThreads=None, res="lg", uvb="FG11"):
     """Run a sequence of CLOUDY models over a parameter grid at a redshift (one redshift per job)."""
     import multiprocessing as mp
 
+    if nThreads is None:
+        # determine automatically
+        nThreads = num_cpus() - 1  # leave just one core free
+
     # config
     densities, temps, metals, redshifts = _getRhoTZzGrid(res=res, uvb=uvb)
 

{temet-0.9.0 → temet-0.9.2}/temet/cosmo/mergertree.py

@@ -531,7 +531,7 @@ def plot_tree(sP, subhaloID, saveFilename, treeName=treeName_default, dpi=100, c
 
     if tree is None:
         # subhalo not in tree
-        return None
+        return False
 
     nrows = tree["count"]
 
@@ -652,8 +652,6 @@ def plot_tree(sP, subhaloID, saveFilename, treeName=treeName_default, dpi=100, c
     if saveFilename is not None:
         fig.savefig(saveFilename, format=output_fmt, dpi=dpi)
         plt.close(fig)
-
-        return True
     else:
         # return image array itself, i.e. draw the canvas then extract the (Nx,Ny,3) array
         canvas = FigureCanvasAgg(fig)

{temet-0.9.0 → temet-0.9.2}/temet/cosmo/stellarPop.py

@@ -12,7 +12,8 @@ from numba import jit
 from scipy.interpolate import interp1d
 from scipy.ndimage import map_coordinates
 
-from ..util.helper import iterable, logZeroMin, logZeroNaN, rootPath, trapsum
+from ..util.extern import tables_path
+from ..util.helper import iterable, logZeroMin, logZeroNaN, num_cpus, trapsum
 from ..util.rotation import rotateCoordinateArray, rotationMatrixFromVec
 from ..util.simParams import simParams
 from ..util.sphMap import sphMap
@@ -315,7 +316,7 @@ def _dust_tau_model_lum_indiv(
 class sps:
     """Use pre-computed FSPS stellar photometrics tables to derive magnitudes for simulation stars."""
 
-    basePath = rootPath + "/tables/fsps/"
+    basePath = tables_path + "fsps/"
 
     imfTypes = {"salpeter": 0, "chabrier": 1, "kroupa": 2}
     isoTracks = ["mist", "padova07", "parsec", "basti", "geneva"]
@@ -1263,7 +1264,8 @@ class sps:
         pxSizeCode = self.sP.units.physicalKpcToCodeLength(pxSize)
         nPixels = np.int32(np.ceil(extentStars / pxSizeCode))[0:2]
 
-        nThreads = 8
+        # determine automatically
+        nThreads = min(num_cpus(), 8)  # all available, at most 8
 
         if pos_in.shape[0] < 1e3 and pos_stars_in.shape[0] < 1e3:
             nThreads = 4
@@ -1558,8 +1560,8 @@ def debug_check_rawspec():
     redshift = 0.8
 
     paths = [
-        rootPath + "/tables/fsps/mags_padova07_chabrier_cf00_bands-143_z=0.5.hdf5",
-        rootPath + "/tables/fsps/mags_padova07_chabrier_cf00_bands-143_z=0.5_em.hdf5",
+        tables_path + "fsps/mags_padova07_chabrier_cf00_bands-143_z=0.5.hdf5",
+        tables_path + "fsps/mags_padova07_chabrier_cf00_bands-143_z=0.5_em.hdf5",
     ]
 
     # start plot

{temet-0.9.0 → temet-0.9.2}/temet/cosmo/xray.py

@@ -11,11 +11,12 @@ from scipy.integrate import cumulative_trapezoid
 from scipy.ndimage import map_coordinates
 
 from ..plot.util import contourf, plothist, plotxy, sampleColorTable
-from ..util.helper import closest, rootPath
+from ..util.extern import tables_path
+from ..util.helper import closest
 from ..util.simParams import simParams
 
 
-basePath = rootPath + "tables/xray/"
+basePath = tables_path + "xray/"
 
 # fmt: off
 apec_elem_names = ["H","He","Li","Be","B","C","N","O","F","Ne","Na","Mg","Al","Si","P","S","Cl","Ar","K","Ca",

temet-0.9.2/temet/plot/__init__.py

@@ -0,0 +1,67 @@
+"""Plotting routines."""
+
+import logging
+import pathlib
+import sys
+from importlib import resources
+
+import matplotlib
+import matplotlib.pyplot as plt
+
+from . import (
+    clustering,
+    cosmoMisc,
+    drivers,
+    driversObs,
+    driversSizes,
+    meta,
+    perf,
+    quantities,
+    snapshot,
+    subhalos,
+    subhalos_evo,
+    util,
+)
+
+
+# check if we are in a Jupyter notebook
+def in_notebook():
+    """Determine if we are inside a Jupyter notebook (or similar), or not."""
+    if "IPython" in sys.modules:
+        IPython = sys.modules["IPython"]
+        ipython = IPython.core.getipython.get_ipython()
+
+        try:
+            shell = ipython.__class__.__name__
+            if shell == "ZMQInteractiveShell":
+                return True  # Jupyter notebook or qtconsole
+            elif shell == "TerminalInteractiveShell":
+                return False  # Terminal running IPython
+            else:
+                return False  # Other type (?)
+        except NameError:
+            return False  # Probably standard Python interpreter
+
+    return False  # no ipython
+
+
+in_notebook = in_notebook()
+
+# set default (non-interactive) backend
+if not in_notebook:
+    matplotlib.use("Agg")
+
+# set default style
+style_path = pathlib.Path(__file__).parent.resolve()
+plt.style.use(str(style_path / "mpl.style"))
+
+# add fonts
+try:
+    font_path = resources.files("temet") / "tables" / "fonts"
+    for font_file in font_path.iterdir():
+        matplotlib.font_manager.fontManager.addfont(font_file)
+except FileNotFoundError:
+    pass
+
+# disable fontTools timestamp warnings
+logging.getLogger("fontTools.ttLib.tables").setLevel(logging.ERROR)

{temet-0.9.0 → temet-0.9.2}/temet/plot/cloudy.py

@@ -15,12 +15,13 @@ from ..cosmo.cloudy import cloudyIon
 from ..cosmo.cloudyGrid import loadUVB
 from ..cosmo.hydrogen import photoCrossSecGray, photoRate, uvbEnergyDensity
 from ..plot.config import figsize, linestyles
-from ..plot.util import contourf, loadColorTable, sampleColorTable
-from ..util.helper import closest, evenlySample, logZeroNaN, rootPath
+from ..plot.util import _finish_plot, contourf, loadColorTable, sampleColorTable
+from ..util.extern import tables_path
+from ..util.helper import closest, evenlySample, logZeroNaN
 from ..util.simParams import simParams
 
 
-def plotUVB(uvb="FG11"):
+def plotUVB(uvb="FG11", saveFilename=None):
     """Debug plots of the UVB(nu) as a function of redshift."""
     # config
     redshifts = [0.0, 2.0, 4.0, 6.0, 7.0, 8.0, 9.0]
@@ -104,8 +105,7 @@ def plotUVB(uvb="FG11"):
     cb.ax.set_ylabel(r"log J$_{\nu}(\nu)$ [ 4 $\pi$ erg / s / cm$^2$ / Hz ]")
 
     # finish
-    fig.savefig("uvb_%s.pdf" % uvb)
-    plt.close(fig)
+    _finish_plot(fig, saveFilename, saveNameDefault="uvb_%s.pdf" % uvb)
 
 
 def plotIonAbundances(res="lg_c17", elements=("Magnesium",)):
@@ -291,7 +291,7 @@ def grackleTable():
     # filename2 = 'grid_cooling_UVB=FG11.hdf5' # my new version (testing)
     ##filename2 = 'CloudyData_UVB=FG2011_shielded.h5' # orig
 
-    filepath = rootPath + "tables/cloudy/"
+    filepath = tables_path + "cloudy/"
     # filename1 = 'grid_cooling_UVB=FG20_unshielded.hdf5'
     filename1 = "grid_cooling_UVB=FG20.hdf5"
     filename2 = "grid_cooling_UVB=FG20_ext.hdf5"

{temet-0.9.0 → temet-0.9.2}/temet/plot/clustering.py

@@ -21,7 +21,7 @@ def galaxyTwoPoint(sPs, cenSatSelects=("all", "cen", "sat"), colorBin=None, cTyp
     rMinMax = [0.01, 100.0]  # log Mpc
     yMinMax = [1e-2, 5e4]
 
-    rLabel = "r [ Mpc ]"
+    rLabel = r"r [ Mpc ]"
     yLabel = r"$\xi(r \pm \Delta r)$  [ real space two-point autocorr ]"
 
     # load/calculate
@@ -35,8 +35,7 @@ def galaxyTwoPoint(sPs, cenSatSelects=("all", "cen", "sat"), colorBin=None, cTyp
             cfs.append({"rad": rad, "xi": xi, "xi_err": xi_err, "css": cenSatSelect, "sP": sP})
 
     # start plot
-    fig = plt.figure(figsize=figsize)
-    ax = fig.add_subplot(111)
+    fig, ax = plt.subplots()
     setAxisColors(ax)
 
     ax.set_xlim(rMinMax)
@@ -99,8 +98,8 @@ def galaxyTwoPointQuantBounds(
     drawSymbols = True
     symSize = 7.0
 
-    rLabel = "r [ log Mpc ]"
-    yLabel = "log %s$\\xi(r)$"
+    rLabel = r"r [ log Mpc ]"
+    yLabel = r"log %s$\xi(r)$"
 
     # load/calculate
     cfs = OrderedDict()
@@ -406,9 +405,7 @@ def conformityWithRedFrac(sP, cenSatSelectSec="all"):
             confs[label][colorLabel] = conf
 
     # start plot
-    fig = plt.figure(figsize=figsize)
-    gs = gridspec.GridSpec(1, 1)
-    ax = fig.add_subplot(gs[0])
+    fig, ax = plt.subplots()
     setAxisColors(ax)
 
     ax.set_xlim(rMinMax)

{temet-0.9.0 → temet-0.9.2}/temet/plot/driversSizes.py

@@ -12,6 +12,7 @@ from scipy.signal import savgol_filter
 
 from ..cosmo.util import cenSatSubhaloIndices
 from ..plot.config import binSize, colors, figsize, linestyles, sKn, sKo
+from ..plot.util import _finish_plot
 from ..util.helper import logZeroNaN, running_median
 from ..util.simParams import simParams
 
@@ -26,6 +27,7 @@ def galaxySizes(
     markersize=12.0,
     xlim=None,
     ylim=None,
+    saveFilename=None,
     pdf=None,
 ):
     """Galaxy sizes (half mass radii) vs stellar mass or halo mass, compared to data.
@@ -41,8 +43,8 @@ def galaxySizes(
       markersize (float): if ``scatterPoints`` then override the default marker size.
       xlim (2-tuple): plot x-limits.
       ylim (2-tuple): plot y-limits.
-      pdf (PdfPages or None): if None, an actual PDF file is written to disk with the figure.
-        If not None, then the figure is added to this existing pdf collection.
+      saveFilename (str): name (and extension, setting format) of output plot. Automatic if None.
+      pdf (PdfPages or None): If not None, then the figure is added to this existing pdf collection.
     """
     from ..load.data import baldry2012SizeMass, lange2016SizeMass, shen2003SizeMass
 
@@ -285,21 +287,18 @@ def galaxySizes(
     # legend
     ax.legend(loc="lower right")
 
-    # finish figure
-    if pdf is not None:
-        pdf.savefig(facecolor=fig.get_facecolor())
-    else:
-        saveFilename = "sizes_%s_z%.1f_halomass=%s_halflight=%s.pdf" % (
-            "-".join([str(sim) for sim in sPs]),
-            simRedshift,
-            vsHaloMass,
-            "-".join([str(s) for s in addHalfLightRad]) if addHalfLightRad is not None else "None",
-        )
-        fig.savefig(saveFilename, facecolor=fig.get_facecolor())
-    plt.close(fig)
+    # finish plot and save
+    saveNameDefault = "sizes_%s_z%.1f_halomass=%s_halflight=%s.pdf" % (
+        "-".join([str(sim) for sim in sPs]),
+        simRedshift,
+        vsHaloMass,
+        "-".join([str(s) for s in addHalfLightRad]) if addHalfLightRad is not None else "None",
+    )
+
+    _finish_plot(fig, saveFilename, saveNameDefault, pdf)
 
 
-def galaxyHISizeMass(sPs, pdf, simRedshift=0.0):
+def galaxyHISizeMass(sPs, simRedshift=0.0, saveFilename=None, pdf=None):
     """Galaxy HI size-mass relation, at redshift zero."""
     # plot setup
     fig, ax = plt.subplots()
@@ -366,13 +365,10 @@ def galaxyHISizeMass(sPs, pdf, simRedshift=0.0):
     handles, labels = ax.get_legend_handles_labels()
     ax.legend(handles, labels, loc="lower right")
 
-    # finish figure
-    if pdf is not None:
-        pdf.savefig(facecolor=fig.get_facecolor())
-    else:
-        saveFilename = "sizes_HI_%s_z%.1f.pdf" % ("-".join([str(sim) for sim in sPs]), simRedshift)
-        fig.savefig(saveFilename, facecolor=fig.get_facecolor())
-    plt.close(fig)
+    # finish plot and save
+    saveNameDefault = "sizes_HI_%s_z%.1f.pdf" % ("-".join([str(sim) for sim in sPs]), simRedshift)
+
+    _finish_plot(fig, saveFilename, saveNameDefault, pdf)
 
 
 def sizeModelsRatios():
@@ -656,7 +652,7 @@ def clumpSizes(sP):
     plt.close(fig)
 
 
-def characteristicSizes(sP, vsHaloMass=False):
+def characteristicSizes(sP, vsHaloMass=False, saveFilename=None):
     """Compare many different 'characteristic' halo/galaxy sizes as a function of mass."""
     from ..load.data import baldry2012SizeMass
 
@@ -792,6 +788,7 @@ def characteristicSizes(sP, vsHaloMass=False):
 
     ax.legend(loc="upper left", ncol=2)
 
-    # finish figure
-    fig.savefig("characteristic_sizes_%s_%d.pdf" % (sP.simName, sP.snap))
-    plt.close(fig)
+    # finish plot and save
+    saveNameDefault = "characteristic_sizes_%s_%d.pdf" % (sP.simName, sP.snap)
+
+    _finish_plot(fig, saveFilename, saveNameDefault)

{temet-0.9.0 → temet-0.9.2}/temet/plot/mpl.style

@@ -90,8 +90,7 @@ font.size           : 14
 #font.serif          : Times New Roman, Bitstream Vera Serif, New Century Schoolbook, Century Schoolbook L, Utopia, ITC Bookman, Bookman, Nimbus Roman No9 L, Times, Palatino, Charter, serif
 #font.sans-serif     : Helvetica, Arial, Verdana, Bitstream Vera Sans, Lucida Grande, Geneva, Lucid, Avant Garde, sans-serif
 
-font.sans-serif      : Roboto
-#Open Sans, Arial, Helvetica, Bitstream Vera Sans, Bebas Neue
+font.sans-serif      : Roboto, Open Sans, Helvetica, Bitstream Vera Sans, Arial
 
 #font.cursive        : Apple Chancery, Textile, Zapf Chancery, Sand, Script MT, Felipa, cursive
 #font.fantasy        : Comic Sans MS, Chicago, Charcoal, Impact, Western, Humor Sans, fantasy
@@ -291,7 +290,7 @@ legend.scatterpoints : 1 # number of scatter points
 #figure.titlesize : medium     # size of the figure title
 #figure.titleweight : normal   # weight of the figure title
 figure.figsize   : 11.2, 8    # figure size in inches
-figure.dpi       : 150      # figure dots per inch
+figure.dpi       : 100      # figure dots per inch
 figure.facecolor : white    # figure facecolor; 0.75 is scalar gray
 #figure.edgecolor : white   # figure edgecolor
 figure.autolayout : False   # When True, automatically adjust subplot