Application: gvSIG desktop

package org.gvsig.fmap.dal.feature.impl;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.util.ArrayList;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Set;

import org.apache.commons.collections4.CollectionUtils;

import org.apache.commons.io.IOUtils;

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

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

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

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

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

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.Persistent;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

public class DALFile implements Persistent {

    private static final String DALFILE_PERSISTENCE_DEFINITION_NAME = "DALFile";

    private static final String CALCULATEDATTRIBUTE_PERSISTENCE_DEFINITION_NAME = "DALFileCalculatedAttribute";

    public static DALFile getDALFile() {

        DALFile f = new DALFile();

        return f;

    }

    public static DALFile getDALFile(File f) {

        DALFile df = new DALFile();

        df.read(f);

        return df;

    }

    public static void registerPersistenceDefinition() {

        PersistenceManager manager = ToolsLocator.getPersistenceManager();

        if (manager.getDefinition(DALFILE_PERSISTENCE_DEFINITION_NAME) == null) {

            DynStruct definition = manager.addDefinition(

                DALFile.class,

                DALFILE_PERSISTENCE_DEFINITION_NAME, 

                DALFILE_PERSISTENCE_DEFINITION_NAME + " Persistent definition", 

                null, 

                null

            );

            definition.addDynFieldList("calculatedAttributes")

                .setClassOfItems(CalculatedAttribute.class)

                .setMandatory(true)

                .setPersistent(true);

        }

        if (manager.getDefinition(CALCULATEDATTRIBUTE_PERSISTENCE_DEFINITION_NAME) == null) {

            DynStruct definition = manager.addDefinition(

                CalculatedAttribute.class,

                CALCULATEDATTRIBUTE_PERSISTENCE_DEFINITION_NAME, 

                CALCULATEDATTRIBUTE_PERSISTENCE_DEFINITION_NAME + " Persistent definition", 

                null, 

                null

            );

            definition.addDynFieldString("name")

                .setMandatory(true)

                .setPersistent(true);

            definition.addDynFieldInt("type")

                .setMandatory(true)

                .setPersistent(true);

            definition.addDynFieldInt("featureTypeIndex")

                .setMandatory(true)

                .setPersistent(true);

            definition.addDynFieldObject("emulator")

                .setClassOfValue(FeatureAttributeEmulator.class)

                .setMandatory(true)

                .setPersistent(true);

        }

    }

    public static class CalculatedAttribute implements Persistent {

        private String name;

        private int type;

        private FeatureAttributeEmulator emulator;

        private int featureTypeIndex;

        public CalculatedAttribute() {

        }

        private CalculatedAttribute(int featureTypeIndex, FeatureAttributeDescriptor attrdesc) {

            this.featureTypeIndex = featureTypeIndex;

            this.name = attrdesc.getName();

            this.type = attrdesc.getType();

            this.emulator = attrdesc.getFeatureAttributeEmulator();

        }

        @Override

        public void saveToState(PersistentState state) throws PersistenceException {

            state.set("name", name);

            state.set("type", type);

            state.set("featureTypeIndex", featureTypeIndex);

            state.set("emulator", emulator);

        }

        @Override

        public void loadFromState(PersistentState state) throws PersistenceException {

            this.name = state.getString("name");

            this.type = state.getInt("type");

            this.featureTypeIndex = state.getInt("featureTypeIndex");

            this.emulator = (FeatureAttributeEmulator) state.get("emulator");

        }

    }

    private final List<CalculatedAttribute> calculatedAttributes;

    public DALFile() {

        this.calculatedAttributes = new ArrayList<>();

    }

    public boolean isEmpty() {

        if( CollectionUtils.isEmpty(this.calculatedAttributes) ) {

            return true;

        }

        return false;

    }

    public void write(File f) {

        FileOutputStream out = null;

        try {

            PersistenceManager manager = ToolsLocator.getPersistenceManager();

            PersistentState state = manager.getState(this);

            out = new FileOutputStream(f);

            manager.saveState(state, out);

        } catch(Exception ex) {

            throw new RuntimeException("Can't write DAL file.",ex);

        } finally {

            IOUtils.closeQuietly(out);

        }

    }

    @SuppressWarnings("UseSpecificCatch")

    public void read(File f) {

        FileInputStream in = null;

        try {

            PersistenceManager manager = ToolsLocator.getPersistenceManager();

            in = new FileInputStream(f);

            DALFile x = (DALFile) manager.getObject(in);

            this.calculatedAttributes.clear();

            this.calculatedAttributes.addAll(x.calculatedAttributes);

        } catch(Exception ex) {

            throw new RuntimeException("Can't read DAL file.",ex);

        } finally {

            IOUtils.closeQuietly(in);

        }

    }

    public void setStore(FeatureStore store) throws DataException {

        this.calculatedAttributes.clear();

        List<FeatureType> types = store.getFeatureTypes();

        for( int n=0; n<types.size(); n++ ) {

            FeatureType type = types.get(n);

                for (FeatureAttributeDescriptor attrdesc : type) {

                    FeatureAttributeEmulator emulator = attrdesc.getFeatureAttributeEmulator();

                    if( emulator!= null && emulator instanceof Persistent ) {

                        CalculatedAttribute calculatedAttribute = new CalculatedAttribute(n,attrdesc);

                        this.calculatedAttributes.add(calculatedAttribute);

                    }

            }

        }

    }

    public void updateStore(FeatureStore store) throws DataException {

        List<FeatureType> types = store.getFeatureTypes();

        Set<DefaultFeatureType> needFixTypes = new HashSet<>();

        for (CalculatedAttribute calculatedAttribute : calculatedAttributes) {

            DefaultFeatureType ft = (DefaultFeatureType) types.get(calculatedAttribute.featureTypeIndex);

            DefaultFeatureAttributeDescriptor attrdesc = new DefaultFeatureAttributeDescriptor(ft);

            attrdesc.setDataType(calculatedAttribute.type);

            attrdesc.setName(calculatedAttribute.name);

            attrdesc.setFeatureAttributeEmulator(calculatedAttribute.emulator);

            ft.add(attrdesc);

            needFixTypes.add(ft);

        }

        for (DefaultFeatureType type : needFixTypes) {

            type.fixAll();

        }

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        state.set("calculatedAttributes", calculatedAttributes);

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        this.calculatedAttributes.clear();

        Iterator<CalculatedAttribute> it = state.getIterator("calculatedAttributes");

        while( it.hasNext() ) {

            CalculatedAttribute calculatedAttribute = it.next();

            this.calculatedAttributes.add(calculatedAttribute);

        }

    }

}