diff options
author | Qiaowei Ren <qiaowei.ren@intel.com> | 2018-01-04 13:43:33 +0800 |
---|---|---|
committer | Qiaowei Ren <qiaowei.ren@intel.com> | 2018-01-05 11:59:39 +0800 |
commit | 812ff6ca9fcd3e629e49d4328905f33eee8ca3f5 (patch) | |
tree | 04ece7b4da00d9d2f98093774594f4057ae561d4 /src/ceph/doc/scripts/gen_state_diagram.py | |
parent | 15280273faafb77777eab341909a3f495cf248d9 (diff) |
initial code repo
This patch creates initial code repo.
For ceph, luminous stable release will be used for base code,
and next changes and optimization for ceph will be added to it.
For opensds, currently any changes can be upstreamed into original
opensds repo (https://github.com/opensds/opensds), and so stor4nfv
will directly clone opensds code to deploy stor4nfv environment.
And the scripts for deployment based on ceph and opensds will be
put into 'ci' directory.
Change-Id: I46a32218884c75dda2936337604ff03c554648e4
Signed-off-by: Qiaowei Ren <qiaowei.ren@intel.com>
Diffstat (limited to 'src/ceph/doc/scripts/gen_state_diagram.py')
-rwxr-xr-x | src/ceph/doc/scripts/gen_state_diagram.py | 214 |
1 files changed, 214 insertions, 0 deletions
diff --git a/src/ceph/doc/scripts/gen_state_diagram.py b/src/ceph/doc/scripts/gen_state_diagram.py new file mode 100755 index 0000000..fccde26 --- /dev/null +++ b/src/ceph/doc/scripts/gen_state_diagram.py @@ -0,0 +1,214 @@ +#!/usr/bin/env python +import re +import sys + + +def do_filter(generator): + return acc_lines(remove_multiline_comments(to_char(remove_single_line_comments(generator)))) + + +def acc_lines(generator): + current = "" + for i in generator: + current += i + if i == ';' or \ + i == '{' or \ + i == '}': + yield current.lstrip("\n") + current = "" + + +def to_char(generator): + for line in generator: + for char in line: + if char is not '\n': + yield char + else: + yield ' ' + + +def remove_single_line_comments(generator): + for i in generator: + if len(i) and i[0] == '#': + continue + yield re.sub(r'//.*', '', i) + + +def remove_multiline_comments(generator): + saw = "" + in_comment = False + for char in generator: + if in_comment: + if saw is "*": + if char is "/": + in_comment = False + saw = "" + if char is "*": + saw = "*" + continue + if saw is "/": + if char is '*': + in_comment = True + saw = "" + continue + else: + yield saw + saw = "" + if char is '/': + saw = "/" + continue + yield char + + +class StateMachineRenderer(object): + def __init__(self): + self.states = {} # state -> parent + self.machines = {} # state-> initial + self.edges = {} # event -> [(state, state)] + + self.context = [] # [(context, depth_encountered)] + self.context_depth = 0 + self.state_contents = {} + self.subgraphnum = 0 + self.clusterlabel = {} + + def __str__(self): + return "-------------------\n\nstates: %s\n\n machines: %s\n\n edges: %s\n\n context %s\n\n state_contents %s\n\n--------------------" % ( + self.states, + self.machines, + self.edges, + self.context, + self.state_contents + ) + + def read_input(self, input_lines): + previous_line = None + for line in input_lines: + self.get_state(line) + self.get_event(line) + # pass two lines at a time to get the context so that regexes can + # match on split signatures + self.get_context(line, previous_line) + previous_line = line + + def get_context(self, line, previous_line): + match = re.search(r"(\w+::)*::(?P<tag>\w+)::\w+\(const (?P<event>\w+)", line) + if match is None and previous_line is not None: + # it is possible that we need to match on the previous line as well, so join + # them to make them one line and try and get this matching + joined_line = ' '.join([previous_line, line]) + match = re.search(r"(\w+::)*::(?P<tag>\w+)::\w+\(\s*const (?P<event>\w+)", joined_line) + if match is not None: + self.context.append((match.group('tag'), self.context_depth, match.group('event'))) + if '{' in line: + self.context_depth += 1 + if '}' in line: + self.context_depth -= 1 + while len(self.context) and self.context[-1][1] == self.context_depth: + self.context.pop() + + def get_state(self, line): + if "boost::statechart::state_machine" in line: + tokens = re.search( + r"boost::statechart::state_machine<\s*(\w*),\s*(\w*)\s*>", + line) + if tokens is None: + raise Exception("Error: malformed state_machine line: " + line) + self.machines[tokens.group(1)] = tokens.group(2) + self.context.append((tokens.group(1), self.context_depth, "")) + return + if "boost::statechart::state" in line: + tokens = re.search( + r"boost::statechart::state<\s*(\w*),\s*(\w*)\s*,?\s*(\w*)\s*>", + line) + if tokens is None: + raise Exception("Error: malformed state line: " + line) + self.states[tokens.group(1)] = tokens.group(2) + if tokens.group(2) not in self.state_contents.keys(): + self.state_contents[tokens.group(2)] = [] + self.state_contents[tokens.group(2)].append(tokens.group(1)) + if tokens.group(3) is not "": + self.machines[tokens.group(1)] = tokens.group(3) + self.context.append((tokens.group(1), self.context_depth, "")) + return + + def get_event(self, line): + if "boost::statechart::transition" in line: + for i in re.finditer(r'boost::statechart::transition<\s*([\w:]*)\s*,\s*(\w*)\s*>', + line): + if i.group(1) not in self.edges.keys(): + self.edges[i.group(1)] = [] + if len(self.context) is 0: + raise Exception("no context at line: " + line) + self.edges[i.group(1)].append((self.context[-1][0], i.group(2))) + i = re.search("return\s+transit<\s*(\w*)\s*>()", line) + if i is not None: + if len(self.context) is 0: + raise Exception("no context at line: " + line) + if self.context[-1][2] is "": + raise Exception("no event in context at line: " + line) + if self.context[-1][2] not in self.edges.keys(): + self.edges[self.context[-1][2]] = [] + self.edges[self.context[-1][2]].append((self.context[-1][0], i.group(1))) + + def emit_dot(self): + top_level = [] + for state in self.machines.keys(): + if state not in self.states.keys(): + top_level.append(state) + print >> sys.stderr, "Top Level States: ", str(top_level) + print """digraph G {""" + print '\tsize="7,7"' + print """\tcompound=true;""" + for i in self.emit_state(top_level[0]): + print '\t' + i + for i in self.edges.keys(): + for j in self.emit_event(i): + print j + print """}""" + + def emit_state(self, state): + if state in self.state_contents.keys(): + self.clusterlabel[state] = "cluster%s" % (str(self.subgraphnum),) + yield "subgraph cluster%s {" % (str(self.subgraphnum),) + self.subgraphnum += 1 + yield """\tlabel = "%s";""" % (state,) + yield """\tcolor = "blue";""" + for j in self.state_contents[state]: + for i in self.emit_state(j): + yield "\t"+i + yield "}" + else: + found = False + for (k, v) in self.machines.items(): + if v == state: + yield state+"[shape=Mdiamond];" + found = True + break + if not found: + yield state+";" + + def emit_event(self, event): + def append(app): + retval = "[" + for i in app: + retval += (i + ",") + retval += "]" + return retval + for (fro, to) in self.edges[event]: + appendix = ['label="%s"' % (event,)] + if fro in self.machines.keys(): + appendix.append("ltail=%s" % (self.clusterlabel[fro],)) + while fro in self.machines.keys(): + fro = self.machines[fro] + if to in self.machines.keys(): + appendix.append("lhead=%s" % (self.clusterlabel[to],)) + while to in self.machines.keys(): + to = self.machines[to] + yield("%s -> %s %s;" % (fro, to, append(appendix))) + + +INPUT_GENERATOR = do_filter(sys.stdin.xreadlines()) +RENDERER = StateMachineRenderer() +RENDERER.read_input(INPUT_GENERATOR) +RENDERER.emit_dot() |