jdplus.sa.base.csv.ArraysSummary Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdplus-sa-base-csv Show documentation
${project.parent.artifactId} - ${project.artifactId}
There is a newer version: 3.3.0
/*
* Copyright 2013 National Bank of Belgium
*
* Licensed under the EUPL, Version 1.1 or – as soon they will be approved 
* by the European Commission - subsequent versions of the EUPL (the "Licence");
* You may not use this work except in compliance with the Licence.
* You may obtain a copy of the Licence at:
*
* http://ec.europa.eu/idabc/eupl
*
* Unless required by applicable law or agreed to in writing, software 
* distributed under the Licence is distributed on an "AS IS" basis,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the Licence for the specific language governing permissions and 
* limitations under the Licence.
 */
package jdplus.sa.base.csv;

import jdplus.toolkit.base.api.data.DoubleSeq;
import jdplus.toolkit.base.api.information.Explorable;
import jdplus.toolkit.base.api.math.matrices.Matrix;
import jdplus.sa.base.api.SaDocument;
import jdplus.toolkit.base.api.util.WildCards;
import java.util.LinkedHashMap;
import java.util.Locale;
import java.util.Map;
import java.util.Set;

/**
 * Supports currently matrices, arrays and doubleseq
 */
public class ArraysSummary {

    private final String name;
    private final Map data = new LinkedHashMap<>();

    public ArraysSummary(String[] items, String name, SaDocument document) {
        this.name = name;
        fillDictionary(items, document.getResults());
    }

     public String getName() {
        return name;
    }

   private void fillDictionary(String[] items, Explorable results) {
        for (String item : items) {
            try {
                item = item.toLowerCase(Locale.ROOT);
                if (results != null) {
                    if (WildCards.hasWildCards(item)) {
                        Map mall = results.searchAll(item, Matrix.class);
                        mall.entrySet().forEach(entry -> data.put(entry.getKey(), DoubleArray.of(entry.getValue())));
                        Map all = results.searchAll(item, double[].class);
                        all.entrySet().forEach(entry -> data.put(entry.getKey(), DoubleArray.of(entry.getValue())));
                        Map vall = results.searchAll(item, DoubleSeq.class);
                        vall.entrySet().forEach(entry -> data.put(entry.getKey(), DoubleArray.of(entry.getValue())));
                    } else {
                        Matrix m = results.getData(item, Matrix.class);
                        if (m != null) {
                            data.put(item, DoubleArray.of(m));
                        } else {
                            double[] v = results.getData(item, double[].class);
                            if (v != null) {
                                data.put(item, DoubleArray.of(v));
                            } else {
                                DoubleSeq s = results.getData(item, DoubleSeq.class);
                                if (s != null) {
                                    data.put(item, DoubleArray.of(s));
                                }
                            }
                        }
                    }
                } else {
                    data.put(item, null);
                }
            } catch (Exception err) {
                data.put(item, null);
            }

        }
    }

    public DoubleArray getArray(String name) {
        return data.get(name);
    }

    void fill(Set set) {
        set.addAll(data.keySet());
    }
}