data-manipulation-utilities 0.2.7__py3-none-any.whl → 0.2.8.dev720__py3-none-any.whl

dmu_data/plotting/tests/high_stat.yaml

@@ -1,22 +0,0 @@
-selection:
-    max_ran_entries : 50000
-    cuts:
-      z : 'z > 0'
-saving:
-    plt_dir : tests/plotting/high_stat
-definitions:
-    z : 'x + y'
-plots:
-    x :
-        binning    : [-5.0, 8.0, 40]
-        yscale     : 'linear'
-        labels     : ['x', 'Entries']
-    y :
-        binning    : [-5.0, 8.0, 40]
-        yscale     : 'linear'
-        labels     : ['y', 'Entries']
-    z :
-        binning    : [-5.0, 8.0, 40]
-        yscale     : 'linear'
-        labels     : ['x + y', 'Normalized']
-        normalized : true

dmu_data/plotting/tests/legend.yaml

@@ -1,12 +0,0 @@
-saving:
-    plt_dir : tests/plotting/legend
-general:
-    size : [20, 10]
-plots:
-    x : 
-        binning : [-5.0, 8.0, 40]
-    y : 
-        binning : [-5.0, 8.0, 40]
-style:
-  legend:
-    bbox_to_anchor : [1.2, 1]

dmu_data/plotting/tests/name.yaml

@@ -1,14 +0,0 @@
-saving:
-    plt_dir : tests/plotting/name
-
-plots:
-    x : 
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear' 
-        labels  : ['x', 'Entries']
-        name    : 'xvar'
-    y : 
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear' 
-        labels  : ['y', 'Entries']
-        name    : 'yvar'

dmu_data/plotting/tests/no_bounds.yaml

@@ -1,12 +0,0 @@
-saving:
-    plt_dir : tests/plotting/no_bounds
-
-plots:
-    x : 
-        binning : [1, 1, 40]
-        yscale  : 'linear' 
-        labels  : ['x', 'Entries']
-    y : 
-        binning : [1, 1, 40]
-        yscale  : 'linear' 
-        labels  : ['y', 'Entries']

dmu_data/plotting/tests/normalized.yaml

@@ -1,9 +0,0 @@
-saving:
-    plt_dir : tests/plotting/normalized
-plots:
-    x : 
-      normalized : true
-      binning    : [-5.0, 8.0, 40]
-    y : 
-      normalized : false 
-      binning    : [-5.0, 8.0, 40]

dmu_data/plotting/tests/plug_fwhm.yaml

@@ -1,24 +0,0 @@
-saving:
-    plt_dir : plotting/pluggins/fwhm
-plots:
-    x : 
-      binning : [-5.0, 8.0, 40]
-      title   : x distribution
-    y : 
-      binning : [-5.0, 8.0, 40]
-      title   : y distribution
-plugin:
-  fwhm:
-    x :
-      plot   : true
-      obs    : [-2, 4]
-      plot   : true
-      format : FWHM={:.3f}
-      add_std: True
-    y :
-      plot   : true
-      obs    : [-4, 8]
-      plot   : true
-      format : FWHM={:.3f}
-      add_std: True
-

dmu_data/plotting/tests/plug_stats.yaml

@@ -1,19 +0,0 @@
-saving:
-    plt_dir : plotting/pluggins/stats
-plots:
-    x : 
-      binning : [-5.0, 8.0, 40]
-      title   : x distribution
-      styling:
-        linestyle : '-'
-    y : 
-      binning : [-5.0, 8.0, 40]
-      title   : y distribution
-      styling:
-        linestyle : '-'
-plugin:
-  stats:
-    x :
-      mean : $\mu$={:.2f}
-      rms  : $\sigma$={:.2f}
-      sum  : $\Sigma$={:.0f}

dmu_data/plotting/tests/simple.yaml

@@ -1,9 +0,0 @@
-saving:
-    plt_dir : tests/plotting/simple
-plots:
-    x : 
-      binning : [-5.0, 8.0, 40]
-      title   : x distribution
-    y : 
-      binning : [-5.0, 8.0, 40]
-      title   : y distribution

dmu_data/plotting/tests/stats.yaml

@@ -1,9 +0,0 @@
-saving:
-    plt_dir : tests/plotting/stats
-plots:
-    x : 
-        binning : [-5.0, 8.0, 40]
-    y : 
-        binning : [-5.0, 8.0, 40]
-stats:
-  nentries : '{:.2e}'

dmu_data/plotting/tests/styling.yaml

