Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2020 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.expressionevaluator.impl.function.dataaccess;

import java.util.Collections;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import java.util.Set;

import org.apache.commons.lang3.Range;

import org.gvsig.expressionevaluator.ExpressionRuntimeException;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.impl.DALFunctions;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.fmap.dal.AbstractStoresRepository;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import static org.gvsig.fmap.dal.DataManager.FUNCTION_CREATE_IN_MEMORY_TABLE;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.StoresRepository;

import org.gvsig.fmap.dal.feature.EditableFeatureType;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.util.UnmodifiableBasicSet;

import org.gvsig.tools.util.UnmodifiableBasicSetAdapter;

/**

 *

 * @author jjdelcerro

 */

public class CreateInMemoryTableFunction extends AbstractFunction {

  public CreateInMemoryTableFunction() {

    super(  DALFunctions.GROUP_DATA_ACCESS,

            FUNCTION_CREATE_IN_MEMORY_TABLE,

            Range.is(2),

            "Create a table in memory with the same structure as the indicated table. Return a reference to the new table.",

            "CREATE IN MEMORY TABLE {{new_table_name}} FROM source_table_name",

            new String[]{

              "new_table_name - Name of the new table to create.",

              "source_table_name - Name of the table to copy their structure."

            },

            "Table",

            false

    );

  }

  @Override

  public boolean allowConstantFolding() {

    return false;

  }

  @Override

  public Object call(Interpreter interpreter, Object[] args) throws Exception {

    DataManager dataManager = DALLocator.getDataManager();

    StoresRepository storesRepository = dataManager.getStoresRepository();

    MemoryStoresRepository repo = (MemoryStoresRepository) storesRepository.getSubrepository("Temporary");

    if( repo == null ) {

      repo = new MemoryStoresRepository("Temporary");

      storesRepository.addRepository(repo);

    }

    String targetName = getStr(args,0);

    String sourceName = getStr(args,1);

    FeatureStore source = (FeatureStore) storesRepository.getStore(sourceName);

    if( source == null ) {

      throw new ExpressionRuntimeException("Can't locate source table '"+sourceName+"'.");

    }

    EditableFeatureType featureType = source.getDefaultFeatureType().getEditable();

    FeatureStore target = dataManager.createMemoryStore(null);

    target.edit();

    target.update(featureType);

    target.finishEditing();

    repo.add(targetName, target);

    return target;

  }

  private class MemoryStoresRepository extends AbstractStoresRepository {

    private final Map<String,FeatureStore> repository;

    public MemoryStoresRepository(String name) {

      super(name);

      this.repository = new HashMap<>();

    }

    public void add(String name, FeatureStore store) {

      this.repository.put(name, store);

    }

    @Override

    protected DataStoreParameters getMyParameters(String name) {

      FeatureStore store = this.repository.get(name);

      return store.getParameters();

    }

    @Override

    protected boolean isEmptyMyRepository() {

      return this.repository.isEmpty();

    }

    @Override

    protected int getMySize() {

      return this.repository.size();

    }

    @Override

    protected UnmodifiableBasicSet<String> getMyKeySet() {

        Set<String> keyset = this.repository.keySet();

        if( keyset == null || keyset.isEmpty() ) {

            return UnmodifiableBasicSet.EMPTY_UNMODIFIABLEBASICSET;

        }

        return new UnmodifiableBasicSetAdapter<>(keyset);

    }

    @Override

    public DataStore getStore(String name) {

      FeatureStore store = this.repository.get(name);

      DisposeUtils.bind(store);

      return store;

    }

    @Override

    public void add(String name, DataStoreParameters parameters) {

    }

    @Override

    public boolean contains(DataStoreParameters parameters) {

      return false;

    }

    @Override

    public boolean containsKey(String key) {

      return this.repository.containsKey(key);

    }

    @Override

    public Iterator<DataStoreParameters> iterator() {

      return Collections.EMPTY_LIST.iterator();

    }

    @Override

    public void remove(String name) {

      this.repository.remove(name);

    }

  }

}