Point Cloud Library (PCL)  1.14.1-dev
device_format.hpp
1 /*
2  * Software License Agreement (BSD License)
3  *
4  * Copyright (c) 2011, Willow Garage, Inc.
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  *
11  * * Redistributions of source code must retain the above copyright
12  * notice, this list of conditions and the following disclaimer.
13  * * Redistributions in binary form must reproduce the above
14  * copyright notice, this list of conditions and the following
15  * disclaimer in the documentation and/or other materials provided
16  * with the distribution.
17  * * Neither the name of Willow Garage, Inc. nor the names of its
18  * contributors may be used to endorse or promote products derived
19  * from this software without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
29  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32  * POSSIBILITY OF SUCH DAMAGE.
33  *
34  * Author: Anatoly Baskeheev, Itseez Ltd, (myname.mysurname@mycompany.com)
35  */
36 
37 #ifndef _PCL_GPU_OCTREE_DEVICE_FORMAT_HPP_
38 #define _PCL_GPU_OCTREE_DEVICE_FORMAT_HPP_
39 
40 #include <pcl/gpu/containers/device_array.h>
41 
42 namespace pcl
43 {
44  namespace gpu
45  {
47  {
50  int max_elems;
51 
53  NeighborIndices(int query_number, int max_elements) : max_elems(0)
54  {
55  create(query_number, max_elements);
56  }
57 
58  void create(int query_number, int max_elements)
59  {
60  max_elems = max_elements;
61  data.create (static_cast<std::size_t>(query_number) * static_cast<std::size_t>(max_elems));
62 
63  if (max_elems != 1)
64  sizes.create(query_number);
65  }
66 
67  void upload(const std::vector<int>& data, const std::vector<int>& sizes, int max_elements)
68  {
69  this->data.upload(data);
70  this->sizes.upload(sizes);
71  max_elems = max_elements;
72  }
73 
74  bool validate(std::size_t cloud_size) const
75  {
76  return (sizes.size() == cloud_size) && (cloud_size * max_elems == data.size());
77  }
78 
79  operator PtrStep<int>() const
80  {
81  return {(int*)data.ptr(), max_elems * sizeof(int)};
82  }
83 
84  std::size_t neighboors_size() const { return data.size()/max_elems; }
85  };
86  }
87 }
88 
89 #endif /* _PCL_GPU_OCTREE_DEVICE_FORMAT_HPP_ */
std::size_t size() const
Returns size in elements.
void create(std::size_t size)
Allocates internal buffer in GPU memory.
T * ptr()
Returns pointer for internal buffer in GPU memory.
NeighborIndices(int query_number, int max_elements)
void upload(const std::vector< int > &data, const std::vector< int > &sizes, int max_elements)
bool validate(std::size_t cloud_size) const
void create(int query_number, int max_elements)
std::size_t neighboors_size() const
DeviceArray< int > sizes
DeviceArray< int > data