vtkDataArrayDispatcher.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkDataArrayDispatcher.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
 
#ifndef vtkDataArrayDispatcher_h
#define vtkDataArrayDispatcher_h
 
#include "vtkType.h" //Required for vtkIdType
#include "vtkDataArray.h" //required for constructor of the vtkDataArrayFunctor
#include <map> //Required for the storage of template params to runtime params
 
// Object that is passed to all functor that are used with this class
// This allows the user the ability to find info about the size
template<typename T>
struct vtkDataArrayDispatcherPointer
{
  typedef T ValueType;
 
  vtkIdType NumberOfTuples;
  vtkIdType NumberOfComponents;
  ValueType* RawPointer;
 
  explicit vtkDataArrayDispatcherPointer(vtkDataArray* array):
    NumberOfTuples(array->GetNumberOfTuples()),
    NumberOfComponents(array->GetNumberOfComponents()),
    RawPointer(static_cast<ValueType*>(array->GetVoidPointer(0)))
    {}
};
 
// class template FunctorDispatcher
template
  <
    class DefaultFunctorType,
    typename ReturnType = void
  >
class vtkDataArrayDispatcher
{
public:
 
  vtkDataArrayDispatcher(DefaultFunctorType& f);
 
  vtkDataArrayDispatcher();
 
  virtual ~vtkDataArrayDispatcher();
 
  ReturnType Go(vtkDataArray* lhs);
 
protected:
  DefaultFunctorType* DefaultFunctor;
  bool OwnsFunctor;
};
 
//We are making all these method non-inline to reduce compile time overhead
 
//----------------------------------------------------------------------------
template<class DefaultFunctorType,typename ReturnType>
vtkDataArrayDispatcher<DefaultFunctorType,ReturnType>::vtkDataArrayDispatcher(DefaultFunctorType& fun):
  DefaultFunctor(&fun),
  OwnsFunctor(false)
  {
  }
 
//----------------------------------------------------------------------------
template<class DefaultFunctorType,typename ReturnType>
vtkDataArrayDispatcher<DefaultFunctorType,ReturnType>::vtkDataArrayDispatcher():
  DefaultFunctor(new DefaultFunctorType()),
  OwnsFunctor(true)
  {
  }
 
//----------------------------------------------------------------------------
template<class DefaultFunctorType,typename ReturnType>
vtkDataArrayDispatcher<DefaultFunctorType,ReturnType>::~vtkDataArrayDispatcher()
  {
  if(OwnsFunctor)
    {
    delete this->DefaultFunctor;
    }
  }
 
//----------------------------------------------------------------------------
template <class DefaultFunctorType,typename ReturnType>
ReturnType vtkDataArrayDispatcher<DefaultFunctorType,ReturnType>
::Go(vtkDataArray* lhs)
  {
  switch(lhs->GetDataType())
      {
      vtkTemplateMacro(return (*this->DefaultFunctor) (
                      vtkDataArrayDispatcherPointer<VTK_TT>(lhs) ));
      }
  return ReturnType();
  }
 
#endif // vtkDataArrayDispatcher_h
// VTK-HeaderTest-Exclude: vtkDataArrayDispatcher.h