internal.spreadsheet.base.api.grid.SheetGrid Maven / Gradle / Ivy

Go to download
/*
 * 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 internal.spreadsheet.base.api.grid;

import jdplus.toolkit.base.api.timeseries.TsCollection;
import ec.util.spreadsheet.Book;
import jdplus.toolkit.base.tsp.grid.GridReader;
import ec.util.spreadsheet.Sheet;
import internal.spreadsheet.base.api.SpreadSheetConnection;

import java.io.File;
import java.io.IOException;
import java.io.UncheckedIOException;
import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

/**
 *
 * @author Philippe Charles
 */
@lombok.AllArgsConstructor(staticName = "of")
public final class SheetGrid implements SpreadSheetConnection {

    @lombok.NonNull
    private final File file;

    @lombok.NonNull
    private final Book.Factory factory;

    @lombok.NonNull
    private final GridReader reader;

    @Override
    public Optional getSheetByName(String name) throws IOException {
        Objects.requireNonNull(name);
        try (Book book = factory.load(file)) {
            return getSheetByName(book, name);
        }
    }

    @Override
    public List getSheetNames() throws IOException {
        try (Book book = factory.load(file)) {
            return getSheetNames(book);
        }
    }

    @Override
    public List getSheets() throws IOException {
        try (Book book = factory.load(file)) {
            return getSheets(book);
        }
    }

    @Override
    public void close() throws IOException {
    }

    private Optional getSheetByName(Book book, String name) throws IOException {
        for (int i = 0; i < book.getSheetCount2(); i++) {
            if (book.getSheetName(i).equals(name)) {
                return Optional.of(readSheet(book.getSheet(i)));
            }
        }
        return Optional.empty();
    }

    private List getSheetNames(Book book) throws IOException {
        String[] result = new String[book.getSheetCount2()];
        for (int i = 0; i < result.length; i++) {
            result[i] = book.getSheetName(i);
        }
        return Arrays.asList(result);
    }

    private List getSheets(Book book) throws IOException {
        TsCollection[] result = new TsCollection[book.getSheetCount2()];
        try {
            book.parallelForEach((sheet, index) -> {
                try {
                    result[index] = readSheet(sheet);
                } catch (IOException ex) {
                    throw new UncheckedIOException(ex);
                }
            });
        } catch (UncheckedIOException ex) {
            throw ex.getCause();
        }
        return Arrays.asList(result);
    }

    private TsCollection readSheet(Sheet sheet) throws IOException {
        return reader.read(SheetGridInput.of(sheet, this::isSupportedDataType));
    }

    public boolean isSupportedDataType(Class type) {
        return (LocalDateTime.class.equals(type) && factory.isSupportedDataType(Date.class))
                || factory.isSupportedDataType(type);
    }
}