Add tool to generate zoneup opcodes if they are offset
Also only accept confident matches if the score is above 0
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
import click
|
||||
import json
|
||||
|
||||
|
||||
@click.command()
|
||||
@click.argument("diff_file", type=click.File("r"))
|
||||
@click.argument("offset", type=int)
|
||||
def generate_diff_offset(diff_file, offset):
|
||||
"""
|
||||
Given a diff file, produces another diff file where the matches
|
||||
are offset by OFFSET.
|
||||
|
||||
Example:
|
||||
python generate_diff_offset.py -- diff.json -1 > offset.diff.json
|
||||
"""
|
||||
diff_json = json.load(diff_file)
|
||||
olds = []
|
||||
news = []
|
||||
for pair in diff_json:
|
||||
if "old" not in pair or "new" not in pair:
|
||||
continue
|
||||
olds.append(pair["old"])
|
||||
news.append(pair["new"])
|
||||
|
||||
if offset < 0:
|
||||
news = [[] for _ in range(-offset)] + news
|
||||
if offset > 0:
|
||||
olds = [[] for _ in range(offset)] + olds
|
||||
|
||||
opcodes_object = []
|
||||
for old, new in zip(olds, news):
|
||||
opcodes_object.append(
|
||||
{
|
||||
"old": old,
|
||||
"new": new,
|
||||
}
|
||||
)
|
||||
|
||||
print(json.dumps(opcodes_object, indent=2))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
generate_diff_offset()
|
||||
Reference in New Issue
Block a user