source: osm/applications/editors/josm/plugins/reltoolbox/src/relcontext/relationfix/AssociatedStreetFixer.java@ 28693

package relcontext.relationfix;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.openstreetmap.josm.command.ChangeCommand;
import org.openstreetmap.josm.command.Command;
import org.openstreetmap.josm.command.SequenceCommand;
import org.openstreetmap.josm.data.osm.Node;
import org.openstreetmap.josm.data.osm.OsmPrimitiveType;
import org.openstreetmap.josm.data.osm.Relation;
import org.openstreetmap.josm.data.osm.RelationMember;
import org.openstreetmap.josm.data.osm.Way;

import static org.openstreetmap.josm.tools.I18n.tr;

public class AssociatedStreetFixer extends RelationFixer {

        public AssociatedStreetFixer() {
                super("associatedStreet");
        }

        @Override
        public boolean isRelationGood(Relation rel) {
                for (RelationMember m : rel.getMembers()) {
                if (m.getType().equals(OsmPrimitiveType.NODE) && !"house".equals(m.getRole()))
                        return false;
                if (m.getType().equals(OsmPrimitiveType.WAY) && !("house".equals(m.getRole()) || "street".equals(m.getRole())))
                        return false;
                if (m.getType().equals(OsmPrimitiveType.RELATION) && !"house".equals(m.getRole()))
                        return false;
        }
                // relation should have name
                if (!rel.hasKey("name")) {
                        return false;
                }
                // check that all street members have same name as relation (???)
                String streetName = rel.get("name");
                for (RelationMember m : rel.getMembers()) {
                        if (m.getRole().equals("street") && !m.getWay().get("name").equals(streetName))
                                return false;
                }
                return true;
        }

        @Override
        public Command fixRelation(Relation source) {
                // any way with highway tag -> street
                // any way/point/relation with addr:housenumber=* or building=* or type=multipolygon -> house
                // name - check which name is most used in street members and add to relation
                // copy this name to the other street members (???)
                Relation rel = new Relation(source);
                boolean fixed = false;
                
                for (int i = 0; i < rel.getMembersCount(); i++) {
                        RelationMember m = rel.getMember(i);
                        
                        if (m.isNode()) {
                                Node node = m.getNode();
                                if (!"house".equals(m.getRole()) && 
                                                (node.hasKey("building") || node.hasKey("addr:housenumber"))) {
                                        fixed = true;
                                        rel.setMember(i, new RelationMember("house", node));
                                }
                        } else if (m.isWay()) {
                                Way way = m.getWay();
                                if (!"street".equals(m.getRole()) && way.hasKey("highway")) {
                                        fixed = true;
                                        rel.setMember(i, new RelationMember("street", way));
                                } else if (!"house".equals(m.getRole()) && 
                                                (way.hasKey("building") || way.hasKey("addr:housenumber"))) {
                                        fixed = true;
                                        rel.setMember(i,  new RelationMember("house", way));
                                }
                        } else if (m.isRelation()) {
                                Relation relation = m.getRelation();
                                if (!"house".equals(m.getRole()) && 
                                                (relation.hasKey("building") || relation.hasKey("addr:housenumber") || "multipolygon".equals(relation.get("type")))) {
                                        fixed = true;
                                        rel.setMember(i, new RelationMember("house", relation));
                                }
                        }
                }
                
                // fill relation name
                Map<String, Integer> streetNames = new HashMap<String, Integer>();
                for (RelationMember m : rel.getMembers()) 
                        if ("street".equals(m.getRole()) && m.isWay()) {
                                String name = m.getWay().get("name");
                                if (name == null || name.isEmpty()) continue;
                                
                                Integer count = streetNames.get(name);
                                
                                streetNames.put(name, count != null? count + 1 : 1);
                        }
                String commonName = "";
                Integer commonCount = 0;
                for (Map.Entry<String, Integer> entry : streetNames.entrySet()) {
                        if (entry.getValue() > commonCount) {
                                commonCount = entry.getValue();
                                commonName = entry.getKey();
                        }
                }
                
                if (!rel.hasKey("name")) {
                        fixed = true;
                        rel.put("name", commonName);
                } else {
                        commonName = ""; // set empty common name - if we already have name on relation, do not overwrite it
                }
                
                List<Command> commandList = new ArrayList<Command>();
                if (fixed) {
                        commandList.add(new ChangeCommand(source, rel));
                }
                
                /*if (!commonName.isEmpty())
                // fill common name to streets
                for (RelationMember m : rel.getMembers()) 
                        if ("street".equals(m.getRole()) && m.isWay()) {
                                String name = m.getWay().get("name");
                                if (commonName.equals(name)) continue;
                                
                                // TODO: ask user if he really wants to overwrite street name??
                                
                                Way oldWay = m.getWay();
                                Way newWay = new Way(oldWay);
                                newWay.put("name", commonName);
                                
                                commandList.add(new ChangeCommand(oldWay, newWay));
                        }
                */
                // return results
                if (commandList.size() == 0)
                        return null;
                if (commandList.size() == 1)
                        return commandList.get(0);
                return new SequenceCommand(tr("fix associatedStreet relation"), commandList);
        }
}