Weighted fields

main.py

@@ -9,6 +9,8 @@ def main():
                         help="Data type, default is '%(default)s'")
     parser.add_argument("-x", metavar="X.STD", required=True, help="X data input file name")
     parser.add_argument("-y", metavar="Y.STD", required=True, help="Y data input file name")
+    parser.add_argument("-wx", metavar="WX.STD", required=False, help="X data weight file name")
+    parser.add_argument("-wy", metavar="WY.STD", required=False, help="Y data weight file name")
     parser.add_argument("-t", "--time", type=int, required=True, help="Length of the time interval")
     parser.add_argument("-k", type=int, default=3,
                         help="Number of singular values, default is %(default)d")
@@ -29,8 +31,14 @@ def main():
 
     X = np.fromfile(args.x, dtype=dtype).reshape(t, -1)
     Y = np.fromfile(args.y, dtype=dtype).reshape(t, -1)
+    WX = None
+    if args.wx is not None:
+        WX = np.fromfile(args.wx, dtype=dtype).reshape(-1)
+    WY = None
+    if args.wy is not None:
+        WY = np.fromfile(args.wy, dtype=dtype).reshape(-1)
 
-    svd = supersvd(X, Y, args.k, args.elim_mean)
+    svd = supersvd(X, Y, args.k, args.elim_mean, WX, WY)
     
     if args.xv is not None:
         svd.x_vect.tofile(args.xv)