###########################################################################
# program: pseudo_inverse_matrix.py
# author: Tom Irvine
# Email: tomirvine@aol.com
# version: 1.0
# date: November 28, 2011
# description:  Calculate the pseudo inverse of a matrix
#
###########################################################################

from sys import stdin

from tompy import read_array,GetInteger2,WriteArray

from numpy import linalg
from numpy import set_printoptions

###########################################################################

set_printoptions(precision=3)

A=read_array("A") 
print " "
print "input matrix"
print(A)  

Ainv=linalg.pinv(A)

print " "
print "pseudo inverse matrix"
print(Ainv)

print " "
print "Write pseudo inverse matrix to file?  1=yes 2=no "

immf=GetInteger2()

if immf==1:
    print " "
    print "Enter the output filename: "
    output_file_path = stdin.readline()
    WriteArray(Ainv,output_file_path)