Selection of feature points by their type corrected

Comparison/src/main/java/cz/fidentis/analyst/visitors/face/HausdorffDistancePrioritized.java

@@ -322,15 +322,18 @@ public class HausdorffDistancePrioritized extends HumanFaceVisitor  {
     public void visitHumanFace(HumanFace humanFace) {
         // Compute the Hasudorff distance using the 'distanceVisitor'
         humanFace.getMeshModel().compute(distanceVisitor, inParallel());
-        
+
         final Map<MeshFacet, List<Double>> hausdorffDistances = distanceVisitor.getDistances();
-                
+
         // Compute priorities of humanFace's vertices for each of the given feature point types
         for (final Map.Entry<FeaturePointType, Double> fpType: featurePointTypes.entrySet()) {
             final FeaturePointType featurePointType = fpType.getKey();
             final double featurePointRadius = fpType.getValue();
             
-            final FeaturePoint featurePoint = humanFace.getFeaturePoints().get(featurePointType.getType()); // Get feature point of desired type
+            final FeaturePoint featurePoint = getFeaturePointByType(humanFace, featurePointType);
+            if (featurePoint == null) {
+                continue;
+            }
             
             final PrioritySphere priorityVisitor = new PrioritySphere(featurePoint.getPosition(), featurePointRadius);
             humanFace.getMeshModel().compute(priorityVisitor, inParallel());
@@ -377,4 +380,14 @@ public class HausdorffDistancePrioritized extends HumanFaceVisitor  {
             }
         }
     }
+    
+    protected FeaturePoint getFeaturePointByType(HumanFace face, FeaturePointType type) {
+        for (final FeaturePoint featurePoint: face.getFeaturePoints()) {
+            if (type.getType() == featurePoint.getFeaturePointType().getType()) {
+                return featurePoint;
+            }
+        }
+        
+        return null;
+    }
 }