StaticBugCheckers/python/CompareDiffsToErrorprone.py at master · sola-da/StaticBugCheckers · GitHub

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
'''

Created on Nov. 30, 2017

@author Andrew Habib

'''

import json
import os
import sys

from Util import load_parsed_diffs, load_parsed_ep, find_msg_by_proj_and_cls, \
                LineMatchesToMessages, CustomEncoder


def match_diff_ep(d, ep_list):
    matches = []
    lines_matches = []
    for inst in ep_list:
        if inst.line in d.lines:
            matches.append(inst)
            lines_matches.append(inst.line)
    return matches, set(lines_matches)


def get_hits_diffs_ep(diffs, ep_res_set):
    ep_count = 0
    ep_all_matches = []
    diffs_match_ep = []
    for d in diffs:

        proj = d.proj
        cls = d.cls

        ep_list = find_msg_by_proj_and_cls(proj, cls, ep_res_set)
        diff_ep, lines = match_diff_ep(d, ep_list)
        if diff_ep:

            ep_count += len(diff_ep)
            ep_all_matches.append(LineMatchesToMessages(lines, diff_ep))
            diffs_match_ep.extend(diff_ep)
#     print(ep_count)

#     return ep_all_matches
    return diffs_match_ep


if __name__ == '__main__':

    """Get lines matches between each tool and  bug fixes diffs"""

    diffs_file = os.path.join(os.getcwd(), sys.argv[1])
    diffs = load_parsed_diffs(diffs_file)

    ep_file = os.path.join(os.getcwd(), sys.argv[2])
    ep_res_set = load_parsed_ep(ep_file)

    diffs_ep = get_hits_diffs_ep(diffs, ep_res_set)

    output_file_name = "ep_diffs_warnings.json"
    with open(output_file_name, "w") as file:
        json.dump(diffs_ep, file, cls=CustomEncoder, indent=4)