/**
* Copyright 2010 JBoss Inc
*
* Licensed 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.drools.base.evaluators;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import org.drools.base.BaseEvaluator;
import org.drools.base.ValueType;
import org.drools.common.InternalWorkingMemory;
import org.drools.rule.VariableRestriction.ObjectVariableContextEntry;
import org.drools.rule.VariableRestriction.VariableContextEntry;
import org.drools.spi.Evaluator;
import org.drools.spi.FieldValue;
import org.drools.spi.InternalReadAccessor;
import org.mvel2.util.Soundex;
/**
* This class defines the soundslike evaluator
*
* @author etirelli
*/
public class SoundslikeEvaluatorsDefinition implements EvaluatorDefinition {
public static final Operator SOUNDSLIKE = Operator.addOperatorToRegistry( "soundslike",
false );
public static final Operator NOT_SOUNDSLIKE = Operator.addOperatorToRegistry( "soundslike",
true );
private static final String[] SUPPORTED_IDS = { SOUNDSLIKE.getOperatorString() };
private EvaluatorCache evaluators = new EvaluatorCache() {
private static final long serialVersionUID = 510l;
{
addEvaluator( ValueType.STRING_TYPE, SOUNDSLIKE, StringSoundsLikeEvaluator.INSTANCE );
addEvaluator( ValueType.STRING_TYPE, NOT_SOUNDSLIKE, StringNotSoundsLikeEvaluator.INSTANCE );
}
};
public void readExternal(ObjectInput in) throws IOException, ClassNotFoundException {
evaluators = (EvaluatorCache)in.readObject();
}
public void writeExternal(ObjectOutput out) throws IOException {
out.writeObject(evaluators);
}
/**
* @inheridDoc
*/
public Evaluator getEvaluator(ValueType type,
Operator operator) {
return this.evaluators.getEvaluator( type,
operator );
}
/**
* @inheridDoc
*/
public Evaluator getEvaluator(ValueType type,
Operator operator,
String parameterText) {
return this.evaluators.getEvaluator( type,
operator );
}
public Evaluator getEvaluator(final ValueType type,
final String operatorId,
final boolean isNegated,
final String parameterText) {
return this.getEvaluator( type,
operatorId,
isNegated,
parameterText,
Target.FACT,
Target.FACT );
}
/**
* @inheritDoc
*/
public Evaluator getEvaluator(final ValueType type,
final String operatorId,
final boolean isNegated,
final String parameterText,
final Target left,
final Target right ) {
return this.evaluators.getEvaluator( type,
Operator.determineOperator( operatorId,
isNegated ) );
}
public String[] getEvaluatorIds() {
return SUPPORTED_IDS;
}
public boolean isNegatable() {
return true;
}
public Target getTarget() {
return Target.FACT;
}
public boolean supportsType(ValueType type) {
return this.evaluators.supportsType( type );
}
/* *********************************************************
* Evaluator Implementations
* *********************************************************
*/
public static class StringSoundsLikeEvaluator extends BaseEvaluator {
private static final long serialVersionUID = 510l;
public final static Evaluator INSTANCE = new StringSoundsLikeEvaluator();
public StringSoundsLikeEvaluator() {
super( ValueType.STRING_TYPE,
SOUNDSLIKE );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor,
final Object object1, final FieldValue object2) {
final String value1 = (String) extractor.getValue( workingMemory, object1 );
final String value2 = (String) object2.getValue();
if ( value1 == null ) {
return false;
}
return Soundex.soundex( value1 ).equals( Soundex.soundex(value2) );
}
public boolean evaluateCachedRight(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final Object left) {
final String value = (String) ((ObjectVariableContextEntry) context).right;
if ( value == null ) {
return false;
}
return Soundex.soundex( value ).equals( Soundex.soundex((String) context.declaration.getExtractor().getValue( workingMemory, left )) );
}
public boolean evaluateCachedLeft(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final Object right) {
final String value = (String) context.extractor.getValue( workingMemory, right );
if ( value == null ) {
return false;
}
return Soundex.soundex(value).equals( Soundex.soundex((String) ((ObjectVariableContextEntry) context).left ));
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor1,
final Object object1,
final InternalReadAccessor extractor2, final Object object2) {
final Object value1 = extractor1.getValue( workingMemory, object1 );
final Object value2 = extractor2.getValue( workingMemory, object2 );
if ( value1 == null ) {
return false;
}
return Soundex.soundex( ((String) value1)).equals( Soundex.soundex( (String) value2 ));
}
public String toString() {
return "Strings sound alike";
}
}
public static class StringNotSoundsLikeEvaluator extends BaseEvaluator {
private static final long serialVersionUID = 510l;
public final static Evaluator INSTANCE = new StringNotSoundsLikeEvaluator();
public StringNotSoundsLikeEvaluator() {
super( ValueType.STRING_TYPE,
NOT_SOUNDSLIKE );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor,
final Object object1, final FieldValue object2) {
final String value1 = (String) extractor.getValue( workingMemory, object1 );
final String value2 = (String) object2.getValue();
if ( value1 == null ) {
return false;
}
return ! Soundex.soundex( value1 ).equals( Soundex.soundex(value2) );
}
public boolean evaluateCachedRight(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final Object left) {
final String value = (String) ((ObjectVariableContextEntry) context).right;
if ( value == null ) {
return false;
}
return ! Soundex.soundex( value ).equals( Soundex.soundex((String) context.declaration.getExtractor().getValue( workingMemory, left )) );
}
public boolean evaluateCachedLeft(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final Object right) {
final String value = (String) context.extractor.getValue( workingMemory, right );
if ( value == null ) {
return false;
}
return ! Soundex.soundex(value).equals( Soundex.soundex((String) ((ObjectVariableContextEntry) context).left ));
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor1,
final Object object1,
final InternalReadAccessor extractor2, final Object object2) {
final Object value1 = extractor1.getValue( workingMemory, object1 );
final Object value2 = extractor2.getValue( workingMemory, object2 );
if ( value1 == null ) {
return false;
}
return ! Soundex.soundex( ((String) value1)).equals( Soundex.soundex( (String) value2 ));
}
public String toString() {
return "Strings not sound alike";
}
}
}