org.jevis.jecalc.calculation.SampleMerger.java Source code

Introduction

Here is the source code for org.jevis.jecalc.calculation.SampleMerger.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 org.jevis.jecalc.calculation;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.jevis.api.JEVisSample;
import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;

/**
 *
 * @author broder
 */
public class SampleMerger {

    private final Logger logger = LogManager.getLogger(SampleMerger.class);
    private final List<List<Sample>> allSamples = new ArrayList<>();
    private final List<Sample> constants = new ArrayList<>();
    private final List<List<Sample>> periodConstants = new ArrayList<>();

    private void addSamples(List<JEVisSample> jevisSamples, String variable) {
        List<Sample> samples = new ArrayList<>();
        for (JEVisSample currentSample : jevisSamples) {
            samples.add(new Sample(currentSample, variable));
        }
        allSamples.add(samples);
    }

    private void addConstant(JEVisSample constant, String variable) {
        Sample sample = new Sample(constant, variable);
        constants.add(sample);
    }

    private void addPeriodConstant(List<JEVisSample> jevisSamples, String variable) {
        List<Sample> samples = new ArrayList<>();
        for (JEVisSample currentSample : jevisSamples) {
            samples.add(new Sample(currentSample, variable));
        }
        periodConstants.add(samples);
    }

    public void addSamples(List<JEVisSample> jevisSamples, String variable, InputType inputType) {
        switch (inputType) {
        case PERIODIC:
            addSamples(jevisSamples, variable);
            break;
        case STATIC:
            addConstant(jevisSamples.get(0), variable);
            break;
        case NON_PERIODIC:
            addPeriodConstant(jevisSamples, variable);
            break;
        }
    }

    public Map<DateTime, List<Sample>> merge() {
        Map<DateTime, List<Sample>> sampleMap = new TreeMap<>();

        insertAllSamples(sampleMap); //value changed for every sample

        insertConstants(sampleMap); //value is always the same

        insertPeriodicConstants(sampleMap); //value changed for specific periods

        int variableSize = allSamples.size() + constants.size() + periodConstants.size();
        Set<DateTime> removableKeys = new HashSet<>();
        for (Map.Entry<DateTime, List<Sample>> sampleEntry : sampleMap.entrySet()) {
            if (sampleEntry.getValue().size() != variableSize) {
                removableKeys.add(sampleEntry.getKey());
            }
        }

        for (DateTime key : removableKeys) {
            logger.debug("not every input data with datetime {}, will delete this datetime from calculation",
                    key.toString(DateTimeFormat.fullDateTime()));
            sampleMap.remove(key);
        }
        return sampleMap;
    }

    private void insertAllSamples(Map<DateTime, List<Sample>> sampleMap) {
        for (List<Sample> currentSamples : allSamples) {
            for (Sample sample : currentSamples) {
                List<Sample> samples = sampleMap.getOrDefault(sample.getDate(), new ArrayList<Sample>());
                samples.add(sample);
                sampleMap.put(sample.getDate(), samples);
            }
        }
    }

    private void insertConstants(Map<DateTime, List<Sample>> sampleMap) {
        for (Map.Entry<DateTime, List<Sample>> entry : sampleMap.entrySet()) {
            for (Sample constant : constants) {
                entry.getValue().add(constant);
            }
        }
    }

    private void insertPeriodicConstants(Map<DateTime, List<Sample>> sampleMap) {
        for (Map.Entry<DateTime, List<Sample>> entry : sampleMap.entrySet()) {
            DateTime currentSampleTime = entry.getKey();
            for (List<Sample> periodicConstants : periodConstants) {
                Sample validConstant = null;
                for (Sample periodicConstant : periodicConstants) {
                    DateTime startDate = periodicConstant.getDate();
                    if (startDate.isBefore(currentSampleTime) || startDate.isEqual(currentSampleTime)) {
                        validConstant = periodicConstant;
                    } else if (startDate.isAfter(currentSampleTime)) {
                        break;
                    }
                }
                if (validConstant != null) {
                    entry.getValue().add(validConstant);
                }
            }
        }
    }

    public enum InputType {

        STATIC, PERIODIC, NON_PERIODIC;
    }

}