1 | // License: GPL. For details, see LICENSE file.
|
---|
2 | package org.openstreetmap.josm.gui.dialogs.relation;
|
---|
3 |
|
---|
4 | import static org.openstreetmap.josm.tools.I18n.tr;
|
---|
5 |
|
---|
6 | import java.awt.datatransfer.Transferable;
|
---|
7 | import java.awt.datatransfer.UnsupportedFlavorException;
|
---|
8 | import java.io.IOException;
|
---|
9 | import java.util.ArrayList;
|
---|
10 | import java.util.Collection;
|
---|
11 |
|
---|
12 | import javax.swing.JComponent;
|
---|
13 | import javax.swing.JTable;
|
---|
14 | import javax.swing.TransferHandler;
|
---|
15 |
|
---|
16 | import org.openstreetmap.josm.Main;
|
---|
17 | import org.openstreetmap.josm.data.osm.OsmPrimitive;
|
---|
18 | import org.openstreetmap.josm.data.osm.RelationMember;
|
---|
19 | import org.openstreetmap.josm.data.osm.RelationMemberData;
|
---|
20 | import org.openstreetmap.josm.gui.datatransfer.RelationMemberTransferable;
|
---|
21 |
|
---|
22 | class MemberTransferHandler extends TransferHandler {
|
---|
23 |
|
---|
24 | @Override
|
---|
25 | public int getSourceActions(JComponent c) {
|
---|
26 | return COPY_OR_MOVE;
|
---|
27 | }
|
---|
28 |
|
---|
29 | @Override
|
---|
30 | protected Transferable createTransferable(JComponent c) {
|
---|
31 | final MemberTable source = (MemberTable) c;
|
---|
32 | return new RelationMemberTransferable(source.getMemberTableModel().getSelectedMembers());
|
---|
33 | }
|
---|
34 |
|
---|
35 | @Override
|
---|
36 | public boolean canImport(TransferSupport support) {
|
---|
37 | support.setShowDropLocation(true);
|
---|
38 | return support.isDataFlavorSupported(RelationMemberTransferable.RELATION_MEMBER_DATA);
|
---|
39 | }
|
---|
40 |
|
---|
41 | @Override
|
---|
42 | public boolean importData(TransferSupport support) {
|
---|
43 | final int insertRow = ((JTable.DropLocation) support.getDropLocation()).getRow();
|
---|
44 | final Collection<RelationMemberData> memberData;
|
---|
45 | try {
|
---|
46 | //noinspection unchecked
|
---|
47 | memberData = (Collection<RelationMemberData>) support.getTransferable().getTransferData(RelationMemberTransferable.RELATION_MEMBER_DATA);
|
---|
48 | } catch (UnsupportedFlavorException | IOException e) {
|
---|
49 | Main.warn(e);
|
---|
50 | return false;
|
---|
51 | }
|
---|
52 | final MemberTable destination = (MemberTable) support.getComponent();
|
---|
53 |
|
---|
54 | try {
|
---|
55 | importRelationMemberData(memberData, destination, insertRow);
|
---|
56 | } catch (Exception e) {
|
---|
57 | Main.warn(e);
|
---|
58 | throw e;
|
---|
59 | }
|
---|
60 | return true;
|
---|
61 | }
|
---|
62 |
|
---|
63 | protected void importRelationMemberData(Collection<RelationMemberData> memberData, MemberTable destination, int insertRow) {
|
---|
64 | final Collection<RelationMember> membersToAdd = new ArrayList<>(memberData.size());
|
---|
65 | for (RelationMemberData member : memberData) {
|
---|
66 | final OsmPrimitive p = destination.getLayer().data.getPrimitiveById(member.getUniqueId(), member.getType());
|
---|
67 | if (p != null) {
|
---|
68 | membersToAdd.add(new RelationMember(member.getRole(), p));
|
---|
69 | } else {
|
---|
70 | Main.warn(tr("Cannot add {0} since it is not part of dataset", member));
|
---|
71 | }
|
---|
72 | }
|
---|
73 | destination.getMemberTableModel().addMembersAtIndex(membersToAdd, insertRow);
|
---|
74 | }
|
---|
75 |
|
---|
76 | @Override
|
---|
77 | protected void exportDone(JComponent sourceComponent, Transferable data, int action) {
|
---|
78 | if (action != MOVE) {
|
---|
79 | return;
|
---|
80 | }
|
---|
81 | final MemberTable source = (MemberTable) sourceComponent;
|
---|
82 | final MemberTableModel model = source.getMemberTableModel();
|
---|
83 | model.remove(source.getSelectedRows());
|
---|
84 | model.selectionChanged(null);
|
---|
85 | }
|
---|
86 | }
|
---|