Пример #1
0
  def updateLabel(self,value):
    if not self.fm:
      return
    self.fm.show(value)
    self.fm.Modified()
    self.fm.Update()

    EditUtil.getLabelImage().DeepCopy(self.fm.GetOutput())
    EditUtil.getLabelImage().Modified()

    EditUtil.markVolumeNodeAsModified(self.sliceLogic.GetLabelLayer().GetVolumeNode())
Пример #2
0
  def updateLabel(self,value):
    if not self.fm:
      return
    self.fm.show(value)
    self.fm.Modified()
    self.fm.Update()

    EditUtil.getLabelImage().DeepCopy(self.fm.GetOutput())
    EditUtil.getLabelImage().Modified()

    EditUtil.markVolumeNodeAsModified(self.sliceLogic.GetLabelLayer().GetVolumeNode())
Пример #3
0
  def percentMaxChanged(self, val):
    labelNode = self.logic.getLabelNode()
    labelImage = EditUtil.getLabelImage()
    spacing = labelNode.GetSpacing()
    dim = labelImage.GetDimensions()
    print dim
    totalVolume = spacing[0]*dim[0]+spacing[1]*dim[1]+spacing[2]*dim[2]

    percentVolumeStr = "%.5f" % (totalVolume*val/100.)
    self.percentVolume.text = '(maximum total volume: '+percentVolumeStr+' mL)'
Пример #4
0
    def percentMaxChanged(self, val):
        labelNode = self.logic.getLabelNode()
        labelImage = EditUtil.getLabelImage()
        spacing = labelNode.GetSpacing()
        dim = labelImage.GetDimensions()
        print dim
        totalVolume = spacing[0] * dim[0] + spacing[1] * dim[1] + spacing[2] * dim[2]

        percentVolumeStr = "%.5f" % (totalVolume * val / 100.0)
        self.percentVolume.text = "(maximum total volume: " + percentVolumeStr + " mL)"
Пример #5
0
  def percentMaxChanged(self, val):
    labelNode = self.logic.getLabelNode()
    labelImage = EditUtil.getLabelImage()
    spacing = labelNode.GetSpacing()
    if vtk.VTK_MAJOR_VERSION <= 5:
      dim = labelImage.GetDimensions()
      totalVolume = spacing[0]*(dim[1]+1)+spacing[1]*(dim[3]+1)+spacing[2]*(dim[5]+1)
    else:
      dim = labelImage.GetDimensions()
      print dim
      totalVolume = spacing[0]*dim[0]+spacing[1]*dim[1]+spacing[2]*dim[2]

      percentVolumeStr = "%.5f" % (totalVolume*val/100.)
    self.percentVolume.text = '(maximum total volume: '+percentVolumeStr+' mL)'
Пример #6
0
    def fastMarching(self, percentMax):

        self.fm = None
        # allocate a new filter each time March is hit
        bgImage = EditUtil.getBackgroundImage()
        labelImage = EditUtil.getLabelImage()

        # collect seeds
        if vtk.VTK_MAJOR_VERSION <= 5:
            dim = bgImage.GetWholeExtent()
        else:
            dim = bgImage.GetDimensions()
            print dim
        # initialize the filter
        self.fm = slicer.vtkPichonFastMarching()
        scalarRange = bgImage.GetScalarRange()
        depth = scalarRange[1] - scalarRange[0]

        # this is more or less arbitrary; large depth values will bring the
        # algorithm to the knees
        scaleValue = 0
        shiftValue = 0

        if depth > 300:
            scaleValue = 300. / depth
        if scalarRange[0] < 0:
            shiftValue = scalarRange[0] * -1

        if scaleValue or shiftValue:
            rescale = vtk.vtkImageShiftScale()
            if vtk.VTK_MAJOR_VERSION <= 5:
                rescale.SetInput(bgImage)
            else:
                rescale.SetInputData(bgImage)
            rescale.SetScale(scaleValue)
            rescale.SetShift(shiftValue)
            rescale.Update()
            bgImage = rescale.GetOutput()
            scalarRange = bgImage.GetScalarRange()
            depth = scalarRange[1] - scalarRange[0]

        print('Input scalar range: ' + str(depth))
        if vtk.VTK_MAJOR_VERSION <= 5:
            self.fm.init(dim[1] + 1, dim[3] + 1, dim[5] + 1, depth, 1, 1, 1)
        else:
            self.fm.init(dim[0], dim[1], dim[2], depth, 1, 1, 1)

        caster = vtk.vtkImageCast()
        caster.SetOutputScalarTypeToShort()
        if vtk.VTK_MAJOR_VERSION <= 5:
            caster.SetInput(bgImage)
            caster.Update()
            self.fm.SetInput(caster.GetOutput())
        else:
            caster.SetInputData(bgImage)
            self.fm.SetInputConnection(caster.GetOutputPort())

        # self.fm.SetOutput(labelImage)

        if vtk.VTK_MAJOR_VERSION <= 5:
            npoints = int(
                (dim[1] + 1) * (dim[3] + 1) * (dim[5] + 1) * percentMax / 100.)
        else:
            npoints = int(dim[0] * dim[1] * dim[2] * percentMax / 100.)

        self.fm.setNPointsEvolution(npoints)
        print('Setting active label to ' + str(EditUtil.getLabel()))
        self.fm.setActiveLabel(EditUtil.getLabel())

        nSeeds = self.fm.addSeedsFromImage(labelImage)
        if nSeeds == 0:
            return 0

        self.fm.Modified()
        self.fm.Update()

        # TODO: need to call show() twice for data to be updated
        self.fm.show(1)
        self.fm.Modified()
        self.fm.Update()

        self.fm.show(1)
        self.fm.Modified()
        self.fm.Update()

        self.undoRedo.saveState()

        EditUtil.getLabelImage().DeepCopy(self.fm.GetOutput())
        EditUtil.markVolumeNodeAsModified(
            self.sliceLogic.GetLabelLayer().GetVolumeNode())
        # print('FastMarching output image: '+str(output))
        print('FastMarching march update completed')

        return npoints
