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;
sascha@3556: import de.intevation.artifactdatabase.state.FacetActivity;
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.CallMeta;
felix@1896: 
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: 
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@3634:     /** State name. */
felix@1896:     public static final String STATIC_STATE_NAME =
felix@1896:         "state.additional_wqkms.interpol.static";
felix@1896: 
felix@3634:     /** Artifact name. */
sascha@3556:     private static final String NAME = "staticwqkmsinterpol";
sascha@3556: 
sascha@3556:     static {
sascha@3556:         // TODO: Move to configuration.
sascha@3556:         FacetActivity.Registry.getInstance()
sascha@3556:             .register(NAME, FacetActivity.INACTIVE);
sascha@3556:     }
sascha@3556: 
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@3634:     /** Return fixed artifact (types) name. */
sascha@3556:     @Override
sascha@3556:     public String getName() {
sascha@3556:         return NAME;
sascha@3556:     }
sascha@3556: 
sascha@3556: 
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@2741:         String code = getDatacageIDValue(data);
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@3634:             logger.debug("WQKmsInterpolArtifact.setup: code " + code);
felix@3634: 
felix@1896:             if (parts.length >= 4) {
sascha@3405:                 int wst = Integer.parseInt(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 {
sascha@3405:                     col = Integer.parseInt(colpos);
felix@1910:                     addStringData("col_pos", parts[2]);
felix@1910:                 }
felix@1896:                 addStringData("wst_id",  parts[3]);
felix@3634:                 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@3585:                 Facet wQFacet = new WQFacet(name,
felix@1896:                     Resources.getMsg(
felix@1896:                         callMeta,
felix@1896:                         wkmsName,
felix@1896:                         wkmsName));
felix@3585:                 fs.add(wQFacet);
felix@1896:                 facets.put(state.getID(), fs);
felix@1896:             }
felix@1896:         }
felix@3634:         else {
felix@3634:             logger.warn("WQKmsInterpolArtifact: no code");
felix@3634:         }
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");
raimund@2132:         FLYSArtifact winfo = (FLYSArtifact) 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@3585:      * @param currentKm the requested km. If NULL, ld_location data
felix@3585:      *        will be used.
felix@1896:      */
felix@3585:     public double [][] getWQAtKm(Double currentKm) {
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@3585:         Double tmp = (currentKm != null)
felix@3585:                      ? currentKm
felix@3585:                      : getDataAsDouble("ld_locations");
felix@3585: 
felix@1906:         double [][] vs = interpolator.interpolateWQColumnwise(
ingo@2701:             tmp != null ? tmp : 0);
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 {
sascha@3405:             return Integer.parseInt(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@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@3270:      * @param idx param is not needed (TODO)
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(
sascha@3405:             Integer.parseInt(getDataAsString("col_pos")),
sascha@3405:             Integer.parseInt(getDataAsString("wst_id")));
felix@1896:     }
felix@1896: }
felix@1896: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :