sobel.cpp

Go to the documentation of this file.

// $Id: sobel.cpp,v 1.1 2003/08/12 04:27:35 sarang Exp $
//
// OpenVL - A Library for working with volumetric datasets.
//          http://openvl.sf.net
//
// Copyright (C) 2000-2003  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 <queue>

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

#include "sobel.h"

typedef vlGenericFactory<vlVolProcessorSobel> vlVolProcessorSobelFactory;
VL_EXPORT_COMPONENT_FACTORY( sobelvol, vlVolProcessorSobelFactory );

vlVolProcessorSobel::vlVolProcessorSobel()
{
  infoRef().setName("Sobel");
  infoRef().setVersion(0,0,1);
  infoRef().setService("Sobel", "Computes a sobel gradient volume.");
}

vlVolProcessorSobel::~vlVolProcessorSobel()
{
}


void vlVolProcessorSobel::init()
{
  configRef().clear();
}


bool vlVolProcessorSobel::run()
{
  // Check all error conditions...

  if(!vol())
    return (false);

  m_cdVolume = new vlVolume(vol()->dim(), Float);

  if(!m_cdVolume->isValid()) {
    std::cout << "Unable to create float volume." << std::endl;
    return (false);
  }

  bool retValue(false);

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

  if(retValue) {
    resultsRef().add(m_cdVolume, "gradientVolume", "A volume of gradients (float)", true);
  }
  
  return (retValue);
}


template <class DataType>
bool vlVolProcessorSobel::runT(DataType & dummy)
{
  vlVolIter<DataType> iter(vol());

  if(!iter.setVoxelOperation(vlVoxelOp::Sobel)) {
    std::cout << "Cannot set voxel operation to Sobel." << std::endl;
    return (false);
  }

  vlVoxelOpValue value;
  vlVolIter<float> fiter(m_cdVolume);
  
  while(!iter.end()) {
    iter.getOp(value);
    fiter.set((float)value.dl());
    iter.nextXYZ();
    fiter.nextXYZ();
  }
  
  return (true);
}

---