Simple-Bayesian-Network-Query-System/main.py at master · fauzanzaid/Simple-Bayesian-Network-Query-System · 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
#! /usr/bin/python2
# -*- coding: utf-8 -*-

import threading

from bayesian_network import BayesianNetwork, get_prob_from_names


class Main(threading.Thread):
	"""docstring for Main"""

	MAX_SEL = 10

	MSG_WELCOME = "Welcome to Bayesian Network solver! Press q to quit"
	MSG_SEL_LMT = "You cannot select more than 10 variables!"
	MSG_WORKING = "Calculating probability..."

	def __init__(self, qu_usr_ip, qu_cmd, input_filename="input1.txt"):
		super(Main, self).__init__()
		self.qu_usr_ip = qu_usr_ip
		self.qu_cmd = qu_cmd

		self.input_filename = input_filename
		self.init_network()

		self.cur_names_qry = []
		self.cur_names_cond = []
		self.cur_name_mrkv = None


	def init_network(self):
		self.bn = BayesianNetwork()
		self.bn.init_from_file(self.input_filename)
		self.node_names = sorted([n.name for n in self.bn.nodes])


	def send_cmd(self, cmd, *args):
		msg = (cmd, args)
		self.qu_cmd.put(msg)


	def run(self):

		self.send_cmd("init_node_names", self.node_names)
		self.send_cmd("draw_base")
		self.send_cmd("display_msg", self.MSG_WELCOME)

		while True:
			usr_ip = self.qu_usr_ip.get()
			t, dev, arg = usr_ip
			# print "Main rcvd", usr_ip

			if dev == "keypress":
				if arg == 'q':
					self.send_cmd("quit")
					break

			else:
				if arg == None:
					pass

				elif arg[0] == "qry" or arg[0] == "cond":

					if arg[0] == "qry":
						if arg[1] in self.cur_names_qry:
							self.send_cmd("off", arg[0], arg[1])
							self.cur_names_qry.remove(arg[1])
						elif arg[1][1:] in self.cur_names_qry:
							self.send_cmd("off", arg[0], arg[1][1:])
							self.cur_names_qry.remove(arg[1][1:])
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_qry.append(arg[1])
						elif "~"+arg[1] in self.cur_names_qry:
							self.send_cmd("off", arg[0], "~"+arg[1])
							self.cur_names_qry.remove("~"+arg[1])
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_qry.append(arg[1])
						elif len(self.cur_names_qry) < self.MAX_SEL:
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_qry.append(arg[1])
						else:
							self.send_cmd("display_msg", self.MSG_SEL_LMT)
							continue

					elif arg[0] == "cond":
						if arg[1] in self.cur_names_cond:
							self.send_cmd("off", arg[0], arg[1])
							self.cur_names_cond.remove(arg[1])
						elif arg[1][1:] in self.cur_names_cond:
							self.send_cmd("off", arg[0], arg[1][1:])
							self.cur_names_cond.remove(arg[1][1:])
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_cond.append(arg[1])
						elif "~"+arg[1] in self.cur_names_cond:
							self.send_cmd("off", arg[0], "~"+arg[1])
							self.cur_names_cond.remove("~"+arg[1])
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_cond.append(arg[1])
						elif len(self.cur_names_cond) < self.MAX_SEL:
							self.send_cmd("on", arg[0], arg[1])
							self.cur_names_cond.append(arg[1])
						else:
							self.send_cmd("display_msg", self.MSG_SEL_LMT)
							continue

					# Calc
					p = get_prob_from_names(self.bn, self.cur_names_qry, self.cur_names_cond)
					self.send_cmd("display_expr", p.to_string())
					self.send_cmd("display_msg", self.MSG_WORKING)
					val = p.evaluate(self.bn)
					self.send_cmd("display_prob", val)

				elif arg[0] == "mrkv":
					if self.cur_name_mrkv != None:
						self.send_cmd("off", arg[0], self.cur_name_mrkv)
					self.send_cmd("on", arg[0], arg[1])
					self.cur_name_mrkv = arg[1]

					cur_node_mrkv = self.bn.nodes_by_name[self.cur_name_mrkv]
					mrkv_blanket = cur_node_mrkv.get_markov_blanket()
					self.send_cmd("draw_mrkv", [n.name for n in mrkv_blanket])