com.joliciel.talismane.machineLearning.features.IfThenElseGenericFeature Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of talismane-machine-learning Show documentation
There is a newer version: 6.1.8
Show newest version
///////////////////////////////////////////////////////////////////////////////
//Copyright (C) 2014 Joliciel Informatique
//
//This file is part of Talismane.
//
//Talismane is free software: you can redistribute it and/or modify
//it under the terms of the GNU Affero General Public License as published by
//the Free Software Foundation, either version 3 of the License, or
//(at your option) any later version.
//
//Talismane 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 Affero General Public License for more details.
//
//You should have received a copy of the GNU Affero General Public License
//along with Talismane.  If not, see .
//////////////////////////////////////////////////////////////////////////////
package com.joliciel.talismane.machineLearning.features;

import com.joliciel.talismane.TalismaneException;

/**
 * Mimics an in-then-else structure - if condition is true return thenFeature
 * result, else return elseFeature result.
 * 
 * @author Assaf Urieli
 *
 */
public class IfThenElseGenericFeature extends AbstractCachableFeature {
  private BooleanFeature condition;
  private Feature thenFeature;
  private Feature elseFeature;

  public IfThenElseGenericFeature(BooleanFeature condition, Feature thenFeature, Feature elseFeature) {
    super();
    this.condition = condition;
    this.thenFeature = thenFeature;
    this.elseFeature = elseFeature;
    this.setName("IfThenElse(" + condition.getName() + "," + thenFeature.getName() + "," + elseFeature.getName() + ")");
  }

  @Override
  protected FeatureResult checkInternal(T context, RuntimeEnvironment env) throws TalismaneException {
    FeatureResult featureResult = null;

    FeatureResult conditionResult = condition.check(context, env);
    if (conditionResult != null) {
      boolean conditionOutcome = conditionResult.getOutcome();
      if (conditionOutcome) {
        FeatureResult thenFeatureResult = thenFeature.check(context, env);
        if (thenFeatureResult != null) {
          Y result = thenFeatureResult.getOutcome();
          featureResult = this.generateResult(result);
        }
      } else {
        FeatureResult elseFeatureResult = elseFeature.check(context, env);
        if (elseFeatureResult != null) {
          Y result = elseFeatureResult.getOutcome();
          featureResult = this.generateResult(result);
        }
      }
    }

    return featureResult;

  }

  public BooleanFeature getCondition() {
    return condition;
  }

  public Feature getThenFeature() {
    return thenFeature;
  }

  public Feature getElseFeature() {
    return elseFeature;
  }

  public void setCondition(BooleanFeature condition) {
    this.condition = condition;
  }

  public void setThenFeature(Feature thenFeature) {
    this.thenFeature = thenFeature;
  }

  public void setElseFeature(Feature elseFeature) {
    this.elseFeature = elseFeature;
  }

  @SuppressWarnings("rawtypes")
  @Override
  public Class getFeatureType() {
    return thenFeature.getFeatureType();
  }

}