/*=========================================================================                                                                               
  Program:   bbtk
  Module:    $RCSfile: bbvtkSegmentationConnectivity.cxx,v $
  Language:  C++
  Date:      $Date: 2008/10/17 08:18:30 $
  Version:   $Revision: 1.4 $
=========================================================================*/

/* ---------------------------------------------------------------------

* Copyright (c) CREATIS-LRMN (Centre de Recherche en Imagerie Medicale)
* Authors : Eduardo Davila, Laurent Guigues, Jean-Pierre Roux
*
*  This software is governed by the CeCILL-B license under French law and 
*  abiding by the rules of distribution of free software. You can  use, 
*  modify and/ or redistribute the software under the terms of the CeCILL-B 
*  license as circulated by CEA, CNRS and INRIA at the following URL 
*  http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html 
*  or in the file LICENSE.txt.
*
*  As a counterpart to the access to the source code and  rights to copy,
*  modify and redistribute granted by the license, users are provided only
*  with a limited warranty  and the software's author,  the holder of the
*  economic rights,  and the successive licensors  have only  limited
*  liability. 
*
*  The fact that you are presently reading this means that you have had
*  knowledge of the CeCILL-B license and that you accept its terms.
* ------------------------------------------------------------------------ */                                                                         

/**
 *  \file 
 *  \brief 
 */


#ifdef _USE_VTK_


#include "bbvtkSegmentationConnectivity.h"
#include "bbvtkPackage.h"

namespace bbvtk
{

  BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,SegmentationConnectivity);
  BBTK_BLACK_BOX_IMPLEMENTATION(SegmentationConnectivity,
				bbtk::AtomicBlackBox);
  
  
  void SegmentationConnectivity::bbUserConstructor() 
  { 
    bbSetInputIn(NULL);
    std::vector<int> position;
    position.push_back(0);
    position.push_back(0);
    position.push_back(0);
    bbSetInputPositionXYZ(position);
    std::vector<int> threshold;
    threshold.push_back(0);
    threshold.push_back(0);
    bbSetInputThresholdMinMax(threshold);
    bbSetOutputOut(NULL);
    thresh2 = vtkImageThreshold::New();
    thresh2->SetInValue(255);
    thresh2->SetOutputScalarTypeToUnsignedShort();
    thresh2->SetOutValue(0);
    cast2 = vtkImageCast::New();
    cast2->SetInput(thresh2->GetOutput());
    cast2->SetOutputScalarTypeToUnsignedChar();
    connect2 = vtkImageSeedConnectivity::New();
    connect2->SetInput(cast2->GetOutput());
    connect2->SetInputConnectValue(255);
    connect2->SetOutputConnectedValue(255);
    connect2->SetOutputUnconnectedValue(0);
    cast4 = vtkImageCast::New();
    cast4->SetInput(connect2->GetOutput());
    cast4->SetOutputScalarTypeToUnsignedShort();
  }
  
  
  void SegmentationConnectivity::bbUserDestructor() 
  { 
    thresh2->Delete();
    cast2->Delete();
    connect2->Delete();
    cast4->Delete();
  }

  void SegmentationConnectivity::DoProcess()
  {
    vtkImageData *imagedata = bbGetInputIn();
    imagedata->UpdateInformation();
    imagedata->SetUpdateExtent(imagedata->GetWholeExtent());
    imagedata->Update();
    thresh2->ThresholdBetween(3000, 3001);
    thresh2->SetInput(imagedata);
    thresh2->ThresholdBetween(bbGetInputThresholdMinMax()[0], bbGetInputThresholdMinMax()[1]);
    thresh2->Update();
    cast2->Update();
    connect2->RemoveAllSeeds ();
    connect2->AddSeed( bbGetInputPositionXYZ()[0] , bbGetInputPositionXYZ()[1] , bbGetInputPositionXYZ()[2] );
    connect2->Update();
    cast4->Update();
    
    bbSetOutputOut(cast4->GetOutput() );	  
  }
  

}// EO namespace bbvtk


#endif //_USE_VTK_