/
AllNodesClassifier.java
64 lines (60 loc) · 2.4 KB
/
AllNodesClassifier.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
/*
* This file is part of GumTree.
*
* GumTree is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* GumTree 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with GumTree. If not, see <http://www.gnu.org/licenses/>.
*
* Copyright 2011-2015 Jean-Rémy Falleri <jr.falleri@gmail.com>
* Copyright 2011-2015 Floréal Morandat <florealm@gmail.com>
*/
package com.github.gumtreediff.actions;
import com.github.gumtreediff.actions.model.*;
import com.github.gumtreediff.actions.model.Delete;
import com.github.gumtreediff.actions.model.Move;
import com.github.gumtreediff.actions.model.Update;
import com.github.gumtreediff.matchers.MappingStore;
/**
* Partition all moved, inserted, updated or deleted nodes.
*/
public class AllNodesClassifier extends AbstractITreeClassifier {
public AllNodesClassifier(Diff diff) {
super(diff);
}
@Override
public void classify() {
for (Action a: diff.editScript) {
if (a instanceof Delete)
srcDelTrees.add(a.getNode());
else if (a instanceof TreeDelete) {
srcDelTrees.add(a.getNode());
srcDelTrees.addAll(a.getNode().getDescendants());
}
else if (a instanceof Insert)
dstAddTrees.add(a.getNode());
else if (a instanceof TreeInsert) {
dstAddTrees.add(a.getNode());
dstAddTrees.addAll(a.getNode().getDescendants());
}
else if (a instanceof Update) {
srcUpdTrees.add(a.getNode());
dstUpdTrees.add(diff.mappings.getDstForSrc(a.getNode()));
}
else if (a instanceof Move) {
srcMvTrees.add(a.getNode());
srcMvTrees.addAll(a.getNode().getDescendants());
dstMvTrees.add(diff.mappings.getDstForSrc(a.getNode()));
dstMvTrees.addAll(diff.mappings.getDstForSrc(a.getNode()).getDescendants());
}
}
}
}