model.SummaryList.java Source code

Introduction

Here is the source code for model.SummaryList.java

Source

/*
 * Copyright 2015 Len Payne <len.payne@lambtoncollege.ca>.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package model;

import au.com.bytecode.opencsv.CSV;
import au.com.bytecode.opencsv.CSVReadProc;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.enterprise.context.ApplicationScoped;
import org.json.simple.JSONArray;

/**
 *
 * @author Len Payne <len.payne@lambtoncollege.ca>
 */
@ApplicationScoped
public class SummaryList {

    private final List<Summary> summaryList = new ArrayList<>();
    private final String CSV_FILENAME = "/CODE-Dataset.csv";

    public SummaryList() {
        BufferedReader inputFile = new BufferedReader(
                new InputStreamReader(getClass().getResourceAsStream(CSV_FILENAME)));
        CSV csv = CSV.separator(',').charset(StandardCharsets.UTF_8).skipLines(1).quote('"').create();
        csv.read(inputFile, new CSVReadProc() {

            @Override
            public void procRow(int i, String... strings) {
                summaryList.add(new Summary(strings[0].trim(), Double.parseDouble(strings[1].trim()),
                        Double.parseDouble(strings[2].trim()), Double.parseDouble(strings[3].trim()),
                        (strings[4].isEmpty()) ? -1 : Double.parseDouble(strings[4].trim())));
            }
        });

    }

    public JSONArray toJSON() {
        JSONArray json = new JSONArray();
        for (Summary s : summaryList) {
            json.add(s.toJSON());
        }
        return json;
    }

    public String toJSONString() {
        summaryList.sort(new Weighted());
        return toJSON().toJSONString();
    }

    public String toJSONString(int stress, int health, int cost) {
        summaryList.sort(new Weighted(stress, health, cost));
        return toJSON().toJSONString();
    }

    private class Weighted implements Comparator<Summary> {

        private final double STRESS_MAX = 37;
        private final double STRESS_MID = 24; // (MAX + MIN) / 2
        private final double STRESS_MIN = 10;
        private final double HEALTH_MAX = 67;
        private final double HEALTH_MID = 55;
        private final double HEALTH_MIN = 43;
        private final double COST_MAX = 102000;
        private final double COST_MID = 83500;
        private final double COST_MIN = 65000;

        private int stressWeight;
        private int healthWeight;
        private int costWeight;

        public Weighted() {
            stressWeight = 33;
            healthWeight = 33;
            costWeight = 33;
        }

        public Weighted(int s, int h, int c) {
            stressWeight = s;
            healthWeight = h;
            costWeight = c;
        }

        private double getWeight(double max, double min, double val) {
            return (val - min) / (max - min) * 2 - 1;
        }

        @Override
        public int compare(Summary o1, Summary o2) {
            int o1Weighted = -1 * (int) Math.round(getWeight(STRESS_MAX, STRESS_MIN, o1.getStress()) * stressWeight)
                    + (int) Math.round(getWeight(HEALTH_MAX, HEALTH_MIN, o1.getHealth()) * healthWeight)
                    + -1 * (int) Math.round(getWeight(COST_MAX, COST_MIN, o1.getExpenses()) * costWeight);
            int o2Weighted = -1 * (int) Math.round(getWeight(STRESS_MAX, STRESS_MIN, o2.getStress()) * stressWeight)
                    + (int) Math.round(getWeight(HEALTH_MAX, HEALTH_MIN, o2.getHealth()) * healthWeight)
                    + -1 * (int) Math.round(getWeight(COST_MAX, COST_MIN, o2.getExpenses()) * costWeight);
            String output = String.format("%s: %d vs %s: %d", o1.getCity(), o1Weighted, o2.getCity(), o2Weighted);
            Logger.getLogger(getClass().getName()).log(Level.INFO, output);
            return o2Weighted - o1Weighted;
        }

    }
}