RandomInAreaTransitionStrategy.java 4.59 KB
Newer Older
1
2
package de.tud.kom.p2psim.impl.topology.movement.modularosm.transition;

3
4
import java.util.HashMap;
import java.util.Map;
5
6
7
import de.tud.kom.p2psim.api.topology.Topology;
import de.tud.kom.p2psim.api.topology.movement.SimLocationActuator;
import de.tud.kom.p2psim.impl.topology.movement.modularosm.attraction.BasicAttractionPoint;
8
import de.tud.kom.p2psim.impl.topology.movement.modularosm.attraction.IAttractionGenerator;
9
import de.tud.kom.p2psim.impl.topology.util.PositionVector;
10
11
12
13
14
15
import de.tudarmstadt.maki.simonstrator.api.Binder;
import de.tudarmstadt.maki.simonstrator.api.Monitor;
import de.tudarmstadt.maki.simonstrator.api.component.sensor.location.AttractionPoint;
import de.tudarmstadt.maki.simonstrator.api.component.sensor.location.Location;

/**
Julian Zobel's avatar
wip    
Julian Zobel committed
16
 * This {@link IAttractionAssigmentStrategy} makes clients move around randomly in a specified area. You can specify the target
Clemens Krug's avatar
Clemens Krug committed
17
18
19
20
 * area center via the {@link #updateTargetAttractionPoint(SimLocationActuator, AttractionPoint)} method. The client will then start
 * to roam the area randomly till a new target area is assigned.
 *
 * @author Clemens Krug
21
22
 * 
 * Made the transition strategy extend the {@link AbstractAttractionBasedTransitionStrategy} for a more generalized approach on 
Julian Zobel's avatar
wip    
Julian Zobel committed
23
 * the {@link IAttractionAssigmentStrategy} implementation for attraction point based strategies.
24
25
26
 * 
 * @author Julian Zobel
 * @version 1.1, January 2019
27
 */
28
29
public class RandomInAreaTransitionStrategy extends AbstractAttractionBasedTransitionStrategy
{   
Clemens Krug's avatar
Clemens Krug committed
30
31
32
    /**
     * These are the current spots inside the target area where the client is currently heading.
     */
33
    private Map<SimLocationActuator, AttractionPoint> currentTarget = new HashMap<>();
Clemens Krug's avatar
Clemens Krug committed
34
    private Map<SimLocationActuator, Integer> currentSearchRadius = new HashMap<>();
35
  
Clemens Krug's avatar
Clemens Krug committed
36
37
38
    /**
     * The radius the target area should have. Should be set via XML config file.
     */
39
40
41
42
    private int defaultRadius;

    @Override
    public void addComponent(SimLocationActuator ms) {
Clemens Krug's avatar
Clemens Krug committed
43

44
45
        if(!assignments.containsKey(ms))
        {
46
            AttractionPoint aPoint = IAttractionGenerator.attractionPoints.iterator().next();
47
48
            assignments.put(ms, aPoint);
            currentTarget.put(ms, nextRandomPosition(aPoint, defaultRadius));
Clemens Krug's avatar
Clemens Krug committed
49
            currentSearchRadius.put(ms, defaultRadius);
50
        }
51
        notifyListenersOfAssignmentUpdate(ms,  currentTarget.get(ms));
52
53
54
    }

    @Override
55
    public void reachedAttractionPoint(SimLocationActuator ms, AttractionPoint attractionPoint) {
Clemens Krug's avatar
Clemens Krug committed
56
        currentTarget.put(ms, nextRandomPosition(assignments.get(ms), currentSearchRadius.get(ms)));
57
        notifyListenersOfAssignmentUpdate(ms,  currentTarget.get(ms));
58
59
60
61
62
    }

    @Override
    public void updateTargetAttractionPoint(SimLocationActuator comp, AttractionPoint attractionPoint) {
        assignments.put(comp, attractionPoint);
Clemens Krug's avatar
Clemens Krug committed
63
        currentTarget.put(comp, nextRandomPosition(attractionPoint, currentSearchRadius.get(comp)));
64
        notifyListenersOfAssignmentUpdate(comp, attractionPoint);
65
66
    }

67
    public void setSearchRadiusForComponent(SimLocationActuator ms, int radius) {
Clemens Krug's avatar
Clemens Krug committed
68
69
70
        currentSearchRadius.put(ms, radius);
    }

71
    public void setDefaultRadius(int radius) {
72
73
74
        this.defaultRadius = radius;
    }

75
76
77
78
79
80
    /**
     * Calculates a random point within a given circular area.
     * @param center The center of the area.
     * @param radius The radius of the area.
     * @return A random position within the area.
     */
81
    private BasicAttractionPoint nextRandomPosition(Location center, int radius) {
Clemens Krug's avatar
Clemens Krug committed
82
        assert radius > 0 : "An area radius must be specified for the RandomInAreaTransitionStrategy! Did you set the 'DefaultRadius' property for this transition?";
Clemens Krug's avatar
Clemens Krug committed
83

84
85
        double x = center.getLongitudeOrX();
        double y = center.getLatitudeOrY();
86
87
88
89
90
91
92
93
94

        double newX = -1;
        double newY = -1;

        int tries = 0;

        while(newX < 0.0 || newX > Binder.getComponentOrNull(Topology.class).getWorldDimensions().getX()
                || newY < 0.0 || newY > Binder.getComponentOrNull(Topology.class).getWorldDimensions().getY())
        {
95
96
            double calcRadius = rnd.nextDouble() * radius;
            double calcAngle = rnd.nextDouble() * 360;
97
98
99
100
101
102
103
104
105
106
107
108
109

            newX = x + Math.sin(calcAngle) * calcRadius;
            newY = y + Math.cos(calcAngle) * calcRadius;
            tries++;

            if(tries > 100) throw new AssertionError("Unable to find a valid target destination within <100 tries.");
        }

        Monitor.log(this.getClass(), Monitor.Level.DEBUG, "Next random position is " + newX + " / " + newY);

        return new BasicAttractionPoint("RNDPOS", new PositionVector(newX, newY));
    }
}