com.joliciel.talismane.machineLearning.ExternalResourceFinderImpl.java Source code

Introduction

Here is the source code for com.joliciel.talismane.machineLearning.ExternalResourceFinderImpl.java

Source

///////////////////////////////////////////////////////////////////////////////
//Copyright (C) 2013 Assaf Urieli
//
//This file is part of Talismane.
//
//Talismane is free software: you can redistribute it and/or modify
//it under the terms of the GNU Affero General Public License as published by
//the Free Software Foundation, either version 3 of the License, or
//(at your option) any later version.
//
//Talismane is distributed in the hope that it will be useful,
//but WITHOUT ANY WARRANTY; without even the implied warranty of
//MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//GNU Affero General Public License for more details.
//
//You should have received a copy of the GNU Affero General Public License
//along with Talismane.  If not, see <http://www.gnu.org/licenses/>.
//////////////////////////////////////////////////////////////////////////////
package com.joliciel.talismane.machineLearning;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.zip.ZipInputStream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.joliciel.talismane.utils.LogUtils;

class ExternalResourceFinderImpl implements ExternalResourceFinder {
    private static final Log LOG = LogFactory.getLog(ExternalResourceFinderImpl.class);
    private Map<String, ExternalResource<?>> resourceMap = new HashMap<String, ExternalResource<?>>();

    @Override
    public ExternalResource<?> getExternalResource(String name) {
        return this.resourceMap.get(name);
    }

    public void addExternalResource(ExternalResource<?> externalResource) {
        LOG.debug("Adding resource with name: " + externalResource.getName());
        this.resourceMap.put(externalResource.getName(), externalResource);
    }

    @Override
    public Collection<ExternalResource<?>> getExternalResources() {
        return resourceMap.values();
    }

    @Override
    public void addExternalResources(File externalResourceFile) {
        try {
            if (externalResourceFile.isDirectory()) {
                File[] files = externalResourceFile.listFiles();
                for (File resourceFile : files) {
                    LOG.debug("Reading " + resourceFile.getName());
                    if (resourceFile.getName().endsWith(".zip")) {
                        ZipInputStream zis = new ZipInputStream(new FileInputStream(resourceFile));
                        zis.getNextEntry();
                        ObjectInputStream ois = new ObjectInputStream(zis);
                        ExternalResource<?> externalResource = (ExternalResource<?>) ois.readObject();
                        this.addExternalResource(externalResource);
                    } else {
                        TextFileResource textFileResource = new TextFileResource(resourceFile);
                        this.addExternalResource(textFileResource);
                    }
                }
            } else {
                LOG.debug("Reading " + externalResourceFile.getName());
                if (externalResourceFile.getName().endsWith(".zip")) {
                    ZipInputStream zis = new ZipInputStream(new FileInputStream(externalResourceFile));
                    zis.getNextEntry();
                    ObjectInputStream ois = new ObjectInputStream(zis);
                    ExternalResource<?> externalResource = (ExternalResource<?>) ois.readObject();
                    this.addExternalResource(externalResource);
                } else {
                    TextFileResource textFileResource = new TextFileResource(externalResourceFile);
                    this.addExternalResource(textFileResource);
                }
            }
        } catch (IOException e) {
            LogUtils.logError(LOG, e);
            throw new RuntimeException(e);
        } catch (ClassNotFoundException e) {
            LogUtils.logError(LOG, e);
            throw new RuntimeException(e);
        }
    }
}