felix@1896: package de.intevation.flys.artifacts;
felix@1896: 
felix@1896: import java.util.ArrayList;
felix@1896: import java.util.List;
felix@1896: 
felix@1896: import org.apache.log4j.Logger;
felix@1896: 
felix@1896: import org.w3c.dom.Document;
felix@1896: 
felix@1896: import de.intevation.artifactdatabase.state.Facet;
felix@1896: import de.intevation.artifactdatabase.state.DefaultOutput;
felix@1896: import de.intevation.artifactdatabase.state.State;
felix@1896: 
felix@1896: import de.intevation.artifacts.Artifact;
felix@1896: import de.intevation.artifacts.ArtifactFactory;
felix@1896: import de.intevation.artifacts.ArtifactNamespaceContext;
felix@1896: import de.intevation.artifacts.CallMeta;
felix@1896: 
felix@1896: import de.intevation.flys.artifacts.model.Calculation;
felix@1896: import de.intevation.flys.artifacts.model.FacetTypes;
felix@1896: import de.intevation.flys.artifacts.model.WQKms;
felix@1896: import de.intevation.flys.artifacts.model.WQFacet;
felix@1896: import de.intevation.flys.artifacts.model.WKmsFactory;
felix@1896: import de.intevation.flys.artifacts.model.WQKmsFactory;
felix@1896: import de.intevation.flys.artifacts.model.WstValueTable;
felix@1896: import de.intevation.flys.artifacts.model.WstValueTableFactory;
felix@1896: 
felix@1896: import de.intevation.flys.artifacts.states.StaticState;
felix@1896: import de.intevation.flys.artifacts.resources.Resources;
felix@1896: 
felix@1896: import de.intevation.flys.utils.FLYSUtils;
felix@1896: import de.intevation.artifacts.common.utils.XMLUtils;
felix@1896: 
felix@1896: /**
felix@1896:  * Artifact to access additional "waterlevel/discharge"-type of data, like
felix@1896:  * fixation measurements, but doing so with costy interpolation.
felix@1896:  *
felix@1896:  * This artifact neglects (Static)FLYSArtifacts capabilities of interaction
felix@1896:  * with the StateEngine by overriding the getState*-methods.
felix@1896:  */
felix@1896: public class WQKmsInterpolArtifact
felix@1896: extends      StaticFLYSArtifact
felix@1896: implements   FacetTypes
felix@1896: {
felix@1896:     /** The logger for this class. */
felix@1896:     private static Logger logger =
felix@1896:         Logger.getLogger(WQKmsInterpolArtifact.class);
felix@1896: 
felix@1896:     /** XPath to access initial parameter. */
felix@1896:     public static final String XPATH_DATA =
felix@1896:         "/art:action/art:ids/@value";
felix@1896: 
felix@1896:     public static final String STATIC_STATE_NAME =
felix@1896:         "state.additional_wqkms.interpol.static";
felix@1896: 
felix@1896:     /** One and only state to be in. */
felix@1896:     protected transient State state = null;
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Trivial Constructor.
felix@1896:      */
felix@1896:     public WQKmsInterpolArtifact() {
felix@1896:         logger.debug("WQKmsInterpolArtifact.WQKmsInterpolArtifact");
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Gets called from factory, to set things up.
felix@1896:      */
felix@1896:     @Override
felix@1896:     public void setup(
felix@1896:         String          identifier,
felix@1896:         ArtifactFactory factory,
felix@1896:         Object          context,
felix@1896:         CallMeta        callMeta,
felix@1896:         Document        data)
felix@1896:     {
felix@1896:         logger.debug("WQKmsInterpolArtifact.setup");
felix@1896: 
felix@1896:         state = new StaticState(STATIC_STATE_NAME);
felix@1896: 
felix@1896:         List<Facet> fs = new ArrayList<Facet>();
felix@1896:         String code = XMLUtils.xpathString(
felix@1896:             data, XPATH_DATA, ArtifactNamespaceContext.INSTANCE);
felix@1896: 
felix@1896:         // TODO Go for JSON, one day.
felix@1896:         //ex.: flood_protection-wstv-114-12
felix@1896:         if (code != null) {
felix@1896:             String [] parts = code.split("-");
felix@1896: 
felix@1896:             if (parts.length >= 4) {
felix@1896:                 int wst = Integer.valueOf(parts[3]);
felix@1910:                 int col = -1;
felix@1910:                 String colpos = parts[2];
felix@1910:                 // Are we interested in a single column or in all columns?
felix@1910:                 if (colpos.equals("A")) {
felix@1910:                     ; // Take all.
felix@1910:                 }
felix@1910:                 else {
felix@1910:                     col = Integer.valueOf(colpos);
felix@1910:                     addStringData("col_pos", parts[2]);
felix@1910:                 }
felix@1896:                 addStringData("wst_id",  parts[3]);
felix@1910:                 String wkmsName = (col > 0)
felix@1910:                                 ? WKmsFactory.getWKmsName(col, wst)
felix@1910:                                 : WKmsFactory.getWKmsName(wst);
felix@1920:                 String name;
felix@1920:                 if (parts[0].startsWith("height")){
felix@1920:                     name = STATIC_WQ_ANNOTATIONS;
felix@1920:                 }
felix@1920:                 else if (parts[0].startsWith("flood")) {
felix@1920:                     name = STATIC_WKMS_INTERPOL;
felix@1920:                 }
felix@1920:                 else {
felix@1920:                     name = STATIC_WQ;
felix@1920:                 }
felix@1914: 
felix@1914:                 Facet facet = new WQFacet(name,
felix@1896:                     Resources.getMsg(
felix@1896:                         callMeta,
felix@1896:                         wkmsName,
felix@1896:                         wkmsName));
felix@1896:                 fs.add(facet);
felix@1896:                 facets.put(state.getID(), fs);
felix@1896:             }
felix@1896:         }
felix@1896: 
felix@1896:         spawnState();
felix@1896:         super.setup(identifier, factory, context, callMeta, data);
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Initialize the static state with output.
felix@1896:      * @return static state
felix@1896:      */
felix@1896:     protected State spawnState() {
felix@1896:         state = new StaticState(STATIC_STATE_NAME);
felix@1896:         List<Facet> fs = facets.get(STATIC_STATE_NAME);
felix@1896:         DefaultOutput output = new DefaultOutput(
felix@1896:             "general",
felix@1896:             "general",
felix@1896:             "image/png",
felix@1896:             fs,
felix@1896:             "chart");
felix@1896: 
felix@1896:         state.getOutputs().add(output);
felix@1896: 
felix@1896:         return state;
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Called via setup.
felix@1896:      *
felix@1896:      * @param artifact The master-artifact.
felix@1896:      */
felix@1896:     @Override
felix@1896:     protected void initialize(
felix@1896:         Artifact artifact,
felix@1896:         Object context,
felix@1896:         CallMeta meta)
felix@1896:     {
felix@1896:         logger.debug("WQKmsInterpolArtifact.initialize");
felix@1896:         WINFOArtifact winfo = (WINFOArtifact) artifact;
felix@1896:         importData(winfo, "river");
felix@1906:         importData(winfo, "ld_locations");
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get a list containing the one and only State.
felix@1896:      * @param  context ignored.
felix@1896:      * @return list with one and only state.
felix@1896:      */
felix@1896:     @Override
felix@1896:     protected List<State> getStates(Object context) {
felix@1896:         ArrayList<State> states = new ArrayList<State>();
felix@1896:         states.add(getState());
felix@1896:         return states;
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get WQ at a given km.
felix@1896:      */
felix@1896:     public double [][] getWQAtKm(double km) {
felix@1910: 
felix@1910:         WstValueTable interpolator = null;
felix@1896:         // Get WstValueTable
felix@1910:         if (getDataAsString("col_pos") != null) {
felix@1910:             interpolator = WstValueTableFactory.getWstColumnTable(
felix@1910:                 getDataAsInt("wst_id"), getDataAsInt("col_pos"));
felix@1910:         }
felix@1910:         else {
felix@1910:             interpolator = WstValueTableFactory.getTable(
felix@1910:                 getDataAsInt("wst_id"));
felix@1910:         }
felix@1896: 
felix@1906:         double [][] vs = interpolator.interpolateWQColumnwise(
felix@1906:             getDataAsDouble("ld_locations"));
felix@1896: 
felix@1896:         for (int x = 0; x < vs[1].length; x++) {
felix@1896:             logger.debug("getWQAtKm: Q/W " + vs[0][x] + " / " + vs[1][x]);
felix@1896:         }
felix@1896: 
felix@1896:         return vs;
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get a DataItem casted to int (0 if fails).
felix@1896:      */
felix@1896:     public int getDataAsInt(String dataName) {
felix@1896:         String val = getDataAsString(dataName);
felix@1896:         try {
felix@1896:             return Integer.valueOf(val);
felix@1896:         }
felix@1896:         catch (NumberFormatException e) {
felix@1896:             logger.warn("Could not get data " + dataName + " as int", e);
felix@1896:             return 0;
felix@1896:         }
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1906:      * Get a DataItem casted to double (0 if fails).
felix@1906:      */
felix@1906:     public double getDataAsDouble(String dataName) {
felix@1906:         String val = getDataAsString(dataName);
felix@1906:         try {
felix@1906:             return Double.valueOf(val);
felix@1906:         }
felix@1906:         catch (NumberFormatException e) {
felix@1906:             logger.warn("Could not get data " + dataName + " as double", e);
felix@1906:             return 0;
felix@1906:         }
felix@1906:     }
felix@1906: 
felix@1906: 
felix@1906:     /**
felix@1896:      * Get the "current" state (there is but one).
felix@1896:      * @param cc ignored.
felix@1896:      * @return the "current" (only possible) state.
felix@1896:      */
felix@1896:     @Override
felix@1896:     public State getCurrentState(Object cc) {
felix@1896:         return getState();
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get the only possible state.
felix@1896:      * @return the state.
felix@1896:      */
felix@1896:     protected State getState() {
felix@1896:         return getState(null, null);
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get the state.
felix@1896:      * @param context ignored.
felix@1896:      * @param stateID ignored.
felix@1896:      * @return the state.
felix@1896:      */
felix@1896:     @Override
felix@1896:     protected State getState(Object context, String stateID) {
felix@1896:         return (state != null)
felix@1896:             ? state
felix@1896:             : spawnState();
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Get WQKms from factory.
felix@1896:      * @param TODO idx param is not needed
felix@1896:      * @return WQKms according to parameterization (can be null);
felix@1896:      */
felix@1896:     public WQKms getWQKms(int idx) {
felix@1896:         logger.debug("WQKmsInterpolArtifact.getWQKms");
felix@1896:         logger.warn("Stub, getWQKms not yet implemented.");
felix@1896: 
felix@1896:         return WQKmsFactory.getWQKms(
felix@1896:             Integer.valueOf(getDataAsString("col_pos")),
felix@1896:             Integer.valueOf(getDataAsString("wst_id")));
felix@1896:     }
felix@1896: 
felix@1896: 
felix@1896:     /**
felix@1896:      * Determines Facets initial disposition regarding activity (think of
felix@1896:      * selection in Client ThemeList GUI). This will be checked one time
felix@1896:      * when the facet enters a collections describe document.
felix@1896:      *
felix@1896:      * @param facetName name of the facet.
felix@1896:      * @param index     index of the facet.
felix@1896:      *
felix@1896:      * @return Always 0. Static Data will enter plots inactive.
felix@1896:      */
felix@1896:     @Override
felix@1896:     public int getInitialFacetActivity(
felix@1896:         String outputName,
felix@1896:         String facetName,
felix@1896:         int index)
felix@1896:     {
felix@1896:         return 0;
felix@1896:     }
felix@1896: }
felix@1896: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :