Пример #1
0
def tag_cloud(dist, id_ = "", class_ = "", width=None, height=None, 
              max_size=None, min_size=None, max_words = None, 
              start_color=None, end_color=None, color_steps=None, 
              sort_order="random"):
    ''' returns a dict with style and body elements. style contains
    defalt styling for the tag cloud, while body contains the html
    markup. '''

    # sort() returns a list of (word, count) tuples in order of decreasing frequency 
    dist = sort(dist) 
    
    # truncate the list of items if max_words was specified
    if max_words:
        max_words = int(max_words)
        dist = dist[:max_words]            

    # get the number of words remaining in dist (ie the number of tags in the
    # tagcloud) to pass back in the metadata 
    words_in_cloud = len(dist)

    # get the equation of the line
    #max_size = 70;
    #min_size = 10;
    if max_size and min_size:
        font_size_fn = min_max_extrapolate(dist, max_size, min_size)
    else:
        if not (width and height):
            width = 600
            height = 800
        font = 'times new roman'
        font_size_fn = fit_to_area(width, height, dist, font)

    # get the equation of the line between min_size and max_size. do this AFTER
    # truncating to max_words and BEFORE shuffling the order around.  

    # determine the sort order. if the sort order is frequency, there's nothing
    # to do since the distribution object is already sorted by frequency. 
    if sort_order not in ['random', 'frequency', 'alphabetical']:
        print 'invalid sort order; using default = random'
        sort_order = 'random'

    if sort_order == 'random':
        # shuffles in place
        random.shuffle(dist)

    if sort_order == 'alphabetical':
        # not yet implemented 
        pass
    
    # assemble the class and id tags for the tag cloud's wrapping div 
    divstyle = '''class="tagcloud"'''
    if class_ != "": 
        divstyle = divstyle[:-1] + " " + class_ + ''' "'''
    
    # the user can specify a unique id for the tag cloud if they want
    # additional styling applied from their own style sheets. 
    if id_ != "": divstyle += ''' id="%s" ''' % id_
    divstyle = divstyle.strip()	  

    body = '''<div %s>''' % divstyle
    for word, freq in dist:
        # each word has a class of 'word' in addition to its frequency so that
        # the user may specify additional styling. **note**: make sure the
        # space after the span is maintained; otherwise the spans within the
        # div won't wrap. 
        body += '''<span title="%d" class="word %s">%s</span> ''' % (freq, num_to_word(freq), word)
    body += '''</div>'''
    #print body

    if start_color and end_color and color_steps:
        num_colors = int(color_steps)
        colors = color_scheme(start_color, end_color, color_steps)
    else:
        num_colors = 5
        colors = color_scheme()

    # get the distinct frequencies and specify a font-size and color for each,
    # that corresponds to its size 
    freqs = []
    for f in [x[1] for x in dist]: 
        if f not in freqs: freqs.append(f)
    style = '''<style type="text/css">
.tagcloud {width: %s; height: %s; text-align: center; }
.word { text-align: center; vertical-align: middle; } ''' % (width, height)
    for f in freqs:
        freq_as_word = num_to_word(f)
        color_index = f % num_colors
        color = colors[color_index]
    	style += ('''
.%s {padding-left: 15px; padding-right: 15px; font-size: %s; color: %s }''' 
% (freq_as_word, font_size_fn(f), color))
    style += '''
</style>'''

   # assemble the response
    oid = pymongo.objectid.ObjectId()
    uid = str(oid)
    long_url = settings.HOME_PAGE + '/cloud/' + uid
    short_url = bitly_shorten(long_url)
    metadata = {
        'utc_created': datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%S"),
        'total_tags' : words_in_cloud,        
        'short_url' : short_url,
    }
    record =  {'_id': oid, 
                'body': body, 
                'style': style, 
                'metadata': metadata} 

    print style

    # save to the database
    con = pymongo.Connection()
    collection = con.wordapi.tagclouds
    collection.insert(record)
    
    return record
