Application: gvSIG desktop » gvSIG tools: Attribute editor

package org.gvsig.app.mainplugin.extension;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.util.Map;

import java.util.zip.CRC32;

import org.apache.commons.io.FileUtils;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.PluginsManager;

import org.gvsig.featureform.swing.FeatureFormDefinitionsProvider;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynClass;

import org.gvsig.tools.dynobject.DynObjectManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.xmlpull.v1.XmlPullParserException;

public class DefaultFeatureFormDefinitionsProvider implements FeatureFormDefinitionsProvider {

    private static final Logger logger = LoggerFactory.getLogger(DefaultFeatureFormDefinitionsProvider.class);

    private File definitionsFolder = null;

    private File getDefinitionsFolder() throws IOException {

        if( this.definitionsFolder == null ) {

            PluginsManager pluginManager = PluginsLocator.getManager();

            PluginServices plugin = pluginManager.getPlugin(AttributeEditorExtension.class);

            File homeFolder = plugin.getPluginHomeFolder();

            File definitionsFolder = new File(homeFolder,"definitions");

            if( !definitionsFolder.exists() ) {

                FileUtils.forceMkdir(definitionsFolder);

            }            

            this.definitionsFolder = definitionsFolder;

        }

        return this.definitionsFolder;

    }

    private long getCRC(FeatureType type) {

        StringBuffer buffer = new StringBuffer();

        for( int i=0; i< type.size(); i++) {

            FeatureAttributeDescriptor x = type.getAttributeDescriptor(i);

            buffer.append(x.getName());

            buffer.append(x.getDataTypeName());

            buffer.append(x.getSize());

        }

        CRC32 crc = new CRC32();

        byte[] data = buffer.toString().getBytes();

        crc.update(data);

        return crc.getValue();

    }

    private String getFilename(FeatureStore store) throws DataException {

        FeatureType type = store.getDefaultFeatureType();

        return store.getName()+"-"+Long.toHexString(getCRC(type))+".xml";

    }

    public void put(FeatureStore store) throws DataException, IOException {

        DynObjectManager dynObjectManager = ToolsLocator.getDynObjectManager();

        FeatureType type = store.getDefaultFeatureType();

        if( !getDefinitionsFolder().exists() ) {

            FileUtils.forceMkdir(getDefinitionsFolder());

        }

        File f = new File(getDefinitionsFolder(), getFilename(store));

        dynObjectManager.exportSimpleDynClassDefinitions(f,type);

    }

    public DynClass get(FeatureStore store) throws DataException {

        DynObjectManager dynObjectManager = ToolsLocator.getDynObjectManager();

        FeatureType type = store.getDefaultFeatureType();

        File f;

        try {

            f = new File(getDefinitionsFolder(), getFilename(store));

        } catch (IOException ex) {

            return type;

        }

        if( !f.exists() ) {

            return type;

        }

        FileInputStream fis;

        try {

            fis = new FileInputStream(f);

        } catch (FileNotFoundException ex) {

            return type;

        }

        Map classes = null;

        try {

            classes = dynObjectManager.importDynClassDefinitions(fis, this.getClass().getClassLoader());

        } catch (XmlPullParserException ex) {

            logger.warn("Can't load xml definition.", ex);

            return type;

        } catch (IOException ex) {

            return type;

        }

        DynClass dynClass = (DynClass) classes.get(type.getName());

        if( dynClass == null ) {

            return type;

        }

        return dynClass;

    }

}