ProteoWizard
Bin.hpp
Go to the documentation of this file.
1 //
2 // $Id: Bin.hpp 2051 2010-06-15 18:39:13Z chambm $
3 //
4 //
5 // Original author: Kate Hoff <katherine.hoff@proteowizard.org>
6 //
7 // Copyright 2009 Center for Applied Molecular Medicine
8 // University of Southern California, Los Angeles, CA
9 //
10 // Licensed under the Apache License, Version 2.0 (the "License");
11 // you may not use this file except in compliance with the License.
12 // You may obtain a copy of the License at
13 //
14 // http://www.apache.org/licenses/LICENSE-2.0
15 //
16 // Unless required by applicable law or agreed to in writing, software
17 // distributed under the License is distributed on an "AS IS" BASIS,
18 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 // See the License for the specific language governing permissions and
20 // limitations under the License.
21 //
22 
23 #ifndef _BIN_HPP_
24 #define _BIN_HPP_
25 
26 #include "boost/shared_ptr.hpp"
27 #include "pwiz/utility/misc/Std.hpp"
28 #include <cmath>
29 
30 namespace pwiz{
31 
32 template <typename T> // T: objects in bins
33 class Bin
34 {
35 
36 public:
37 
38  Bin(){}
39  Bin(const vector<pair<pair<double,double>, T> >& objects, double binSizeX, double binSizeY);
40  Bin(const vector<pair<pair<double,double>, boost::shared_ptr<T> > >& objects, double binSizeX, double binSizeY);
41  Bin(const Bin<T>& b) : _allContents(b.getAllContents()), _objects(b.getObjects()), _data(b.getData()), _binSizeX(b.getBinSizes().first), _binSizeY(b.getBinSizes().second) {}
42 
43  double getBinSizeX()
44  {
45  return _binSizeX;
46  }
47 
48  double getBinSizeY()
49  {
50  return _binSizeY;
51  }
52 
53  void update(const T& t, pair<double,double> coordinates);
54  void erase(const T& t, pair<double,double> coordinates);
55  void rebin(const double& binSizeX, const double& binSizeY);
56  size_t size() const { return _data.size();}
57  size_t count(pair<int, int> coordinates) const
58  {
59  return _data.count(coordinates);
60  }
61 
62  void getBinContents(const pair<int, int>& coordinates, vector<T>& result) const;
63  void getBinContents(const pair<double,double>& coordinates, vector<boost::shared_ptr<T> >& result) const;
64  void getAdjacentBinContents(pair<double,double> coordinates, vector<boost::shared_ptr<T> >& result) ; // gets bin and all adjacent bin coordinates
65 
66  // accessors
67  const vector<boost::shared_ptr<T> >& getAllContents() const { return _allContents;}
68  const vector<pair<pair<double,double>, boost::shared_ptr<T> > >& getObjects() const { return _objects;}
69  const multimap<const pair<int, int>,boost::shared_ptr<T> >& getData() const { return _data;}
70  pair<double,double> getBinSizes() const { return make_pair(_binSizeX, _binSizeY);}
71 
72  bool operator==(const Bin& that);
73  bool operator!=(const Bin& that);
74 
75 private:
76 
77  vector<boost::shared_ptr<T> > _allContents;
78  vector<pair<pair<double,double>, boost::shared_ptr<T> > > _objects;
79 
80  multimap<const pair<int,int>, boost::shared_ptr<T> > _data;
81  double _binSizeX;
82  double _binSizeY;
83 };
84 
85 ///
86 /// inline implementation
87 ///
88 
89 template <typename T>
90 Bin<T>::Bin(const vector<pair<pair<double,double>, T> >& objects, double binSizeX, double binSizeY)
91  : _binSizeX(binSizeX), _binSizeY(binSizeY)
92 {
93  typename vector<pair<pair<double,double>, T> >::const_iterator it = objects.begin();
94  for(; it!= objects.end(); ++it)
95  {
96  int binXCoord = int(floor(it->first.first/_binSizeX));
97  int binYCoord = int(floor(it->first.second/_binSizeY));
98 
99  boost::shared_ptr<T> t(new T(it->second));
100  boost::shared_ptr<T> sp_t(new T(it->second));
101 
102  _data.insert(pair<pair<int,int>, boost::shared_ptr<T> >(pair<int,int>(binXCoord,binYCoord), t));
103  _allContents.push_back(sp_t);
104  _objects.push_back(make_pair(it->first, t));
105 
106  }
107 
108 }
109 
110 template <typename T>
111 Bin<T>::Bin(const vector<pair<pair<double,double>, boost::shared_ptr<T> > >& objects, double binSizeX, double binSizeY) : _binSizeX(binSizeX), _binSizeY(binSizeY)
112 {
113  typename vector<pair<pair<double,double>, boost::shared_ptr<T> > >::const_iterator it = objects.begin();
114  for(; it!= objects.end(); ++it)
115  {
116  int binXCoord = int(floor(it->first.first/_binSizeX));
117  int binYCoord = int(floor(it->first.second/_binSizeY));
118 
119  _data.insert(pair<pair<int,int>, boost::shared_ptr<T> >(pair<int,int>(binXCoord,binYCoord), it->second));
120  _allContents.push_back(it->second);
121  _objects.push_back(*it);
122 
123  }
124 
125 }
126 
127 template <typename T>
128 void Bin<T>::update(const T& t, pair<double,double> coordinates)
129 {
130  int binXCoord = int(floor(coordinates.first/_binSizeX));
131  int binYCoord = int(floor(coordinates.second/_binSizeY));
132 
133  boost::shared_ptr<T> tp_sp(new T(t));
134  const pair<const pair<int,int>, boost::shared_ptr<T> >& entry = make_pair(make_pair(binXCoord,binYCoord), tp_sp);
135  _data.insert(entry);
136  _objects.push_back(entry);
137  _allContents.push_back(tp_sp);
138 
139 }
140 
141 template <typename T>
142 struct SecondIs
143 {
144  T _t;
145  SecondIs(const T& t) : _t(t) {}
146  bool operator()(pair<const pair<double,double>,boost::shared_ptr<T> > entry) { return (*(entry.second) == _t); }
147 
148 };
149 
150 template <typename T>
151 struct IsObject
152 {
153  T _t;
154  IsObject(const T& t) : _t(t) {}
155  bool operator()(boost::shared_ptr<T> entry) { return (*entry == _t); }
156 
157 };
158 
159 template <typename T>
160 void Bin<T>::erase(const T& t, pair<double,double> coordinates)
161 {
162  int binXCoord = int(floor(coordinates.first/_binSizeX));
163  int binYCoord = int(floor(coordinates.second/_binSizeY));
164 
165  pair<int,int> intCoordinates = make_pair(binXCoord, binYCoord);
166  pair<typename multimap<const pair<int,int>, boost::shared_ptr<T> >::iterator, typename multimap<const pair<int,int>,boost::shared_ptr<T> >::iterator> its = _data.equal_range(intCoordinates);
167 
168  typename multimap<const pair<int,int>,boost::shared_ptr<T> >::iterator search_it = find_if(its.first, its.second, SecondIs<T>(t));
169  if (search_it != its.second)
170  {
171  _data.erase(search_it);
172 
173  ////////////////////////
174  typename vector<pair<pair<double,double>, boost::shared_ptr<T> > >::iterator objects_eraser = find_if(_objects.begin(), _objects.end(), SecondIs<T>(t));
175  _objects.erase(objects_eraser);
176 
177  ////////////////////////
178  typename vector<boost::shared_ptr<T> >::iterator allContents_erase = find_if(_allContents.begin(), _allContents.end(), IsObject<T>(t));
179  _allContents.erase(allContents_erase);
180 
181  }
182 
183  else cerr << "[Bin<T>::erase] Object to erase was not found." << endl;
184 
185 }
186 
187 template <typename T>
188 void Bin<T>::rebin(const double& binSizeX, const double& binSizeY)
189 {
190  Bin<T> rebinned(_objects, binSizeX, binSizeY);
191  _data.clear();
192  _data = rebinned.getData();
193 
194  _binSizeX = binSizeX;
195  _binSizeY = binSizeY;
196 
197  return;
198 
199 }
200 
201 template <typename T>
202 void Bin<T>::getBinContents(const pair<int, int>& coordinates,
203  vector<T>& result) const
204 {
205  pair<typename multimap<const pair<int,int>,boost::shared_ptr<T> >::const_iterator, typename multimap<const pair<int,int>,boost::shared_ptr<T> >::const_iterator> its = _data.equal_range(coordinates);
206 
207  typename multimap<const pair<int, int>, boost::shared_ptr<T> >::const_iterator it = its.first;
208  for(; it != its.second; ++it)
209  {
210  result.push_back(*(it->second));
211 
212  }
213 
214  return;
215 
216 }
217 
218 template <typename T>
219 void Bin<T>::getBinContents(const pair<double,double>& coordinates,
220  vector<boost::shared_ptr<T> >& result) const
221 {
222  int binXCoord = int(floor(coordinates.first/_binSizeX));
223  int binYCoord = int(floor(coordinates.second/_binSizeY));
224  pair<int,int> intCoordinates = make_pair(binXCoord, binYCoord);
225 
226  pair<typename multimap<const pair<int,int>,boost::shared_ptr<T> >::const_iterator, typename multimap<const pair<int,int>,boost::shared_ptr<T> >::const_iterator> its = _data.equal_range(intCoordinates);
227 
228  typename multimap<const pair<int, int>, boost::shared_ptr<T> >::const_iterator it = its.first;
229  for(; it != its.second; ++it)
230  {
231  result.push_back((it->second));
232 
233  }
234 
235  return;
236 
237 }
238 
239 template <typename T>
240 void Bin<T>::getAdjacentBinContents(pair<double,double> coordinates, vector<boost::shared_ptr<T> >& result)
241 {
242 
243  // 8 adjacent bins:
244  // dasher | dancer | prancer
245  // blitzen | coordinates | vixen
246  // donner | cupid | comet
247 
248  pair<double,double> dasher = make_pair(coordinates.first - _binSizeX, coordinates.second + _binSizeY);
249  pair<double,double> dancer = make_pair(coordinates.first + _binSizeX, coordinates.second);
250  pair<double,double> prancer = make_pair(coordinates.first + _binSizeX, coordinates.second + _binSizeY);
251  pair<double,double> vixen = make_pair(coordinates.first, coordinates.second + _binSizeY);
252  pair<double,double> comet = make_pair(coordinates.first + _binSizeX, coordinates.second - _binSizeY);
253  pair<double,double> cupid = make_pair(coordinates.first, coordinates.second - _binSizeY);
254  pair<double,double> donner = make_pair(coordinates.first - _binSizeX, coordinates.second - _binSizeY);
255  pair<double,double> blitzen = make_pair(coordinates.first - _binSizeX, coordinates.second);
256 
257  vector<pair<double,double> > sleigh;
258  sleigh.push_back(dasher);
259  sleigh.push_back(dancer);
260  sleigh.push_back(prancer);
261  sleigh.push_back(vixen);
262  sleigh.push_back(comet);
263  sleigh.push_back(cupid);
264  sleigh.push_back(donner);
265  sleigh.push_back(blitzen);
266 
267  vector<boost::shared_ptr<T> > santa;
268  getBinContents(coordinates, santa);
269  copy(santa.begin(), santa.end(), back_inserter(result));
270 
271  vector<pair<double,double> >::iterator it = sleigh.begin();
272  for(; it!= sleigh.end(); ++it)
273  {
274  vector<boost::shared_ptr<T> > rudolph;
275  getBinContents(*it, rudolph);
276  copy(rudolph.begin(), rudolph.end(), back_inserter(result));
277 
278  }
279 
280  return;
281 
282 }
283 template <typename T>
284 bool Bin<T>::operator==(const Bin& that)
285 {
286  return _allContents == that._allContents &&
287  _objects == that._objects &&
288  _data == that._data &&
289  _binSizeX == that._binSizeX &&
290  _binSizeY == that._binSizeY;
291 
292 }
293 
294 template <typename T>
295 bool Bin<T>::operator!=(const Bin& that)
296 {
297  return !(*this == that);
298 
299 }
300 
301 } // namespace pwiz
302 
303 #endif // _BIN_HPP_
304 
Generated on Mon Nov 26 2012 18:05:46 for ProteoWizard by   doxygen 1.8.1.1