Пример #2
0
def word_cloud(dist, css_id = "", css_class = "", layout = "svg", width=800, 
		height=600, max_words = None, start_color=None, end_color=None, 
		color_steps=None, sort_order="random", equn="linear", slope=0.15, 
		link_prefix=None):

	''' returns a dict with style and body elements. style contains
	defalt styling for the tag cloud, while body contains the html
	markup. '''

	# sort() returns a list of (word, count) tuples in order of decreasing frequency 
	dist = sort(dist) 

	# truncate the list of items if max_words was specified
	if max_words:
		max_words = int(max_words)
		dist = dist[:max_words]            

	# get the number of words remaining in dist (ie the number of tags in the
	# tagcloud) to pass back in the metadata 
	words_in_cloud = len(dist)

	# the equation of the line used here determines the *relative* sizes of
	# different frequency words. actual font sizes are computed with javascript
	# to dynamically fit the words into the specified area. 

	# get the equation of the line AFTER truncating to max_words and BEFORE
	# shuffling the order around.  

	if equn == "log":
		# shifting the logarithm up by 1 gives a word of frequency one size one. 
		font_size_fn = lambda freq: 10*math.log(freq,2) + 1
	elif equn == "exp":  
		font_size_fn = lambda freq: math.pow(freq, 1.2)
	else: # linear
		min_size = 1
		font_size_fn = extrapolate_linear(dist, min_size, slope)

	# determine the sort order. if the sort order is frequency, there's nothing
	# to do since the distribution object is already sorted by frequency. 
	if sort_order == 'random': 
		# shuffles in place
		random.shuffle(dist)
	if sort_order == 'alphabetical':
		dist.sort()

	# assemble the class and id tags for the tag cloud's wrapping div 
	divstyle = '''id="tagcloud"'''
	if css_class != "": 
		divstyle = divstyle[:-1] + " " + class_ + ''' "'''

	# the user can specify a unique id for the tag cloud if they want
	# additional styling applied from their own style sheets. 
	if css_id != "": divstyle += ''' id="%s" ''' % id_
	divstyle = divstyle.strip()	  

	if layout == "svg":
		body = svg_based_layout(divstyle, dist)
	else:
		body = div_based_layout(divstyle, dist)


	if start_color and end_color and color_steps:
		num_colors = int(color_steps)
		colors = color_scheme(start_color, end_color, color_steps)
	else:
		num_colors = 5
		colors = color_scheme()

	# get the distinct frequencies and specify a font-size and color for each,
	# that corresponds to its size 
	freqs = []
	for f in [x[1] for x in dist]: 
		if f not in freqs: freqs.append(f)
	style = '''<style type="text/css">
// important clearfix for divs which wrap floating elements
// see http://nicolasgallagher.com/micro-clearfix-hack/
#hidden-resizer { zoom: 1; }
#hidden-resizer:before, #hidden-resizer:after { content: ""; display: table; }
#hidden-resizer:after { clear: both; }
#tagcloud { font-size: 10px; text-align: center; zoom: 1; }
#tagcloud:before, #tagcloud:after { content: ""; display: table; }
#tagcloud:after { clear: both; }
.word { text-align: center; vertical-align: middle; line-height:1; padding-right:5px; float:left; } '''
	for f in freqs:
		freq_as_word = num_to_word(f)
		color_index = f % num_colors
		color = colors[color_index]
		style += ('''
.%s {font-size: %sem; color: %s }''' 
% (freq_as_word, font_size_fn(f), color))
	style += '''
</style>'''

   # assemble the response (if you get an error on this line, probably mongo is
   # not running)
	oid = pymongo.objectid.ObjectId()
	uid = str(oid)
	long_url = settings.HOME_PAGE + '/cloud/' + uid
	if settings.DEBUG:
		short_url = long_url
	else:
		short_url = bitly_shorten(long_url)
	metadata = {
			'utc_created': datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%S"),
			'total_tags' : words_in_cloud,        
			'short_url' : short_url,
			}
	record =  {
			'_id': oid, 
			'body': body, 
			'style': style, 
			# hidden div to pass target width and height information onto javascript resizer
			# TODO not all tagclouds will necessarily have width and height?
			'wordcloud_size': '''
<div id="tagcloud_size" style="width:%dpx; height:%dpx; position: absolute; left:-999em;top:-999em;"></div>''' % (width, height),
			'script': '''<script type="text/javascript" src="http://code.jquery.com/jquery-1.7.1.min.js"></script>
			<script type="text/javascript" src="/media/js/wordcloud.js"></script>
			''',
			'metadata': metadata
			} 

	print style

	# save to the database
	con = pymongo.Connection()
	collection = con.wordapi.tagclouds
	collection.insert(record)

	return record