diff options
Diffstat (limited to 'generate_graph.py')
| -rw-r--r-- | generate_graph.py | 42 |
1 files changed, 33 insertions, 9 deletions
diff --git a/generate_graph.py b/generate_graph.py index d47e345..1715404 100644 --- a/generate_graph.py +++ b/generate_graph.py @@ -11,7 +11,17 @@ subnode_threshold = float(sys.argv[10]) nodes = {} -nodes['n0'] = {'width': base_size, 'height': base_size, 'color': 'rgba(255,255,255,0.01)', 'mass': base_size, 'fixed': True, 'x': 0, 'y': 0, 'image': 'photo' + str(random.randint(1, 5)) + '.jpg'} +nodes['n0'] = { + 'width': base_size, + 'height': base_size, + 'color': 'rgba(255,255,255,0.01)', + 'mass': base_size, + 'fixed': True, + 'x': 0, + 'y': 0, + 'image': 'photo' + str(random.randint(1, 5)) + '.jpg', + 'main': True +} node_count = 1 node_sizes = [] @@ -24,8 +34,8 @@ node_colors = { 0.8: 'rgba(52,149,255,0.33)', 0.7: 'rgba(52,149,255,0.33)', 0.3: 'rgba(255,0,0,0.33)', - 0.2: 'yellow', - 0.1: 'white' + 0.2: 'rgba(255,255,0,0.33)', + 0.1: 'rgba(255,255,255,0.33)' } angular_positions = [2 * pi / len(node_sizes) * (n - 1) for n in range(1, len(node_sizes)+1)] @@ -40,9 +50,16 @@ for node_size in node_sizes: node_angular_position = angular_positions[node_count-1] node_position = { 'x': 0.8 * cos(node_angular_position), 'y': 0.8 * sin(node_angular_position) } - nodes[node_id] = {'width': node_dimension, 'height': node_dimension, 'mass': node_mass, 'color': node_colors[node_size], 'fixed': True, 'x': node_position['x'], 'y': node_position['y']} - if node_size > 0.2: - nodes[node_id]['image'] = 'photo' + str(random.randint(1, 5)) + '.jpg' + nodes[node_id] = { + 'width': node_dimension, + 'height': node_dimension, + 'mass': node_mass, + 'color': node_colors[node_size], + 'fixed': True, + 'x': node_position['x'], + 'y': node_position['y'], + 'image': 'photo' + str(random.randint(1, 5)) + '.jpg' + } edges['n0'][node_id] = {} edges[node_id] = {} node_count += 1 @@ -52,9 +69,16 @@ for node_size in node_sizes: subnode_id = node_id + '_' + str(subnode_count) subnode_dimension = max(5, int(ceil(base_size * node_size * subnode_size * size_quotient * size_quotient))) subnode_mass = int(ceil(base_size * mass_quotient)) - nodes[subnode_id] = {'width': subnode_dimension, 'height': subnode_dimension, 'mass': subnode_mass, 'color': node_colors[subnode_size], 'fixed': True, 'x': node_position['x'] * (0.995 + random.random() / 100), 'y': node_position['y'] * (0.995 + random.random() / 100)} - if subnode_size > 0.2: - nodes[subnode_id]['image'] = 'photo' + str(random.randint(1, 5)) + '.jpg' + nodes[subnode_id] = { + 'width': subnode_dimension, + 'height': subnode_dimension, + 'mass': subnode_mass, + 'color': node_colors[subnode_size], + 'fixed': True, + 'x': node_position['x'] * (0.995 + random.random() / 100), + 'y': node_position['y'] * (0.995 + random.random() / 100), + 'image': 'photo' + str(random.randint(1, 5)) + '.jpg' + } edges[node_id][subnode_id] = {} subnode_count += 1 |