#!/home/bfiedler/anaconda3/bin/python3
# 2025 January 16
import glob
import os, sys
import numpy as np
import argparse

# Define the argument parser
parser = argparse.ArgumentParser(description="Process npy files with optional prefix and plotting flag.")

# Positional arguments (like npyfile1 and optional npyfile2)
parser.add_argument("npyfile1", help="First required npy file")
parser.add_argument("npyfile2", nargs="?", help="Optional second npy file")  # '?' makes it optional

# Optional arguments
parser.add_argument("-p", dest="pre", type=str, default="gfs1degnpy/2025", help="Path prefix")
parser.add_argument("-i", action="store_true", dest="plotit", help="enable plotting")

# Version argument
parser.add_argument("--version", action="version", version="0.1")

# Parse arguments
args = parser.parse_args()
#opts = parser.parse_opts()
print(args)

# Access parsed values
print(f"Path Prefix: {args.pre}")
print(f"Plotting Enabled: {args.plotit}")
print(f"First NPY File: {args.npyfile1}")
if args.npyfile2:
    print(f"Second NPY File: {args.npyfile2}")


v = np.load(args.pre+args.npyfile1)
if args.npyfile2:
    v = v-np.load(args.pre+args.npyfile2)

print("max=",v.max(),"  min=",v.min())

if args.plotit:
    import matplotlib.pyplot as plt
    plt.imshow(v)
    plt.show()