#!/usr/bin/env python3


import sys

import numpy as np

import pyro.util.io_pyro as io

usage = """
      usage: ./compare.py file1 file2 (rtol)

      where rtol is an (optional) relative tolerance parameter to use when
      comparing the data
"""

errors = {"gridbad": "grids don't agree",
          "namesbad": "variable lists don't agree",
          "varerr": "one or more variables don't agree"}



[docs]
def compare(data1, data2, rtol=1.e-12):
    """
    given two CellCenterData2d objects, compare the data, zone-by-zone
    and output any errors

    Parameters
    ----------
    data1, data2 : CellCenterData2d object
        Two data grids to compare
    rtol : float
        relative tolerance to use to compare grids

    """

    # compare the grids
    if not data1.grid == data2.grid:
        return "gridbad"

    # compare the data
    if not sorted(data1.names) == sorted(data2.names):
        return "namesbad"

    print(" ")
    print("variable comparisons:")

    result = 0

    for name in data1.names:

        d1 = data1.get_var(name)
        d2 = data2.get_var(name)

        abs_err = np.max(np.abs(d1.v() - d2.v()))

        if not np.any(d2.v() == 0):
            rel_err = np.max(np.abs(d1.v() - d2.v()) / np.abs(d2.v()))
            print(f"{name:20s} absolute error = {abs_err:10.10g}, relative error = {rel_err:10.10g}")
        else:
            print(f"{name:20s} absolute error = {abs_err:10.10g}")

        if not np.allclose(d1.v(), d2.v(), rtol=rtol):
            result = "varerr"

    return result




[docs]
def main():
    if not (len(sys.argv) == 3 or len(sys.argv) == 4):
        print(usage)
        sys.exit(2)

    file1 = sys.argv[1]
    file2 = sys.argv[2]

    s1 = io.read(file1)
    s2 = io.read(file2)

    if len(sys.argv) == 3:
        result = compare(s1.cc_data, s2.cc_data)
    else:
        result = compare(s1.cc_data, s2.cc_data, rtol=float(sys.argv[3]))

    if result == 0:
        print("SUCCESS: files agree")
    else:
        print("ERROR: ", errors[result])



if __name__ == "__main__":
    main()