Application: gvSIG desktop

package org.gvsig.xml2db.lib.impl.xmlinfo;

import java.nio.charset.Charset;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Iterator;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Locale;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.IProjection;

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

import org.gvsig.fmap.geom.DataTypes;

import org.gvsig.tools.util.CompareUtils;

import org.gvsig.xml2db.lib.api.xmlinfo.XMLInfo;

import org.gvsig.xml2db.lib.api.xmlinfo.XMLTableInfo;

/**

 *

 * @author jjdelcerro

 */

public class XMLInfoImpl implements XMLInfo {

    private final Map<String,XMLAttributeInfoImpl> tags;

    private final Map<String,XMLTableInfo> tables;

    private IProjection srid;

    private long countLines;

    private String repositoryID;

    private Charset charset;

    private Locale locale;

    private String tablePrefix;

    public XMLInfoImpl() {

        this.tags = new LinkedHashMap<>();

        this.tables = new LinkedHashMap<>();

        this.countLines = -1;

    }

    @Override

    public IProjection getSrid() {

        return this.srid;

    }

    @Override

    public void setSrid(IProjection srid) {

        this.srid = srid;

    }

    public XMLAttributeInfoImpl getTag(String path) {

        return this.tags.get(path);

    }

    public Collection<String> getTagsPaths() {

        return this.tags.keySet();

    }

    public void addTag(XMLAttributeInfoImpl info) {

        tags.put(info.getPath(), info);

    }

    public XMLAttributeInfoImpl getTagInfo(String path) {

        return this.tags.get(path);

    }

    public void addTable(XMLTableInfoImpl tableInfo) {

        tables.put(tableInfo.getPath(), tableInfo);

    }

    public XMLTableInfoImpl getTableByPath(String path) {

        return (XMLTableInfoImpl) this.tables.get(path);

    }

    public boolean existsTableByPath(String path) {

        return this.tables.containsKey(path);

    }

    @Override

    public int size() {

        return this.tables.size();

    }

    @Override

    public XMLTableInfo get(String name) {

        for (XMLTableInfo tableInfo : this) {

            if( StringUtils.equals(name, tableInfo.getName()) ) {

                return tableInfo;

            }

        }

        for (XMLTableInfo tableInfo : this) {

            if( StringUtils.equals(name, this.getNameWithPrefix(tableInfo.getName())) ) {

                return tableInfo;

            }

        }

        return null;

    }

//    public XMLTableInfo getByOriginalName(String name) {

//        for (XMLTableInfo tableInfo : this) {

//            if( StringUtils.equals(name, ((XMLTableInfoImpl)tableInfo).getOriginalName()) ) {

//                return tableInfo;

//            }

//        }

//        return null;

//    }

//

    @Override

    public List<String> getKeys() {

        List<String> names = new ArrayList<>(this.tables.keySet());

        names.sort(CompareUtils.EQUALS_IGNORECASE_COMPARATOR);

        return names;

    }

    @Override

    public Iterator<XMLTableInfo> iterator() {

        return this.tables.values().iterator();

    }

    @Override

    public boolean isEmpty() {

        return this.tables.isEmpty();

    }

    public void setCountLines(long countlines) {

        this.countLines = countlines;

    }

    public long getCountLines() {

        return this.countLines;

    }

    @Override

    public String getRepositoryID() {

        return this.repositoryID;

    }

    @Override

    public void setRepositoryID(String id) {

        this.repositoryID = id;

    }

    @Override

    public Charset getCharset() {

        return this.charset;

    }

    public void setCharset(Charset charset) {

        this.charset = charset;

    }

    @Override

    public Locale getLocale() {

        return this.locale;

    }

    public void setLocale(Locale locale) {

        this.locale = locale;

    }

    @Override

    public boolean hasGeometries() {

        for (XMLTableInfo tableInfo : this) {

            for (FeatureAttributeDescriptor ft : tableInfo.getFeatureType()) {

                if( ft.getType()==DataTypes.GEOMETRY ) {

                    return true;

                }

            }

        }

        return false;

    }

    @Override

    public void setTablePrefix(String tablePrefix) {

        this.tablePrefix = tablePrefix;

    }

    @Override

    public String getTablePrefix() {

        return this.tablePrefix;

    }

    @Override

    public String getNameWithPrefix(String name) {

        return StringUtils.join(this.tablePrefix,name);

    }

}