001 /* 002 * Copyright (C) 2008-2010 by Holger Arndt 003 * 004 * This file is part of the Universal Java Matrix Package (UJMP). 005 * See the NOTICE file distributed with this work for additional 006 * information regarding copyright ownership and licensing. 007 * 008 * UJMP is free software; you can redistribute it and/or modify 009 * it under the terms of the GNU Lesser General Public License as 010 * published by the Free Software Foundation; either version 2 011 * of the License, or (at your option) any later version. 012 * 013 * UJMP is distributed in the hope that it will be useful, 014 * but WITHOUT ANY WARRANTY; without even the implied warranty of 015 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 016 * GNU Lesser General Public License for more details. 017 * 018 * You should have received a copy of the GNU Lesser General Public 019 * License along with UJMP; if not, write to the 020 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, 021 * Boston, MA 02110-1301 USA 022 */ 023 024 package org.ujmp.core.doublematrix.calculation.basic; 025 026 import org.ujmp.core.Matrix; 027 import org.ujmp.core.doublematrix.calculation.AbstractDoubleCalculation; 028 import org.ujmp.core.exceptions.MatrixException; 029 import org.ujmp.core.util.MathUtil; 030 031 public class Mtimes extends AbstractDoubleCalculation { 032 private static final long serialVersionUID = 4170937261358240120L; 033 034 private boolean ignoreNaN = false; 035 036 private long[] size = null; 037 038 public Mtimes(boolean ignoreNaN, Matrix m1, Matrix m2) { 039 super(m1, m2); 040 this.ignoreNaN = ignoreNaN; 041 this.size = new long[] { m1.getRowCount(), m2.getColumnCount() }; 042 } 043 044 public Mtimes() { 045 super(); 046 } 047 048 public double getDouble(long... coordinates) throws MatrixException { 049 final Matrix m1 = getSources()[0]; 050 final Matrix m2 = getSources()[1]; 051 052 final long row = coordinates[ROW]; 053 final long col = coordinates[COLUMN]; 054 055 double sum = 0.0; 056 if (ignoreNaN) { 057 for (long k = m1.getColumnCount(); --k >= 0;) { 058 sum += MathUtil.ignoreNaN(m1.getAsDouble(row, k)) 059 * MathUtil.ignoreNaN(m2.getAsDouble(k, col)); 060 } 061 } else { 062 for (long k = m1.getColumnCount(); --k >= 0;) { 063 sum += m1.getAsDouble(row, k) * m2.getAsDouble(k, col); 064 } 065 } 066 067 return sum; 068 } 069 070 public long[] getSize() { 071 return size; 072 } 073 074 }