#==============================================================================
#
# Project:   SharpImage
# Module:    MorphologicalCloseByReconstruction.py
# Language:  IronPython
# Author:    Dan Mueller
# Date:      $Date: 2007-07-13 06:30:23 +1000 (Fri, 13 Jul 2007) $
# Revision:  $Revision: 8 $
#
# Copyright (c) Queensland University of Technology (QUT) 2007.
# All rights reserved.
#
# This software is distributed WITHOUT ANY WARRANTY; without even
# the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
# PURPOSE.  See the above copyright notices for more information.
#
#==============================================================================

# Import the base script class
import ImageToImageScript
from ImageToImageScript import *

# Add reference and import Morphology library
clr.AddReference("ManagedITK.MorphologicalFilters")
from itk import *

class MorphologicalCloseByReconstructionScript(ImageToImageScriptObject):
    # -------------------------------------------------------------------------
    Name = "MorphologicalCloseByReconstruction"
    Help = """Computes the greyscale morphological closing by reconstruction.
              This fitler is similar to MorpholicalClose, except that closing
              by reconstruction preserves the shape of the components.
              The morphological closing by reconstruction of an image "I" is defined as:
              \tClosingByReconstruction(I) = ErodeByReconstruction( DilateByReconstruction(I) )"""
    Parameters = """(itkSize) KernelRadius = the radius of the ball. (itkSize(1,1,.))
                    (itkFlatStructuringElement) Kernel = the structuring element. (Ball)"""
    KernelRadius = None
    Kernel = None
    # -------------------------------------------------------------------------

    def ThreadedDoWork(self):
        """ Perform the main functions of the script on a background thread. """
        try:
            self.StartedWork()
            self.WriteInputName()
            filterClose = itkClosingByReconstructionImageFilter.New( self.Input, self.Output )
            self.AddEventHandlersToProcessObject( filterClose )
            filterClose.SetInput( self.Input )
            if (self.KernelRadius == None):
                self.KernelRadius = itkSize( self.Input.Dimension )
                self.KernelRadius.Fill( 1 )
            if (self.Kernel == None):
                self.Kernel = itkFlatStructuringElement.Ball( self.KernelRadius );
            filterClose.Kernel = self.Kernel
            filterClose.FullyConnected = False
            filterClose.PreserveIntensities = True
            self.WriteLineToConsole( "Kernel=" + self.Kernel.ToString() )
            self.WriteLineToConsole( "FullyConnected=" + filterClose.FullyConnected.ToString() )
            self.WriteLineToConsole( "PreserveIntensities=" + filterClose.PreserveIntensities.ToString() )
            filterClose.UpdateLargestPossibleRegion()
            filterClose.GetOutput( self.Output )
            self.DisconnectInputAndOutput()
            self.DisposeOfObject( filterClose )
            self.WriteOutputName()
            self.FinishedWork( True )
        except Exception, ex:
            self.HandleException( ex )
            self.FinishedWork( False )
            self.Finalise( )