1 | /*********************************************************************** |
---|
2 | * Software License Agreement (BSD License) |
---|
3 | * |
---|
4 | * Copyright 2008-2011 Marius Muja (mariusm@cs.ubc.ca). All rights reserved. |
---|
5 | * Copyright 2008-2011 David G. Lowe (lowe@cs.ubc.ca). 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 | * 1. Redistributions of source code must retain the above copyright |
---|
12 | * notice, this list of conditions and the following disclaimer. |
---|
13 | * 2. Redistributions in binary form must reproduce the above copyright |
---|
14 | * notice, this list of conditions and the following disclaimer in the |
---|
15 | * documentation and/or other materials provided with the distribution. |
---|
16 | * |
---|
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
---|
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
---|
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
---|
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
---|
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
---|
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
---|
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
---|
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
---|
25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
---|
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
---|
27 | *************************************************************************/ |
---|
28 | |
---|
29 | |
---|
30 | #ifndef MPI_QUERIES_H_ |
---|
31 | #define MPI_QUERIES_H_ |
---|
32 | |
---|
33 | #include <flann/mpi/matrix.h> |
---|
34 | #include <boost/archive/binary_iarchive.hpp> |
---|
35 | #include <boost/archive/binary_oarchive.hpp> |
---|
36 | #include <boost/asio.hpp> |
---|
37 | |
---|
38 | namespace flann |
---|
39 | { |
---|
40 | |
---|
41 | template<typename T> |
---|
42 | struct Request |
---|
43 | { |
---|
44 | flann::Matrix<T> queries; |
---|
45 | int nn; |
---|
46 | int checks; |
---|
47 | |
---|
48 | template<typename Archive> |
---|
49 | void serialize(Archive& ar, const unsigned int version) |
---|
50 | { |
---|
51 | ar & queries & nn & checks; |
---|
52 | } |
---|
53 | }; |
---|
54 | |
---|
55 | template<typename T> |
---|
56 | struct Response |
---|
57 | { |
---|
58 | flann::Matrix<int> indices; |
---|
59 | flann::Matrix<T> dists; |
---|
60 | |
---|
61 | template<typename Archive> |
---|
62 | void serialize(Archive& ar, const unsigned int version) |
---|
63 | { |
---|
64 | ar & indices & dists; |
---|
65 | } |
---|
66 | }; |
---|
67 | |
---|
68 | |
---|
69 | using boost::asio::ip::tcp; |
---|
70 | |
---|
71 | template <typename T> |
---|
72 | void read_object(tcp::socket& sock, T& val) |
---|
73 | { |
---|
74 | uint32_t size; |
---|
75 | boost::asio::read(sock, boost::asio::buffer(&size, sizeof(size))); |
---|
76 | size = ntohl(size); |
---|
77 | |
---|
78 | boost::asio::streambuf archive_stream; |
---|
79 | boost::asio::read(sock, archive_stream, boost::asio::transfer_at_least(size)); |
---|
80 | |
---|
81 | boost::archive::binary_iarchive archive(archive_stream); |
---|
82 | archive >> val; |
---|
83 | } |
---|
84 | |
---|
85 | template <typename T> |
---|
86 | void write_object(tcp::socket& sock, const T& val) |
---|
87 | { |
---|
88 | boost::asio::streambuf archive_stream; |
---|
89 | boost::archive::binary_oarchive archive(archive_stream); |
---|
90 | archive << val; |
---|
91 | |
---|
92 | uint32_t size = archive_stream.size(); |
---|
93 | size = htonl(size); |
---|
94 | boost::asio::write(sock, boost::asio::buffer(&size, sizeof(size))); |
---|
95 | boost::asio::write(sock, archive_stream); |
---|
96 | |
---|
97 | } |
---|
98 | |
---|
99 | } |
---|
100 | |
---|
101 | |
---|
102 | |
---|
103 | #endif /* MPI_QUERIES_H_ */ |
---|