Commit 4c2896e7 authored by Daniel Schramm's avatar Daniel Schramm
Browse files

Prioritized distance computation parallelized

parent 5f27f6f3
Loading
Loading
Loading
Loading
+21 −3
Original line number Diff line number Diff line
@@ -15,6 +15,8 @@ import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import javax.vecmath.Point3d;

/**
@@ -86,12 +88,28 @@ public class HausdorffDistancePrioritized extends HumanFaceVisitor {
    @Override
    public void visitHumanFace(HumanFace humanFace) {
        final FeaturePoint featurePoint = humanFace.getFeaturePoints().get(featurePointType.getType());
        final List<MeshFacet> facets = humanFace.getMeshModel().getFacets();
        
        final ExecutorService executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
        
        final PrioritySphere priorityVisitor = new PrioritySphere(featurePoint.getPosition(), 1 /*TODO TEMPORARY - sphere radius needs to be computed dynamically*/);
        for (final MeshFacet comparedFacet: humanFace.getMeshModel().getFacets()) {
        for (final MeshFacet comparedFacet: facets) {
            if (inParallel() && distanceVisitor.isThreadSafe()) {
                executor.execute(() -> comparedFacet.accept(distanceVisitor));
            } else {
                comparedFacet.accept(distanceVisitor);
            }
            if (inParallel() && priorityVisitor.isThreadSafe()) {
                executor.execute(() -> comparedFacet.accept(priorityVisitor));
            } else {
                comparedFacet.accept(priorityVisitor);
            }
        }
        
        if (inParallel()) {
            executor.shutdown();
            while (!executor.isTerminated()){}
        }
        
        synchronized(this) {
            priorities.putAll(priorityVisitor.getPriorities());