-
Notifications
You must be signed in to change notification settings - Fork 49
/
index.rb
343 lines (319 loc) · 12.1 KB
/
index.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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
module Picky
# = Picky Indexes
#
# A Picky Index defines
# * what backend it uses.
# * where its data comes from (a data source).
# * how this data it is indexed.
# * a number of categories that may or may not map directly to data categories.
#
# == Howto
#
# This is a step-by-step description on how to create an index.
#
# Start by choosing an <tt>Index</tt> or an <tt>Index</tt>.
# In the example, we will be using an in-memory index, <tt>Index</tt>.
#
# books = Index.new(:books)
#
# That in itself won't do much good, that's why we add a data source:
#
# books = Index.new(:books) do
# source Sources::CSV.new(:title, :author, file: 'data/books.csv')
# end
#
# In the example, we use an explicit <tt>Sources::CSV</tt> of Picky.
# However, anything that responds to <tt>#each</tt>, and returns an object that
# answers to <tt>#id</tt>, works.
#
# For example, a 3.0 ActiveRecord class:
#
# books = Index.new(:books) do
# source Book.order('isbn ASC')
# end
#
# Now we know where the data comes from, but not, how to categorize it.
#
# Let's add a few categories:
#
# books = Index.new(:books) do
# source Book.order('isbn ASC')
# category :title
# category :author
# category :isbn
# end
#
# Categories offer quite a few options, see <tt>Indexes::Base#category</tt> for details.
#
# After adding more options, it might look like this:
#
# books = Index.new(:books) do
# source Book.order('isbn ASC')
# category :title,
# partial: Partial::Substring.new(from: 1),
# similarity: Similarity::DoubleMetaphone.new(3),
# qualifiers: [:t, :title, :titulo]
# category :author,
# similarity: Similarity::Metaphone.new(2)
# category :isbn,
# partial: Partial::None.new,
# from: :legacy_isbn_name
# end
#
# For this to work, a <tt>Book</tt> should support methods <tt>#title</tt>, <tt>#author</tt> and <tt>#legacy_isbn_name</tt>.
#
# If it uses <tt>String</tt> ids, use <tt>#key_format</tt> to define a formatting method:
#
# books = Index.new(:books) do
# key_format :to_s
# source Book.order('isbn ASC')
# category :title
# category :author
# category :isbn
# end
#
# Finally, use the index for a <tt>Search</tt>:
#
# route %r{^/media$} => Search.new(books, dvds, mp3s)
#
# This class defines the indexing and index API that is exposed to the user
# as the #index method inside the Application class.
#
# It provides a single front for both indexing and index options. We suggest to always use the index API.
#
# Note: An Index holds both an *Indexed*::*Index* and an *Indexing*::*Index*.
#
class Index
attr_reader :name,
:categories
delegate :[],
:dump,
:each_category,
:to => :categories
# Create a new index with a given source.
#
# === Parameters
# * name: A name that will be used for the index directory and in the Picky front end.
#
# === Options (all are used in the block - not passed as a Hash, see examples)
# * source: Where the data comes from, e.g. Sources::CSV.new(...). Optional, can be defined in the block using #source.
# * result_identifier: Use if you'd like a different identifier/name in the results than the name of the index.
# * after_indexing: As of this writing only used in the db source. Executes the given after_indexing as SQL after the indexing process.
# * indexing: Call and pass either a tokenizer (responds to #tokenize) or the options for a tokenizer..
# * key_format: Call and pass in a format method for the ids (default is #to_i).
#
# Example:
# my_index = Index.new(:my_index) do
# source Sources::CSV.new(file: 'data/index.csv')
# key_format :to_sym
# category :bla
# result_identifier :my_special_results
# end
#
def initialize name
@name = name.intern
@categories = Categories.new
# Centralized registry.
#
Indexes.register self
instance_eval(&Proc.new) if block_given?
end
# API method.
#
# Sets/returns the backend used.
# Default is @Backends::Memory.new@.
#
def backend backend = nil
if backend
@backend = backend
else
@backend || Backends::Memory.new
end
end
# TODO Symbols.
#
# # API method.
# #
# # Tells Picky to use Symbols internally.
# #
# def use_symbols
# @symbols = true
# end
# def use_symbols?
# @symbols
# end
# API method.
#
# Defines a searchable category on the index.
#
# === Parameters
# * category_name: This identifier is used in the front end, but also to categorize query text. For example, “title:hobbit” will narrow the hobbit query on categories with the identifier :title.
#
# === Options
# * partial: Partial::None.new or Partial::Substring.new(from: starting_char, to: ending_char). Default is Partial::Substring.new(from: -3, to: -1).
# * similarity: Similarity::None.new or Similarity::DoubleMetaphone.new(similar_words_searched). Default is Similarity::None.new.
# * qualifiers: An array of qualifiers with which you can define which category you’d like to search, for example “title:hobbit” will search for hobbit in just title categories. Example: qualifiers: [:t, :titre, :title] (use it for example with multiple languages). Default is the name of the category.
# * qualifier: Convenience options if you just need a single qualifier, see above. Example: qualifiers => :title. Default is the name of the category.
# * source: Use a different source than the index uses. If you think you need that, there might be a better solution to your problem. Please post to the mailing list first with your application.rb :)
# * from: Take the data from the data category with this name. Example: You have a source Sources::CSV.new(:title, file:'some_file.csv') but you want the category to be called differently. The you use from: define_category(:similar_title, :from => :title).
#
def category category_name, options = {}
new_category = Category.new category_name.intern, self, options
categories << new_category
new_category = yield new_category if block_given?
new_category
end
alias define_category category
# Make this category range searchable with a fixed range. If you need other
# ranges, define another category with a different range value.
#
# Example:
# You have data values inside 1..100, and you want to have Picky return
# not only the results for 47 if you search for 47, but also results for
# 45, 46, or 47.2, 48.9, in a range of 2 around 47, so (45..49).
#
# Then you use:
# ranged_category :values_inside_1_100, 2
#
# Optionally, you give it a precision value to reduce the error margin
# around 47 (Picky is a bit liberal).
# Index.new :range do
# ranged_category :values_inside_1_100, 2, precision: 5
# end
#
# This will force Picky to maximally be wrong 5% of the given range value
# (5% of 2 = 0.1) instead of the default 20% (20% of 2 = 0.4).
#
# We suggest not to use much more than 5 as a higher precision is more
# performance intensive for less and less precision gain.
#
# == Protip 1
#
# Create two ranged categories to make an area search:
# Index.new :area do
# ranged_category :x, 1
# ranged_category :y, 1
# end
#
# Search for it using for example:
# x:133, y:120
#
# This will search this square area (* = 133, 120: The "search" point entered):
#
# 132 134
# | |
# --|---------|-- 121
# | |
# | * |
# | |
# --|---------|-- 119
# | |
#
# Note: The area does not need to be square, but can be rectangular.
#
# == Protip 2
#
# Create three ranged categories to make a volume search.
#
# Or go crazy and use 4 ranged categories for a space/time search! ;)
#
# === Parameters
# * category_name: The category_name as used in #define_category.
# * range: The range (in the units of your data values) around the query point where we search for results.
#
# -----|<- range ->*------------|-----
#
# === Options
# * precision: Default is 1 (20% error margin, very fast), up to 5 (5% error margin, slower) makes sense.
# * anchor: Where to anchor the geo grid.
# * ... all options of #define_category.
#
def ranged_category category_name, range, options = {}
precision = options.delete(:precision) || 1
anchor = options.delete(:anchor) || 0.0
# Note: :key_format => :to_f ?
#
options = { partial: Partial::None.new }.merge options
define_category category_name, options do |category|
Wrappers::Category::Location.wrap category, range, precision, anchor
end
end
alias define_ranged_category ranged_category
# HIGHLY EXPERIMENTAL Not correctly working yet. Try it if you feel "beta".
#
# Also a range search see #ranged_category, but on the earth's surface.
#
# Parameters:
# * lat_name: The latitude's name as used in #define_category.
# * lng_name: The longitude's name as used in #define_category.
# * radius: The distance (in km) around the query point which we search for results.
#
# Note: Picky uses a square, not a circle. That should be ok for most usages.
#
# -----------------------------
# | |
# | |
# | |
# | |
# | |
# | *<- radius ->|
# | |
# | |
# | |
# | |
# | |
# -----------------------------
#
# Options
# * precision: Default 1 (20% error margin, very fast), up to 5 (5% error margin, slower) makes sense.
# * lat_from: The data category to take the data for the latitude from.
# * lng_from: The data category to take the data for the longitude from.
#
# TODO Will have to write a wrapper that combines two categories that are
# indexed simultaneously, since lat/lng are correlated.
#
def geo_categories lat_name, lng_name, radius, options = {} # :nodoc:
# Extract lat/lng specific options.
#
lat_from = options.delete :lat_from
lng_from = options.delete :lng_from
# One can be a normal ranged_category.
#
ranged_category lat_name, radius*0.00898312, options.merge(from: lat_from)
# The other needs to adapt the radius depending on the one.
#
# Depending on the latitude, the radius of the longitude
# needs to enlarge, the closer we get to the pole.
#
# In our simplified case, the radius is given as if all the
# locations were on the 45 degree line.
#
# This calculates km -> longitude (degrees).
#
# A degree on the 45 degree line is equal to ~222.6398 km.
# So a km on the 45 degree line is equal to 0.01796624 degrees.
#
ranged_category lng_name, radius*0.01796624, options.merge(from: lng_from)
end
alias define_geo_categories geo_categories
def to_stats # :nodoc:
stats = <<-INDEX
#{name} (#{self.class}):
#{"source: #{source}".indented_to_s}
#{"categories: #{categories.map(&:name).join(', ')}".indented_to_s}
INDEX
stats << " result identifier: \"#{result_identifier}\"".indented_to_s unless result_identifier.to_s == name.to_s
stats
end
# Identifier used for technical output.
#
def identifier
name
end
#
#
def to_s
"#{self.class}(#{name}, result_id: #{result_identifier}, source: #{@source}, categories: #{categories})"
end
end
end