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