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.util.SerializationUtil;
029    
030    public class PlusMatrixBenchmarkTask extends AbstractBenchmarkTask {
031    
032            public PlusMatrixBenchmarkTask(long benchmarkSeed, Class<? extends DoubleMatrix2D> matrixClass,
033                            BenchmarkConfig config) {
034                    super(benchmarkSeed, matrixClass, config.getPlusSizes(), config);
035            }
036    
037            @Override
038            public BenchmarkResult task(Class<? extends Matrix> matrixClass, long benchmarkSeed, int run,
039                            long[] size) {
040                    final long t0, t1, e0, e1;
041                    final DoubleMatrix2D m0, m1;
042                    Matrix r;
043                    try {
044                            m0 = BenchmarkUtil.createMatrix(matrixClass, size);
045                            m1 = BenchmarkUtil.createMatrix(matrixClass, size);
046                            if (!m0.getClass().getName().startsWith("org.ujmp.core")
047                                            && m0.getClass().getDeclaredMethod("plus", Matrix.class) == null) {
048                                    System.out.print("-");
049                                    System.out.flush();
050                                    return BenchmarkResult.NOTAVAILABLE;
051                            }
052                            BenchmarkUtil.rand(benchmarkSeed, run, 0, m0);
053                            BenchmarkUtil.rand(benchmarkSeed, run, 1, m1);
054                            BenchmarkUtil.purgeMemory(getConfig());
055                            e0 = Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory();
056                            t0 = System.nanoTime();
057                            r = m0.plus(m1);
058                            t1 = System.nanoTime();
059                            e1 = Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory();
060                            if (r == null) {
061                                    System.out.print("e");
062                                    System.out.flush();
063                                    return BenchmarkResult.ERROR;
064                            }
065                            long mem = e1 - e0 - SerializationUtil.sizeOf(r);
066                            mem = mem > 0 ? mem : 0;
067                            return new BenchmarkResult((t1 - t0) / 1000000.0, mem);
068                    } catch (Throwable e) {
069                            System.out.print("e");
070                            System.out.flush();
071                            return BenchmarkResult.ERROR;
072                    }
073            }
074    
075            @Override
076            public String getTaskName() {
077                    return "plusMatrix";
078            }
079    
080    }