com.js.quickestquail.ui.stats.YearStat.java Source code

Introduction

Here is the source code for com.js.quickestquail.ui.stats.YearStat.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.js.quickestquail.ui.stats;

import com.js.quickestquail.model.DriveManager;
import com.js.quickestquail.model.Movie;
import com.js.quickestquail.imdb.CachedMovieProvider;

import java.awt.*;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import javax.swing.JPanel;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.category.DefaultCategoryDataset;

/**
 *
 * @author joris
 */
public class YearStat extends JPanel {

    public YearStat() {
        setLayout(new BorderLayout());
        add(new ChartPanel(generateChart()), BorderLayout.CENTER);
    }

    private CategoryDataset generateDataset() {
        Map<Integer, Integer> yearFrequency = new HashMap<>();
        for (String id : DriveManager.get().getSelected().values()) {
            Movie mov = CachedMovieProvider.get().getMovieByID(id);
            int year = mov.getYear();

            if (!yearFrequency.containsKey(year)) {
                yearFrequency.put(year, 1);
            } else {
                yearFrequency.put(year, yearFrequency.get(year) + 1);
            }

        }

        List<Entry<Integer, Integer>> entries = new ArrayList<>(yearFrequency.entrySet());
        java.util.Collections.sort(entries, new Comparator<Entry<Integer, Integer>>() {
            @Override
            public int compare(Entry<Integer, Integer> o1, Entry<Integer, Integer> o2) {
                return o1.getKey().compareTo(o2.getKey());
            }
        });

        // convert to proper format
        DefaultCategoryDataset dataset = new DefaultCategoryDataset();
        for (Entry<Integer, Integer> en : entries) {
            dataset.addValue(en.getValue(),
                    java.util.ResourceBundle.getBundle("i18n/i18n").getString("stat.year.xaxis"), en.getKey());
        }

        // return
        return dataset;
    }

    private JFreeChart generateChart() {
        JFreeChart lineChart = ChartFactory.createLineChart(
                java.util.ResourceBundle.getBundle("i18n/i18n").getString("stat.year.title"),
                java.util.ResourceBundle.getBundle("i18n/i18n").getString("stat.year.xaxis"),
                java.util.ResourceBundle.getBundle("i18n/i18n").getString("stat.year.yaxis"), generateDataset(),
                PlotOrientation.VERTICAL, true, true, false);
        lineChart.setBackgroundPaint(new Color(0xFF, 0xFF, 0xFF, 0));
        lineChart.getPlot().setBackgroundPaint(Color.WHITE);
        return lineChart;
    }
}