Mercurial > repos > stevecassidy > textgrid

annotate query_textgrids.py @ 0:e9c8e6204679 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
author stevecassidy
date Mon, 15 Aug 2016 23:47:30 -0400
parents
children be28ced5c4e0
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
1 from __future__ import print_function
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
2 import json
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
3 import argparse
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
4 import pyalveo
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
5 import sys
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
6 import os
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
7 import tgt
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
8
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
9 API_URL = 'https://app.alveo.edu.au' # TODO: export constants to a separate module
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
10
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
11 def parser():
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
12 parser = argparse.ArgumentParser(description="Find matching segments in a TextGrid")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
13 parser.add_argument('--textgrid', required=True, action="store", type=str, help="TextGrid files (comma separated)")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
14 parser.add_argument('--identifier', required=True, action="store", type=str, help="Dataset identifiers (comma separated)")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
15 parser.add_argument('--tier', required=True, action="store", type=str, help="TextGrid Tier to search")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
16 parser.add_argument('--regex', required=True, action="store", type=str, help="Regular expression matching segments")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
17 parser.add_argument('--output_path', required=True, action="store", type=str, help="Path to output file")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
18 return parser.parse_args()
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
19
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
20 def main():
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
21 args = parser()
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
22
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
23 tgfiles = args.textgrid.split(',')
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
24 identifiers = args.identifier.split(',')
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
25 assert len(tgfiles) == len(identifiers), "number of textgrids must match number of identifiers"
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
26
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
27 pairs = zip(tgfiles, identifiers)
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
28
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
29 rows = []
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
30 for tgfile, identifier in pairs:
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
31 tg = tgt.read_textgrid(tgfile)
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
32 tier = tg.get_tier_by_name(args.tier)
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
33 matches = tier.get_annotations_with_text(args.regex, regex=True)
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
34
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
35 for m in matches:
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
36 rows.append((str(m.start_time), str(m.end_time), str(m.duration()), m.text, identifier))
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
37
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
38 with open(args.output_path, 'w') as out:
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
39 out.write("start\tend\tduration\tlabel\tidentifier\n")
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
40 for row in rows:
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
41 out.write('\t'.join(row) + '\n')
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
42
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
43
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
44
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
45 if __name__ == '__main__':
e9c8e6204679 planemo upload commit 0fca33c3b7285bd31f6c7380393d08bbdad4e4d6
stevecassidy
parents:
diff changeset
46 main()