package de.intevation.flys.artifacts.model;

import org.w3c.dom.Document;
import org.w3c.dom.Element;

import java.util.List;
import java.util.ArrayList;

public class Calculation
{
    public static class Problem {

        protected Double km;
        protected String msg;

        public Problem() {
        }

        public Problem(String msg) {
            this.msg = msg;
        }

        public Problem(double km, String msg) {
            this.km  = km;
            this.msg = msg;
        }

        public Element toXML(Document document) {
            Element problem = document.createElement("problem");
            if (km != null) {
                problem.setAttribute("km", String.valueOf(km));
            }
            problem.setTextContent(msg);
            return problem;
        }
    } // class Problem

    protected List<Problem> problems;

    public Calculation() {
    }

    protected List<Problem> checkProblems() {
        if (problems == null) {
            problems = new ArrayList<Problem>();
        }
        return problems;
    }

    public void addProblem(String msg) {
        checkProblems().add(new Problem(msg));
    }

    public void addProblem(double km, String msg) {
        checkProblems().add(new Problem(km, msg));
    }

    public boolean hasProblems() {
        return problems != null && !problems.isEmpty();
    }

    public List<Problem> getProblems() {
        return problems;
    }

    public void toXML(Document document) {

        Element root = document.createElement("problems");

        if (hasProblems()) {
            for (Problem problem: problems) {
                root.appendChild(problem.toXML(document));
            }
        }

        document.appendChild(root);
    }
}
// vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :