1140

pending completion

Build # 1140

Build Type

Pull #261

travis-ci-com

Committed by

web-flow

Commit Message

Merge 353ffb33f into 91c1c61f6

Pull Request Pull Request #261: Wip/cikm camera ready

Run Details

6663 of 12523 branches covered (53.21%)

Branch coverage included in aggregate %.

67 of 67 new or added lines in 6 files covered. (100.0%)

16 existing lines in 6 files now uncovered.

15281 of 24149 relevant lines covered (63.28%)

1236139.25 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

96.24

/lib/dataset-tools/requoting.js

// -*- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// This file is part of Genie
//
// Copyright 2020 The Board of Trustees of the Leland Stanford Junior University
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Author: Giovanni Campagna <gcampagn@cs.stanford.edu>
"use strict";

function findSubstring(sequence, substring) {
    for (let i = 0; i < sequence.length - substring.length + 1; i++) {
        let found = true;
        for (let j = 0; j < substring.length; j++) {
            if (sequence[i+j] !== substring[j]) {
                found = false;
                break;
            }
        }
        if (found)
            return i;
    }
    return -1;
}

const ENTITY_MATCH_REGEX = /^([A-Z].*)_[0-9]+$/;

function findSpanType(program, begin_index, end_index) {
    let spanType;
    if (begin_index > 1 && program[begin_index-2] === 'location:') {
        spanType = 'LOCATION';
    } else if (end_index === program.length - 1 || !program[end_index+1].startsWith('^^')) {
        spanType = 'QUOTED_STRING';
    } else {
        switch (program[end_index+1]) {
        case '^^tt:hashtag':
            spanType = 'HASHTAG';
            break;
        case '^^tt:username':
            spanType = 'USERNAME';
            break;
        default:
            spanType = 'GENERIC_ENTITY_' + program[end_index+1].substring(2);
        }
        end_index++;
    }
    return [spanType, end_index];
}

function* requoteSentence(id, sentence, program) {
    sentence = sentence.split(' ');
    program = program.split(' ');

    const spans = [];
    let in_string = false;
    let begin_index = null;
    let end_index = null;
    for (let i = 0; i < program.length; i++) {
        let token = program[i];
        if (token === '"') {
            in_string = !in_string;
            if (in_string) {
                begin_index = i+1;
            } else {
                end_index = i;
                const substring = program.slice(begin_index, end_index);
                const idx = findSubstring(sentence, substring);
                if (idx < 0)
                    throw new Error(`Cannot find span ${substring.join(' ')} in sentence id ${id}`);

                const spanBegin = idx;
                const spanEnd = idx+end_index-begin_index;

                let spanType;
                [spanType, end_index] = findSpanType(program, begin_index, end_index);
                spans.push([spanBegin, spanEnd, spanType]);
                i = end_index;
            }
        }
    }

    if (spans.length === 0) {
        for (let i = 0; i < sentence.length; i++) {
            let word = sentence[i];
            // if the current word is a QUOTED_STRING, HASHTAG, ENTITY or GENERIC_ENTITY_, remove the last _<id> part
            const entityMatch = ENTITY_MATCH_REGEX.exec(word);
            if (entityMatch !== null)
                yield entityMatch[1];
            else
                yield word;
        }
        return;
    }

    spans.sort((a, b) => {
        const [abegin, aend] = a;
        const [bbegin, bend] = b;
        if (abegin < bbegin)
            return -1;
        if (bbegin < abegin)
            return +1;
        if (aend < bend)
            return -1;
        if (bend < aend)
            return +1;
        return 0;
    });
    let current_span_idx = 0;
    let current_span = spans[0];
    for (let i = 0; i < sentence.length; i++) {
        let word = sentence[i];
        if (current_span === null || i < current_span[0]) {
            // if the current word is a QUOTED_STRING, HASHTAG, ENTITY or GENERIC_ENTITY_, remove the last _<id> part
            //
            // this means that the sentences
            // " tweet foo bar "
            // and
            // " tweet QUOTED_STRING_0 "
            //
            // will both turn into " tweet QUOTED_STRING "
            // and therefore match
            //
            // (this is only relevant if some portion of the data is left quoted
            // after augmentation)
            //
            // for simplicity, we do it only all entities
            // it has no effect on other entities because ids are assigned in increasing
            // number in the sentence
            const entityMatch = ENTITY_MATCH_REGEX.exec(word);
            if (entityMatch !== null)
                yield entityMatch[1];
            else
                yield word;
        } else if (i === current_span[0]) {
            yield current_span[2];
        } else if (i >= current_span[1]) {
            yield word;
            current_span_idx += 1;
            current_span = current_span_idx < spans.length ? spans[current_span_idx] : null;
        }
    }
}

