Initial revision

include/ExtractG.hh

+/*=========================================================================
+
+  Program:   Visualization Library
+  Module:    ExtractG.hh
+  Language:  C++
+  Date:      $Date$
+  Version:   $Revision$
+
+Description:
+---------------------------------------------------------------------------
+This file is part of the Visualization Library. No part of this file
+or its contents may be copied, reproduced or altered in any way
+without the express written consent of the authors.
+
+Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen 1993, 1994 
+
+=========================================================================*/
+// .NAME vlExtractGeomtry - extract cells entirely within a apcecified sphere
+// .SECTION Description
+// vlExtractGeometry extracts from its input dataset all cells that are
+// entirely within a specified sphere.
+
+#ifndef __vlExtractGeometry_h
+#define __vlExtractGeometry_h
+
+#include "DS2UGrid.hh"
+
+class vlExtractGeometry : public vlDataSetToUnstructuredGridFilter
+{
+public:
+  vlExtractGeometry();
+  ~vlExtractGeometry() {};
+  char *GetClassName() {return "vlExtractGeometry";};
+  void PrintSelf(ostream& os, vlIndent indent);
+
+  vlSetClampMacro(Radius,float, 0.0, LARGE_FLOAT);
+  vlGetMacro(Radius,float);
+
+  vlSetVector3Macro(Center,float);
+  vlGetVectorMacro(Center,float);
+
+protected:
+  // Usual data generation method
+  void Execute();
+
+  float Radius;
+  float Center[3];
+};
+
+#endif
+
+

src/ExtractG.cc

+/*=========================================================================
+
+  Program:   Visualization Library
+  Module:    ExtractG.cc
+  Language:  C++
+  Date:      $Date$
+  Version:   $Revision$
+
+Description:
+---------------------------------------------------------------------------
+This file is part of the Visualization Library. No part of this file
+or its contents may be copied, reproduced or altered in any way
+without the express written consent of the authors.
+
+Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen 1993, 1994 
+
+=========================================================================*/
+#include "ExtractG.hh"
+#include "vlMath.hh"
+
+void vlExtractGeometry::Execute()
+{
+  int ptId, numPts, i, cellId;
+  vlIdList *cellPts, *pointMap;
+  vlCell *cell;
+  vlMath math;
+  int numCellPts, newId;
+  vlPointData *pd;
+  float *x;
+  float r2=this->Radius*this->Radius;
+  vlFloatPoints *newPoints;
+  vlIdList newCellPts(MAX_CELL_SIZE);
+
+  vlDebugMacro(<< "Extracting geometry in sphere");
+
+  this->Initialize();
+//
+// Loop over all points determining whether they are inside sphere. Copy if
+// they are.
+//
+  numPts = this->Input->GetNumberOfPoints();
+  pointMap = new vlIdList(numPts); // maps old point ids into new
+  for (i=0; i < numPts; i++) pointMap->SetId(i,-1);
+
+  pd = this->Input->GetPointData();
+  this->PointData.CopyAllocate(pd);
+  
+  for ( ptId=0; ptId < numPts; ptId++ )
+    {
+    x = this->Input->GetPoint(ptId);
+    if ( math.Distance2BetweenPoints(x,this->Center) <= r2 )
+      {
+      newId = newPoints->InsertNextPoint(x);
+      pointMap->SetId(ptId,newId);
+      this->PointData.CopyData(pd,ptId,newId);
+      }
+    }
+//
+// Now loop over all cells to see whether they are inside sphere. Copy if
+// they are.
+//
+  for (cellId=0; cellId < this->Input->GetNumberOfCells(); cellId++)
+    {
+    cell = this->Input->GetCell(cellId);
+    cellPts = cell->GetPointIds();
+    numCellPts = cell->GetNumberOfPoints();
+
+    for ( i=0; i < numCellPts; i++)
+      {
+      ptId = cellPts->GetId(i);
+      if ( (newId=pointMap->GetId(ptId)) < 0 ) break;
+      newCellPts.SetId(i,newId);
+      }
+
+    if ( i >= numCellPts )
+      {
+      this->InsertNextCell(cell->GetCellType(),newCellPts);
+      }
+    }
+//
+// Update ourselves
+//
+  delete pointMap;
+
+  this->SetPoints(newPoints);
+  this->Squeeze();
+}
+
+void vlExtractGeometry::PrintSelf(ostream& os, vlIndent indent)
+{
+  if (this->ShouldIPrint(vlExtractGeometry::GetClassName()))
+    {
+    vlDataSetToUnstructuredGridFilter::PrintSelf(os,indent);
+
+    os << indent << "Radius: " << this->Radius << "\n";
+    os << indent << "Center: (" << this->Center[0] << ", "
+       << this->Center[1] << ", " << this->Center[2] << ")\n";
+    }
+}