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.benchmark;
025    
026    import org.ujmp.core.Matrix;
027    import org.ujmp.core.doublematrix.DoubleMatrix2D;
028    import org.ujmp.core.doublematrix.impl.DefaultDenseDoubleMatrix2D;
029    import org.ujmp.core.util.SerializationUtil;
030    
031    public class SolveTallBenchmarkTask extends AbstractBenchmarkTask {
032    
033            public SolveTallBenchmarkTask(long benchmarkSeed, Class<? extends DoubleMatrix2D> matrixClass,
034                            BenchmarkConfig config) {
035                    super(benchmarkSeed, matrixClass, config.getSolveTallSizes(), config);
036            }
037    
038            @Override
039            public BenchmarkResult task(Class<? extends Matrix> matrixClass, long benchmarkSeed, int run,
040                            long[] size) {
041                    final long t0, t1, m0, m1;
042                    final DoubleMatrix2D a, x;
043                    Matrix b1 = null, b2 = null, result = null;
044                    try {
045                            a = BenchmarkUtil.createMatrix(matrixClass, size);
046                            if (!a.getClass().getName().startsWith("org.ujmp.core.")
047                                            && a.getClass().getDeclaredMethod("solve", Matrix.class) == null) {
048                                    System.out.print("-");
049                                    System.out.flush();
050                                    return BenchmarkResult.NOTAVAILABLE;
051                            }
052                            x = DoubleMatrix2D.factory.zeros(size[1], size[0]);
053                            BenchmarkUtil.rand(benchmarkSeed, run, 0, a);
054                            BenchmarkUtil.rand(benchmarkSeed, run, 1, x);
055                            b1 = new DefaultDenseDoubleMatrix2D(a).mtimes(new DefaultDenseDoubleMatrix2D(x));
056                            b2 = BenchmarkUtil.createMatrix(matrixClass, b1);
057                            BenchmarkUtil.purgeMemory(getConfig());
058                            m0 = Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory();
059                            t0 = System.nanoTime();
060                            result = a.solve(b2);
061                            t1 = System.nanoTime();
062                            m1 = Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory();
063                            if (result == null) {
064                                    System.out.print("e");
065                                    System.out.flush();
066                                    return BenchmarkResult.ERROR;
067                            }
068                            double diff = BenchmarkUtil.difference(result, x);
069                            result = null;
070                            long mem = m1 - m0 - SerializationUtil.sizeOf(result);
071                            mem = mem > 0 ? mem : 0;
072                            return new BenchmarkResult((t1 - t0) / 1000000.0, diff, mem);
073                    } catch (Throwable e) {
074                            System.out.print("e");
075                            System.out.flush();
076                            return BenchmarkResult.ERROR;
077                    }
078            }
079    
080            @Override
081            public String getTaskName() {
082                    return "solveTall";
083            }
084    
085    }