def generate_data(self, obj):
"""Generate data to insert into the GroupClass table."""
data = {
'repr': srz.encode_group_class(obj),
'size': sum(obj.values())
}
yield data
def add_data(self):
"""Add data from a trial to the database."""
# Graph
G = self.random_connected_graph()
graph_id = srz.graph_to_int(G)
if not self.handlers['Graph'].exists(id=graph_id):
self.handlers['Graph'].insert(G)
# Subset of graph's nodes
nodes = sorted(G.nodes)
starting_nodes, node_encoding = randobj.random_subsequence(nodes, encoding=True)
methods = {'bfs': G.bfs, 'dfs': G.dfs}
for mtd, method in methods.iteritems():
# PermGroup
group = generate_group(method, starting_nodes)
group_repr = srz.encode_objects(group, srz.perm_to_int)
group_id = self.handlers['PermGroup'].exists(repr=group_repr)
if group_id == None:
# Permutation
for perm in group:
perm_id = srz.perm_to_int(perm)
if self.handlers['Permutation'].exists(id=perm_id) == None:
self.handlers['Permutation'].insert(perm)
# GroupClass
group_class = fctn.get_fingerprint(group)
group_class_repr = srz.encode_group_class(group_class)
group_class_id = self.handlers['GroupClass'].exists(repr=group_class_repr)
if group_class_id == None:
group_class_id = self.handlers['GroupClass'].insert(group_class)[0][0]
# Histogram
histogram = {'id': group_class_id, 'fingerprint': group_class}
self.handlers['Histogram'].insert(histogram)
group_data = {'repr': group_repr, 'cls': group_class_id}
group_id = self.handlers['PermGroup'].insert(group_data)[0][0]
# GroupElement
self.handlers['GroupElement'].insert({'id': group_id, 'elements': group})
# Trial
trial_data = {
'graph': graph_id, 'nodes': node_encoding,
'method': mtd, 'grp': group_id
}
self.handlers['Trial'].insert(trial_data)