001// License: GPL. For details, see LICENSE file.
002package org.openstreetmap.josm.actions.relation;
003
004import static org.openstreetmap.josm.tools.I18n.tr;
005
006import java.awt.event.ActionEvent;
007import java.util.Collection;
008import java.util.HashSet;
009import java.util.Set;
010
011import org.openstreetmap.josm.Main;
012import org.openstreetmap.josm.data.osm.OsmPrimitive;
013import org.openstreetmap.josm.data.osm.Relation;
014import org.openstreetmap.josm.gui.dialogs.relation.DownloadRelationMemberTask;
015import org.openstreetmap.josm.io.OnlineResource;
016import org.openstreetmap.josm.tools.ImageProvider;
017import org.openstreetmap.josm.tools.Predicate;
018import org.openstreetmap.josm.tools.Utils;
019
020/**
021 * Action for downloading incomplete members of selected relations
022 * @since 5793
023 */
024public class DownloadSelectedIncompleteMembersAction extends AbstractRelationAction {
025
026    /**
027     * Constructs a new <code>DownloadSelectedIncompleteMembersAction</code>.
028     */
029    public DownloadSelectedIncompleteMembersAction() {
030        putValue(SHORT_DESCRIPTION, tr("Download incomplete members of selected relations"));
031        putValue(SMALL_ICON, ImageProvider.get("dialogs/relation", "downloadincompleteselected"));
032        putValue(NAME, tr("Download incomplete members"));
033    }
034
035    /**
036     * Returns the set of incomplete members of the given relations.
037     * @param rels The relations to inspect.
038     * @return The set of incomplete members of the given relations.
039     */
040    public Set<OsmPrimitive> buildSetOfIncompleteMembers(Collection<Relation> rels) {
041        Set<OsmPrimitive> ret = new HashSet<>();
042        for (Relation r : rels) {
043            ret.addAll(r.getIncompleteMembers());
044        }
045        return ret;
046    }
047
048    @Override
049    public void actionPerformed(ActionEvent e) {
050        if (!isEnabled() || relations.isEmpty() || !Main.isDisplayingMapView()) return;
051        Main.worker.submit(new DownloadRelationMemberTask(
052                relations,
053                buildSetOfIncompleteMembers(relations),
054                Main.main.getEditLayer()));
055    }
056
057    @Override
058    public void setPrimitives(Collection<? extends OsmPrimitive> primitives) {
059        // selected relations with incomplete members
060        this.relations = Utils.filter(getRelations(primitives), new Predicate<Relation>(){
061            @Override public boolean evaluate(Relation r) {
062                return !r.isNew() && r.hasIncompleteMembers();
063            }});
064        updateEnabledState();
065    }
066
067    @Override
068    protected void updateEnabledState() {
069        setEnabled(!relations.isEmpty() && !Main.isOffline(OnlineResource.OSM_API));
070    }
071}