Пример #7
0
  def fastMarching(self,percentMax):

    self.fm = None
    # allocate a new filter each time March is hit
    bgImage = EditUtil.getBackgroundImage()
    labelImage = EditUtil.getLabelImage()

    # collect seeds
    if vtk.VTK_MAJOR_VERSION <= 5:
      dim = bgImage.GetWholeExtent()
    else:
      dim = bgImage.GetDimensions()
      print dim
    # initialize the filter
    self.fm = slicer.vtkPichonFastMarching()
    scalarRange = bgImage.GetScalarRange()
    depth = scalarRange[1]-scalarRange[0]

    # this is more or less arbitrary; large depth values will bring the
    # algorithm to the knees
    scaleValue = 0
    shiftValue = 0

    if depth>300:
      scaleValue = 300./depth
    if scalarRange[0] < 0:
      shiftValue = scalarRange[0]*-1

    if scaleValue or shiftValue:
      rescale = vtk.vtkImageShiftScale()
      if vtk.VTK_MAJOR_VERSION <= 5:
        rescale.SetInput(bgImage)
      else:
        rescale.SetInputData(bgImage)
      rescale.SetScale(scaleValue)
      rescale.SetShift(shiftValue)
      rescale.Update()
      bgImage = rescale.GetOutput()
      scalarRange = bgImage.GetScalarRange()
      depth = scalarRange[1]-scalarRange[0]

    print('Input scalar range: '+str(depth))
    if vtk.VTK_MAJOR_VERSION <= 5:
      self.fm.init(dim[1]+1, dim[3]+1, dim[5]+1, depth, 1, 1, 1)
    else:
      self.fm.init(dim[0], dim[1], dim[2], depth, 1, 1, 1)

    caster = vtk.vtkImageCast()
    caster.SetOutputScalarTypeToShort()
    if vtk.VTK_MAJOR_VERSION <= 5:
      caster.SetInput(bgImage)
      caster.Update()
      self.fm.SetInput(caster.GetOutput())
    else:
      caster.SetInputData(bgImage)
      self.fm.SetInputConnection(caster.GetOutputPort())

    # self.fm.SetOutput(labelImage)

    if vtk.VTK_MAJOR_VERSION <= 5:
      npoints = int((dim[1]+1)*(dim[3]+1)*(dim[5]+1)*percentMax/100.)
    else:
      npoints = int(dim[0]*dim[1]*dim[2]*percentMax/100.)

    self.fm.setNPointsEvolution(npoints)
    print('Setting active label to '+str(EditUtil.getLabel()))
    self.fm.setActiveLabel(EditUtil.getLabel())

    nSeeds = self.fm.addSeedsFromImage(labelImage)
    if nSeeds == 0:
      return 0

    self.fm.Modified()
    self.fm.Update()

    # TODO: need to call show() twice for data to be updated
    self.fm.show(1)
    self.fm.Modified()
    self.fm.Update()

    self.fm.show(1)
    self.fm.Modified()
    self.fm.Update()

    self.undoRedo.saveState()

    EditUtil.getLabelImage().DeepCopy(self.fm.GetOutput())
    EditUtil.markVolumeNodeAsModified(self.sliceLogic.GetLabelLayer().GetVolumeNode())
    # print('FastMarching output image: '+str(output))
    print('FastMarching march update completed')

    return npoints