Subversion Repositories mkgmap

/*
 * Copyright (C) 2008 Steve Ratcliffe
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  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.
 *
 *
 * Author: Steve Ratcliffe
 * Create date: 15-Nov-2008
 */
package uk.me.parabola.mkgmap.osmstyle.actions;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import uk.me.parabola.mkgmap.reader.osm.Element;

/**
 * Add a tag, optionally changing it if it already exists.  The value that
 * the tag is set to can have replacements from the current tags.
 *
 * @author Steve Ratcliffe
 */
public class AddTagAction implements Action {
        private final boolean modify;
        private final String tag;
        private final List<ValueBuilder> values = new ArrayList<ValueBuilder>();

        // The tags used to build the value.
        private Element valueTags;

        /**
         * Create an action to add the given tag with a value.
         * If the modify flag is true, then we change the tag if it
         * already exists.
         */
        public AddTagAction(String tag, String value, boolean modify) {
                this.modify = modify;
                this.tag = tag;
                this.values.add(new ValueBuilder(value));
        }

        public void perform(Element el) {
                String tv = el.getTag(tag);
                if (tv != null && !modify)
                        return;

                Element tags = valueTags!=null? valueTags: el;

                for (ValueBuilder value : values) {
                        String newval = value.build(tags, el);
                        if (newval != null) {
                                el.addTag(tag, newval);
                                break;
                        }
                }
        }

        public void add(String value) {
                values.add(new ValueBuilder(value));
        }

        public void setValueTags(Element valueTags) {
                this.valueTags = valueTags;
        }

        public Set<String> getUsedTags() {
                Set<String> set = new HashSet<String>();

                if (values != null) {
                        for (ValueBuilder vb : values) {
                                set.addAll(vb.getUsedTags());
                        }
                }
                return set;
        }

        public String toString() {
                StringBuilder sb = new StringBuilder();
                sb.append(modify ? "set " : "add ");
                sb.append(tag);
                sb.append("=");
                for (int i = 0; i < values.size(); i++) {
                        sb.append(values.get(i));
                        if (i < values.size() - 1)
                                sb.append(" | ");
                }
                sb.append(';');
                return sb.toString();
        }
}