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;

    }

}

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.app.mainplugin.extension;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.ViewManager;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

/**

 *

 * @author jjdelcerro

 */

public class CreateDefaultDefinitionExtension extends Extension {

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

    public void initialize() {

    }

    public void execute(String actionCommand) {

        if( "tools-attributeditor-create-default-definition".equalsIgnoreCase(actionCommand) ) {

            ApplicationManager application = ApplicationLocator.getManager();

            ViewDocument viewDoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

            if( viewDoc == null ) {

                return;

            }

            MapContext mapContext = viewDoc.getMapContext();

            if( !mapContext.hasActiveVectorLayers() ) {

                return;

            }

            DataSwingManager featureFormManager = DALSwingLocator.getSwingManager();

            FLayer[] layers = mapContext.getLayers().getActives();

            for( int i=0; i<layers.length; i++ ) {

                if( layers[i] instanceof FLyrVect ) {

                    FLyrVect layer = (FLyrVect) layers[i];

                    try {

                        featureFormManager.getFeatureFormDefinitionsProvider().put(layer.getFeatureStore());

                    } catch (Exception ex) {

                        logger.warn("Can't save form definition for layer '"+layer.getName()+"'.",ex);

                    }

                }

            }

        }

    }

    public boolean isEnabled() {

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewDoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if( viewDoc == null ) {

            return false;

        }

        MapContext mapContext = viewDoc.getMapContext();

        return mapContext.hasActiveVectorLayers();

    }

    public boolean isVisible() {

        return true;

    }

}

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.DataSwingManager;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

    private static Logger logger = LoggerFactory.getLogger(

            AttributeEditorExtension.class);

        IView vw = (IView) ApplicationLocator.getManager().getActiveComponent(ViewDocument.class);

        if (vw != null) {

            FLayer[] act_lyr = vw.getMapControl().getMapContext().getLayers().getActives();

            if (act_lyr == null || act_lyr.length != 1

                    || !(act_lyr[0] instanceof FLyrVect)) {

                //do nothing

            } else {

                FLyrVect lyr = (FLyrVect) act_lyr[0];

                MapControl mapControl = vw.getMapControl();

        DataSwingManager featureFormManager = DALSwingLocator.getSwingManager();

        featureFormManager.registerFeatureFormDefinitionsProvider(new DefaultFeatureFormDefinitionsProvider());

        ViewDocument vw = actWin();

        if (vw != null) {

            FLayer[] act_lyr = vw.getMapContext().getLayers().getActives();

            if (act_lyr == null || act_lyr.length != 1

                    || !(act_lyr[0] instanceof FLyrVect)) {

                return false;

            } else {

                FLyrVect vect = (FLyrVect) act_lyr[0];

                if (!vect.isAvailable()) {

                    /*

                     * This can happen when opening a persisted project

                     * and there is a "slow" layer (GeoDB)

                     */

                    return false;

                }

                return true;

            }

        } else {