im/edge.py at master · heyjusang/im · 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
class Edge:

  #id
  #src
  #dest
  #prob
  #valid
  #status

  STATUS_ACTIVATION = "activation"
  STATUS_INHIBITION = "inhibition"

  def __init__(self, id, status):
    from node import Node
    self.id = id
    self.prob = 1
    self.valid = False
    self.status = status

  def setSrc(self, src):
    self.src = src

  def setDest(self, dest):
    self.dest = dest

  def setProb(self, prob):
    self.prob = prob

  def validate(self):
    self.valid = True

  def invalidate(self):
    self.valid = False

  def getId(self):
    return self.id

  def getSrc(self):
    return self.src

  def getDest(self):
    return self.dest

  def getProb(self):
    return self.prob

  def isActivation(self):
    return self.status == Edge.STATUS_ACTIVATION

  def isInhibition(self):
    return self.status == Edge.STATUS_INHIBITION

  def isValid(self):
    return self.valid