/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.sysml.hops.codegen.cplan;
import java.util.Arrays;
import org.apache.commons.lang.StringUtils;
import org.apache.sysml.parser.Expression.DataType;
public class CNodeUnary extends CNode
{
public enum UnaryType {
LOOKUP_R, LOOKUP_C, LOOKUP_RC, LOOKUP0, //codegen specific
ROW_SUMS, ROW_MINS, ROW_MAXS, //codegen specific
VECT_EXP, VECT_POW2, VECT_MULT2, VECT_SQRT, VECT_LOG,
VECT_ABS, VECT_ROUND, VECT_CEIL, VECT_FLOOR, VECT_SIGN,
EXP, POW2, MULT2, SQRT, LOG, LOG_NZ,
ABS, ROUND, CEIL, FLOOR, SIGN,
SIN, COS, TAN, ASIN, ACOS, ATAN,
SELP, SPROP, SIGMOID;
public static boolean contains(String value) {
for( UnaryType ut : values() )
if( ut.name().equals(value) )
return true;
return false;
}
public String getTemplate(boolean sparse) {
switch( this ) {
case ROW_SUMS:
case ROW_MINS:
case ROW_MAXS: {
String vectName = StringUtils.capitalize(this.toString().substring(4,7).toLowerCase());
return sparse ? " double %TMP% = LibSpoofPrimitives.vect"+vectName+"(%IN1v%, %IN1i%, %POS1%, %LEN%);\n":
" double %TMP% = LibSpoofPrimitives.vect"+vectName+"(%IN1%, %POS1%, %LEN%);\n";
}
case VECT_EXP:
case VECT_POW2:
case VECT_MULT2:
case VECT_SQRT:
case VECT_LOG:
case VECT_ABS:
case VECT_ROUND:
case VECT_CEIL:
case VECT_FLOOR:
case VECT_SIGN: {
String vectName = getVectorPrimitiveName();
return sparse ? " double[] %TMP% = LibSpoofPrimitives.vect"+vectName+"Write(%IN1v%, %IN1i%, %POS1%, %LEN%);\n" :
" double[] %TMP% = LibSpoofPrimitives.vect"+vectName+"Write(%IN1%, %POS1%, %LEN%);\n";
}
case EXP:
return " double %TMP% = FastMath.exp(%IN1%);\n";
case LOOKUP_R:
return " double %TMP% = getValue(%IN1%, rowIndex);\n";
case LOOKUP_C:
return " double %TMP% = getValue(%IN1%, colIndex);\n";
case LOOKUP_RC:
return " double %TMP% = getValue(%IN1%, rowIndex*n+colIndex);\n";
case LOOKUP0:
return " double %TMP% = %IN1%[0];\n" ;
case POW2:
return " double %TMP% = %IN1% * %IN1%;\n" ;
case MULT2:
return " double %TMP% = %IN1% + %IN1%;\n" ;
case ABS:
return " double %TMP% = Math.abs(%IN1%);\n";
case SIN:
return " double %TMP% = FastMath.sin(%IN1%);\n";
case COS:
return " double %TMP% = FastMath.cos(%IN1%);\n";
case TAN:
return " double %TMP% = FastMath.tan(%IN1%);\n";
case ASIN:
return " double %TMP% = FastMath.asin(%IN1%);\n";
case ACOS:
return " double %TMP% = FastMath.acos(%IN1%);\n";
case ATAN:
return " double %TMP% = Math.atan(%IN1%);\n";
case SIGN:
return " double %TMP% = FastMath.signum(%IN1%);\n";
case SQRT:
return " double %TMP% = Math.sqrt(%IN1%);\n";
case LOG:
return " double %TMP% = FastMath.log(%IN1%);\n";
case ROUND:
return " double %TMP% = Math.round(%IN1%);\n";
case CEIL:
return " double %TMP% = FastMath.ceil(%IN1%);\n";
case FLOOR:
return " double %TMP% = FastMath.floor(%IN1%);\n";
case SELP:
return " double %TMP% = (%IN1%>0) ? %IN1% : 0;\n";
case SPROP:
return " double %TMP% = %IN1% * (1 - %IN1%);\n";
case SIGMOID:
return " double %TMP% = 1 / (1 + FastMath.exp(-%IN1%));\n";
case LOG_NZ:
return " double %TMP% = (%IN1%==0) ? 0 : FastMath.log(%IN1%);\n";
default:
throw new RuntimeException("Invalid unary type: "+this.toString());
}
}
public boolean isVectorScalarPrimitive() {
return this == VECT_EXP || this == VECT_POW2
|| this == VECT_MULT2 || this == VECT_SQRT
|| this == VECT_LOG || this == VECT_ABS
|| this == VECT_ROUND || this == VECT_CEIL
|| this == VECT_FLOOR || this == VECT_SIGN;
}
public UnaryType getVectorAddPrimitive() {
return UnaryType.valueOf("VECT_"+getVectorPrimitiveName().toUpperCase()+"_ADD");
}
public String getVectorPrimitiveName() {
String [] tmp = this.name().split("_");
return StringUtils.capitalize(tmp[1].toLowerCase());
}
}
private UnaryType _type;
public CNodeUnary( CNode in1, UnaryType type ) {
_inputs.add(in1);
_type = type;
setOutputDims();
}
public UnaryType getType() {
return _type;
}
public void setType(UnaryType type) {
_type = type;
}
@Override
public String codegen(boolean sparse) {
if( _generated )
return "";
StringBuilder sb = new StringBuilder();
//generate children
sb.append(_inputs.get(0).codegen(sparse));
//generate unary operation
boolean lsparse = sparse && (_inputs.get(0) instanceof CNodeData);
String var = createVarname();
String tmp = _type.getTemplate(lsparse);
tmp = tmp.replaceAll("%TMP%", var);
String varj = _inputs.get(0).getVarname();
//replace sparse and dense inputs
tmp = tmp.replaceAll("%IN1v%", varj+"vals");
tmp = tmp.replaceAll("%IN1i%", varj+"ix");
tmp = tmp.replaceAll("%IN1%", varj );
//replace start position of main input
String spos = (!varj.startsWith("b")
&& _inputs.get(0) instanceof CNodeData
&& _inputs.get(0).getDataType().isMatrix()) ? varj+"i" : "0";
tmp = tmp.replaceAll("%POS1%", spos);
tmp = tmp.replaceAll("%POS2%", spos);
sb.append(tmp);
//mark as generated
_generated = true;
return sb.toString();
}
@Override
public String toString() {
switch(_type) {
case ROW_SUMS: return "u(R+)";
case ROW_MINS: return "u(Rmin)";
case ROW_MAXS: return "u(Rmax)";
case VECT_EXP:
case VECT_POW2:
case VECT_MULT2:
case VECT_SQRT:
case VECT_LOG:
case VECT_ABS:
case VECT_ROUND:
case VECT_CEIL:
case VECT_FLOOR:
case VECT_SIGN: return "u(v"+_type.name().toLowerCase()+")";
case LOOKUP_R: return "u(ixr)";
case LOOKUP_C: return "u(ixc)";
case LOOKUP_RC: return "u(ixrc)";
case LOOKUP0: return "u(ix0)";
case POW2: return "^2";
default: return "u("+_type.name().toLowerCase()+")";
}
}
@Override
public void setOutputDims() {
switch(_type) {
case VECT_EXP:
case VECT_POW2:
case VECT_MULT2:
case VECT_SQRT:
case VECT_LOG:
case VECT_ABS:
case VECT_ROUND:
case VECT_CEIL:
case VECT_FLOOR:
case VECT_SIGN:
_rows = _inputs.get(0)._rows;
_cols = _inputs.get(0)._cols;
_dataType= DataType.MATRIX;
break;
case ROW_SUMS:
case ROW_MINS:
case ROW_MAXS:
case EXP:
case LOOKUP_R:
case LOOKUP_C:
case LOOKUP_RC:
case LOOKUP0:
case POW2:
case MULT2:
case ABS:
case SIN:
case COS:
case TAN:
case ASIN:
case ACOS:
case ATAN:
case SIGN:
case SQRT:
case LOG:
case ROUND:
case CEIL:
case FLOOR:
case SELP:
case SPROP:
case SIGMOID:
case LOG_NZ:
_rows = 0;
_cols = 0;
_dataType= DataType.SCALAR;
break;
default:
throw new RuntimeException("Operation " + _type.toString() + " has no "
+ "output dimensions, dimensions needs to be specified for the CNode " );
}
}
@Override
public int hashCode() {
if( _hash == 0 ) {
int h1 = super.hashCode();
int h2 = _type.hashCode();
_hash = Arrays.hashCode(new int[]{h1,h2});
}
return _hash;
}
@Override
public boolean equals(Object o) {
if( !(o instanceof CNodeUnary) )
return false;
CNodeUnary that = (CNodeUnary) o;
return super.equals(that)
&& _type == that._type;
}
}