Application: gvSIG desktop » gvSIG tools: ToolBox

package es.unex.sextante.tridimensional.grid3DFromFunction;

import org.nfunk.jep.JEP;

import es.unex.sextante.additionalInfo.AdditionalInfoNumericalValue;

import es.unex.sextante.core.GeoAlgorithm;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.I3DRasterLayer;

import es.unex.sextante.dataObjects.IRasterLayer;

import es.unex.sextante.exceptions.GeoAlgorithmExecutionException;

import es.unex.sextante.exceptions.RepeatedParameterNameException;

public class Grid3DFromFunctionAlgorithm

         extends

            GeoAlgorithm {

   public static final String MINX    = "MINX";

   public static final String MINY    = "MINY";

   public static final String MINZ    = "MINZ";

   public static final String MAXX    = "MAXX";

   public static final String MAXY    = "MAXY";

   public static final String MAXZ    = "MAXZ";

   public static final String FORMULA = "FORMULA";

   public static final String RESULT  = "RESULT";

   @Override

   public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

      int x, y, z;

      int iNX, iNY, iNZ;

      double dx, dy, dz;

      double dMinX, dMinY, dMinZ, dMaxX, dMaxY, dMaxZ;

      double dRangeX, dRangeY;

      final double dRangeZ;

      double dValue;

      String sFormula;

      final JEP jep = new JEP();

      I3DRasterLayer result;

      jep.addStandardConstants();

      jep.addStandardFunctions();

      jep.addVariable("x", 0);

      jep.addVariable("y", 0);

      jep.addVariable("z", 0);

      dMinX = m_Parameters.getParameterValueAsInt(MINX);

      dMinY = m_Parameters.getParameterValueAsInt(MINY);

      dMinZ = m_Parameters.getParameterValueAsInt(MINY);

      dMaxX = m_Parameters.getParameterValueAsInt(MAXX);

      dMaxY = m_Parameters.getParameterValueAsInt(MAXY);

      dMaxZ = m_Parameters.getParameterValueAsInt(MAXZ);

      sFormula = m_Parameters.getParameterValueAsString(FORMULA).toLowerCase();

      dRangeX = dMaxX - dMinX;

      dRangeY = dMaxY - dMinY;

      dRangeZ = dMaxZ - dMinZ;

      result = getNew3DRasterLayer(RESULT, Sextante.getText("3D_Layer_from_function"), IRasterLayer.RASTER_DATA_TYPE_FLOAT);

      iNX = result.getLayerExtent().getNX();

      iNY = result.getLayerExtent().getNY();

      iNZ = result.getLayerExtent().getNZ();

      jep.parseExpression(sFormula);

      if (!jep.hasError()) {

         for (z = 0; (z < iNZ) && setProgress(z, iNZ); z++) {

            dz = dMinZ + dRangeZ * ((double) z / (double) iNZ);

            jep.addVariable("z", dz);

            for (y = 0; (y < iNY) && !m_Task.isCanceled(); y++) {

               dy = dMinY + dRangeY * ((double) y / (double) iNY);

               jep.addVariable("y", dy);

               for (x = 0; x < iNX; x++) {

                  dx = dMinX + dRangeX * ((double) x / (double) iNX);

                  jep.addVariable("x", dx);

                  dValue = jep.getValue();

                  result.setCellValue(x, y, z, dValue);

               }

            }

         }

      }

      else {

         throw new GeoAlgorithmExecutionException(Sextante.getText("Wrong_formula"));

      }

      return !m_Task.isCanceled();

   }

   @Override

   public void defineCharacteristics() {

      setName(Sextante.getText("3D_Layer_from_function"));

      setGroup(Sextante.getText("3D"));

      setUserCanDefineAnalysisExtent(true);

      try {

         m_Parameters.addNumericalValue(MINX, "X min", -1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addNumericalValue(MAXX, "X Max", 1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addNumericalValue(MINY, "Y Min", -1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addNumericalValue(MAXY, "Y Max", 1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addNumericalValue(MINZ, "Z Min", -1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addNumericalValue(MAXZ, "Z Max", 1, AdditionalInfoNumericalValue.NUMERICAL_VALUE_DOUBLE);

         m_Parameters.addString(FORMULA, Sextante.getText("Formula"));

         addOutput3DRasterLayer(RESULT, Sextante.getText("3D_Layer_from_function"));

      }

      catch (final RepeatedParameterNameException e) {

         Sextante.addErrorToLog(e);

      }

   }

   @Override

   public boolean is3D() {

      return true;

   }

   @Override

   public boolean isActive() {

      return false;

   }

}