import sys

class Args:
    def __init__(self, filename, func, var=[None], mval=[None], oname=[None], out=[None], ndim = None):
        if out == "plotter_lib_pngs/":
            print("Output directory name can't be equal to " + out)
            sys.exit(-1)
    
        self.filename = filename
        self.var = var
        self.mval = mval
        self.oname = oname
        self.out = out
        self.func = func
        self.ndim = ndim

        self.CorrVals()

    def CorrVals(self):
        if type(self.oname) is not list:
            self.oname = [self.oname]
        if None in self.oname:
            self.oname = None

        if type(self.var) is not list:
            self.var = [self.var]
        if None in self.var:
            self.var = None

        if type(self.mval) is not list:
            self.mval = [self.mval]
        if None in self.mval:
            self.mval = None