Application: gvSIG desktop » Library: org.gvsig.tools

/**

 * 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.tools.namestranslator;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

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

/**

 *

 * @author jjdelcerro

 */

public class TrimNamesTranslator

        extends BaseNamesTranslator {

    protected int maxNameLen;

    public TrimNamesTranslator() {

    }

    protected TrimNamesTranslator(int maxNameLen) {

        if (maxNameLen < 8) {

            throw new IllegalArgumentException("Invalid max name len, can be greater than 7.");

        }

        this.maxNameLen = maxNameLen;

    }

    @Override

    public NamesTranslator clone() throws CloneNotSupportedException {

        TrimNamesTranslator clone = (TrimNamesTranslator) super.clone();

        return clone;

    }

    @Override

    public boolean isValid(String name) {

        String sourceName = name;

        if (StringUtils.isBlank(sourceName)) {

            sourceName = "field";

        }

        int len = sourceName.length();

        return len <= this.maxNameLen && !this.translatedNames.contains(sourceName);

    }

    @Override

    public String getSuggestion(String name) {

        String sourceName = name;

        if (StringUtils.isBlank(sourceName)) {

            sourceName = "field";

        }

        int len = sourceName.length();

        if (len <= this.maxNameLen && !this.translatedNames.contains(sourceName)) {

            return sourceName;

        }

        String translatedName;

        for (int i = 0; i < 255; i++) {

            if (len <= this.maxNameLen) {

                if (i <= 9) {

                    if (len == this.maxNameLen) {

                        translatedName = sourceName.substring(0, this.maxNameLen - 3) + "_" + i;

                    } else {

                        translatedName = sourceName + "_" + i;

                    }

                } else if (i <= 99) {

                    if (len == this.maxNameLen) {

                        translatedName = sourceName.substring(0, this.maxNameLen - 3) + i;

                    } else {

                        translatedName = sourceName + i;

                    }

                } else {

                    if (len == this.maxNameLen - 1) {

                        translatedName = sourceName.substring(0, this.maxNameLen - 4) + i;

                    } else {

                        translatedName = sourceName + i;

                    }

                }

            } else {

                if (i == 0) {

                    translatedName = sourceName.substring(0, this.maxNameLen - 1);

                } else if (i <= 9) {

                    translatedName = sourceName.substring(0, this.maxNameLen - 3) + "_" + i;

                } else if (i <= 99) {

                    translatedName = sourceName.substring(0, this.maxNameLen - 3) + i;

                } else {

                    translatedName = sourceName.substring(0, this.maxNameLen - 4) + i;

                }

            }

            if (!this.translatedNames.contains(translatedName)) {

                return translatedName;

            }

        }

        /*

                 * Should not get here

         */

        return sourceName.substring(0, this.maxNameLen - 7) + "_" + (System.currentTimeMillis() % 1000000);

    }

    @Override

    public int setTranslation(String sourceName, String translatedName) {

        translatedName = StringUtils.left(translatedName, this.maxNameLen);

        return super.setTranslation(sourceName, translatedName);

    }

    public static void registerPersistence() {

        PersistenceManager manager = ToolsLocator.getPersistenceManager();

        if (manager.getDefinition("TrimNamesTranslator") == null) {

            DynStruct definition = manager.addDefinition(TrimNamesTranslator.class,

                    "TrimNamesTranslator", "TrimNamesTranslator persistence definition", null, null);

            definition.addDynFieldInt("maxNameLen");

            definition.extend(manager.getDefinition("BaseNamesTranslator"));

        }

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        super.saveToState(state);

        state.set("maxNameLen", maxNameLen);

    }

    @Override

    public void loadFromState(PersistentState state) throws PersistenceException {

        super.loadFromState(state);

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

    }

}