Download as
Browse Code »

Commit e2d5f585ac42ef131f543e81762860ef4517328d

Authored by Thiago Franco de Moraes
1 parent 525d68ff
Exists in select_part

Strip white spaces and better doc in Mask create_mask class

Showing 1 changed file with 11 additions and 6 deletions   Show diff stats
invesalius/data/mask.py
  Diff comments   View file @e2d5f58
... ... @@ -126,7 +126,7 @@ class EditionHistory(object):
126 126 h[self.index].commit_history(mvolume)
127 127 self._reload_slice(self.index)
128 128 Publisher.sendMessage("Enable redo", True)
129   -
  129 +
130 130 if self.index == 0:
131 131 Publisher.sendMessage("Enable undo", False)
132 132 print "AT", self.index, len(self.history), self.history[self.index].filename
... ... @@ -154,7 +154,7 @@ class EditionHistory(object):
154 154 h[self.index].commit_history(mvolume)
155 155 self._reload_slice(self.index)
156 156 Publisher.sendMessage("Enable undo", True)
157   -
  157 +
158 158 if self.index == len(h) - 1:
159 159 Publisher.sendMessage("Enable redo", False)
160 160 print "AT", self.index, len(h), h[self.index].filename
... ... @@ -174,7 +174,7 @@ class EditionHistory(object):
174 174 v_undo = False
175 175 elif self.index == len(self.history) - 1:
176 176 v_redo = False
177   -
  177 +
178 178 Publisher.sendMessage("Enable undo", v_undo)
179 179 Publisher.sendMessage("Enable redo", v_redo)
180 180  
... ... @@ -229,7 +229,7 @@ class Mask():
229 229  
230 230 def SavePlist(self, dir_temp, filelist):
231 231 mask = {}
232   - filename = u'mask_%d' % self.index
  232 + filename = u'mask_%d' % self.index
233 233 mask_filename = u'%s.dat' % filename
234 234 mask_filepath = os.path.join(dir_temp, mask_filename)
235 235 filelist[self.temp_file] = mask_filename
... ... @@ -304,7 +304,12 @@ class Mask():
304 304 Mask.general_index = index
305 305  
306 306 def create_mask(self, shape):
307   - print "Creating a mask"
  307 + """
  308 + Creates a new mask object. This method do not append this new mask into the project.
  309 +
  310 + Parameters:
  311 + shape(int, int, int): The shape of the new mask.
  312 + """
308 313 self.temp_file = tempfile.mktemp()
309 314 shape = shape[0] + 1, shape[1] + 1, shape[2] + 1
310 315 self.matrix = numpy.memmap(self.temp_file, mode='w+', dtype='uint8', shape=shape)
... ... @@ -329,7 +334,7 @@ class Mask():
329 334 new_mask.threshold_range = self.threshold_range
330 335 new_mask.edition_threshold_range = self.edition_threshold_range
331 336 new_mask.is_shown = self.is_shown
332   -
  337 +
333 338 new_mask.create_mask(shape=[i-1 for i in self.matrix.shape])
334 339 new_mask.matrix[:] = self.matrix[:]
335 340  
... ...