function* requoteProgram(program) {
    if (typeof program === 'string')
        program = program.split(' ');

    let inString = false;
    let begin_index;
    for (let i = 0; i < program.length; i++) {
        const token = program[i];

        if (token === '"') {
            inString = !inString;
            if (inString) {
                begin_index = i+1;
            } else {
                const [spanType, end_index] = findSpanType(program, begin_index, i);
                yield spanType;
                i = end_index;
            }
        } else if (!inString) {
            const entityMatch = ENTITY_MATCH_REGEX.exec(token);
            if (entityMatch !== null)
                yield entityMatch[1];
            else if (token !== 'location:')
                yield token;
        }
    }
}

function* getFunctions(program) {
    if (typeof program === 'string')
        program = program.split(' ');
    let inString = false;
    let isDialoguePolicy = false;
    for (let token of program) {
        if (token === '"') {
            inString = !inString;
        } else if (!inString && token.startsWith('$dialogue')) {
            isDialoguePolicy = true;
        } else if (!inString && token.startsWith('@')) {
            if (isDialoguePolicy) {
                isDialoguePolicy = false;
                continue; // discard. It's the policy, not a function
            }
            yield token;
        }
    }
}

function* getDevices(program) {
    for (let fn of getFunctions(program)) {
        let dot = fn.lastIndexOf('.');
        yield fn.substring(0, dot);
    }
}

module.exports = {
    requoteSentence,
    requoteProgram,
    getFunctions,
    getDevices
};

