dive4elements/river: flys-artifacts/src/main/java/de/intevation/flys/artifacts/model/fixings/FixCalculation.java annotate

author	Sascha L. Teichmann <sascha.teichmann@intevation.de>
date	Tue, 22 May 2012 17:09:27 +0000
parents
children	c1f2e792704a

rev	line source
2729 a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	1 package de.intevation.flys.artifacts.model.fixings;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	2
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	3 import de.intevation.flys.artifacts.FixationArtifactAccess;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	4
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	5 import de.intevation.flys.artifacts.math.fitting.Function;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	6 import de.intevation.flys.artifacts.math.fitting.FunctionFactory;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	7
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	8 import de.intevation.flys.artifacts.model.Calculation;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	9 import de.intevation.flys.artifacts.model.CalculationResult;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	10 import de.intevation.flys.artifacts.model.FixingsColumn;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	11 import de.intevation.flys.artifacts.model.FixingsColumnFactory;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	12 import de.intevation.flys.artifacts.model.FixingsOverview;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	13 import de.intevation.flys.artifacts.model.FixingsOverviewFactory;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	14 import de.intevation.flys.artifacts.model.Parameters;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	15
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	16 import de.intevation.flys.artifacts.model.FixingsOverview.Fixing;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	17 import de.intevation.flys.artifacts.model.FixingsOverview.IdFilter;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	18
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	19 import de.intevation.flys.utils.DoubleUtil;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	20
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	21 import java.util.ArrayList;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	22 import java.util.List;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	23
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	24 import org.apache.commons.math.MathException;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	25
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	26 import org.apache.commons.math.optimization.fitting.CurveFitter;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	27
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	28 import org.apache.commons.math.optimization.general.LevenbergMarquardtOptimizer;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	29
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	30 import org.apache.log4j.Logger;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	31
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	32 public class FixCalculation
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	33 extends Calculation
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	34 {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	35 private static Logger logger = Logger.getLogger(FixCalculation.class);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	36
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	37 protected String river;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	38 protected double from;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	39 protected double to;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	40 protected double step;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	41 protected boolean preprocessing;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	42 protected String function;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	43 protected int [] events;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	44
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	45 public FixCalculation() {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	46 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	47
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	48 public FixCalculation(FixationArtifactAccess access) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	49
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	50 String river = access.getRiver();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	51 Double from = access.getFrom();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	52 Double to = access.getTo();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	53 Double step = access.getStep();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	54 String function = access.getFunction();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	55 int [] events = access.getEvents();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	56
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	57 if (river == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	58 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	59 addProblem("fix.missing.river");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	60 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	61
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	62 if (from == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	63 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	64 addProblem("fix.missing.from");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	65 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	66
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	67 if (to == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	68 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	69 addProblem("fix.missing.to");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	70 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	71
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	72 if (step == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	73 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	74 addProblem("fix.missing.step");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	75 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	76
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	77 if (function == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	78 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	79 addProblem("fix.missing.function");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	80 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	81
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	82 if (events == null \|\| events.length < 1) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	83 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	84 addProblem("fix.missing.events");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	85 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	86
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	87 if (!hasProblems()) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	88 this.river = river;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	89 this.from = from;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	90 this.to = to;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	91 this.step = step;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	92 this.function = function;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	93 this.events = events;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	94 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	95 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	96
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	97 public CalculationResult calculate() {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	98
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	99 FixingsOverview overview =
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	100 FixingsOverviewFactory.getOverview(river);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	101
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	102 if (overview == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	103 addProblem("fix.no.overview.available");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	104 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	105
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	106 Function func = FunctionFactory.getInstance()
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	107 .getFunction(function);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	108
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	109 if (func == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	110 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	111 addProblem("fix.invalid.function.name");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	112 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	113
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	114 if (hasProblems()) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	115 return new CalculationResult(this);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	116 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	117
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	118 FixingsColumnFactory fcf = FixingsColumnFactory.getInstance();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	119
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	120 List<FixingsColumn> dataColumns =
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	121 new ArrayList<FixingsColumn>(events.length);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	122
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	123 for (int eventId: events) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	124 IdFilter idFilter = new IdFilter(eventId);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	125
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	126 List<Fixing.Column> columns = overview.filter(null, idFilter);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	127 if (columns.isEmpty()) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	128 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	129 addProblem("fix.missing.column", eventId);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	130 continue;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	131 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	132 FixingsColumn dataColumn = fcf.getColumnData(columns.get(0));
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	133 if (dataColumn == null) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	134 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	135 addProblem("fix.cannot.load.data", eventId);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	136 continue;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	137 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	138 dataColumns.add(dataColumn);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	139 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	140
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	141 if (dataColumns.size() < 2) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	142 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	143 addProblem("fix.too.less.data.columns");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	144 return new CalculationResult(this);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	145 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	146
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	147 double [] kms = DoubleUtil.explode(from, to, step);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	148
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	149 double [] ws = new double[dataColumns.size()];
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	150 double [] qs = new double[ws.length];
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	151
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	152 String [] parameterNames = func.getParameterNames();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	153
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	154 Parameters results =
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	155 new Parameters(createColumnNames(parameterNames));
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	156
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	157 boolean invalid = false;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	158
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	159 for (int i = 0; i < kms.length; ++i) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	160 double km = kms[i];
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	161
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	162 for (int j = 0; j < ws.length; ++j) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	163 FixingsColumn column = dataColumns.get(j);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	164 qs[j] = column.getQ(km);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	165 boolean interpolated = column.getW(km, ws, j);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	166 // TODO: mark as interpolated.
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	167 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	168
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	169 // TODO: Do preprocessing here!
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	170 double [] parameters = fit(func, km, ws, qs);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	171 if (parameters == null) { // Problems are reported already.
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	172 continue;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	173 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	174
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	175 int row = results.newRow();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	176
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	177 results.set(row, "km", km);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	178 for (int j = 0; j < parameters.length; ++j) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	179 if (Double.isNaN(parameters[j])) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	180 invalid = true;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	181 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	182 else {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	183 results.set(row, parameterNames[j], parameters[j]);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	184 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	185 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	186 // TODO: Calculate statistics, too!
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	187 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	188
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	189 if (invalid) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	190 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	191 addProblem("fix.invalid.values");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	192 results.removeNaNs();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	193 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	194
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	195 // TODO: Calculate Delta W/t, too.
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	196 return new CalculationResult(results, this);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	197 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	198
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	199 protected static String [] createColumnNames(String [] parameters) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	200 String [] result = new String[parameters.length + 1];
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	201 result[0] = "km";
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	202 // TODO: Add statistic columns, too.
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	203 System.arraycopy(parameters, 0, result, 1, parameters.length);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	204 return result;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	205 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	206
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	207 protected double [] fit(
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	208 Function function,
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	209 double km,
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	210 double [] ws,
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	211 double [] qs
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	212 ) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	213 LevenbergMarquardtOptimizer lmo = new LevenbergMarquardtOptimizer();
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	214 CurveFitter cf = new CurveFitter(lmo);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	215
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	216 boolean missingWs = false;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	217 boolean missingQs = false;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	218
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	219 for (int i = 0; i < ws.length; ++i) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	220 boolean ignore = false;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	221 if (Double.isNaN(ws[i])) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	222 ignore = true;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	223 if (!missingWs) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	224 missingWs = true;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	225 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	226 addProblem(km, "fix.missing.w");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	227 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	228 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	229 if (Double.isNaN(qs[i])) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	230 ignore = true;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	231 if (!missingQs) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	232 missingQs = true;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	233 // TODO: i18n
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	234 addProblem(km, "fix.missing.q");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	235 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	236 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	237 if (!ignore) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	238 cf.addObservedPoint(ws[i], qs[i]);
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	239 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	240 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	241
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	242 try {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	243 return cf.fit(function, function.getInitialGuess());
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	244 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	245 catch (MathException me) {
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	246 addProblem(km, "fix.fitting.failed");
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	247 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	248
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	249 return null;
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	250 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	251 }
a441be7f1589 Added Fix calculation. Sascha L. Teichmann <sascha.teichmann@intevation.de> parents: diff changeset	252 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

2729

a441be7f1589 Added Fix calculation.