@@ -1,11 +0,0 @@
-saving:
-    plt_dir : tests/plotting/styling
-plots:
-  x :
-    binning : [-5.0, 8.0, 40]
-    title   : x distribution
-    styling :
-      histtype : step
-  y :
-    binning : [-5.0, 8.0, 40]
-    title   : y distribution

dmu_data/plotting/tests/title.yaml

@@ -1,14 +0,0 @@
-saving:
-    plt_dir : tests/plotting/title
-
-plots:
-    x :
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear'
-        labels  : ['x', 'Entries']
-        title   : 'Title for X plot'
-    y :
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear'
-        labels  : ['y', 'Entries']
-        title   : 'Title for Y plot'

dmu_data/plotting/tests/weights.yaml

@@ -1,13 +0,0 @@
-saving:
-    plt_dir : tests/plotting/weights
-plots:
-    x :
-        weights : weights
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear'
-        labels  : ['x', 'Entries']
-    y :
-        weights : weights
-        binning : [-5.0, 8.0, 40]
-        yscale  : 'linear'
-        labels  : ['y', 'Entries']

dmu_data/text/transform.toml

@@ -1,4 +0,0 @@
-[trf]
-[trf.append]
-'primes are'=['2', '3', '5']
-'days are'=['Monday', 'Tuesday', 'Wednesday']

dmu_data/text/transform.txt

@@ -1,6 +0,0 @@
-the
-first
-primes are
-and
-the first
-days are

dmu_data/text/transform_set.toml

@@ -1,8 +0,0 @@
-[settings]
-as_substring=true
-format      ='--> {} <--'
-
-[trf]
-[trf.append]
-'primes are'=['2', '3', '5']
-'days are'=['Monday', 'Tuesday', 'Wednesday']

dmu_data/text/transform_set.txt

@@ -1,6 +0,0 @@
-the
-first
-primes are:
-and
-the first
-days are:

dmu_data/text/transform_trf.txt

@@ -1,12 +0,0 @@
-the
-first
-primes are
-2
-3
-5
-and
-the first
-days are
-Monday
-Tuesday
-Wednesday

dmu_scripts/git/publish

@@ -1,89 +0,0 @@
-#!/usr/bin/env bash
-
-# --------------------------
-display_help()
-{
-    echo "Script meant to:"
-    echo ""
-    echo "1. Check if version in pyproject.toml has been modified"
-    echo "2. If it has create new tag following version name"
-    echo "3. Push to remote "
-}
-# --------------------------
-get_opts()
-{
-    while getopts :hf: option; do 
-        case "${option}" in
-            h)  
-                display_help
-                exit 0
-                ;;  
-           \?)  echo "Invalid option: -${OPTARG}"
-                display_help
-                exit 1
-                ;;  
-            :)  echo "$0: Arguments needed"
-                display_help
-                exit 1
-                ;;  
-        esac
-    done
-}
-# --------------------------
-# Picks up version from pyproject.toml
-get_version()
-{
-    if [[ ! -f pyproject.toml ]];then
-        echo "Cannot find pyproject.toml"
-        exit 1
-    fi
-
-    VERSION_LINE=$(grep version pyproject.toml)
-
-    if [[ $? -ne 0 ]];then
-        ehco "Could not extract version from pyproject.toml"
-        exit 1
-    fi
-
-    if [[ "$VERSION_LINE" =~ .*([0-9]\.[0-9]\.[0-9]).* ]];then
-        VERSION=${BASH_REMATCH[1]}
-        echo "Using version: $VERSION"
-        return
-    fi
-
-    echo "Could not extract version from: $VERSION_LINE"
-    exit 1
-}
-# --------------------------
-create_tag()
-{
-    git tag -n | grep $VERSION
-
-    if [[ $? -eq 0 ]];then
-        echo "Version found among tags, not tagging"
-        return
-    fi
-
-    echo "Version $VERSION not found among tags, creating new tag"
-
-    git tag -a $VERSION
-}
-# --------------------------
-push_all()
-{
-    for REMOTE in $(git remote);do
-        echo "Pushing tags and commits to remote: $REMOTE"
-        git add pyproject.toml
-        git commit -m "Publication commit"
-
-        git pull    $REMOTE HEAD
-        git push -u $REMOTE HEAD
-        git push    $REMOTE --tags
-    done
-}
-# --------------------------
-get_opts "$@"
-
-get_version
-create_tag
-push_all

dmu_scripts/physics/check_truth.py