1	// -- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 --
2	//
3	// This file is part of Genie
4	//
5	// Copyright 2020 The Board of Trustees of the Leland Stanford Junior University
6	//
7	// Licensed under the Apache License, Version 2.0 (the "License");
8	// you may not use this file except in compliance with the License.
9	// You may obtain a copy of the License at
10	//
11	// http://www.apache.org/licenses/LICENSE-2.0
12	//
13	// Unless required by applicable law or agreed to in writing, software
14	// distributed under the License is distributed on an "AS IS" BASIS,
15	// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16	// See the License for the specific language governing permissions and
17	// limitations under the License.
18	//
19	// Author: Giovanni Campagna <gcampagn@cs.stanford.edu>
20	"use strict";
21
22	function findSubstring(sequence, substring) {
23	for (let i = 0; i < sequence.length - substring.length + 1; i++) {	17,509✔
24	let found = true;	210,125✔
25	for (let j = 0; j < substring.length; j++) {	210,125✔
26	if (sequence[i+j] !== substring[j]) {	228,635✔
27	found = false;	192,616✔
28	break;	192,616✔
29	}
30	}
31	if (found)	210,125✔
32	return i;	17,509✔
33	}
34	return -1;	×
35	}
36
37	const ENTITY_MATCH_REGEX = /^([A-Z].*)_[0-9]+$/;	56✔
38
39	function findSpanType(program, begin_index, end_index) {
40	let spanType;
41	if (begin_index > 1 && program[begin_index-2] === 'location:') {	17,929✔
42	spanType = 'LOCATION';	228✔
43	} else if (end_index === program.length - 1 \|\| !program[end_index+1].startsWith('^^')) {	17,701✔
44	spanType = 'QUOTED_STRING';	17,158✔
45	} else {
46	switch (program[end_index+1]) {	543✔
47	case '^^tt:hashtag':
48	spanType = 'HASHTAG';	2✔
49	break;	2✔
50	case '^^tt:username':
51	spanType = 'USERNAME';	48✔
52	break;	48✔
53	default:
54	spanType = 'GENERIC_ENTITY_' + program[end_index+1].substring(2);	493✔
55	}
56	end_index++;	543✔
57	}
58	return [spanType, end_index];	17,929✔
59	}
60
61	function* requoteSentence(id, sentence, program) {
62	sentence = sentence.split(' ');	26,974✔
63	program = program.split(' ');	26,974✔
64
65	const spans = [];	26,974✔
66	let in_string = false;	26,974✔
67	let begin_index = null;	26,974✔
68	let end_index = null;	26,974✔
69	for (let i = 0; i < program.length; i++) {	26,974✔
70	let token = program[i];	562,639✔
71	if (token === '"') {	562,639✔
72	in_string = !in_string;	35,018✔
73	if (in_string) {	35,018✔
74	begin_index = i+1;	17,509✔
75	} else {
76	end_index = i;	17,509✔
77	const substring = program.slice(begin_index, end_index);	17,509✔
78	const idx = findSubstring(sentence, substring);	17,509✔
79	if (idx < 0)	17,509!
80	throw new Error(`Cannot find span ${substring.join(' ')} in sentence id ${id}`);	×
81
82	const spanBegin = idx;	17,509✔
83	const spanEnd = idx+end_index-begin_index;	17,509✔
84
85	let spanType;
86	[spanType, end_index] = findSpanType(program, begin_index, end_index);	17,509✔
87	spans.push([spanBegin, spanEnd, spanType]);	17,509✔
88	i = end_index;	17,509✔
89	}
90	}
91	}
92
93	if (spans.length === 0) {	26,974✔
94	for (let i = 0; i < sentence.length; i++) {	13,723✔
95	let word = sentence[i];	183,307✔
96	// if the current word is a QUOTED_STRING, HASHTAG, ENTITY or GENERIC_ENTITY_, remove the last _<id> part
97	const entityMatch = ENTITY_MATCH_REGEX.exec(word);	183,307✔
98	if (entityMatch !== null)	183,307✔
99	yield entityMatch[1];	1,824✔
100	else
101	yield word;	181,483✔
102	}
103	return;	13,723✔
104	}
105
106	spans.sort((a, b) => {	13,251✔
107	const [abegin, aend] = a;	4,352✔
108	const [bbegin, bend] = b;	4,352✔
109	if (abegin < bbegin)	4,352✔
110	return -1;	1,472✔
111	if (bbegin < abegin)	2,880✔
112	return +1;	2,490✔
113	if (aend < bend)	390✔
114	return -1;	4✔
115	if (bend < aend)	386!
UNCOV 116	return +1;	×
117	return 0;	386✔
118	});
119	let current_span_idx = 0;	13,251✔
120	let current_span = spans[0];	13,251✔
121	for (let i = 0; i < sentence.length; i++) {	13,251✔
122	let word = sentence[i];	219,174✔
123	if (current_span === null \|\| i < current_span[0]) {	219,174✔
124	// if the current word is a QUOTED_STRING, HASHTAG, ENTITY or GENERIC_ENTITY_, remove the last _<id> part
125	//
126	// this means that the sentences
127	// " tweet foo bar "
128	// and
129	// " tweet QUOTED_STRING_0 "
130	//
131	// will both turn into " tweet QUOTED_STRING "
132	// and therefore match
133	//
134	// (this is only relevant if some portion of the data is left quoted
135	// after augmentation)
136	//
137	// for simplicity, we do it only all entities
138	// it has no effect on other entities because ids are assigned in increasing
139	// number in the sentence
140	const entityMatch = ENTITY_MATCH_REGEX.exec(word);	169,311✔
141	if (entityMatch !== null)	169,311✔
142	yield entityMatch[1];	4,189✔
143	else
144	yield word;	165,122✔
145	} else if (i === current_span[0]) {	49,863✔
146	yield current_span[2];	17,116✔
147	} else if (i >= current_span[1]) {	32,747✔
148	yield word;	14,540✔
149	current_span_idx += 1;	14,540✔
150	current_span = current_span_idx < spans.length ? spans[current_span_idx] : null;	14,540✔
151	}
152	}
153	}
154
155	function* requoteProgram(program) {
156	if (typeof program === 'string')	635!
157	program = program.split(' ');	635✔
158
159	let inString = false;	635✔
160	let begin_index;
161	for (let i = 0; i < program.length; i++) {	635✔
162	const token = program[i];	11,365✔
163
164	if (token === '"') {	11,365✔
165	inString = !inString;	840✔
166	if (inString) {	840✔
167	begin_index = i+1;	420✔
168	} else {
169	const [spanType, end_index] = findSpanType(program, begin_index, i);	420✔
170	yield spanType;	420✔
171	i = end_index;	420✔
172	}
173	} else if (!inString) {	10,525✔
174	const entityMatch = ENTITY_MATCH_REGEX.exec(token);	9,366✔
175	if (entityMatch !== null)	9,366✔
176	yield entityMatch[1];	114✔
177	else if (token !== 'location:')	9,252✔
178	yield token;	9,233✔
179	}
180	}
181	}
182
183	function* getFunctions(program) {
184	if (typeof program === 'string')	630!
185	program = program.split(' ');	630✔
186	let inString = false;	630✔
187	let isDialoguePolicy = false;	630✔
188	for (let token of program) {	630✔
189	if (token === '"') {	11,440✔
190	inString = !inString;	838✔
191	} else if (!inString && token.startsWith('$dialogue')) {	10,602✔
192	isDialoguePolicy = true;	5✔
193	} else if (!inString && token.startsWith('@')) {	10,597✔
194	if (isDialoguePolicy) {	1,170✔
195	isDialoguePolicy = false;	5✔
196	continue; // discard. It's the policy, not a function	5✔
197	}
198	yield token;	1,165✔
199	}
200	}
201	}
202
203	function* getDevices(program) {
204	for (let fn of getFunctions(program)) {	5✔
205	let dot = fn.lastIndexOf('.');	8✔
206	yield fn.substring(0, dot);	8✔
207	}
208	}
209
210	module.exports = {	56✔
211	requoteSentence,
212	requoteProgram,
213	getFunctions,
214	getDevices
215	};

stanford-oval / genie-toolkit / 1140

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous