CElementWise.java

Go to the documentation of this file.

/*
 * @cond LICENSE
 * ######################################################################################
 * # LGPL License                                                                       #
 * #                                                                                    #
 * # This file is part of the LightJason AgentSpeak(L++)                                #
 * # Copyright (c) 2015-19, LightJason (info@lightjason.org)                            #
 * # This program is free software: you can redistribute it and/or modify               #
 * # it under the terms of the GNU Lesser General Public License as                     #
 * # published by the Free Software Foundation, either version 3 of the                 #
 * # License, or (at your option) any later version.                                    #
 * #                                                                                    #
 * # This program is distributed in the hope that it will be useful,                    #
 * # but WITHOUT ANY WARRANTY; without even the implied warranty of                     #
 * # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                      #
 * # GNU Lesser General Public License for more details.                                #
 * #                                                                                    #
 * # You should have received a copy of the GNU Lesser General Public License           #
 * # along with this program. If not, see http://www.gnu.org/licenses/                  #
 * ######################################################################################
 * @endcond
 */

package org.lightjason.agentspeak.action.builtin.math.blas;

import cern.colt.function.tdouble.DoubleDoubleFunction;
import cern.colt.matrix.tdouble.DoubleMatrix1D;
import cern.colt.matrix.tdouble.DoubleMatrix2D;
import cern.jet.math.tdouble.DoubleFunctions;
import com.codepoetics.protonpack.StreamUtils;
import org.lightjason.agentspeak.action.builtin.IBuiltinAction;
import org.lightjason.agentspeak.language.CCommon;
import org.lightjason.agentspeak.language.CRawTerm;
import org.lightjason.agentspeak.language.ITerm;
import org.lightjason.agentspeak.language.execution.IContext;
import org.lightjason.agentspeak.language.fuzzy.CFuzzyValue;
import org.lightjason.agentspeak.language.fuzzy.IFuzzyValue;

import javax.annotation.Nonnegative;
import javax.annotation.Nonnull;
import java.util.List;
import java.util.function.BiFunction;


public final class CElementWise extends IBuiltinAction
{
    private static final long serialVersionUID = -2655464156364927632L;

    public CElementWise()
    {
        super( 3 );
    }

    @Nonnegative
    @Override
    public final int minimalArgumentNumber()
    {
        return 3;
    }

    @Nonnull
    @Override
    public final IFuzzyValue<Boolean> execute( final boolean p_parallel, @Nonnull final IContext p_context,
                                               @Nonnull final List<ITerm> p_argument, @Nonnull final List<ITerm> p_return )
    {
        return CFuzzyValue.from(
                StreamUtils.windowed(
                CCommon.flatten( p_argument ),
            3,
                3
            ).allMatch( i ->
            {

                switch ( i.get( 1 ).<String>raw().trim() )
                {
                    case "+" :
                        return CElementWise.apply( i.get( 0 ), i.get( 2 ), DoubleFunctions.plus, ( n, m ) -> n + m, p_return );

                    case "|+|" :
                        return CElementWise.apply( i.get( 0 ), i.get( 2 ),  DoubleFunctions.plusAbs, ( n, m ) -> Math.abs( n + m ), p_return );

                    case "-" :
                        return CElementWise.apply( i.get( 0 ), i.get( 2 ), DoubleFunctions.minus, ( n, m ) -> n - m, p_return );

                    case "*" :
                        return CElementWise.apply( i.get( 0 ), i.get( 2 ), DoubleFunctions.mult, ( n, m ) -> n * m, p_return );

                    case "/" :
                        return CElementWise.apply( i.get( 0 ), i.get( 2 ), DoubleFunctions.div, ( n, m ) -> n / m, p_return );

                    default:
                        return false;
                }

            } )
        );
    }


    private static boolean apply( final ITerm p_left, final ITerm p_right,
                                  final DoubleDoubleFunction p_matrixfunction, final BiFunction<Double, Double, Double> p_scalarfunction,
                                  final List<ITerm> p_return )
    {
        // operation for matrix
        if ( CCommon.rawvalueAssignableTo( p_left, DoubleMatrix2D.class ) )
        {
            final DoubleMatrix2D l_assign = p_left.<DoubleMatrix2D>raw().copy();

            if ( CCommon.rawvalueAssignableTo( p_right, DoubleMatrix2D.class ) )
            {
                l_assign.assign( p_right.<DoubleMatrix2D>raw(), p_matrixfunction );
                p_return.add( CRawTerm.from( l_assign ) );
                return true;
            }

            if ( CCommon.rawvalueAssignableTo( p_right, Number.class ) )
            {
                l_assign.assign( i -> p_scalarfunction.apply( i, p_right.<Number>raw().doubleValue() ) );
                p_return.add( CRawTerm.from( l_assign ) );
                return true;
            }
        }

        // operation for vector
        if ( CCommon.rawvalueAssignableTo( p_left, DoubleMatrix1D.class ) )
        {
            final DoubleMatrix1D l_assign = p_left.<DoubleMatrix1D>raw().copy();

            if ( CCommon.rawvalueAssignableTo( p_right, DoubleMatrix1D.class ) )
            {
                l_assign.assign( p_right.<DoubleMatrix1D>raw(), p_matrixfunction );
                p_return.add( CRawTerm.from( l_assign ) );
                return true;
            }

            if ( CCommon.rawvalueAssignableTo( p_right, Number.class ) )
            {
                l_assign.assign( i -> p_scalarfunction.apply( i, p_right.<Number>raw().doubleValue() ) );
                p_return.add( CRawTerm.from( l_assign ) );
                return true;
            }
        }

        return false;
    }



}