ingo@392: package de.intevation.flys.exports;
ingo@392:
ingo@392: import org.apache.log4j.Logger;
ingo@392:
felix@1083: import java.util.ArrayList;
felix@1083: import java.util.List;
felix@1083:
ingo@392: import org.w3c.dom.Document;
ingo@392:
ingo@414: import org.jfree.chart.JFreeChart;
ingo@414: import org.jfree.chart.title.TextTitle;
ingo@923: import org.jfree.data.xy.XYSeries;
felix@1083: import org.jfree.chart.axis.ValueAxis;
felix@1083: import org.jfree.chart.plot.XYPlot;
felix@1083: import org.jfree.ui.TextAnchor;
ingo@414:
ingo@392: import de.intevation.artifacts.Artifact;
ingo@392:
ingo@695: import de.intevation.artifactdatabase.state.Facet;
ingo@695:
ingo@696: import de.intevation.flys.artifacts.FLYSArtifact;
ingo@696: import de.intevation.flys.artifacts.model.FacetTypes;
ingo@393: import de.intevation.flys.artifacts.model.WQKms;
felix@1083: import de.intevation.flys.jfree.StickyAxisAnnotation;
ingo@393:
felix@1083: import de.intevation.flys.model.MainValue;
ingo@392:
ingo@392: /**
ingo@392: * An OutGenerator that generates discharge curves.
ingo@392: *
ingo@392: * @author Ingo Weinzierl
ingo@392: */
ingo@696: public class ComputedDischargeCurveGenerator
ingo@696: extends DischargeCurveGenerator
ingo@696: implements FacetTypes
ingo@696: {
felix@1081: /** The logger used in this generator. */
ingo@392: private static Logger logger =
ingo@392: Logger.getLogger(ComputedDischargeCurveGenerator.class);
ingo@392:
ingo@392:
ingo@408: public static final String I18N_CHART_TITLE =
ingo@408: "chart.computed.discharge.curve.title";
ingo@408:
ingo@414: public static final String I18N_CHART_SUBTITLE =
ingo@414: "chart.computed.discharge.curve.subtitle";
ingo@414:
ingo@408: public static final String I18N_YAXIS_LABEL =
ingo@408: "chart.computed.discharge.curve.yaxis.label";
ingo@408:
ingo@408: public static final String I18N_CHART_TITLE_DEFAULT = "Abflusskurve";
ingo@408: public static final String I18N_YAXIS_LABEL_DEFAULT = "W [NN + m]";
ingo@408:
felix@1083: /** List of MainValues (Annotations in plot). */
felix@1083: protected static List mainValues;
felix@1083:
felix@1083:
felix@1083: /** Trivial Constructor. */
felix@1083: public ComputedDischargeCurveGenerator () {
felix@1083: super();
felix@1083: mainValues = new ArrayList();
felix@1083: }
felix@1083:
ingo@408:
ingo@392: @Override
ingo@400: protected String getChartTitle() {
ingo@408: return msg(I18N_CHART_TITLE, I18N_CHART_TITLE_DEFAULT);
ingo@400: }
ingo@400:
ingo@400:
ingo@400: @Override
ingo@414: protected void addSubtitles(JFreeChart chart) {
ingo@414: double[] dist = getRange();
ingo@414:
ingo@414: Object[] args = new Object[] {
ingo@414: getRiverName(),
ingo@414: dist[0]
ingo@414: };
ingo@414:
ingo@414: String subtitle = msg(I18N_CHART_SUBTITLE, "", args);
ingo@414: chart.addSubtitle(new TextTitle(subtitle));
ingo@414: }
ingo@414:
ingo@414:
ingo@414: @Override
ingo@400: protected String getYAxisLabel() {
ingo@408: return msg(I18N_YAXIS_LABEL, I18N_YAXIS_LABEL_DEFAULT);
ingo@400: }
ingo@400:
ingo@400:
ingo@400: @Override
ingo@695: public void doOut(Artifact artifact, Facet facet, Document attr) {
felix@1081: String name = (facet != null) ? facet.getName() : null;
ingo@393:
ingo@696: logger.debug("ComputedDischargeCurveGenerator.doOut: " + name);
ingo@393:
ingo@696: FLYSArtifact flys = (FLYSArtifact) artifact;
ingo@696: Facet f = flys.getNativeFacet(facet);
ingo@696:
felix@1083: if (name.equals(COMPUTED_DISCHARGE_Q)) {
ingo@924: doQOut((WQKms) f.getData(artifact, context), attr);
ingo@696: }
felix@1083: else if (name.equals(COMPUTED_DISCHARGE_MAINVALUES)) {
felix@1083: doMainValueAnnotations(f.getData(artifact, context), attr);
felix@1083: }
ingo@696: else {
ingo@696: logger.warn("Unknown facet type for computed discharge: " + name);
ingo@696: return;
ingo@696: }
ingo@696: }
ingo@696:
ingo@696:
felix@1083: /**
felix@1083: * Add MainValues as annotations to plot.
felix@1083: */
felix@1083: protected void doMainValueAnnotations(Object o, Document theme) {
felix@1083: this.mainValues = (List) o;
felix@1083: }
felix@1083:
felix@1083:
felix@1083: /** Generate Chart with annotations. */
felix@1083: @Override
felix@1083: public JFreeChart generateChart() {
felix@1083: JFreeChart c = super.generateChart();
felix@1083: XYPlot p = (XYPlot) c.getPlot();
felix@1083: redoAnnotations(p, p.getDomainAxis());
felix@1083: return c;
felix@1083: }
felix@1083:
felix@1083:
felix@1083: /**
felix@1083: * Recalculate some annotation positions and add them to plot.
felix@1083: * Annotations represent MainValues.
felix@1083: * @param plot Plot to add annotations to.
felix@1083: * @param valueAxis ignored.
felix@1083: */
felix@1083: protected void redoAnnotations(XYPlot plot, ValueAxis axis) {
felix@1083: plot.clearAnnotations();
felix@1083: ValueAxis yAxis = plot.getRangeAxis();
felix@1083: float posY = 140.f;
felix@1083: if (yAxis != null) {
felix@1083: posY = (float) yAxis.getRange().getLowerBound();
felix@1083: // Add some (2%) space between Text and axis.
felix@1083: // TODO have all the position logic in StickyAxisAnnotation
felix@1083: // (then merge LongitudinalSectioNGenerator).
felix@1083: posY += 0.02f * (yAxis.getRange().getUpperBound()
felix@1083: - yAxis.getRange().getLowerBound());
felix@1083: }
felix@1083: // Add all MainValues as annotations.
felix@1083: // TODO Implement and handle second facet for annotations on
felix@1083: // vertical axis.
felix@1083: for (MainValue mv: mainValues) {
felix@1083: if (mv.getMainValue().getType().getName() == "W") {
felix@1083: continue;
felix@1083: }
felix@1083: float posX = mv.getValue().floatValue();
felix@1083:
felix@1083: String text = mv.getMainValue().getName();
felix@1083: StickyAxisAnnotation ta = new StickyAxisAnnotation(text, posX, posY);
felix@1083: plot.getRenderer().addAnnotation(ta);
felix@1083: }
felix@1083: }
felix@1083:
felix@1083:
felix@1083: /**
felix@1083: * Add Q-Series to plot.
felix@1083: * @param wqkms actual data
felix@1083: * @param theme theme to use.
felix@1083: */
ingo@924: protected void doQOut(WQKms wqkms, Document theme) {
ingo@393: int size = wqkms.size();
ingo@393:
ingo@393: double[] res = new double[3];
ingo@393:
ingo@924: XYSeries series = new StyledXYSeries(getSeriesName(wqkms), theme);
ingo@393: for (int i = 0; i < size; i++) {
ingo@393: res = wqkms.get(i, res);
ingo@923: series.add(res[1], res[0]);
ingo@393: }
ingo@393:
ingo@923: addFirstAxisSeries(series);
ingo@393: }
ingo@393:
ingo@393:
ingo@456: protected String getSeriesName(WQKms wqkms) {
ingo@456: Object[] args = new Object[] {
ingo@456: getRiverName(),
ingo@456: wqkms.getName()
ingo@456: };
ingo@456:
ingo@456: return msg(
ingo@456: "chart.computed.discharge.curve.curve.label",
ingo@456: "",
ingo@456: args);
ingo@456: }
ingo@392: }
ingo@392: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :