cudabelieft.h
Go to the documentation of this file.
00001 /*
00002  * Copyright (c) 2011, Mårten Björkman (celle@csc.kth.se) 
00003  * All rights reserved.
00004  *
00005  * Redistribution and use in source and binary forms, with or without
00006  * modification, are permitted provided that the following conditions are
00007  * met:
00008  *
00009  *  1.Redistributions of source code must retain the above copyright
00010  *    notice, this list of conditions and the following disclaimer.
00011  *  2.Redistributions in binary form must reproduce the above
00012  *    copyright notice, this list of conditions and the following
00013  *    disclaimer in the documentation and/or other materials provided
00014  *    with the distribution.  
00015  *  3.The name of Mårten Björkman may not be used to endorse or
00016  *    promote products derived from this software without specific
00017  *    prior written permission.
00018  *
00019  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00020  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00021  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
00022  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
00023  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
00024  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
00025  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
00026  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
00027  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
00028  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
00029  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00030  */
00031 
00032 #ifndef CUDABELIEF_H
00033 #define CUDABELIEF_H
00034 
00037 
00038 
00041 template<int dim>
00042 class CudaBelief {
00043 private:
00045   bool copyFromHost;
00047   unsigned char *allocated_d;
00049   float *msgl_d[2*dim], *msgr_d[2*dim];
00051   float *msgu_d[2*dim], *msgd_d[2*dim];
00053   float *prior_d[dim], *belief_d[dim];
00055   int width, height;
00056 private:
00058 
00061   void ComputeBeliefs(int loop);
00063 
00068   void CollectMessages(float *costh_d, float *costv_d, int loop);
00069 public:
00071 
00076   CudaBelief(int w, int h, bool copyFromHost = true);
00077   ~CudaBelief();
00079 
00085   void Execute(float **priors, float **beliefs, float *costh_d, float *costv_d);
00087 
00095   void Execute(float **priors, float **beliefs, float *costh_d, float *costv_d, int loops, int depth);
00096 };
00097 
00098 #include "cudabelieft.cu"
00099 
00100 #endif // CUDABELIEF_H
00101  
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Properties Friends Defines


active_realtime_segmentation
Author(s): Mårten Björkman. Maintained by Jeannette Bohg
autogenerated on Wed Jan 23 2013 15:46:43