Download as
Browse Code »

Commit de653d326ee28fc452a65361292d6a161da7357f

Authored by Thiago Franco de Moraes
1 parent daaeeb87
Exists in mask_greatest_component

Added Intesection and XOR

Showing 3 changed files with 18 additions and 4 deletions   Show diff stats
invesalius/constants.py
  Diff comments   View file @de653d3
@@ -571,3 +571,5 @@ PROJECTION_MIP_SIZE=2 @@ -571,3 +571,5 @@ PROJECTION_MIP_SIZE=2
571 # ------------- Boolean operations ------------------ 571 # ------------- Boolean operations ------------------
572 BOOLEAN_UNION = 1 572 BOOLEAN_UNION = 1
573 BOOLEAN_DIFF = 2 573 BOOLEAN_DIFF = 2
  574 +BOOLEAN_AND = 3
  575 +BOOLEAN_XOR = 4
invesalius/data/slice_.py
  Diff comments   View file @de653d3
@@ -1222,7 +1222,9 @@ class Slice(object): @@ -1222,7 +1222,9 @@ class Slice(object):
1222 1222
1223 def do_boolean_op(self, op, m1, m2): 1223 def do_boolean_op(self, op, m1, m2):
1224 name_ops = {const.BOOLEAN_UNION: _(u"Union"), 1224 name_ops = {const.BOOLEAN_UNION: _(u"Union"),
1225 - const.BOOLEAN_DIFF: _(u"Diff")} 1225 + const.BOOLEAN_DIFF: _(u"Diff"),
  1226 + const.BOOLEAN_AND: _(u"Intersection"),
  1227 + const.BOOLEAN_XOR: _(u"XOR")}
1226 1228
1227 1229
1228 name = u"%s_%s_%s" % (name_ops[op], m1.name, m2.name) 1230 name = u"%s_%s_%s" % (name_ops[op], m1.name, m2.name)
@@ -1250,6 +1252,12 @@ class Slice(object): @@ -1250,6 +1252,12 @@ class Slice(object):
1250 elif op == const.BOOLEAN_DIFF: 1252 elif op == const.BOOLEAN_DIFF:
1251 m[:] = ((m1 > 2) - (m2 > 2)) * 255 1253 m[:] = ((m1 > 2) - (m2 > 2)) * 255
1252 1254
  1255 + elif op == const.BOOLEAN_AND:
  1256 + m[:] = ((m1 > 2) & (m2 > 2)) * 255
  1257 +
  1258 + elif op == const.BOOLEAN_XOR:
  1259 + m[:] = numpy.logical_xor((m1 > 2), (m2 > 2)) * 255
  1260 +
1253 future_mask.was_edited = True 1261 future_mask.was_edited = True
1254 self._add_mask_into_proj(future_mask) 1262 self._add_mask_into_proj(future_mask)
1255 1263
invesalius/gui/dialogs.py
  Diff comments   View file @de653d3
@@ -1407,7 +1407,9 @@ class MaskBooleanDialog(wx.Dialog): @@ -1407,7 +1407,9 @@ class MaskBooleanDialog(wx.Dialog):
1407 self.mask2.SetSelection(0) 1407 self.mask2.SetSelection(0)
1408 1408
1409 op_choices = ((u"Union", const.BOOLEAN_UNION), 1409 op_choices = ((u"Union", const.BOOLEAN_UNION),
1410 - (u"Difference", const.BOOLEAN_DIFF)) 1410 + (u"Difference", const.BOOLEAN_DIFF),
  1411 + (u"Intersection", const.BOOLEAN_AND),
  1412 + (u"XOR", const.BOOLEAN_XOR))
1411 self.op_boolean = wx.ComboBox(self, -1, op_choices[0][0], choices=[]) 1413 self.op_boolean = wx.ComboBox(self, -1, op_choices[0][0], choices=[])
1412 1414
1413 for n, i in op_choices: 1415 for n, i in op_choices:
@@ -1443,6 +1445,8 @@ class MaskBooleanDialog(wx.Dialog): @@ -1443,6 +1445,8 @@ class MaskBooleanDialog(wx.Dialog):
1443 m1 = self.mask1.GetClientData(self.mask1.GetSelection()) 1445 m1 = self.mask1.GetClientData(self.mask1.GetSelection())
1444 m2 = self.mask2.GetClientData(self.mask2.GetSelection()) 1446 m2 = self.mask2.GetClientData(self.mask2.GetSelection())
1445 1447
1446 - print op, m1.name, m2.name  
1447 -  
1448 Publisher.sendMessage('Do boolean operation', (op, m1, m2)) 1448 Publisher.sendMessage('Do boolean operation', (op, m1, m2))
  1449 + Publisher.sendMessage('Reload actual slice')
  1450 +
  1451 + self.Close()
  1452 + self.Destroy()