Series.java

/*
 * WBI Indicator Explorer
 *
 * Copyright 2015 Sebastian Nogara <snogaraleal@gmail.com>
 *
 * This file is part of WBI.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package models;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.persistence.Entity;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;

import rpc.shared.data.Type;

@SuppressWarnings("serial")
@Entity
public class Series extends Model {
    public static Finder<Series> objects =
        new Finder<Series>(Series.class);

    @ManyToOne
    protected Indicator indicator;

    @ManyToOne
    protected Country country;

    @OneToMany
    protected List<Point> points;

    public Series() {
        super();
    }

    public Series(Indicator indicator, Country country) {
        this();
        this.indicator = indicator;
        this.country = country;
    }

    public Series(Indicator indicator, Country country, List<Point> points) {
        this(indicator, country);
        this.points = points;
    }

    public Indicator getIndicator() {
        return indicator;
    }

    public Country getCountry() {
        return country;
    }

    public List<Point> getPoints() {
        return points;
    }

    public void setPoints(List<Point> points) {
        this.points = points;
    }

    public Double getAverage() {
        if (points.isEmpty()) {
            return null;
        }

        double average = 0.0;

        for (Point point : points) {
            average += point.getValue();
        }

        average /= points.size();
        return average;
    }

    private Map<Integer, Double> pointsMap;

    public Map<Integer, Double> getPointsMap() {
        if (pointsMap == null) {
            pointsMap = new HashMap<Integer, Double>();

            for (Point point : points) {
                pointsMap.put(point.getYear(), point.getValue());
            }
        }
        return pointsMap;
    }

    public Double getPointValue(Integer year) {
        return getPointsMap().get(year);
    }

    /*
     * {@code Serializable} implementation
     */

    public static final String FIELD_COUNTRY = "country";
    public static final String FIELD_POINTS = "points";

    @Override
    public Object get(String field) {
        if (field.equals(FIELD_COUNTRY)) return country;
        if (field.equals(FIELD_POINTS)) return points;
        return super.get(field);
    }

    @SuppressWarnings("unchecked")
    @Override
    public void set(String field, Object value) {
        super.set(field, value);
        if (field.equals(FIELD_COUNTRY)) country = (Country) value;
        if (field.equals(FIELD_POINTS)) points = (List<Point>) value;
    }

    private static Map<String, Type> fields;

    @Override
    public Map<String, Type> fields() {
        if (fields == null) {
            fields = super.fields();
            fields.put(FIELD_COUNTRY, Type.get(Country.class));
            fields.put(FIELD_POINTS, Type.get(
                List.class, Type.get(Point.class)));
        }
        return fields;
    }
}