-
Notifications
You must be signed in to change notification settings - Fork 0
/
composition.py
54 lines (46 loc) · 1.89 KB
/
composition.py
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
# -*- coding: utf-8 -*-
from graphviz import Digraph
import random
class Composition(object):
def graph(data):
print("Printing composition")
dot = Digraph(comment='Kanjis')
dot.engine = 'neato'
dot.format = 'svg'
dot.attr(overlap="false", sep="-0.1")
dot.attr('node', fontsize='30')
allkanjis = []
for kanji in data.kanjis:
if kanji in data.spotlight or random.random() < 0.1:
allkanjis.append(kanji)
modif = True
while modif:
modif = False
for kanji in allkanjis:
for newkanji in data.components[kanji]:
if newkanji not in allkanjis:
allkanjis.append(newkanji)
modif = True
for kanji in allkanjis:
if kanji in data.radicals:
continue
#try:
k = kanji#.decode('utf-8')
label = ("" + kanji + "\n " + data.descriptions[kanji])#.decode('utf-8')
color = data.colors[kanji]
if (len(data.components[kanji]) == 0 and data.colors[kanji] != '0.6 0.8 1.0'):
color = "lightgrey"
if not (len(data.anticomponents[kanji]) == 0):
shape = "circle"
else:
shape = "doublecircle"
dot.node(data.descriptions[kanji], label=kanji, fillcolor=color, style='filled', shape=shape)
for component in data.components[kanji]:
if component not in data.radicals and component in allkanjis:
source = component
if component in data.descriptions:
source = data.descriptions[component]
dot.edge(source, data.descriptions[kanji], constraint='true')#.decode('utf-8')
#except Exception:
# print("encoding fial")
return dot