@@ -1,121 +0,0 @@
-'''
-Script meant to do truth matching checks
-'''
-import os
-import copy
-import argparse
-
-import yaml
-import mplhep
-import matplotlib.pyplot as plt
-
-from ROOT import RDataFrame
-
-from dmu.logging.log_store   import LogStore
-from dmu.plotting.plotter_1d import Plotter1D as Plotter
-
-log=LogStore.add_logger('dmu:physics:check_truth')
-# ----------------------------------
-def _set_logs() -> None:
-    LogStore.set_level('dmu:plotting:Plotter'  , 30)
-    LogStore.set_level('dmu:plotting:Plotter1D', 30)
-# ----------------------------------
-def _get_args() -> argparse.Namespace:
-    '''
-    Parse args
-    '''
-    parser = argparse.ArgumentParser(description='Script used to carry out checks on truth matching mechanisms for MC')
-    parser.add_argument('-c', '--conf' , type=str, help='Path to config file', required=True)
-    args = parser.parse_args()
-
-    return args
-# ----------------------------------
-def _get_config(args : argparse.Namespace) -> dict:
-    path = args.conf
-    if not os.path.isfile(path):
-        raise FileNotFoundError(f'Cannot find {path}')
-
-    with open(path, encoding='utf-8') as ifile:
-        cfg = yaml.safe_load(ifile)
-
-    return cfg
-# ----------------------------------
-def _get_rdf(file_path : str, tree_path : str) -> RDataFrame:
-    log.debug(f'Picking inputs from: {file_path}/{tree_path}')
-    rdf = RDataFrame(tree_path, file_path)
-
-    nentries = rdf.Count().GetValue()
-    log.debug(f'Found {nentries} entries')
-
-    return rdf
-# ----------------------------------
-def _preprocess_rdf(rdf : RDataFrame, cfg : dict) -> RDataFrame:
-    if 'max_entries' in cfg:
-        max_entries = cfg['max_entries']
-        rdf = rdf.Range(max_entries)
-
-    return rdf
-# ----------------------------------
-def _check(cfg : dict) -> None:
-    log.info(110 * '-')
-    log.info(f'{"Sample":<20}{"Method":<20}{"Initial":<15}{"":<15}{"Final":<15}{"":15}{"Efficiency":<10}')
-    log.info(110 * '-')
-
-    for sample_name in cfg['samples']:
-        file_path = cfg['samples'][sample_name]['file_path']
-        tree_path = cfg['samples'][sample_name]['tree_path']
-        rdf = _get_rdf(file_path, tree_path)
-        rdf = _preprocess_rdf(rdf, cfg)
-
-        d_cut_true = {}
-        d_cut_fake = {}
-        for method, cut in cfg['samples'][sample_name]['methods'].items():
-            _check_kind(rdf, sample_name, method, cut)
-
-            d_cut_true[method] = cut
-            d_cut_fake[method] = f'({cut}) == 0'
-        log.info('')
-
-        _plot_distributions(cfg, sample_name, rdf, d_cut_true, kind='matched')
-        _plot_distributions(cfg, sample_name, rdf, d_cut_fake, kind='anti_matched')
-# ----------------------------------
-def _plot_distributions(cfg : dict, sample_name : str, rdf : RDataFrame, d_cut : dict[str,str], kind : str) -> None:
-    cfg     = copy.deepcopy(cfg)
-    cfg_plt = cfg['samples'][sample_name]['plot']
-    cfg_plt = _add_suffix(cfg_plt, sample_name, kind)
-    d_rdf   = { method : rdf.Filter(cut) for method, cut in d_cut.items() }
-
-    ptr=Plotter(d_rdf=d_rdf, cfg=cfg_plt)
-    ptr.run()
-# ----------------------------------
-def _add_suffix(cfg : dict, sample_name : str, kind : str) -> dict:
-    d_var = cfg['plots']
-    for var in d_var:
-        d_var[var]['name']  = f'{var}_{kind}'
-        d_var[var]['title'] = f'{sample_name}; {kind}'
-
-    cfg['plots'] = d_var
-
-    return cfg
-# ----------------------------------
-def _check_kind(rdf : RDataFrame, sample : str, name : str, cut : str) -> RDataFrame:
-    nini = rdf.Count().GetValue()
-    rdf  = rdf.Filter(cut, name)
-    nfnl = rdf.Count().GetValue()
-    eff  = nfnl / nini * 100
-
-    log.info(f'{sample:<20}{name:<20}{nini:<15}{"":<15}{nfnl:<15}{"-->":15}{eff:10.2f}')
-# ----------------------------------
-def main():
-    '''
-    Script starts here
-    '''
-    _set_logs()
-    args = _get_args()
-    cfg  = _get_config(args)
-    plt.style.use(mplhep.style.LHCb2)
-
-    _check(cfg)
-# ----------------------------------
-if __name__ == '__main__':
-    main()