scpy2/common.py at master · ruoyu0088/scpy2 · GitHub

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
# -*- coding: utf-8 -*-
from __future__ import absolute_import
import inspect


def issimple(obj):
    return isinstance(obj, (str, unicode, int, long, float))


def isempty(obj):
    if obj is None:
        return True
    if isinstance(obj, list) and obj == []:
        return True
    return False


def isproperty(obj, attr):
    try:
        return isinstance(getattr(obj.__class__, attr), property)
    except:
        return False


def trans(key):
    import string
    return str(key).translate(None, string.punctuation)


class Graphviz(object):

    def node(self, obj):
        if isempty(obj): return
        self.visited_obj.append(obj)
        if isinstance(obj, (list, tuple)):
            label="|".join("<f%d> %s" % (i, repr(v) if issimple(v) else "*") for i, v in enumerate(obj))
            color = "gray"
        elif isinstance(obj, dict):
            label="|".join("<f%s> %s" % (key, trans(key) + ":" + repr(val)
                                         if issimple(val) else key) for key, val in obj.items())
            color = "gray"
        else:
            label = obj.__class__.__name__
            format_func = getattr(self, "format_%s" % label, None)
            if format_func is not None:
                text = format_func(obj)
                self.result.append(text)
                return
            color = "white"
        text = 'obj_%d[label="%s",fillcolor=%s];' % (id(obj), label, color)
        self.result.append(text)

    def link(self, obj1, obj2, attr, is_property=False):
        if isempty(obj2): return
        style = "dashed" if is_property else "solid"
        text = 'obj_%d -> obj_%d[label="%s",style=%s];' % (id(obj1), id(obj2), attr, style)
        self.result.append(text)
        if issimple(obj2):
            text = 'obj_%d[label="%s", fillcolor=white]' % (id(obj2), repr(obj2))
            self.result.append(text)

    def list_link(self, alist, idx, obj):
        if isempty(obj): return
        if issimple(obj): return
        text = "obj_%d:f%d -> obj_%d;" % (id(alist), idx, id(obj))
        self.result.append(text)

    def dict_link(self, adict, key, obj):
        if isempty(obj): return
        if issimple(obj): return
        text = "obj_%d:f%s -> obj_%d;" % (id(adict), trans(key), id(obj))
        self.result.append(text)

    def __init__(self):
        self.checked_ids = set()
        self.expanded = set()
        self.visited_obj = []

    def _graphviz(self, obj):
        if obj is None: return
        if id(obj) in self.checked_ids: return
        if issimple(obj): return
        self.node(obj)
        self.checked_ids.add(id(obj))
        if isinstance(obj, (list, tuple)):
            for idx, inner in enumerate(obj):
                self._graphviz(inner)
                self.list_link(obj, idx, inner)
            return
        if isinstance(obj,  dict):
            for key, inner in obj.iteritems():
                self._graphviz(inner)
                self.dict_link(obj, key, inner)
            return

        class_names = set([cls.__name__ for cls in obj.__class__.mro()])
        klass = obj.__class__.__name__
        if (len(class_names & self.expand_classes) > 0 or
           klass in self.expand_once_classes and klass not in self.expanded):
            self.expanded.add( klass )
            for attr in self.check_attributes:
                    if hasattr(obj, attr):
                        subobj = getattr(obj, attr)
                        if inspect.isroutine(subobj):
                            continue
                        self._graphviz(subobj)
                        self.link(obj, subobj, attr, isproperty(obj, attr))

    @classmethod
    def graphviz(klass, obj):
        self = klass()
        self.checked_ids = set()
        self.expanded = set()
        self.result = [
        """digraph structs {
rankdir="LR";
node [shape=record,style=filled];
edge [fontsize=10, penwidth=0.5];"""
        ]
        self._graphviz(obj)
        self.result.append("}")
        return "\n".join(self.result)


class GraphvizDataFrame(Graphviz):

    check_attributes = ["index", "columns", "values", "name", "_data",
                        "blocks", "shape", "_engine", "mapping", "labels", "levels"]

    expand_classes = {"tuple"}
    expand_once_classes = {"Index", "DataFrame", "BlockManager", "MultiIndex",
                           "FloatBlock", "ObjectBlock", "ObjectEngine", "DatetimeBlock"}


class GraphvizMatplotlib(Graphviz):

    check_attributes = [
        "patch", "axes", "lines", "patches", "texts", "artists",
        "xaxis", "yaxis", "majorTicks", "label", "tick1line", "tick2line"
    ]

    expand_classes = {"Figure", "Axes", "Line2D", "Text"}
    expand_once_classes = {"XAxis", "YAxis",  "XTick", "YTick"}


class GraphvizMPLTransform(Graphviz):

    check_attributes = [
        "_boxout", "_bbox", "_transform", "_points", "_mtx", "_boxin", "_child", "_x", "_y"]

    expand_classes = {"TransformNode"}
    expand_once_classes = {}

    def format_ndarray(self, obj):
        if obj.ndim == 2:
            html = "".join("<tr>%s</tr>" % "".join("<td>%g</td>" % v for v in row) for row in obj)
            html = '<table border="0">%s</table>' % html
        text = '''obj_%d[label=<%s>,fillcolor=white, fontsize=9];''' % (id(obj), html)
        return text