Created a basic tabplot2.py

src/scripts/python/tabplot2.py

@@ -12,8 +12,10 @@
 #import pdb
 #pdb.set_trace()
 
+isScatter = False
+numlines = 0
 
-keyval = [
+keyval = [  # list of key values
     "in=???\n       input fits cube",
     "xcol=1\n       X column(s) (1=first column)",
     "ycol=2\n       Y column(s)",
@@ -27,25 +29,79 @@
     "VERSION=0.1\n  3-apr-2023 PJT",
 ]
 
-usage = """
+usage = """  
 plot a table like tabplot
 """
 
 p = getparam.Param(keyval,usage)
 
-# get CLI parameters
+# get CLI parameters using getparam
 infile = p.get("in")
-xcol = int(p.get("xcol"))
-ycol = int(p.get("ycol"))
+xinput = p.get("xcol")
+yinput = p.get("ycol")
+colorinput = p.get("color")
+lineinput = p.get("line")
+pointinput = p.get("point")
+
+# get indexes for xcol and ycol
+try:
+    xcols = [int(x) for x in xinput.split(',')]
+    ycols = [int(y) for y in yinput.split(',')]
+except ValueError:
+    print("ERROR: excpected 'ints' or 'ints,ints")
+    exit(1)
+if(len(xcols) > len(ycols)):
+    numlines = len(xcols)
+else:
+    numlines = len(ycols)
+
+# get CLI parameters
+colors = colorinput.split(',')
+linewidths = lineinput.split(',')
+pointsizes = [1] * numlines
+try:
+    pointsizes = [int(p) for p in pointinput.split(',')]
+except ValueError:
+    pointsizes = [1] * numlines
+
+# if no xcol or ycol specified, assume col 1 is x and col 2 is y
+if len(xcols) == 0:
+    xcols = [1]
+if len(ycols) == 0:
+    ycols = [2]
+
+# if no color or width specified, set color to black and width 1
+if colors[0] == "":
+    colors = ["black"] * numlines
+if linewidths[0] == "":
+    linewidths = [1] * numlines
 
 # gather data
 data = np.loadtxt(infile).T
 
-xdata = data[xcol-1]
-ydata = data[ycol-1]
+xdata = [0] * len(xcols)
+ydata = [0] * len(ycols)
 
-# plot!
+for i in range(len(xcols)): # Read x data
+    xdata[i] = data[xcols[i]-1]
 
+for i in range(len(ycols)): # Read y data
+    ydata[i] = data[ycols[i]-1]
+
+# Plotting
 plt.figure()
-plt.plot(xdata, ydata)
+if len(xdata) == 1: # Case: only 1 xcol, plot each y against the only x
+    for i in range (len(ydata)):
+        plt.plot(xdata[0], ydata[i], color=colors[i], linewidth=linewidths[i], marker = '.', markersize=pointsizes[i])
+elif len(ydata) == 1: # Case: only 1 ycol, plot each x against the only y
+    for i in range (len(xdata)):
+        plt.plot(xdata[i], ydata[0], color=colors[i], linewidth=linewidths[i], marker = '.', markersize=pointsizes[i])
+else:               # Case: more than 1 xcol and ycol, plot each ycol with its xcol at the same index
+    for i in range (len(xdata)):
+        try:
+            plt.plot(xdata[i], ydata[i], color=colors[i], linewidth=linewidths[i], marker = '.', markersize=pointsizes[i])
+        except IndexError:
+            print("ERROR: xcols and ycols mismatch")
+            sys.exit()
+
 plt.show()