def do_query(qtype, query, max_zoom=4):
query = query.lower()
if qtype == "neumes":
query_stmt = 'neumes:{0}'.format(query.replace(' ', '_'))
elif qtype == "pnames" or qtype == "pnames-invariant":
if not search_utils.valid_pitch_sequence(query):
raise LiberSearchException("The query you provided is not a valid pitch sequence")
real_query = query if qtype == 'pnames' else ' OR '.join(search_utils.get_transpositions(query))
query_stmt = 'pnames:{0}'.format(real_query)
elif qtype == "contour":
query_stmt = 'contour:{0}'.format(query)
elif qtype == "text":
query_stmt = 'text:"{0}"'.format(query)
elif qtype == "intervals":
query_stmt = 'intervals:{0}'.format(query.replace(' ', '_'))
elif qtype == "incipit":
query_stmt = "incipit:{0}*".format(query)
else:
raise LiberSearchException("Invalid query type provided")
if qtype == "pnames-invariant":
response = solrconn.query(query_stmt, score=False, sort="pagen asc", q_op="OR", rows=1000000)
else:
response = solrconn.query(query_stmt, score=False, sort="pagen asc", rows=1000000)
numfound = response.numFound
results = []
boxes = []
# get only the longest ngram in the results
if qtype == "neumes":
notegrams_num = search_utils.get_neumes_length(query)
response = [r for r in response if len(r['pnames']) == notegrams_num]
for d in response:
page_number = d['pagen']
locations = json.loads(d['location'].replace("'", '"'))
box_id = "m-" + d['id']
search_data = {
'semitones': d['semitones'],
'pnames': d['pnames'],
'neumes': d['neumes'],
'contour': d['contour'],
'intervals': d['intervals']
}
if isinstance(locations, types.DictType):
box_w = locations['width']
box_h = locations['height']
box_x = locations['ulx']
box_y = locations['uly']
boxes.append({'p': page_number, 'w': box_w, 'h': box_h, 'x': box_x, 'y': box_y, 'id': box_id, 'results': search_data})
else:
for location in locations:
box_w = location['width']
box_h = location['height']
box_x = location['ulx']
box_y = location['uly']
boxes.append({'p': page_number, 'w': box_w, 'h': box_h, 'x': box_x, 'y': box_y, 'id': box_id, 'results': search_data})
boxes_sorted = sorted(boxes, key=itemgetter('p', 'y'))
return boxes_sorted
def do_query(qtype, query, zoom_level, project_id, max_zoom=5):
query = query.lower()
if qtype == "neumes":
query_stmt = 'neumes:{0}'.format(query.replace(' ', '_'))
elif qtype == "pnames" or qtype == "pnames-invariant":
if not search_utils.valid_pitch_sequence(query):
raise LiberSearchException("The query you provided is not a valid pitch sequence")
real_query = query if qtype == 'pnames' else ' OR '.join(search_utils.get_transpositions(query))
query_stmt = 'pnames:{0}'.format(real_query)
elif qtype == "contour":
query_stmt = 'contour:{0}'.format(query)
elif qtype == "text":
query_stmt = 'text:"{0}"'.format(query)
elif qtype == "intervals":
query_stmt = 'intervals:{0}'.format(query.replace(' ', '_'))
elif qtype == "incipit":
query_stmt = "incipit:{0}*".format(query)
else:
raise LiberSearchException("Invalid query type provided")
query_stmt = '(%s) AND (project:%d)' % (query_stmt, project_id)
if qtype == "pnames-invariant":
response = solrconn.query(query_stmt, score=False, sort="pagen asc", q_op="OR", rows=1000000)
else:
response = solrconn.query(query_stmt, score=False, sort="pagen asc", rows=1000000)
numfound = response.numFound
results = []
boxes = []
# get only the longest ngram in the results
if qtype == "neumes":
notegrams_num = search_utils.get_neumes_length(query)
response = [r for r in response if len(r['pnames']) == notegrams_num]
for d in response:
page_number = d['pagen']
locations = json.loads(d['location'].replace("'", '"'))
if isinstance(locations, types.DictType):
box_w = locations['width']
box_h = locations['height']
box_x = locations['ulx']
box_y = locations['uly']
boxes.append({'p': page_number, 'w': box_w, 'h': box_h, 'x': box_x, 'y': box_y})
else:
for location in locations:
box_w = location['width']
box_h = location['height']
box_x = location['ulx']
box_y = location['uly']
boxes.append({'p': page_number, 'w': box_w, 'h': box_h, 'x': box_x, 'y': box_y})
zoom_diff = max_zoom - int(zoom_level)
real_boxes = []
for box in boxes:
#incorporate zoom
box['w'] = search_utils.incorporate_zoom(box['w'], zoom_diff)
box['h'] = search_utils.incorporate_zoom(box['h'], zoom_diff)
box['x'] = search_utils.incorporate_zoom(box['x'], zoom_diff)
box['y'] = search_utils.incorporate_zoom(box['y'], zoom_diff)
if box['w'] > 0 and box['h'] > 0:
real_boxes.append(box)
boxes_sorted = sorted(real_boxes, key=itemgetter('p', 'y'))
return boxes_sorted
