source: proiecte/ptvs/src/vnsim/vehicular/routePlan/cityRouting/CrossRoadCongestion.java @ 31

package vnsim.vehicular.routePlan.cityRouting;

import java.util.ArrayList;

public class CrossRoadCongestion {

        private IntersectionNode intersectionNode;
        // for every segments in the intersection
        private ArrayList<WeekCongestion> lastWeek;
        // for every segments in the intersection
        private ArrayList<ArrayList<Double>> realTimeCongestion;

        public CrossRoadCongestion(IntersectionNode inode) {
                realTimeCongestion = new ArrayList<ArrayList<Double>>();
                lastWeek = new ArrayList<WeekCongestion>();
                for (int i = 0; i < inode.getSegmentsNb(); i++) {
                        realTimeCongestion.add(new ArrayList<Double>());
                        lastWeek.add(new WeekCongestion());
                }
                intersectionNode = inode;
        }

        public IntersectionNode getIntersectionNode() {
                return intersectionNode;
        }

        // get last congestion on the segment
        public Double getCongestion(int segmentIndex, int day, int hour) {
                ArrayList<Double> data = realTimeCongestion.get(segmentIndex);
                if (data.size() == 0) {
                        return null;
                }
                return data.get(data.size() - 1);
        }

        public Double getLastWeekCongestion(int segmentIndex, int day, int hour) {
                return lastWeek.get(segmentIndex).getCongestion(day, hour);
        }

        public void setCongestion(int segmentIndex, Double congestion) {
                realTimeCongestion.get(segmentIndex).add(congestion);
        }

        public void setLastWeekCongestion(int segmentIndex, int day, int hour,
                        Double congestion) {
                lastWeek.get(segmentIndex).setCongestion(day, hour, congestion);
        }

        // for all segments mediate the real time data and set the value to lastWeek
        // clears the real-time list;
        public void updateLastWeekCongestion(int day, int hour) {
                Double congestion;
                for (int i = 0; i < lastWeek.size(); i++) {
                        congestion = mediate(realTimeCongestion.get(i));
                        lastWeek.get(i).setCongestion(day, hour, congestion);
                        realTimeCongestion.get(i).clear();
                }
        }

        private static Double mediate(ArrayList<Double> list) {
                if (list == null) {
                        return null;
                }
                if (list.size() == 0) {
                        return new Double(0);
                }
                Double m = new Double(0);
                for (int i = 0; i < list.size(); i++) {
                        m = m + list.get(i);
                }
                return m / list.size();
        }

        public ArrayList<WeekCongestion> cloneLastWeek() {
                ArrayList<WeekCongestion> cloned = new ArrayList<WeekCongestion>();
                for (int i = 0; i < lastWeek.size(); i++) {
                        cloned.add(lastWeek.get(i).cloneWeekCongestion());
                }
                return cloned;
        }

        // last week congestion
        public String toString() {
                String str = "segments_nb " + lastWeek.size() + "\n";
                for (int i = 0; i < lastWeek.size(); i++) {
                        str = str + "segment " + i + "\n";
                        str = str + lastWeek.get(i).toString() + "\n";
                }
                return str;
        }
}