VCG Library
selection.h
1 /****************************************************************************
2 * VCGLib o o *
3 * Visual and Computer Graphics Library o o *
4 * _ O _ *
5 * Copyright(C) 2004-2016 \/)\/ *
6 * Visual Computing Lab /\/| *
7 * ISTI - Italian National Research Council | *
8 * \ *
9 * All rights reserved. *
10 * *
11 * This program is free software; you can redistribute it and/or modify *
12 * it under the terms of the GNU General Public License as published by *
13 * the Free Software Foundation; either version 2 of the License, or *
14 * (at your option) any later version. *
15 * *
16 * This program is distributed in the hope that it will be useful, *
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
19 * GNU General Public License (http://www.gnu.org/licenses/gpl.txt) *
20 * for more details. *
21 * *
22 ****************************************************************************/
23 #ifndef __VCG_TRI_UPDATE_SELECTION
24 #define __VCG_TRI_UPDATE_SELECTION
25 
26 namespace vcg {
27 namespace tri {
30 
34 template <class ComputeMeshType>
36 {
37  typedef typename ComputeMeshType::template PerVertexAttributeHandle< bool > vsHandle;
38  typedef typename ComputeMeshType::template PerEdgeAttributeHandle< bool > esHandle;
39  typedef typename ComputeMeshType::template PerFaceAttributeHandle< bool > fsHandle;
40 
41 public:
42  SelectionStack(ComputeMeshType &m)
43  {
44  _m=&m;
45  }
46 
47  bool push()
48  {
49  vsHandle vsH = Allocator<ComputeMeshType>::template AddPerVertexAttribute< bool >(*_m);
50  esHandle esH = Allocator<ComputeMeshType>::template AddPerEdgeAttribute< bool >(*_m);
51  fsHandle fsH = Allocator<ComputeMeshType>::template AddPerFaceAttribute< bool > (*_m);
52  typename ComputeMeshType::VertexIterator vi;
53  for(vi = _m->vert.begin(); vi != _m->vert.end(); ++vi)
54  if( !(*vi).IsD() ) vsH[*vi] = (*vi).IsS() ;
55 
56  typename ComputeMeshType::EdgeIterator ei;
57  for(ei = _m->edge.begin(); ei != _m->edge.end(); ++ei)
58  if( !(*ei).IsD() ) esH[*ei] = (*ei).IsS() ;
59 
60  typename ComputeMeshType::FaceIterator fi;
61  for(fi = _m->face.begin(); fi != _m->face.end(); ++fi)
62  if( !(*fi).IsD() ) fsH[*fi] = (*fi).IsS() ;
63 
64  vsV.push_back(vsH);
65  esV.push_back(esH);
66  fsV.push_back(fsH);
67  return true;
68  }
69 
70  bool popOr()
71  {
72  return pop(true,false);
73  }
74 
75  bool popAnd()
76  {
77  return pop(false,true);
78  }
79 
84  bool pop(bool orFlag=false, bool andFlag=false)
85  {
86  if(vsV.empty()) return false;
87  if(orFlag && andFlag) return false;
88 
89  vsHandle vsH = vsV.back();
90  esHandle esH = esV.back();
91  fsHandle fsH = fsV.back();
92  if(! (Allocator<ComputeMeshType>::template IsValidHandle(*_m, vsH))) return false;
93 
94  for(auto vi = _m->vert.begin(); vi != _m->vert.end(); ++vi)
95  if( !(*vi).IsD() )
96  {
97  if(vsH[*vi]) {
98  if(!andFlag) (*vi).SetS();
99  } else {
100  if(!orFlag) (*vi).ClearS();
101  }
102  }
103 
104  for(auto ei = _m->edge.begin(); ei != _m->edge.end(); ++ei)
105  if( !(*ei).IsD() )
106  {
107  if(esH[*ei]) {
108  if(!andFlag) (*ei).SetS();
109  } else {
110  if(!orFlag) (*ei).ClearS();
111  }
112  }
113 
114 
115  for(auto fi = _m->face.begin(); fi != _m->face.end(); ++fi)
116  if( !(*fi).IsD() )
117  {
118  if(fsH[*fi]) {
119  if(!andFlag) (*fi).SetS();
120  } else {
121  if(!orFlag) (*fi).ClearS();
122  }
123  }
124 
125  Allocator<ComputeMeshType>::template DeletePerVertexAttribute<bool>(*_m,vsH);
126  Allocator<ComputeMeshType>::template DeletePerEdgeAttribute<bool>(*_m,esH);
127  Allocator<ComputeMeshType>::template DeletePerFaceAttribute<bool>(*_m,fsH);
128  vsV.pop_back();
129  esV.pop_back();
130  fsV.pop_back();
131  return true;
132  }
133 
134 private:
135  ComputeMeshType *_m;
136  std::vector<vsHandle> vsV;
137  std::vector<esHandle> esV;
138  std::vector<fsHandle> fsV;
139 };
140 
142 
144 
146 
150 template <class ComputeMeshType>
152 {
153 
154 public:
155 typedef ComputeMeshType MeshType;
156 typedef typename MeshType::ScalarType ScalarType;
157 typedef typename MeshType::VertexType VertexType;
158 typedef typename MeshType::VertexPointer VertexPointer;
159 typedef typename MeshType::VertexIterator VertexIterator;
160 typedef typename MeshType::EdgeIterator EdgeIterator;
161 typedef typename MeshType::FaceType FaceType;
162 typedef typename MeshType::FacePointer FacePointer;
163 typedef typename MeshType::FaceIterator FaceIterator;
164 typedef typename vcg::Box3<ScalarType> Box3Type;
165 
167 static size_t VertexAll(MeshType &m)
168 {
169  for(VertexIterator vi = m.vert.begin(); vi != m.vert.end(); ++vi)
170  if( !(*vi).IsD() ) (*vi).SetS();
171  return m.vn;
172 }
173 
175 static size_t EdgeAll(MeshType &m)
176 {
177  for(EdgeIterator ei = m.edge.begin(); ei != m.edge.end(); ++ei)
178  if( !(*ei).IsD() ) (*ei).SetS();
179  return m.fn;
180 }
182 static size_t FaceAll(MeshType &m)
183 {
184  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
185  if( !(*fi).IsD() ) (*fi).SetS();
186  return m.fn;
187 }
188 
190 static size_t VertexClear(MeshType &m)
191 {
192  for(VertexIterator vi = m.vert.begin(); vi != m.vert.end(); ++vi)
193  if( !(*vi).IsD() ) (*vi).ClearS();
194  return 0;
195 }
196 
198 static size_t EdgeClear(MeshType &m)
199 {
200  for(EdgeIterator ei = m.edge.begin(); ei != m.edge.end(); ++ei)
201  if( !(*ei).IsD() ) (*ei).ClearS();
202  return 0;
203 }
204 
206 static size_t FaceClear(MeshType &m)
207 {
208  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
209  if( !(*fi).IsD() ) (*fi).ClearS();
210  return 0;
211 }
212 
214 static void Clear(MeshType &m)
215 {
216  VertexClear(m);
217  EdgeClear(m);
218  FaceClear(m);
219 }
220 
222 static size_t FaceCount(MeshType &m)
223 {
224  size_t selCnt=0;
225  for(FaceIterator fi=m.face.begin();fi!=m.face.end();++fi)
226  if(!(*fi).IsD() && (*fi).IsS()) ++selCnt;
227  return selCnt;
228 }
229 
231 static size_t EdgeCount(MeshType &m)
232 {
233  size_t selCnt=0;
234  for(EdgeIterator ei=m.edge.begin();ei!=m.edge.end();++ei)
235  if(!(*ei).IsD() && (*ei).IsS()) ++selCnt;
236  return selCnt;
237 }
238 
240 static size_t VertexCount(MeshType &m)
241 {
242  size_t selCnt=0;
243  for(VertexIterator vi=m.vert.begin();vi!=m.vert.end();++vi)
244  if(!(*vi).IsD() && (*vi).IsS()) ++selCnt;
245  return selCnt;
246 }
247 
249 static size_t FaceInvert(MeshType &m)
250 {
251  size_t selCnt=0;
252  for(FaceIterator fi=m.face.begin();fi!=m.face.end();++fi)
253  if(!(*fi).IsD())
254  {
255  if((*fi).IsS()) (*fi).ClearS();
256  else {
257  (*fi).SetS();
258  ++selCnt;
259  }
260  }
261  return selCnt;
262 }
263 
265 static size_t EdgeInvert(MeshType &m)
266 {
267  size_t selCnt=0;
268  for(EdgeIterator ei=m.edge.begin();ei!=m.edge.end();++ei)
269  if(!(*ei).IsD())
270  {
271  if((*ei).IsS()) (*ei).ClearS();
272  else {
273  (*ei).SetS();
274  ++selCnt;
275  }
276  }
277  return selCnt;
278 }
279 
281 static size_t VertexInvert(MeshType &m)
282 {
283  size_t selCnt=0;
284  for(VertexIterator vi=m.vert.begin();vi!=m.vert.end();++vi)
285  if(!(*vi).IsD())
286  {
287  if((*vi).IsS()) (*vi).ClearS();
288  else {
289  (*vi).SetS();
290  ++selCnt;
291  }
292  }
293  return selCnt;
294 }
295 
297 static size_t VertexFromFaceLoose(MeshType &m, bool preserveSelection=false)
298 {
299  size_t selCnt=0;
300 
301  if(!preserveSelection) VertexClear(m);
302  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
303  if( !(*fi).IsD() && (*fi).IsS())
304  for(int i = 0; i < (*fi).VN(); ++i)
305  if( !(*fi).V(i)->IsS()) { (*fi).V(i)->SetS(); ++selCnt; }
306  return selCnt;
307 }
308 
310 static size_t VertexFromEdgeLoose(MeshType &m, bool preserveSelection=false)
311 {
312  size_t selCnt=0;
313 
314  if(!preserveSelection) VertexClear(m);
315  for(EdgeIterator ei = m.edge.begin(); ei != m.edge.end(); ++ei)
316  if( !(*ei).IsD() && (*ei).IsS())
317  {
318  if( !(*ei).V(0)->IsS()) { (*ei).V(0)->SetS(); ++selCnt; }
319  if( !(*ei).V(1)->IsS()) { (*ei).V(1)->SetS(); ++selCnt; }
320  }
321  return selCnt;
322 }
323 
325 
327 static size_t VertexFromFaceStrict(MeshType &m, bool preserveSelection=false)
328 {
330  if(preserveSelection) ss.push();
332  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
333  if( !(*fi).IsD() && !(*fi).IsS())
334  for(int i = 0; i < (*fi).VN(); ++i)
335  (*fi).V(i)->ClearS();
336 
337  if(preserveSelection) ss.popOr();
338  return VertexCount(m);
339 }
340 
342 static size_t FaceFromVertexStrict(MeshType &m, bool preserveSelection=false)
343 {
345  if(preserveSelection) ss.push();
346  size_t selCnt=0;
347  FaceClear(m);
348  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
349  if( !(*fi).IsD())
350  {
351  bool selFlag=true;
352  for(int i = 0; i < (*fi).VN(); ++i)
353  if(!(*fi).V(i)->IsS())
354  selFlag =false;
355  if(selFlag)
356  {
357  (*fi).SetS();
358  ++selCnt;
359  }
360  }
361 
362  if(preserveSelection) ss.popOr();
363  return selCnt;
364 }
365 
367 static size_t FaceFromVertexLoose(MeshType &m, bool preserveSelection=false)
368 {
369  size_t selCnt=0;
370  if(!preserveSelection) FaceClear(m);
371  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
372  if( !(*fi).IsD())
373  {
374  bool selVert=false;
375  for(int i = 0; i < (*fi).VN(); ++i)
376  if((*fi).V(i)->IsS())
377  selVert=true;
378 
379  if(selVert) {
380  (*fi).SetS();
381  ++selCnt;
382  }
383  }
384  return selCnt;
385 }
388 static size_t FaceDilate(MeshType &m)
389 {
392 }
393 
396 static size_t FaceErode(MeshType &m)
397 {
400 }
401 
402 
404 static size_t VertexFromBorderFlag(MeshType &m, bool preserveSelection=false)
405 {
406  size_t selCnt=0;
407  if(!preserveSelection) VertexClear(m);
408  for(VertexIterator vi = m.vert.begin(); vi != m.vert.end(); ++vi)
409  if( !(*vi).IsD() )
410  {
411  if((*vi).IsB() )
412  {
413  (*vi).SetS();
414  ++selCnt;
415  }
416  }
417  return selCnt;
418 }
419 
421 static size_t FaceFromBorderFlag(MeshType &m, bool preserveSelection=false)
422 {
423  tri::RequireTriangularMesh(m);
424  size_t selCnt=0;
425  if(!preserveSelection) FaceClear(m);
426  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
427  if( !(*fi).IsD() )
428  {
429  bool bordFlag=false;
430  for(int i = 0; i < 3; ++i)
431  if((*fi).IsB(i)) bordFlag=true;
432  if(bordFlag)
433  {
434  (*fi).SetS();
435  ++selCnt;
436  }
437  }
438  return selCnt;
439 }
440 
443 static size_t FaceOutOfRangeEdge(MeshType &m, ScalarType MinEdgeThr, ScalarType MaxEdgeThr=(std::numeric_limits<ScalarType>::max)(), bool preserveSelection=false)
444 {
445  if(!preserveSelection) FaceClear(m);
446  size_t selCnt = 0;
447  MinEdgeThr=MinEdgeThr*MinEdgeThr;
448  MaxEdgeThr=MaxEdgeThr*MaxEdgeThr;
449  for(FaceIterator fi=m.face.begin(); fi!=m.face.end();++fi)
450  if(!(*fi).IsD())
451  {
452  for(int i=0;i<(*fi).VN();++i)
453  {
454  const ScalarType squaredEdge=SquaredDistance((*fi).V0(i)->cP(),(*fi).V1(i)->cP());
455  if((squaredEdge<=MinEdgeThr) || (squaredEdge>=MaxEdgeThr) )
456  {
457  selCnt++;
458  (*fi).SetS();
459  break; // skip the rest of the edges of the tri
460  }
461  }
462  }
463  return selCnt;
464 }
465 
467 static size_t FaceConnectedFF(MeshType &m)
468 {
469  // it also assumes that the FF adjacency is well computed.
470  RequireFFAdjacency(m);
472 
473  std::deque<FacePointer> visitStack;
474  size_t selCnt=0;
475  for(FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi)
476  if( !(*fi).IsD() && (*fi).IsS() && !(*fi).IsV() )
477  visitStack.push_back(&*fi);
478 
479  while(!visitStack.empty())
480  {
481  FacePointer fp = visitStack.front();
482  visitStack.pop_front();
483  assert(!fp->IsV());
484  fp->SetV();
485  for(int i=0;i<fp->VN();++i) {
486  FacePointer ff = fp->FFp(i);
487  if(! ff->IsS())
488  {
489  ff->SetS();
490  ++selCnt;
491  visitStack.push_back(ff);
492  assert(!ff->IsV());
493  }
494  }
495  }
496  return selCnt;
497 }
499 static size_t FaceFromQualityRange(MeshType &m,float minq, float maxq, bool preserveSelection=false)
500 {
501  size_t selCnt=0;
502  if(!preserveSelection) FaceClear(m);
503  RequirePerFaceQuality(m);
504  for(FaceIterator fi=m.face.begin();fi!=m.face.end();++fi)
505  if(!(*fi).IsD())
506  {
507  if( (*fi).Q()>=minq && (*fi).Q()<=maxq )
508  {
509  (*fi).SetS();
510  ++selCnt;
511  }
512  }
513  return selCnt;
514 }
515 
517 static size_t VertexFromQualityRange(MeshType &m,float minq, float maxq, bool preserveSelection=false)
518 {
519  size_t selCnt=0;
520  if(!preserveSelection) VertexClear(m);
521  RequirePerVertexQuality(m);
522  for(VertexIterator vi=m.vert.begin();vi!=m.vert.end();++vi)
523  if(!(*vi).IsD())
524  {
525  if( (*vi).Q()>=minq && (*vi).Q()<=maxq )
526  {
527  (*vi).SetS();
528  ++selCnt;
529  }
530  }
531  return selCnt;
532 }
533 
535 static size_t VertexInBox( MeshType & m, const Box3Type &bb, bool preserveSelection=false)
536 {
537  if(!preserveSelection) VertexClear(m);
538  int selCnt=0;
539  for (VertexIterator vi = m.vert.begin(); vi != m.vert.end(); ++vi) if(!(*vi).IsD())
540  {
541  if(bb.IsIn((*vi).cP()) ) {
542  (*vi).SetS();
543  ++selCnt;
544  }
545  }
546  return selCnt;
547 }
548 
552 static size_t VertexCornerBorder(MeshType &m, ScalarType angleRad, bool preserveSelection=false)
553 {
554  if(!preserveSelection) VertexClear(m);
555  SimpleTempData<typename MeshType::VertContainer, ScalarType > angleSumH(m.vert,0);
556  int selCnt=0;
557  for(auto vi=m.vert.begin();vi!=m.vert.end();++vi) if(!(*vi).IsD())
558  angleSumH[vi]=0;
559 
560  for(auto fi=m.face.begin();fi!=m.face.end();++fi) if(!(*fi).IsD())
561  {
562  for(int i=0;i<(*fi).VN();++i)
563  angleSumH[fi->V(i)] += face::WedgeAngleRad(*fi,i);
564  }
565 
566  for(auto vi=m.vert.begin();vi!=m.vert.end();++vi) if(!(*vi).IsD())
567  {
568  if(angleSumH[vi]<angleRad && vi->IsB())
569  {
570  (*vi).SetS();
571  ++selCnt;
572  }
573  }
574  return selCnt;
575 }
576 
577 
578 void VertexNonManifoldEdges(MeshType &m, bool preserveSelection=false)
579 {
580  tri::RequireFFAdjacency(m);
581 
582  if(!preserveSelection) VertexClear(m);
583  for (FaceIterator fi = m.face.begin(); fi != m.face.end(); ++fi) if (!fi->IsD())
584  {
585  for(int i=0;i<fi->VN();++i)
586  if(!IsManifold(*fi,i)){
587  (*fi).V0(i)->SetS();
588  (*fi).V1(i)->SetS();
589  }
590  }
591 }
592 
593 }; // end class
594 
595 } // End namespace
596 } // End namespace
597 
598 
599 #endif