Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.expressionevaluator.impl.symboltable;

import java.util.AbstractMap;

import java.util.AbstractSet;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Set;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.Expression;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.function.programming.CreateFnFunction;

import org.gvsig.expressionevaluator.impl.function.programming.VarFunction;

import org.gvsig.expressionevaluator.spi.AbstractSymbolTable;

import org.gvsig.tools.bookmarksandhistory.Bookmark;

import org.gvsig.tools.bookmarksandhistory.Bookmarks;

import org.gvsig.tools.exception.BaseException;

/**

 *

 * @author gvSIG Team

 */

public class BookmarksSymbolTable

        extends AbstractSymbolTable

        implements SymbolTable {

    static final String NAME = "Bookmarks";

    private static class BookmarksFunctionsEntrySet

            extends AbstractSet<Map.Entry<String, Function>>

            implements Set<Map.Entry<String, Function>> {

        @Override

        public int size() {

            Bookmarks<Expression> bookmarks = ExpressionEvaluatorLocator.getExpressionEvaluatorManager().getBookmarks();

            int count = 0;

            for (Bookmark<Expression> bookmark : bookmarks) {

                if (StringUtils.startsWith(bookmark.getName(), "$")) {

                    count++;

                }

            }

            return count;

        }

        @Override

        public Iterator<Map.Entry<String, Function>> iterator() {

            final Bookmarks<Expression> bookmarks = ExpressionEvaluatorLocator.getExpressionEvaluatorManager().getBookmarks();

            Iterator<Map.Entry<String, Function>> it = new Iterator<Map.Entry<String, Function>>() {

                private final Iterator<Bookmark<Expression>> bookmarksIt = bookmarks.iterator();

                private Bookmark<Expression> current = null;

                @Override

                public boolean hasNext() {

                    while (bookmarksIt.hasNext()) {

                        this.current = this.bookmarksIt.next();

                        if (StringUtils.startsWith(this.current.getName(), "$")) {

                            return true;

                        }

                    }

                    return false;

                }

                @Override

                public Map.Entry<String, Function> next() {

                    if (this.current == null) {

                        return null;

                    }

                    Code code = this.current.getValue().getCode();

                    List<String> argNames = new ArrayList<>();

                    try {

                        code.accept((Object o) -> {

                            if (((Code) o).code() == Code.CALLABLE) {

                                Code.Callable c = (Code.Callable) o;

                                if(StringUtils.equalsIgnoreCase(c.name(), VarFunction.NAME)){

                                    argNames.add(((Code.Identifier) c.parameters().get(0)).name());

                                }

                            }

                        });

                    } catch (BaseException ex) {

                    }

                    Function function = new CreateFnFunction.UserFunction(

                            NAME,

                            this.current.getName(),

                            argNames,

                            code

                    );

                    return new AbstractMap.SimpleEntry(

                            this.current.getName().toUpperCase(),

                            function

                    );

                }

            };

            return it;

        }

    }

    public BookmarksSymbolTable() {

        super(NAME);

        this.functions = new AbstractMap<String, Function>() {

            @Override

            public Set<Map.Entry<String, Function>> entrySet() {

                return new BookmarksFunctionsEntrySet();

            }

        };

    }

    @Override

    protected Map<String, Object> getVars() {

        return Collections.EMPTY_MAP;

    }

}