Add calculate node normals

Author: nids2001

src/Modules/Legacy/Fields/CalculateNormals.cc

@@ -35,18 +35,119 @@ using namespace SCIRun::Core::Datatypes;
 using namespace SCIRun::Dataflow::Networks;
 
 /// @class CalculateNormals
-/// @brief This module splits out a string.
+/// @brief Make a new vector field that points to the input point.
 
 MODULE_INFO_DEF(CalculateNormals, ChangeFieldData, SCIRun) ;
 
-CalculateNormals::CalculateNormals() : Module(staticInfo_)
+CalculateNormals::CalculateNormals() : Module(staticInfo_, false)
 {
   INITIALIZE_PORT(InputField);
-  INITIALIZE_PORT(InputField2);
+  INITIALIZE_PORT(InputPoint);
   INITIALIZE_PORT(OutputField);
 }
 
 void
 CalculateNormals::execute()
 {
+  auto ifieldhandle = getRequiredInput(InputField);
+  auto ipointhandle = getRequiredInput(InputPoint);
+
+  if ( inputs_changed_ || !oport_cached("Output Field"))
+  {
+    update_state(Executing);
+
+    VField* point_field = ipointhandle->vfield();
+    VMesh*  point_mesh  = ipointhandle->vmesh();
+
+    if (point_mesh->num_nodes() != 1)
+    {
+      error("Input Point needs to have a single node only");
+      return;
+    }
+
+    Point attract_point;
+    Vector dir;
+    bool has_vector = false;
+    point_mesh->get_center(attract_point,VMesh::Node::index_type(0));
+
+    if (!(point_field->is_pointcloudmesh()))
+    {
+      error("Input field was not a valid point cloud.");
+      return;
+    }
+
+    if (point_field->is_vector())
+    {
+      has_vector = true;
+      point_field->get_value(dir,0);
+    }
+
+    VField* field = ifieldhandle->vfield();
+
+    VField::size_type num_values = field->num_values();
+
+    FieldInformation fi(ifieldhandle);
+    if (fi.field_basis_order() < 1)
+    {
+      fi.make_lineardata();
+      fi.make_vector();
+    }
+
+    fi.make_vector();
+
+    auto ofieldhandle = CreateField(fi,ifieldhandle->mesh());
+    VField* ofield = ofieldhandle->vfield();
+    ofield->resize_values();
+
+    int cnt = 0;
+
+    if (field->is_scalar() && field->basis_order() == 1)
+    {
+      Point c; Vector vec; Vector diff; double val;
+      for (VField::index_type idx = 0; idx<num_values;idx++)
+      {
+        field->get_center(c,idx);
+        if (has_vector)
+        {
+           diff = c - attract_point;
+           vec = (dir*Dot(dir,diff)-diff);
+           vec.safe_normalize();
+        }
+        else
+        {
+          vec = (attract_point - c);
+          vec.safe_normalize();
+        }
+
+        field->get_value(val,idx);
+        if (val == 0.0) val = 1e-3;
+        vec = vec*val;
+        ofield->set_value(vec,idx);
+        cnt++; if (cnt == 400) { cnt=0; update_progress(idx,num_values); }
+      }
+    }
+    else
+    {
+      Point c; Vector vec; Vector diff;
+      for (VField::index_type idx = 0; idx<num_values;idx++)
+      {
+        field->get_center(c,idx);
+        if (has_vector)
+        {
+           diff = c - attract_point;
+           vec = (dir*Dot(dir,diff)-diff);
+           vec.safe_normalize();
+        }
+        else
+        {
+          vec = (attract_point - c);
+          vec.safe_normalize();
+        }
+        ofield->set_value(vec,idx);
+        cnt++; if (cnt == 400) { cnt=0; update_progress(idx,num_values); }
+      }
+    }
+
+    sendOutput(OutputField, ofield);
+  }
 }

src/Modules/Legacy/Fields/CalculateNormals.h

@@ -47,18 +47,17 @@ namespace Fields {
   class SCISHARE CalculateNormals : public SCIRun::Dataflow::Networks::Module,
     public Has1OutputPort<FieldPortTag>,
     public Has2InputPorts<FieldPortTag, FieldPortTag>
-    //public Has1OutputPort<FieldPortTag>
   {
   public:
     // these functions are required for all modules
     CalculateNormals();
+
     virtual void execute();
     virtual void setStateDefaults(){}
 
     //name the ports and datatype.
     INPUT_PORT(0, InputField, Field);
-    INPUT_PORT(1, InputField2, Field);
-    // OUTPUT_PORT(0, OutputField, Field);
+    INPUT_PORT(1, InputPoint, Field);
     OUTPUT_PORT(0, OutputField, Field);
 
     // this is needed for the module factory