CUDAWatershed.h Source File

00001 /***************************************************************************
00002  *cr
00003  *cr            (C) Copyright 1995-2019 The Board of Trustees of the
00004  *cr                        University of Illinois
00005  *cr                         All Rights Reserved
00006  *cr
00007  ***************************************************************************/
00008 /***************************************************************************
00009  * RCS INFORMATION:
00010  *
00011  *      $RCSfile: CUDAWatershed.h,v $
00012  *      $Author: johns $        $Locker:  $             $State: Exp $
00013  *      $Revision: 1.10 $        $Date: 2020/02/26 19:26:47 $
00014  *
00015  ***************************************************************************/
00021 #ifndef CUDA_WATERSHED_H
00022 #define CUDA_WATERSHED_H
00023 
00024 #include "Watershed.h"
00025 
00026 template <typename GROUP_T, typename IMAGE_T>
00027 bool init_gpu(state_t<GROUP_T, IMAGE_T>& state, int* eq_and_lower, watershed_gpu_state_t<GROUP_T, IMAGE_T>& gpu_state,
00028               unsigned int w, unsigned int h, unsigned int d);
00029 
00030 template <class GROUP_T, typename IMAGE_T>
00031 bool init_gpu_on_device(watershed_gpu_state_t<GROUP_T, IMAGE_T> &gpu_state, 
00032                         IMAGE_T* image, int imageongpu, 
00033                         unsigned int w, unsigned int h, unsigned int d);
00034  
00035 template <typename GROUP_T, typename IMAGE_T>
00036 void destroy_gpu(watershed_gpu_state_t<GROUP_T, IMAGE_T>& gpu_state);
00037 
00038 template <typename GROUP_T, typename IMAGE_T>
00039 void update_cuda(watershed_gpu_state_t<GROUP_T, IMAGE_T>& gpu_state, GROUP_T* segments_d);
00040 
00041 void guassian3D_gpu();
00042 
00043 #endif