1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
|
############################################################################
##
## Copyright (C) 2006-2007 University of Utah. All rights reserved.
##
## This file is part of VisTrails.
##
## This file may be used under the terms of the GNU General Public
## License version 2.0 as published by the Free Software Foundation
## and appearing in the file LICENSE.GPL included in the packaging of
## this file. Please review the following to ensure GNU General Public
## Licensing requirements will be met:
## http://www.opensource.org/licenses/gpl-license.php
##
## If you are unsure which license is appropriate for your use (for
## instance, you are interested in developing a commercial derivative
## of VisTrails), please contact us at contact@vistrails.org.
##
## This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
## WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
##
############################################################################
import core.modules
import core.modules.module_registry
from core.modules.vistrails_module import Module, ModuleError
from SciPy import SciPy
from Matrix import *
import math
import scipy
class MatrixOperation(SciPy):
def compute(self):
pass
class GetReals(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
self.setResult("OutputMatrix", m.Reals())
class GetImaginaries(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
self.setResult("OutputMatrix", m.Imaginaries())
class Conjugate(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
self.setResult("OutputMatrix", m.Conjugate())
class GetRow(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
c = self.getInputFromPort("RowIndex")
row = m.GetRow(c)
mat = SparseMatrix()
mat.matrix = row
self.setResult("OutputMatrix", mat)
class GetCol(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
c = self.getInputFromPort("ColIndex")
col = m.GetCol(c)
mat = SparseMatrix()
mat.matrix = col
self.setResult("OutputMatrix", mat)
class MatrixMultiply(MatrixOperation):
def compute(self):
a = self.getInputFromPort("InputMatrix1")
b = self.getInputFromPort("InputMatrix2")
c = a.matrix * b.matrix
out = SparseMatrix()
out.matrix = c
return out
class ScalarMultiply(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
s = self.getInputFromPort("Scalar")
m.matrix = m.matrix * s
self.setResult("OutputMatrix", m)
class ElementMultiply(MatrixOperation):
def compute(self):
from numpy import array
a = self.getInputFromPort("InputMatrix1")
b = self.getInputFromPort("InputMatrix2")
if a.matrix.shape != b.matrix.shape:
raise ModuleError(self, 'Mismatching input dimensions!')
c = SparseMatrix()
c.matrix = sparse.csc_matrix(array(a.matrix.toarray())*array(b.matrix.toarray()))
self.setResult("OutputMatrix", c)
class Transpose(MatrixOperation):
def compute(self):
m = self.getInputFromPort("InputMatrix")
m.matrix = m.matrix.transpose()
self.setResult("OutputMatrix", m)
class PrintMatrix(Matrix):
def compute(self):
m = self.getInputFromPort("InputMatrix")
s = m.matrix.__str__()
self.setResult("Output", s)
class GetColumnRange(Matrix):
def compute(self):
m = self.getInputFromPort("InputMatrix")
start = self.getInputFromPort("StartIndex")
end = self.getInputFromPort("EndIndex")
size = end - start
out = SparseMatrix()
out.matrix = sparse.csc_matrix((m.matrix.shape[0], size))
i = 0
j = 0
while j < m.matrix.shape[0]:
while i < size:
val = m.matrix[j,start+i]
out.matrix[j,i] = val
i = i + 1
j = j + 1
self.setResult("Output", out)
class ATan2(MatrixOperation):
def compute(self):
if self.hasInputFromPort("InputMatrix"):
m = self.getInputFromPort("InputMatrix")
r = m.Reals().matrix
im = m.Imaginaries().matrix
else:
r = self.getInputFromPort("RealMatrix").matrix
im = self.getInputFromPort("ImaginaryMatrix").matrix
out = SparseMatrix()
out.matrix = sparse.csc_matrix(scipy.arctan2(im.toarray(),r.toarray()))
self.setResult("Output", out)
|