-
Notifications
You must be signed in to change notification settings - Fork 1
/
combine_recode.rb
118 lines (93 loc) · 3.61 KB
/
combine_recode.rb
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
require 'Datavyu_API'
# $input_dir = "~/code/work/seedlings/reliability_data/by_month/07/recode_and_orig_opfs_partial"
# $output_dir = "~/code/work/seedlings/reliability_data/by_month/07/converge_out"
$input_dir = "/Volumes/pn-opus/Seedlings/Working_Files/annot_id/video/recode_and_orig_opfs/"
$output_dir = "/Volumes/pn-opus/Seedlings/Working_Files/annot_id/video/converge_out/"
def cells_equal(cell1, cell2)
if cell1.get_codes() == cell2.get_codes() &&
cell1.ordinal == cell2.ordinal &&
cell1.onset == cell2.onset && cell1.offset == cell2.offset
return true
else
return false
end
end
def combine(in_dir, groups)
groups.each_value { |files|
prefix = files["orig"][0..4]
puts("****************************#{prefix}*********")
$db, $pj = load_db(File.join(in_dir, files["recode"]))
recode_col = get_column("recode")
$db, $pj = load_db(File.join(in_dir, files["orig"]))
orig_col = get_column("recode")
new_orig_col = create_new_column("original", "object", "utterance_type", "object_present", "speaker", "id", "original_ordinal")
new_recode_col = create_new_column("recode", "object", "utterance_type", "object_present", "speaker", "id", "original_ordinal")
for cell in orig_col.cells
found_match = false
for reco_cell in recode_col.cells
# puts("#{cell.get_codes()}\n#{reco_cell.get_codes()}\n\n")
if cells_equal(cell, reco_cell)
# puts("#{cell.get_codes()}\n#{reco_cell.get_codes()}\n\n")
# puts("found a match\n\n")
found_match = true
break
end
end
if !found_match
newcell = new_orig_col.make_new_cell()
newcell.change_code("object", cell.object)
newcell.change_code("utterance_type", cell.utterance_type)
newcell.change_code("object_present", cell.object_present)
newcell.change_code("speaker", cell.speaker)
newcell.change_code("id", cell.id)
newcell.change_code("onset", cell.onset)
newcell.change_code("offset", cell.offset)
newcell.change_code("ordinal", cell.ordinal)
newcell.change_code("original_ordinal", cell.original_ordinal)
not_matched = recode_col.cells[cell.ordinal-1]
newrecode = new_recode_col.make_new_cell()
newrecode.change_code("object", not_matched.object)
newrecode.change_code("utterance_type", not_matched.utterance_type)
newrecode.change_code("object_present", not_matched.object_present)
newrecode.change_code("speaker", not_matched.speaker)
newrecode.change_code("id", cell.id)
newrecode.change_code("onset", not_matched.onset)
newrecode.change_code("offset", not_matched.offset)
newrecode.change_code("ordinal", not_matched.ordinal)
newrecode.change_code("original_ordinal", not_matched.original_ordinal)
end
end
delete_variable(orig_col)
set_column(new_orig_col)
set_column(new_recode_col)
if (new_recode_col.cells.size != 0)
save_db(File.join(File.expand_path($output_dir), "#{prefix}_converge_rel.opf"))
end
}
end
begin
in_dir = File.expand_path($input_dir)
filenames = Dir.new(in_dir).entries
groups = Hash.new
for file in filenames
if file.end_with? ".opf"
prefix = file[0..4]
if groups.has_key?(prefix)
if file.include? "_orig"
groups[prefix]["orig"] = file
else
groups[prefix]["recode"] = file
end
else
groups[prefix] = Hash.new
if file.include? "_orig"
groups[prefix]["orig"] = file
else
groups[prefix]["recode"] = file
end
end
end
end
puts(groups)
combine(in_dir, groups)
end