summaryrefslogtreecommitdiffstats
path: root/languagetool/python_rules/enAvsAnRule.py
blob: 110756ad6d92cfedc271425c64f918a4d6d38755 (plain)
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
# Rule that checks the use of 'a' vs. 'an' (English only)
# (c) 2003,2004 Daniel Naber <daniel.naber@t-online.de>
#
#$rcs = ' $Id$ ' ;
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import os
import sys

import Rules
import Tools

class enAvsAnRule(Rules.Rule):
	"""Check if the determiner (if any) before a word is:
	-'an' if the next word starts with a vowel
	-'a' if the next word does not start with a vowel
	This rule knows about some exceptions (e.g. 'an hour')."""

	requires_a_file = os.path.join("data", "det_a.txt")
	requires_an_file = os.path.join("data", "det_an.txt")
	
	def __init__(self):
		Rules.Rule.__init__(self, "DET", "Use of 'a' vs. use of 'an'.", 0, None)
		self.requires_a = self.loadWords(self.requires_a_file) 
		self.requires_an = self.loadWords(self.requires_an_file)
		return

	def loadWords(self, filename):
		f = open(filename)
		l = []
		while 1:
			line = f.readline()
			if not line:
				break
			if line.startswith("#") or line.strip() == '':
				continue
			l.append(line.strip().lower())
		f.close()
		return l
		
	def match(self, tagged_words, chunks, position_fix=0, line_fix=0, column_fix=0):
		matches = []
		text_length = 0
		line_breaks = 0
		column = 0
		i = 0
		while 1:
			if i >= len(tagged_words)-2:
				break
			org_word = tagged_words[i][0]
			org_word_next = tagged_words[i+2][0]
			line_breaks_cur = Tools.Tools.countLinebreaks(org_word)
			if line_breaks_cur > 0:
				column = 0
			line_breaks = line_breaks + line_breaks_cur
			if org_word.lower() == 'a':
				err = 0
				if org_word_next.lower() in self.requires_an:
					err = 1
				elif len(org_word_next) > 0 and org_word_next[0].lower() in ('a', 'e', 'i', 'o', 'u') and \
					not org_word_next.lower() in self.requires_a:
					err = 1
				if err:
					matches.append(Rules.RuleMatch(self.rule_id,
						text_length+position_fix, text_length+len(org_word)+position_fix, 
						line_breaks+line_fix, column+column_fix,
						"<message>Use <em>an</em> instead of <em>a</em> if the following "+
						"word starts with a vowel sound, e.g. 'an article', "+
						"'an hour'</message>", org_word))
			elif org_word.lower() == 'an':
				err = 0
				if org_word_next.lower() in self.requires_a:
					err = 1
				elif len(org_word_next) > 0 and \
					(not org_word_next[0].lower() in ('a', 'e', 'i', 'o', 'u')) and \
					not org_word_next.lower() in self.requires_an:
					err = 1
				if err:
					matches.append(Rules.RuleMatch(self.rule_id,
						text_length+position_fix,
						text_length+len(org_word)+position_fix,
						line_breaks+line_fix, column+column_fix,
						"<message>Use <em>a</em> instead of <em>an</em> if the following "+
						"word doesn't start with a vowel sound, e.g. 'a test', "+
						"'a university'</message>", org_word))
				pass
			if line_breaks_cur == 0:
				column = column + len(org_word)
			text_length = text_length + len(org_word)
			i = i + 1
		return matches