thresholding.cpp

Go to the documentation of this file.

// $Id: thresholding.cpp,v 1.18 2003/01/16 18:36:52 sarang Exp $
//
// OpenVL - A Library for working with volumetric datasets.
//          http://openvl.sf.net
//
// Copyright (C) 2000-2002  Sarang Lakare <sarang#users.sf.net>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Library General Public
// License as published by the Free Software Foundation; either
// version 2 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Library General Public License for more details.
//
// You should have received a copy of the GNU Library General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA.
//
// Please report all bugs and problems to openvl-devel@lists.sf.net or
// at the OpenVL website.
//

#include <iostream>

#include "vlvarlist.h"
#include "vlvolume.h"
#include "vlgenericfactory.h"
#include "vlvoliterator.h"

#include "thresholding.h"


typedef vlGenericFactory<vlVolProcessorThresh> vlVolProcessorThreshFactory;
VL_EXPORT_COMPONENT_FACTORY( thresholder, vlVolProcessorThreshFactory );

vlVolProcessorThresh::vlVolProcessorThresh()
{
  infoRef().setVersion(0, 1, 0);
  infoRef().setName("Thresholder");
  infoRef().setService("Thresholder", "Performs thresholding.");
  infoRef().addAuthor("Sarang Lakare");
  infoRef().setCopyrightText("2000-2002 Sarang Lakare");
}


vlVolProcessorThresh::~vlVolProcessorThresh()
{
}


void vlVolProcessorThresh::init()
{
  configRef().clear();
  configRef().add(vol()->dataType(), "thresholdHigh", "Threshold Value - High");
  configRef().add(vol()->dataType(), "thresholdLow", "Threshold Value - Low");
  vlVolume *vol;
  configRef().add(vol, "maskVolume", "Mask volume.");
  uint8 inMask(0xFF), outMask(0x00);
  configRef().add(inMask, "insideMask", "Mask for voxels inside the region.", true);
  configRef().add(outMask, "outsideMask", "Mask for voxels outside the region.", true);
}

template <>
bool vlVolProcessorThresh::runT<uint8>(uint8 & dummy)
{
  return (thresholding(dummy));
}

template <>
bool vlVolProcessorThresh::runT<int8>(int8 & dummy)
{
  return (thresholding(dummy));
}

template <>
bool vlVolProcessorThresh::runT<uint16>(uint16 & dummy)
{
  return (thresholding(dummy));
}

template <>
bool vlVolProcessorThresh::runT<int16>(int16 & dummy)
{
  return (thresholding(dummy));
}


template <>
bool vlVolProcessorThresh::runT<uint32>(uint32 & dummy)
{
  return (thresholding(dummy));
}


template <>
bool vlVolProcessorThresh::runT<int32>(int32 & dummy)
{
  return (thresholding(dummy));
}

//template <>
//bool vlVolProcessorThresh::runT<float>(float & dummy)
//{
//  return (thresholding(dummy));
//}
//
//template <>
//bool vlVolProcessorThresh::runT<double>(double & dummy)
//{
//  return (thresholding(dummy));
//}


bool vlVolProcessorThresh::run()
{
  if(!vol())
    return (false);
  
  bool retValue(false);

  callFunctionOnDataTypeNoArgs(vol()->dataType(), retValue, runT);

  return (retValue);
}


template <class DataType>
bool vlVolProcessorThresh::runT(DataType & dummy)
{
  std::cout << "Cannot apply thresholding on this datatype [" << typeid(DataType).name() << "]" << std::endl;
  return (false);
}


template <class DataType>
bool vlVolProcessorThresh::thresholding(DataType & dummy)
{
  std::cout << "About to run thresholding.. " << std::endl;

  vlVolume *maskVol;

  if(!configRef().get("maskVolume", maskVol)) {
    std::cout << "ERROR : No mask volume specified." << std::endl;
    return (false);
  }

  if(vol()->dim() != maskVol->dim()) {
    std::cout << "ERROR : Mask volume dimensions not same as the volume dimension." << std::endl;
    return (false);
  }

  if(maskVol->dataType() != UnsignedInt8) {
    std::cout << "ERROR : Mask volume has to be of unsigned int8 data type." << std::endl;
    return (false);
  }
  
  DataType threshLow, threshHigh;
  uint8 inVoxelMask(0x00);
  uint8 outVoxelMask(0x00);

  uint64 regionVoxelCount(0); // count the number of voxels which will belong to te region

  if(!configRef().get("thresholdHigh", threshHigh)) {
    std::cout << "ERROR : Higher threshold not specified." << std::endl;
    return (false);
  }

  if(!configRef().get("thresholdLow", threshLow)) {
    std::cout << "ERROR : Lower threshold not specified." << std::endl;
    return (false);
  }

  configRef().get("insideMask", inVoxelMask);
  configRef().get("outsideMask", outVoxelMask);

  // make sure thresholds are proper
  if(threshHigh < threshLow) {
    std::cout << "ERROR : thresholdHigh < thresholdLow. This is not allowed." << std::endl;
    return (false);
  }
  
  std::cout << "Threshold value(s) set to : " << (int)threshLow << " -- " << (int)threshHigh << std::endl;

  vlVolIter<DataType> iter(vol());
  vlVolIter<uint8> maskIter(maskVol);

  do {
    DataType voxelValue = iter.get();

    if (threshLow <= voxelValue && voxelValue <= threshHigh) {
      // inside the region
      ++regionVoxelCount;
      maskIter.set(inVoxelMask);
    } else { // outside the region
      maskIter.set(outVoxelMask);
    }
  } while(iter.nextXYZ() && maskIter.nextXYZ());

  resultsRef().add(regionVoxelCount, "voxelCount", "Count of the voxels in the threshold interval", true);

  std::cout << "Total number of voxels in the region : " << regionVoxelCount << std::endl;

  std::cout << "\n---- End of Thresholding ---- " << std::endl;

  return (true);
}

---