/*
* Copyright 2010 Red Hat, Inc. and/or its affiliates.
*
* 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.core.base.evaluators;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectOutput;
import org.drools.core.base.BaseEvaluator;
import org.drools.core.base.ValueType;
import org.drools.core.common.InternalFactHandle;
import org.drools.core.common.InternalWorkingMemory;
import org.drools.core.rule.VariableRestriction.ObjectVariableContextEntry;
import org.drools.core.rule.VariableRestriction.VariableContextEntry;
import org.drools.core.spi.Evaluator;
import org.drools.core.spi.FieldValue;
import org.drools.core.spi.InternalReadAccessor;
import org.mvel2.util.Soundex;
/**
* This class defines the soundslike evaluator
*/
public class SoundslikeEvaluatorsDefinition implements EvaluatorDefinition {
protected static final String soundsLikeOp = "soundslike";
public static Operator SOUNDSLIKE;
public static Operator NOT_SOUNDSLIKE;
private static String[] SUPPORTED_IDS;
{ init(); }
static void init() {
if ( SUPPORTED_IDS == null ) {
SOUNDSLIKE = Operator.addOperatorToRegistry( soundsLikeOp, false );
NOT_SOUNDSLIKE = Operator.addOperatorToRegistry( soundsLikeOp, true );
SUPPORTED_IDS = new String[] { soundsLikeOp };
}
}
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 );
addEvaluator( ValueType.OBJECT_TYPE, SOUNDSLIKE, StringSoundsLikeEvaluator.INSTANCE );
addEvaluator( ValueType.OBJECT_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 );
}
private static boolean soundslike(final String value1,
final String value2) {
final String soundex1;
final String soundex2;
if (value1 == null || value2 == null) {
return false;
}
soundex1 = Soundex.soundex(value1);
soundex2 = Soundex.soundex(value2);
if (soundex1 == null) {
return false;
}
return soundex1.equals(soundex2);
}
/* *********************************************************
* Evaluator Implementations
* *********************************************************
*/
public static class StringSoundsLikeEvaluator extends BaseEvaluator {
private static final long serialVersionUID = 510l;
public final static Evaluator INSTANCE = new StringSoundsLikeEvaluator();
{
SoundslikeEvaluatorsDefinition.init();
}
public StringSoundsLikeEvaluator() {
super( ValueType.STRING_TYPE,
SOUNDSLIKE );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor,
final InternalFactHandle handle1, final FieldValue handle2) {
final String value1 = (String) extractor.getValue( workingMemory, handle1.getObject() );
final String value2 = (String) handle2.getValue();
return soundslike(value1,value2);
}
public boolean evaluateCachedRight(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final InternalFactHandle left) {
final String value = (String) ((ObjectVariableContextEntry) context).right;
return soundslike( value, (String) context.declaration.getExtractor().getValue( workingMemory, left.getObject() ) );
}
public boolean evaluateCachedLeft(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final InternalFactHandle rightHandle) {
final String value = (String) context.extractor.getValue( workingMemory, rightHandle.getObject() );
return soundslike(value, (String) ((ObjectVariableContextEntry) context).left );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor1,
final InternalFactHandle handle1,
final InternalReadAccessor extractor2, final InternalFactHandle handle2) {
final Object value1 = extractor1.getValue( workingMemory, handle1.getObject() );
final Object value2 = extractor2.getValue( workingMemory, handle2.getObject() );
return soundslike( (String) value1, (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();
{
SoundslikeEvaluatorsDefinition.init();
}
public StringNotSoundsLikeEvaluator() {
super( ValueType.STRING_TYPE,
NOT_SOUNDSLIKE );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor,
final InternalFactHandle handle1, final FieldValue object2) {
final String value1 = (String) extractor.getValue( workingMemory, handle1.getObject() );
final String value2 = (String) object2.getValue();
return ! soundslike( value1, value2 );
}
public boolean evaluateCachedRight(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final InternalFactHandle left) {
final String value = (String) ((ObjectVariableContextEntry) context).right;
return ! soundslike( value, (String) context.declaration.getExtractor().getValue( workingMemory, left.getObject() ) );
}
public boolean evaluateCachedLeft(InternalWorkingMemory workingMemory,
final VariableContextEntry context, final InternalFactHandle right) {
final String value = (String) context.extractor.getValue( workingMemory, right.getObject() );
return ! soundslike( value, (String) ((ObjectVariableContextEntry) context).left );
}
public boolean evaluate(InternalWorkingMemory workingMemory,
final InternalReadAccessor extractor1,
final InternalFactHandle handl1,
final InternalReadAccessor extractor2, final InternalFactHandle handl2) {
final Object value1 = extractor1.getValue( workingMemory, handl1.getObject() );
final Object value2 = extractor2.getValue( workingMemory, handl2.getObject() );
return ! soundslike( (String) value1, (String) value2 );
}
public String toString() {
return "Strings not sound alike";
}
}
}