Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.expressionevaluator.suggestions;

import java.math.BigDecimal;

import java.util.HashMap;

import java.util.Map;

import static java.util.concurrent.Executors.callable;

import org.apache.commons.lang.StringUtils;

import org.apache.commons.lang3.mutable.MutableBoolean;

import org.apache.commons.text.similarity.LevenshteinDistance;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.expressionevaluator.ExpressionUtils;

import org.gvsig.expressionevaluator.swing.SuggestionProvider;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.util.Factory;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DivSuggestionProviderFactory implements Factory {

  private FeatureType featureType;

  public DivSuggestionProviderFactory(FeatureStore store) {

    try {

      this.featureType = store.getDefaultFeatureType();

    } catch (DataException ex) {

      this.featureType = null;

    }            

  }

  @Override

  public String getName() {

    return "DivSuggestion";

  }

  @Override

  public Object create(Object... parameters) {

    SuggestionProvider suggestionProvider = new DivSuggestionProvider();

    return suggestionProvider;

  }

  private class DivSuggestionProvider implements SuggestionProvider {

    @Override

    public String getSuggestion(String expression) {

      if ( StringUtils.isBlank(expression)) {

        return null;

      }

      MutableBoolean showSuggestion = new MutableBoolean(false);

      try {

        Code code = ExpressionUtils.compile(expression);

        if (code == null) {

          return null;

        }

        code.accept((Object obj) -> {

          if (obj instanceof Code.Callable) {

            Code.Callable callable = (Code.Callable) obj;

            String name = callable.name();

            if( !StringUtils.equalsIgnoreCase(name, ExpressionBuilder.OPERATOR_DIV)) {

                return;

            }

            boolean isInt1 = false;

            boolean isDec1 = false;

            boolean isInt2 = false;

            boolean isDec2 = false;

            boolean isCast = false;

            Code p1 = callable.parameters().get(0);

            Code p2 = callable.parameters().get(1);

            if( p1 instanceof Code.Constant ) {

                Code.Constant c1 = (Code.Constant)p1;

                Object v1 = c1.value();

                isInt1 = (v1 instanceof Integer || v1 instanceof Long || v1 instanceof Short);

                isDec1 = (v1 instanceof Double || v1 instanceof Float || v1 instanceof BigDecimal);

            } else if( p1 instanceof Code.Identifier ) {

                Code.Identifier i1 = (Code.Identifier)p1;

                FeatureAttributeDescriptor attr = featureType.getAttributeDescriptor(i1.name());

                if( attr != null ) {

                    int type1 = attr.getDataType().getType();

                    isInt1 = (type1 == DataTypes.INT || type1 == DataTypes.LONG );

                    isDec1 = (type1 == DataTypes.DECIMAL || type1 == DataTypes.DOUBLE || type1 == DataTypes.FLOAT );

                }

            } else if( p1 instanceof Code.Callable && StringUtils.equalsIgnoreCase(((Code.Callable)p1).name(), ExpressionBuilder.FUNCTION_CAST) ) {

                isCast=true;

            }

            if( p2 instanceof Code.Constant ) {

                Code.Constant c2 = (Code.Constant)p2;

                Object v2 = c2.value();

                isInt2 = (v2 instanceof Integer || v2 instanceof Long || v2 instanceof Short);

                isDec2 = (v2 instanceof Double || v2 instanceof Float || v2 instanceof BigDecimal);

            } else if( p2 instanceof Code.Identifier ) {

                Code.Identifier i2 = (Code.Identifier)p2;

                FeatureAttributeDescriptor attr = featureType.getAttributeDescriptor(i2.name());

                if( attr != null ) {

                    int type2 = attr.getDataType().getType();

                    isInt2 = (type2 == DataTypes.INT || type2 == DataTypes.LONG );

                    isDec2 = (type2 == DataTypes.DECIMAL || type2 == DataTypes.DOUBLE || type2 == DataTypes.FLOAT );

                }

            } else if( p2 instanceof Code.Callable && StringUtils.equalsIgnoreCase(((Code.Callable)p2).name(), ExpressionBuilder.FUNCTION_CAST) ) {

                isCast=true;

            }

            if( isCast ) {

                return;

            }

            if( isDec1 || isDec2 ) {

                return;

            }

            if( isInt1 || isInt2 || (!isInt1 && !isInt2) ) {

                showSuggestion.setValue(true);

                return;

            }

//            if( (isInt1 && !isInt2) || (!isInt1 && isInt2)) {

//                showSuggestion.setValue(true);

//            }

          }

        });

      } catch (Throwable th) {

      }

      if( showSuggestion.isFalse() ) {

        return null;

      }

      I18nManager i18n = ToolsLocator.getI18nManager();

      StringBuilder suggestion = new StringBuilder();

      suggestion.append(i18n.getTranslation("_Division_of_two_integer_values_returns_an_integer_value_You_can_force_an_operand_to_decimal_to_get_decimal_values"));

      return suggestion.toString();

    }

  }

}