Commit b768fe9411ceed100a37d70ed710138e98bf00ed
1 parent
ca96fb31f8
Exists in
master
bugfix in document, more stopwords
Showing 2 changed files with 3 additions and 2 deletions Inline Diff
lib/mirimiri/document.rb
1 | #!/usr/bin/env ruby | 1 | #!/usr/bin/env ruby |
2 | 2 | ||
3 | #-- | 3 | #-- |
4 | # This file is a part of the mirimiri library | 4 | # This file is a part of the mirimiri library |
5 | # | 5 | # |
6 | # Copyright (C) 2010-2011 Romain Deveaud <romain.deveaud@gmail.com> | 6 | # Copyright (C) 2010-2011 Romain Deveaud <romain.deveaud@gmail.com> |
7 | # | 7 | # |
8 | # This program is free software: you can redistribute it and/or modify | 8 | # This program is free software: you can redistribute it and/or modify |
9 | # it under the terms of the GNU General Public License as published by | 9 | # it under the terms of the GNU General Public License as published by |
10 | # the Free Software Foundation, either version 3 of the License, or | 10 | # the Free Software Foundation, either version 3 of the License, or |
11 | # (at your option) any later version. | 11 | # (at your option) any later version. |
12 | # | 12 | # |
13 | # This program is distributed in the hope that it will be useful, | 13 | # This program is distributed in the hope that it will be useful, |
14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
16 | # GNU General Public License for more details. | 16 | # GNU General Public License for more details. |
17 | # | 17 | # |
18 | # You should have received a copy of the GNU General Public License | 18 | # You should have received a copy of the GNU General Public License |
19 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | 19 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
20 | #++ | 20 | #++ |
21 | 21 | ||
22 | 22 | ||
23 | # General module | 23 | # General module |
24 | module Mirimiri | 24 | module Mirimiri |
25 | 25 | ||
26 | # A Document is a bag of words and is constructed from a string. | 26 | # A Document is a bag of words and is constructed from a string. |
27 | class Document | 27 | class Document |
28 | attr_reader :words, :doc_content, :count_words | 28 | attr_reader :words, :doc_content, :count_words |
29 | 29 | ||
30 | # Any non-word characters are removed from the words (see http://perldoc.perl.org/perlre.html | 30 | # Any non-word characters are removed from the words (see http://perldoc.perl.org/perlre.html |
31 | # and the \\W special escape). | 31 | # and the \\W special escape). |
32 | # | 32 | # |
33 | # Protected function, only meant to by called at the initialization. | 33 | # Protected function, only meant to by called at the initialization. |
34 | def format_words | 34 | def format_words |
35 | wo = [] | 35 | wo = [] |
36 | 36 | ||
37 | @doc_content.split.each do |w| | 37 | @doc_content.split.each do |w| |
38 | w.split(/\W/).each do |sw| | 38 | w.split(/\W/).each do |sw| |
39 | wo.push(sw.downcase) if sw =~ /[a-zA-Z]/ | 39 | wo.push(sw.downcase) if sw =~ /[a-zA-Z]/ |
40 | end | 40 | end |
41 | end | 41 | end |
42 | 42 | ||
43 | wo | 43 | wo |
44 | end | 44 | end |
45 | 45 | ||
46 | # Returns an Array containing the +n+-grams (words) from the current Document. | 46 | # Returns an Array containing the +n+-grams (words) from the current Document. |
47 | # | 47 | # |
48 | # ngrams(2) #=> ["the free", "free encyclopedia", "encyclopedia var", "var skin", ...] | 48 | # ngrams(2) #=> ["the free", "free encyclopedia", "encyclopedia var", "var skin", ...] |
49 | def ngrams(n) | 49 | def ngrams(n) |
50 | window = [] | 50 | window = [] |
51 | ngrams_array = [] | 51 | ngrams_array = [] |
52 | 52 | ||
53 | @words.each do |w| | 53 | @words.each do |w| |
54 | window.push(w) | 54 | window.push(w) |
55 | if window.size == n | 55 | if window.size == n |
56 | ngrams_array.push window.join(" ") | 56 | ngrams_array.push window.join(" ") |
57 | window.delete_at(0) | 57 | window.delete_at(0) |
58 | end | 58 | end |
59 | end | 59 | end |
60 | 60 | ||
61 | ngrams_array.uniq | 61 | ngrams_array.uniq |
62 | end | 62 | end |
63 | 63 | ||
64 | # Returns a Hash containing the words and their associated counts in the current Document. | 64 | # Returns a Hash containing the words and their associated counts in the current Document. |
65 | # | 65 | # |
66 | # count_words #=> { "guitar"=>1, "bass"=>3, "album"=>20, ... } | 66 | # count_words #=> { "guitar"=>1, "bass"=>3, "album"=>20, ... } |
67 | def count_words | 67 | def count_words |
68 | counts = Hash.new { |h,k| h[k] = 0 } | 68 | counts = Hash.new { |h,k| h[k] = 0 } |
69 | @words.each { |w| counts[w] += 1 } | 69 | @words.each { |w| counts[w] += 1 } |
70 | 70 | ||
71 | counts | 71 | counts |
72 | end | 72 | end |
73 | 73 | ||
74 | # Computes the entropy of a given string +s+ inside the document. | 74 | # Computes the entropy of a given string +s+ inside the document. |
75 | # | 75 | # |
76 | # If the string parameter is composed of many words (i.e. tokens separated | 76 | # If the string parameter is composed of many words (i.e. tokens separated |
77 | # by whitespace(s)), it is considered as an ngram. | 77 | # by whitespace(s)), it is considered as an ngram. |
78 | # | 78 | # |
79 | # entropy("guitar") #=> 0.00432114812727959 | 79 | # entropy("guitar") #=> 0.00432114812727959 |
80 | # entropy("dillinger escape plan") #=> 0.265862076325102 | 80 | # entropy("dillinger escape plan") #=> 0.265862076325102 |
81 | def entropy(s) | 81 | def entropy(s) |
82 | en = 0.0 | 82 | en = 0.0 |
83 | 83 | ||
84 | s.split.each do |w| | 84 | s.split.each do |w| |
85 | p_wi = @count_words[w].to_f/@words.count.to_f | 85 | p_wi = @count_words[w].to_f/@words.count.to_f |
86 | en += p_wi*Math.log2(p_wi) | 86 | en += p_wi*Math.log2(p_wi) |
87 | end | 87 | end |
88 | 88 | ||
89 | en *= -1 | 89 | en *= -1 |
90 | en | 90 | en |
91 | end | 91 | end |
92 | 92 | ||
93 | # Computes the term frequency of a given *word* +s+. | 93 | # Computes the term frequency of a given *word* +s+. |
94 | # | 94 | # |
95 | # tf("guitar") #=> 0.000380372765310004 | 95 | # tf("guitar") #=> 0.000380372765310004 |
96 | def tf(s) | 96 | def tf(s) |
97 | @count_words[s].to_f/@words.size.to_f | 97 | @count_words[s].to_f/@words.size.to_f |
98 | end | 98 | end |
99 | 99 | ||
100 | 100 | ||
101 | def initialize(content="") | 101 | def initialize(content="") |
102 | @doc_content = content | 102 | @doc_content = content |
103 | @words = format_words | 103 | @words = format_words |
104 | @count_words = count_words | 104 | @count_words = count_words |
105 | end | 105 | end |
106 | 106 | ||
107 | protected :format_words, :count_words | 107 | protected :format_words, :count_words |
108 | end | 108 | end |
109 | 109 | ||
110 | # A WebDocument is a Document with a +url+. | 110 | # A WebDocument is a Document with a +url+. |
111 | class WebDocument < Document | 111 | class WebDocument < Document |
112 | attr_reader :url | 112 | attr_reader :url |
113 | 113 | ||
114 | # Returns the HTML text from the page of a given +url+. | 114 | # Returns the HTML text from the page of a given +url+. |
115 | def self.get_content(url) | 115 | def self.get_content(url) |
116 | require 'net/http' | 116 | require 'net/http' |
117 | Net::HTTP.get(URI.parse(url)) | 117 | Net::HTTP.get(URI.parse(url)) |
118 | end | 118 | end |
119 | 119 | ||
120 | # WebDocument constructor, the content of the Document is the HTML page | 120 | # WebDocument constructor, the content of the Document is the HTML page |
121 | # without the tags. | 121 | # without the tags. |
122 | def initialize(url,only_tags=nil) | 122 | def initialize(url,only_tags=nil) |
123 | @url = url | 123 | @url = url |
124 | content = only_tags.nil? ? WebDocument.get_content(url) : WebDocument.get_content(url).extract_xmltags_values(only_tags).join("") | 124 | content = only_tags.nil? ? WebDocument.get_content(url) : WebDocument.get_content(url).extract_xmltags_values(only_tags).join("") |
125 | super content.strip_javascripts.strip_xml_tags | 125 | super content.strip_javascripts.strip_xml_tags |
126 | end | 126 | end |
127 | end | 127 | end |
128 | 128 | ||
129 | # A WikipediaPage is a WebDocument. | 129 | # A WikipediaPage is a WebDocument. |
130 | class WikipediaPage < WebDocument | 130 | class WikipediaPage < WebDocument |
131 | require 'rexml/document' | 131 | require 'rexml/document' |
132 | require 'net/http' | 132 | require 'net/http' |
133 | require 'kconv' | 133 | require 'kconv' |
134 | 134 | ||
135 | 135 | ||
136 | def self.search_wikipedia_titles(name) | 136 | def self.search_wikipedia_titles(name) |
137 | raise ArgumentError, "Bad encoding", name unless name.isutf8 | 137 | raise ArgumentError, "Bad encoding", name unless name.isutf8 |
138 | 138 | ||
139 | res = REXML::Document.new(Net::HTTP.get( URI.parse "http://en.wikipedia.org/w/api.php?action=query&list=search&srsearch=#{URI.escape name}&format=xml" ).unaccent.toutf8).elements['api/query/search'] | 139 | res = REXML::Document.new(Net::HTTP.get( URI.parse "http://en.wikipedia.org/w/api.php?action=query&list=search&srsearch=#{URI.escape name}&format=xml" ).unaccent.toutf8).elements['api/query/search'] |
140 | 140 | ||
141 | res.collect { |e| e.attributes['title'] } unless res.nil? | 141 | res.collect { |e| e.attributes['title'] } unless res.nil? |
142 | end | 142 | end |
143 | 143 | ||
144 | def self.get_url(name) | 144 | def self.get_url(name) |
145 | raise ArgumentError, "Bad encoding", name unless name.isutf9 | 145 | raise ArgumentError, "Bad encoding", name unless name.isutf8 |
146 | 146 | ||
147 | atts = REXML::Document.new(Net::HTTP.get( URI.parse "http://en.wikipedia.org/w/api.php?action=query&titles=#{URI.escape name}&inprop=url&prop=info&format=xml" ).unaccent.toutf8).elements['api/query/pages/page'].attributes | 147 | atts = REXML::Document.new(Net::HTTP.get( URI.parse "http://en.wikipedia.org/w/api.php?action=query&titles=#{URI.escape name}&inprop=url&prop=info&format=xml" ).unaccent.toutf8).elements['api/query/pages/page'].attributes |
148 | 148 | ||
149 | atts['fullurl'] if atts['missing'].nil? | 149 | atts['fullurl'] if atts['missing'].nil? |
150 | end | 150 | end |
151 | 151 | ||
152 | def self.search_homepage(name) | 152 | def self.search_homepage(name) |
153 | title = WikipediaPage.search_wikipedia_titles name | 153 | title = WikipediaPage.search_wikipedia_titles name |
154 | 154 | ||
155 | WikipediaPage.get_url(title[0]) unless title.nil? || title.empty? | 155 | WikipediaPage.get_url(title[0]) unless title.nil? || title.empty? |
156 | end | 156 | end |
157 | 157 | ||
158 | end | 158 | end |
159 | end | 159 | end |
160 | 160 |
lib/mirimiri/string.rb
1 | #!/usr/bin/env ruby | 1 | #!/usr/bin/env ruby |
2 | 2 | ||
3 | #-- | 3 | #-- |
4 | # This file is a part of the mirimiri library | 4 | # This file is a part of the mirimiri library |
5 | # | 5 | # |
6 | # Copyright (C) 2010-2011 Romain Deveaud <romain.deveaud@gmail.com> | 6 | # Copyright (C) 2010-2011 Romain Deveaud <romain.deveaud@gmail.com> |
7 | # | 7 | # |
8 | # This program is free software: you can redistribute it and/or modify | 8 | # This program is free software: you can redistribute it and/or modify |
9 | # it under the terms of the GNU General Public License as published by | 9 | # it under the terms of the GNU General Public License as published by |
10 | # the Free Software Foundation, either version 3 of the License, or | 10 | # the Free Software Foundation, either version 3 of the License, or |
11 | # (at your option) any later version. | 11 | # (at your option) any later version. |
12 | # | 12 | # |
13 | # This program is distributed in the hope that it will be useful, | 13 | # This program is distributed in the hope that it will be useful, |
14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | 14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
16 | # GNU General Public License for more details. | 16 | # GNU General Public License for more details. |
17 | # | 17 | # |
18 | # You should have received a copy of the GNU General Public License | 18 | # You should have received a copy of the GNU General Public License |
19 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | 19 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
20 | #++ | 20 | #++ |
21 | 21 | ||
22 | module Mirimiri | 22 | module Mirimiri |
23 | 23 | ||
24 | # These are the default stopwords provided by Lemur. | 24 | # These are the default stopwords provided by Lemur. |
25 | Stoplist = [ | 25 | Stoplist = [ |
26 | "a","about","above","according","across","after","afterwards","again","against", | 26 | "a","about","above","according","across","after","afterwards","again","against", |
27 | "albeit","all","almost","alone","along","already","also","although","always","am", | 27 | "albeit","all","almost","alone","along","already","also","although","always","am", |
28 | "among","amongst","an","and","another","any","anybody","anyhow","anyone","anything", | 28 | "among","amongst","an","and","another","any","anybody","anyhow","anyone","anything", |
29 | "anyway","anywhere","apart","are","around","as","at","av","be","became","because", | 29 | "anyway","anywhere","apart","are","around","as","at","av","be","became","because", |
30 | "become","becomes","becoming","been","before","beforehand","behind","being","below", | 30 | "become","becomes","becoming","been","before","beforehand","behind","being","below", |
31 | "beside","besides","between","beyond","both","but","by","can","cannot","canst", | 31 | "beside","besides","between","beyond","both","but","by","can","cannot","canst", |
32 | "certain","cf","choose","contrariwise","cos","could","cu","day","do","does","doesn't", | 32 | "certain","cf","choose","contrariwise","cos","could","cu","day","do","does","doesn't", |
33 | "doing","dost","doth","double","down","dual","during","each","either","else", | 33 | "doing","dost","doth","double","down","dual","during","each","either","else", |
34 | "elsewhere","enough","et","etc","even","ever","every","everybody","everyone", | 34 | "elsewhere","enough","et","etc","even","ever","every","everybody","everyone", |
35 | "everything","everywhere","except","excepted","excepting","exception","exclude", | 35 | "everything","everywhere","except","excepted","excepting","exception","exclude", |
36 | "excluding","exclusive","far","farther","farthest","few","ff","first","for", | 36 | "excluding","exclusive","far","farther","farthest","few","ff","first","for", |
37 | "formerly","forth","forward","from","front","further","furthermore","furthest","get", | 37 | "formerly","forth","forward","from","front","further","furthermore","furthest","get", |
38 | "go","had","halves","hardly","has","hast","hath","have","he","hence","henceforth", | 38 | "go","had","halves","hardly","has","hast","hath","have","he","hence","henceforth", |
39 | "her","here","hereabouts","hereafter","hereby","herein","hereto","hereupon","hers", | 39 | "her","here","hereabouts","hereafter","hereby","herein","hereto","hereupon","hers", |
40 | "herself","him","himself","hindmost","his","hither","hitherto","how","however", | 40 | "herself","him","himself","hindmost","his","hither","hitherto","how","however", |
41 | "howsoever","i","ie","if","in","inasmuch","inc","include","included","including", | 41 | "howsoever","i","ie","if","in","inasmuch","inc","include","included","including", |
42 | "indeed","indoors","inside","insomuch","instead","into","inward","inwards","is", | 42 | "indeed","indoors","inside","insomuch","instead","into","inward","inwards","is", |
43 | "it","its","itself","just","kind","kg","km","last","latter","latterly","less","lest", | 43 | "it","its","itself","just","kind","kg","km","last","latter","latterly","less","lest", |
44 | "let","like","little","ltd","many","may","maybe","me","meantime","meanwhile","might", | 44 | "let","like","little","ltd","many","may","maybe","me","meantime","meanwhile","might", |
45 | "moreover","most","mostly","more","mr","mrs","ms","much","must","my","myself", | 45 | "moreover","most","mostly","more","mr","mrs","ms","much","must","my","myself", |
46 | "namely","need","neither","never","nevertheless","next","no","nobody","none", | 46 | "namely","need","neither","never","nevertheless","next","no","nobody","none", |
47 | "nonetheless","noone","nope","nor","not","nothing","notwithstanding","now","nowadays", | 47 | "nonetheless","noone","nope","nor","not","nothing","notwithstanding","now","nowadays", |
48 | "nowhere","of","off","often","ok","on","once","one","only","onto","or","other", | 48 | "nowhere","of","off","often","ok","on","once","one","only","onto","or","other", |
49 | "others","otherwise","ought","our","ours","ourselves","out","outside","over","own", | 49 | "others","otherwise","ought","our","ours","ourselves","out","outside","over","own", |
50 | "per","perhaps","plenty","provide","quite","rather","really","round","said","sake", | 50 | "per","perhaps","plenty","provide","quite","rather","really","round","said","sake", |
51 | "same","sang","save","saw","see","seeing","seem","seemed","seeming","seems","seen", | 51 | "same","sang","save","saw","see","seeing","seem","seemed","seeming","seems","seen", |
52 | "seldom","selves","sent","several","shalt","she","should","shown","sideways","since", | 52 | "seldom","selves","sent","several","shalt","she","should","shown","sideways","since", |
53 | "slept","slew","slung","slunk","smote","so","some","somebody","somehow","someone", | 53 | "slept","slew","slung","slunk","smote","so","some","somebody","somehow","someone", |
54 | "something","sometime","sometimes","somewhat","somewhere","spake","spat","spoke", | 54 | "something","sometime","sometimes","somewhat","somewhere","spake","spat","spoke", |
55 | "spoken","sprang","sprung","stave","staves","still","such","supposing","than","that", | 55 | "spoken","sprang","sprung","stave","staves","still","such","supposing","than","that", |
56 | "the","thee","their","them","themselves","then","thence","thenceforth","there", | 56 | "the","thee","their","them","themselves","then","thence","thenceforth","there", |
57 | "thereabout","thereabouts","thereafter","thereby","therefore","therein","thereof", | 57 | "thereabout","thereabouts","thereafter","thereby","therefore","therein","thereof", |
58 | "thereon","thereto","thereupon","these","they","this","those","thou","though", | 58 | "thereon","thereto","thereupon","these","they","this","those","thou","though", |
59 | "thrice","through","throughout","thru","thus","thy","thyself","till","to","together", | 59 | "thrice","through","throughout","thru","thus","thy","thyself","till","to","together", |
60 | "too","toward","towards","ugh","unable","under","underneath","unless","unlike", | 60 | "too","toward","towards","ugh","unable","under","underneath","unless","unlike", |
61 | "until","up","upon","upward","upwards","us","use","used","using","very","via","vs", | 61 | "until","up","upon","upward","upwards","us","use","used","using","very","via","vs", |
62 | "want","was","we","week","well","were","what","whatever","whatsoever","when","whence", | 62 | "want","was","we","week","well","were","what","whatever","whatsoever","when","whence", |
63 | "whenever","whensoever","where","whereabouts","whereafter","whereas","whereat", | 63 | "whenever","whensoever","where","whereabouts","whereafter","whereas","whereat", |
64 | "whereby","wherefore","wherefrom","wherein","whereinto","whereof","whereon", | 64 | "whereby","wherefore","wherefrom","wherein","whereinto","whereof","whereon", |
65 | "wheresoever","whereto","whereunto","whereupon","wherever","wherewith","whether", | 65 | "wheresoever","whereto","whereunto","whereupon","wherever","wherewith","whether", |
66 | "whew","which","whichever","whichsoever","while","whilst","whither","who","whoa", | 66 | "whew","which","whichever","whichsoever","while","whilst","whither","who","whoa", |
67 | "whoever","whole","whom","whomever","whomsoever","whose","whosoever","why","will", | 67 | "whoever","whole","whom","whomever","whomsoever","whose","whosoever","why","will", |
68 | "wilt","with","within","without","worse","worst","would","wow","ye","yet","year", | 68 | "wilt","with","within","without","worse","worst","would","wow","ye","yet","year", |
69 | "yippee","you","your","yours","yourself","yourselves" | 69 | "yippee","you","your","yours","yourself","yourselves", |
70 | "edit", "new", "page", "article", "http", "www", "com", "org", "wikipedia", "en" | ||
70 | ] | 71 | ] |
71 | 72 | ||
72 | Transmap = { | 73 | Transmap = { |
73 | "\xC3\x80" => "A", "\xC3\x81" => "A", "\xC3\x82" => "A", "\xC3\x83" => "A", | 74 | "\xC3\x80" => "A", "\xC3\x81" => "A", "\xC3\x82" => "A", "\xC3\x83" => "A", |
74 | "\xC3\x84" => "A", "\xC3\x85" => "A", "\xC3\x86" => "AE","\xC3\x87" => "C", | 75 | "\xC3\x84" => "A", "\xC3\x85" => "A", "\xC3\x86" => "AE","\xC3\x87" => "C", |
75 | "\xC3\x88" => "E", "\xC3\x89" => "E", "\xC3\x8A" => "E", "\xC3\x8B" => "E", | 76 | "\xC3\x88" => "E", "\xC3\x89" => "E", "\xC3\x8A" => "E", "\xC3\x8B" => "E", |
76 | "\xC3\x8C" => "I", "\xC3\x8D" => "I", "\xC3\x8E" => "I", "\xC3\x8F" => "I", | 77 | "\xC3\x8C" => "I", "\xC3\x8D" => "I", "\xC3\x8E" => "I", "\xC3\x8F" => "I", |
77 | "\xC3\x90" => "D", "\xC3\x91" => "N", "\xC3\x92" => "O", "\xC3\x93" => "O", | 78 | "\xC3\x90" => "D", "\xC3\x91" => "N", "\xC3\x92" => "O", "\xC3\x93" => "O", |
78 | "\xC3\x94" => "O", "\xC3\x95" => "O", "\xC3\x96" => "O", "\xC3\x98" => "O", | 79 | "\xC3\x94" => "O", "\xC3\x95" => "O", "\xC3\x96" => "O", "\xC3\x98" => "O", |
79 | "\xC3\x99" => "U", "\xC3\x9A" => "U", "\xC3\x9B" => "U", "\xC3\x9C" => "U", | 80 | "\xC3\x99" => "U", "\xC3\x9A" => "U", "\xC3\x9B" => "U", "\xC3\x9C" => "U", |
80 | "\xC3\x9D" => "Y", "\xC3\x9E" => "P", "\xC3\x9F" => "ss", | 81 | "\xC3\x9D" => "Y", "\xC3\x9E" => "P", "\xC3\x9F" => "ss", |
81 | "\xC3\xA0" => "a", "\xC3\xA1" => "a", "\xC3\xA2" => "a", "\xC3\xA3" => "a", | 82 | "\xC3\xA0" => "a", "\xC3\xA1" => "a", "\xC3\xA2" => "a", "\xC3\xA3" => "a", |
82 | "\xC3\xA4" => "a", "\xC3\xA5" => "a", "\xC3\xA6" => "ae","\xC3\xA7" => "c", | 83 | "\xC3\xA4" => "a", "\xC3\xA5" => "a", "\xC3\xA6" => "ae","\xC3\xA7" => "c", |
83 | "\xC3\xA8" => "e", "\xC3\xA9" => "e", "\xC3\xAA" => "e", "\xC3\xAB" => "e", | 84 | "\xC3\xA8" => "e", "\xC3\xA9" => "e", "\xC3\xAA" => "e", "\xC3\xAB" => "e", |
84 | "\xC3\xAC" => "i", "\xC3\xAD" => "i", "\xC3\xAE" => "i", "\xC3\xAF" => "i", | 85 | "\xC3\xAC" => "i", "\xC3\xAD" => "i", "\xC3\xAE" => "i", "\xC3\xAF" => "i", |
85 | "\xC3\xB0" => "o", "\xC3\xB1" => "n", "\xC3\xB2" => "o", "\xC3\xB3" => "o", | 86 | "\xC3\xB0" => "o", "\xC3\xB1" => "n", "\xC3\xB2" => "o", "\xC3\xB3" => "o", |
86 | "\xC3\xB4" => "o", "\xC3\xB5" => "o", "\xC3\xB6" => "o", "\xC3\xB8" => "o", | 87 | "\xC3\xB4" => "o", "\xC3\xB5" => "o", "\xC3\xB6" => "o", "\xC3\xB8" => "o", |
87 | "\xC3\xB9" => "u", "\xC3\xBA" => "u", "\xC3\xBB" => "u", "\xC3\xBC" => "u", | 88 | "\xC3\xB9" => "u", "\xC3\xBA" => "u", "\xC3\xBB" => "u", "\xC3\xBC" => "u", |
88 | "\xC3\xBD" => "y", "\xC3\xBE" => "p", "\xC3\xBF" => "y", | 89 | "\xC3\xBD" => "y", "\xC3\xBE" => "p", "\xC3\xBF" => "y", |
89 | "\xC4\x80" => "A", "\xC4\x81" => "a", "\xC4\x82" => "A", "\xC4\x83" => "a", | 90 | "\xC4\x80" => "A", "\xC4\x81" => "a", "\xC4\x82" => "A", "\xC4\x83" => "a", |
90 | "\xC4\x84" => "A", "\xC4\x85" => "a", "\xC4\x86" => "C", "\xC4\x87" => "c", | 91 | "\xC4\x84" => "A", "\xC4\x85" => "a", "\xC4\x86" => "C", "\xC4\x87" => "c", |
91 | "\xC4\x88" => "C", "\xC4\x89" => "c", "\xC4\x8A" => "C", "\xC4\x8B" => "c", | 92 | "\xC4\x88" => "C", "\xC4\x89" => "c", "\xC4\x8A" => "C", "\xC4\x8B" => "c", |
92 | "\xC4\x8C" => "C", "\xC4\x8D" => "c", "\xC4\x8E" => "D", "\xC4\x8F" => "d", | 93 | "\xC4\x8C" => "C", "\xC4\x8D" => "c", "\xC4\x8E" => "D", "\xC4\x8F" => "d", |
93 | "\xC4\x90" => "D", "\xC4\x91" => "d", "\xC4\x92" => "E", "\xC4\x93" => "e", | 94 | "\xC4\x90" => "D", "\xC4\x91" => "d", "\xC4\x92" => "E", "\xC4\x93" => "e", |
94 | "\xC4\x94" => "E", "\xC4\x95" => "e", "\xC4\x96" => "E", "\xC4\x97" => "e", | 95 | "\xC4\x94" => "E", "\xC4\x95" => "e", "\xC4\x96" => "E", "\xC4\x97" => "e", |
95 | "\xC4\x98" => "E", "\xC4\x99" => "e", "\xC4\x9A" => "E", "\xC4\x9B" => "e", | 96 | "\xC4\x98" => "E", "\xC4\x99" => "e", "\xC4\x9A" => "E", "\xC4\x9B" => "e", |
96 | "\xC4\x9C" => "G", "\xC4\x9D" => "g", "\xC4\x9E" => "G", "\xC4\x9F" => "g", | 97 | "\xC4\x9C" => "G", "\xC4\x9D" => "g", "\xC4\x9E" => "G", "\xC4\x9F" => "g", |
97 | "\xC4\xA0" => "G", "\xC4\xA1" => "g", "\xC4\xA2" => "G", "\xC4\xA3" => "g", | 98 | "\xC4\xA0" => "G", "\xC4\xA1" => "g", "\xC4\xA2" => "G", "\xC4\xA3" => "g", |
98 | "\xC4\xA4" => "H", "\xC4\xA5" => "h", "\xC4\xA6" => "H", "\xC4\xA7" => "h", | 99 | "\xC4\xA4" => "H", "\xC4\xA5" => "h", "\xC4\xA6" => "H", "\xC4\xA7" => "h", |
99 | "\xC4\xA8" => "I", "\xC4\xA9" => "i", "\xC4\xAA" => "I", "\xC4\xAB" => "i", | 100 | "\xC4\xA8" => "I", "\xC4\xA9" => "i", "\xC4\xAA" => "I", "\xC4\xAB" => "i", |
100 | "\xC4\xAC" => "I", "\xC4\xAD" => "i", "\xC4\xAE" => "I", "\xC4\xAF" => "i", | 101 | "\xC4\xAC" => "I", "\xC4\xAD" => "i", "\xC4\xAE" => "I", "\xC4\xAF" => "i", |
101 | "\xC4\xB0" => "I", "\xC4\xB1" => "i", "\xC4\xB2" => "IJ","\xC4\xB3" => "ij", | 102 | "\xC4\xB0" => "I", "\xC4\xB1" => "i", "\xC4\xB2" => "IJ","\xC4\xB3" => "ij", |
102 | "\xC4\xB4" => "J", "\xC4\xB5" => "j", "\xC4\xB6" => "K", "\xC4\xB7" => "k", | 103 | "\xC4\xB4" => "J", "\xC4\xB5" => "j", "\xC4\xB6" => "K", "\xC4\xB7" => "k", |
103 | "\xC4\xB8" => "k", "\xC4\xB9" => "L", "\xC4\xBA" => "l", "\xC4\xBB" => "L", | 104 | "\xC4\xB8" => "k", "\xC4\xB9" => "L", "\xC4\xBA" => "l", "\xC4\xBB" => "L", |
104 | "\xC4\xBC" => "l", "\xC4\xBD" => "L", "\xC4\xBE" => "l", "\xC4\xBF" => "L", | 105 | "\xC4\xBC" => "l", "\xC4\xBD" => "L", "\xC4\xBE" => "l", "\xC4\xBF" => "L", |
105 | "\xC5\x80" => "l", "\xC5\x81" => "L", "\xC5\x82" => "l", "\xC5\x83" => "N", | 106 | "\xC5\x80" => "l", "\xC5\x81" => "L", "\xC5\x82" => "l", "\xC5\x83" => "N", |
106 | "\xC5\x84" => "n", "\xC5\x85" => "N", "\xC5\x86" => "n", "\xC5\x87" => "N", | 107 | "\xC5\x84" => "n", "\xC5\x85" => "N", "\xC5\x86" => "n", "\xC5\x87" => "N", |
107 | "\xC5\x88" => "n", "\xC5\x89" => "n", "\xC5\x8A" => "N", "\xC5\x8B" => "n", | 108 | "\xC5\x88" => "n", "\xC5\x89" => "n", "\xC5\x8A" => "N", "\xC5\x8B" => "n", |
108 | "\xC5\x8C" => "O", "\xC5\x8D" => "o", "\xC5\x8E" => "O", "\xC5\x8F" => "o", | 109 | "\xC5\x8C" => "O", "\xC5\x8D" => "o", "\xC5\x8E" => "O", "\xC5\x8F" => "o", |
109 | "\xC5\x90" => "O", "\xC5\x91" => "o", "\xC5\x92" => "CE","\xC5\x93" => "ce", | 110 | "\xC5\x90" => "O", "\xC5\x91" => "o", "\xC5\x92" => "CE","\xC5\x93" => "ce", |
110 | "\xC5\x94" => "R", "\xC5\x95" => "r", "\xC5\x96" => "R", "\xC5\x97" => "r", | 111 | "\xC5\x94" => "R", "\xC5\x95" => "r", "\xC5\x96" => "R", "\xC5\x97" => "r", |
111 | "\xC5\x98" => "R", "\xC5\x99" => "r", "\xC5\x9A" => "S", "\xC5\x9B" => "s", | 112 | "\xC5\x98" => "R", "\xC5\x99" => "r", "\xC5\x9A" => "S", "\xC5\x9B" => "s", |
112 | "\xC5\x9C" => "S", "\xC5\x9D" => "s", "\xC5\x9E" => "S", "\xC5\x9F" => "s", | 113 | "\xC5\x9C" => "S", "\xC5\x9D" => "s", "\xC5\x9E" => "S", "\xC5\x9F" => "s", |
113 | "\xC5\xA0" => "S", "\xC5\xA1" => "s", "\xC5\xA2" => "T", "\xC5\xA3" => "t", | 114 | "\xC5\xA0" => "S", "\xC5\xA1" => "s", "\xC5\xA2" => "T", "\xC5\xA3" => "t", |
114 | "\xC5\xA4" => "T", "\xC5\xA5" => "t", "\xC5\xA6" => "T", "\xC5\xA7" => "t", | 115 | "\xC5\xA4" => "T", "\xC5\xA5" => "t", "\xC5\xA6" => "T", "\xC5\xA7" => "t", |
115 | "\xC5\xA8" => "U", "\xC5\xA9" => "u", "\xC5\xAA" => "U", "\xC5\xAB" => "u", | 116 | "\xC5\xA8" => "U", "\xC5\xA9" => "u", "\xC5\xAA" => "U", "\xC5\xAB" => "u", |
116 | "\xC5\xAC" => "U", "\xC5\xAD" => "u", "\xC5\xAE" => "U", "\xC5\xAF" => "u", | 117 | "\xC5\xAC" => "U", "\xC5\xAD" => "u", "\xC5\xAE" => "U", "\xC5\xAF" => "u", |
117 | "\xC5\xB0" => "U", "\xC5\xB1" => "u", "\xC5\xB2" => "U", "\xC5\xB3" => "u", | 118 | "\xC5\xB0" => "U", "\xC5\xB1" => "u", "\xC5\xB2" => "U", "\xC5\xB3" => "u", |
118 | "\xC5\xB4" => "W", "\xC5\xB5" => "w", "\xC5\xB6" => "Y", "\xC5\xB7" => "y", | 119 | "\xC5\xB4" => "W", "\xC5\xB5" => "w", "\xC5\xB6" => "Y", "\xC5\xB7" => "y", |
119 | "\xC5\xB8" => "Y", "\xC5\xB9" => "Z", "\xC5\xBA" => "z", "\xC5\xBB" => "Z", | 120 | "\xC5\xB8" => "Y", "\xC5\xB9" => "Z", "\xC5\xBA" => "z", "\xC5\xBB" => "Z", |
120 | "\xC5\xBC" => "z", "\xC5\xBD" => "Z", "\xC5\xBE" => "z", "\xC6\x8F" => "E", | 121 | "\xC5\xBC" => "z", "\xC5\xBD" => "Z", "\xC5\xBE" => "z", "\xC6\x8F" => "E", |
121 | "\xC6\xA0" => "O", "\xC6\xA1" => "o", "\xC6\xAF" => "U", "\xC6\xB0" => "u", | 122 | "\xC6\xA0" => "O", "\xC6\xA1" => "o", "\xC6\xAF" => "U", "\xC6\xB0" => "u", |
122 | "\xC7\x8D" => "A", "\xC7\x8E" => "a", "\xC7\x8F" => "I", | 123 | "\xC7\x8D" => "A", "\xC7\x8E" => "a", "\xC7\x8F" => "I", |
123 | "\xC7\x90" => "i", "\xC7\x91" => "O", "\xC7\x92" => "o", "\xC7\x93" => "U", | 124 | "\xC7\x90" => "i", "\xC7\x91" => "O", "\xC7\x92" => "o", "\xC7\x93" => "U", |
124 | "\xC7\x94" => "u", "\xC7\x95" => "U", "\xC7\x96" => "u", "\xC7\x97" => "U", | 125 | "\xC7\x94" => "u", "\xC7\x95" => "U", "\xC7\x96" => "u", "\xC7\x97" => "U", |
125 | "\xC7\x98" => "u", "\xC7\x99" => "U", "\xC7\x9A" => "u", "\xC7\x9B" => "U", | 126 | "\xC7\x98" => "u", "\xC7\x99" => "U", "\xC7\x9A" => "u", "\xC7\x9B" => "U", |
126 | "\xC7\x9C" => "u", | 127 | "\xC7\x9C" => "u", |
127 | "\xC7\xBA" => "A", "\xC7\xBB" => "a", "\xC7\xBC" => "AE","\xC7\xBD" => "ae", | 128 | "\xC7\xBA" => "A", "\xC7\xBB" => "a", "\xC7\xBC" => "AE","\xC7\xBD" => "ae", |
128 | "\xC7\xBE" => "O", "\xC7\xBF" => "o", | 129 | "\xC7\xBE" => "O", "\xC7\xBF" => "o", |
129 | "\xC9\x99" => "e", | 130 | "\xC9\x99" => "e", |
130 | "\xC2\x82" => ",", # High code comma | 131 | "\xC2\x82" => ",", # High code comma |
131 | "\xC2\x84" => ",,", # High code double comma | 132 | "\xC2\x84" => ",,", # High code double comma |
132 | "\xC2\x85" => "...", # Tripple dot | 133 | "\xC2\x85" => "...", # Tripple dot |
133 | "\xC2\x88" => "^", # High carat | 134 | "\xC2\x88" => "^", # High carat |
134 | "\xC2\x91" => "\x27", # Forward single quote | 135 | "\xC2\x91" => "\x27", # Forward single quote |
135 | "\xC2\x92" => "\x27", # Reverse single quote | 136 | "\xC2\x92" => "\x27", # Reverse single quote |
136 | "\xC2\x93" => "\x22", # Forward double quote | 137 | "\xC2\x93" => "\x22", # Forward double quote |
137 | "\xC2\x94" => "\x22", # Reverse double quote | 138 | "\xC2\x94" => "\x22", # Reverse double quote |
138 | "\xC2\x96" => "-", # High hyphen | 139 | "\xC2\x96" => "-", # High hyphen |
139 | "\xC2\x97" => "--", # Double hyphen | 140 | "\xC2\x97" => "--", # Double hyphen |
140 | "\xC2\xA6" => "|", # Split vertical bar | 141 | "\xC2\xA6" => "|", # Split vertical bar |
141 | "\xC2\xAB" => "<<", # Double less than | 142 | "\xC2\xAB" => "<<", # Double less than |
142 | "\xC2\xBB" => ">>", # Double greater than | 143 | "\xC2\xBB" => ">>", # Double greater than |
143 | "\xC2\xBC" => "1/4", # one quarter | 144 | "\xC2\xBC" => "1/4", # one quarter |
144 | "\xC2\xBD" => "1/2", # one half | 145 | "\xC2\xBD" => "1/2", # one half |
145 | "\xC2\xBE" => "3/4", # three quarters | 146 | "\xC2\xBE" => "3/4", # three quarters |
146 | "\xCA\xBF" => "\x27", # c-single quote | 147 | "\xCA\xBF" => "\x27", # c-single quote |
147 | "\xCC\xA8" => "", # modifier - under curve | 148 | "\xCC\xA8" => "", # modifier - under curve |
148 | "\xCC\xB1" => "", # modifier - under line | 149 | "\xCC\xB1" => "", # modifier - under line |
149 | # /\W/ => "" | 150 | # /\W/ => "" |
150 | } | 151 | } |
151 | 152 | ||
152 | end | 153 | end |
153 | 154 | ||
154 | # Extention of the standard class String with useful function. | 155 | # Extention of the standard class String with useful function. |
155 | class String | 156 | class String |
156 | include Mirimiri | 157 | include Mirimiri |
157 | 158 | ||
158 | def unaccent | 159 | def unaccent |
159 | # force_encoding is needed with ruby1.9 | 160 | # force_encoding is needed with ruby1.9 |
160 | Transmap.inject(self.force_encoding("ASCII-8BIT")) { |str, (utf8, asc)| str.gsub(utf8, asc) } | 161 | Transmap.inject(self.force_encoding("ASCII-8BIT")) { |str, (utf8, asc)| str.gsub(utf8, asc) } |
161 | end | 162 | end |
162 | 163 | ||
163 | # Returns +true+ if +self+ belongs to Rir::Stoplist, +false+ otherwise. | 164 | # Returns +true+ if +self+ belongs to Rir::Stoplist, +false+ otherwise. |
164 | def is_stopword? | 165 | def is_stopword? |
165 | Stoplist.include?(self.downcase) | 166 | Stoplist.include?(self.downcase) |
166 | end | 167 | end |
167 | 168 | ||
168 | # Do not use. | 169 | # Do not use. |
169 | # TODO: rewamp. find why this function is here. | 170 | # TODO: rewamp. find why this function is here. |
170 | def remove_special_characters | 171 | def remove_special_characters |
171 | self.split.collect { |w| w.gsub(/\W/,' ').split.collect { |w| w.gsub(/\W/,' ').strip.sub(/\A.\z/, '')}.join(' ').strip.sub(/\A.\z/, '')}.join(' ') | 172 | self.split.collect { |w| w.gsub(/\W/,' ').split.collect { |w| w.gsub(/\W/,' ').strip.sub(/\A.\z/, '')}.join(' ').strip.sub(/\A.\z/, '')}.join(' ') |
172 | end | 173 | end |
173 | 174 | ||
174 | # Removes all XML-like tags from +self+. | 175 | # Removes all XML-like tags from +self+. |
175 | # | 176 | # |
176 | # s = "<html><body>test</body></html>" | 177 | # s = "<html><body>test</body></html>" |
177 | # s.strip_xml_tags! | 178 | # s.strip_xml_tags! |
178 | # s #=> "test" | 179 | # s #=> "test" |
179 | def strip_xml_tags! | 180 | def strip_xml_tags! |
180 | replace strip_with_pattern /<\/?[^>]*>/ | 181 | replace strip_with_pattern /<\/?[^>]*>/ |
181 | end | 182 | end |
182 | 183 | ||
183 | # Removes all XML-like tags from +self+. | 184 | # Removes all XML-like tags from +self+. |
184 | # | 185 | # |
185 | # s = "<html><body>test</body></html>" | 186 | # s = "<html><body>test</body></html>" |
186 | # s.strip_xml_tags #=> "test" | 187 | # s.strip_xml_tags #=> "test" |
187 | # s #=> "<html><body>test</body></html>" | 188 | # s #=> "<html><body>test</body></html>" |
188 | def strip_xml_tags | 189 | def strip_xml_tags |
189 | dup.strip_xml_tags! | 190 | dup.strip_xml_tags! |
190 | end | 191 | end |
191 | 192 | ||
192 | # Removes all Javascript sources from +self+. | 193 | # Removes all Javascript sources from +self+. |
193 | # | 194 | # |
194 | # s = "<script type='text/javascript'> | 195 | # s = "<script type='text/javascript'> |
195 | # var skin='vector', | 196 | # var skin='vector', |
196 | # stylepath='http://bits.wikimedia.org/skins-1.5' | 197 | # stylepath='http://bits.wikimedia.org/skins-1.5' |
197 | # </script> | 198 | # </script> |
198 | # | 199 | # |
199 | # test" | 200 | # test" |
200 | # s.strip_javascripts! | 201 | # s.strip_javascripts! |
201 | # s #=> "test" | 202 | # s #=> "test" |
202 | def strip_javascripts! | 203 | def strip_javascripts! |
203 | replace strip_with_pattern /<script type="text\/javascript">(.+?)<\/script>/m | 204 | replace strip_with_pattern /<script type="text\/javascript">(.+?)<\/script>/m |
204 | end | 205 | end |
205 | 206 | ||
206 | # Removes all Javascript sources from +self+. | 207 | # Removes all Javascript sources from +self+. |
207 | # | 208 | # |
208 | # s = "<script type='text/javascript'> | 209 | # s = "<script type='text/javascript'> |
209 | # var skin='vector', | 210 | # var skin='vector', |
210 | # stylepath='http://bits.wikimedia.org/skins-1.5' | 211 | # stylepath='http://bits.wikimedia.org/skins-1.5' |
211 | # </script> | 212 | # </script> |
212 | # | 213 | # |
213 | # test" | 214 | # test" |
214 | # s.strip_javascripts #=> "test" | 215 | # s.strip_javascripts #=> "test" |
215 | def strip_javascripts | 216 | def strip_javascripts |
216 | dup.strip_javascripts! | 217 | dup.strip_javascripts! |
217 | end | 218 | end |
218 | 219 | ||
219 | def strip_stylesheets! | 220 | def strip_stylesheets! |
220 | # TODO: rewamp. dunno what is it. | 221 | # TODO: rewamp. dunno what is it. |
221 | replace strip_with_pattern /<style type="text\/css">(.+?)<\/style>/m | 222 | replace strip_with_pattern /<style type="text\/css">(.+?)<\/style>/m |
222 | end | 223 | end |
223 | 224 | ||
224 | def strip_stylesheets | 225 | def strip_stylesheets |
225 | dup.strip_stylesheets! | 226 | dup.strip_stylesheets! |
226 | end | 227 | end |
227 | 228 | ||
228 | # Removes punctuation from +self+. | 229 | # Removes punctuation from +self+. |
229 | # | 230 | # |
230 | # s = "hello, world. how are you?!" | 231 | # s = "hello, world. how are you?!" |
231 | # s.strip_punctuation! | 232 | # s.strip_punctuation! |
232 | # s # => "hello world how are you" | 233 | # s # => "hello world how are you" |
233 | def strip_punctuation! | 234 | def strip_punctuation! |
234 | replace strip_with_pattern /[^a-zA-Z0-9\-\s]/ | 235 | replace strip_with_pattern /[^a-zA-Z0-9\-\s]/ |
235 | end | 236 | end |
236 | 237 | ||
237 | # Removes punctuation from +self+. | 238 | # Removes punctuation from +self+. |
238 | # | 239 | # |
239 | # s = "hello, world. how are you?!" | 240 | # s = "hello, world. how are you?!" |
240 | # s.strip_punctuation # => "hello world how are you" | 241 | # s.strip_punctuation # => "hello world how are you" |
241 | def strip_punctuation | 242 | def strip_punctuation |
242 | dup.strip_punctuation! | 243 | dup.strip_punctuation! |
243 | end | 244 | end |
244 | 245 | ||
245 | # Returns the text values inside all occurences of a XML tag in +self+ | 246 | # Returns the text values inside all occurences of a XML tag in +self+ |
246 | # | 247 | # |
247 | # s = "four-piece in <a href='#'>Indianapolis</a>, <a href='#'>Indiana</a> at the Murat Theatre" | 248 | # s = "four-piece in <a href='#'>Indianapolis</a>, <a href='#'>Indiana</a> at the Murat Theatre" |
248 | # s.extract_xmltags_values 'a' #=> ["Indianapolis", "Indiana"] | 249 | # s.extract_xmltags_values 'a' #=> ["Indianapolis", "Indiana"] |
249 | def extract_xmltags_values(tag_name) | 250 | def extract_xmltags_values(tag_name) |
250 | self.scan(/<#{tag_name}.*?>(.+?)<\/#{tag_name}>/).flatten | 251 | self.scan(/<#{tag_name}.*?>(.+?)<\/#{tag_name}>/).flatten |
251 | end | 252 | end |
252 | 253 | ||
253 | def strip_with_pattern(pattern) | 254 | def strip_with_pattern(pattern) |
254 | require 'cgi' | 255 | require 'cgi' |
255 | 256 | ||
256 | CGI::unescapeHTML(self.gsub(pattern,"")).unaccent.encode("UTF-8", {:invalid => :replace, :undef => :replace, :replace => " "}) | 257 | CGI::unescapeHTML(self.gsub(pattern,"")).unaccent.encode("UTF-8", {:invalid => :replace, :undef => :replace, :replace => " "}) |
257 | end | 258 | end |
258 | 259 | ||
259 | private :strip_with_pattern | 260 | private :strip_with_pattern |
260 | end | 261 | end |
261 | 262 |