Source code for rid.utils.files

import json
import pickle
from typing import Dict, List, Union
import numpy as np


def read_txt(
        fname: str
    ) -> str:
    with open(fname, "r") as fn:
        fcont = fn.read()
    return fcont


def read_binary(
        fname: str
    ) -> str:
    with open(fname, "rb") as fn:
        fcont = fn.read()
    return fcont
    

def write_txt(
        fname: str, 
        fcont: str
    ):
    with open(fname, "w") as fn:
        fn.write(fcont)


def write_binary(
        fname: str, 
        fcont: str
    ):
    with open(fname, "wb") as fn:
        fn.write(fcont)


def load_txt(
        fname: str,
        dtype = float,
        comments: List[str] = ["#"]
    ):
    data = np.loadtxt(fname, comments=comments, dtype=dtype)
    return data

def save_txt(
        fname: str,
        fcont: Union[np.ndarray, List],
        fmt: str = "%.6e"
    ):
    np.savetxt(fname, fcont, fmt=fmt)


def load_json(
        fname: str
    ) -> Dict:
    with open(fname, "r") as fn:
        jdata = json.load(fn)
    return jdata


def dump_json(
        fname: str, 
        fcont: Dict,
        indent: int = 4
    ):
    with open(fname, "w") as fn:
        json.dump(fcont, fn, indent=indent)


def load_pkl(
        fname: str
    ):
    with open(fname, "rb") as ff:
        data = pickle.load(ff) 
    return data


def save_pkl(
        fname,
        obj
    ):
    with open(fname, "wb") as ff:
        pickle.dump(obj, fname)