RandomInAreaTransitionStrategy.java 4.67 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;

/**
Clemens Krug's avatar
Clemens Krug committed
16
17
18
19
20
 * This {@link ITransitionStrategy} makes clients move around randomly in a specified area. You can specify the target
 * 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
23
24
25
26
 * 
 * Made the transition strategy extend the {@link AbstractAttractionBasedTransitionStrategy} for a more generalized approach on 
 * the {@link ITransitionStrategy} implementation for attraction point based strategies.
 * 
 * @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
  
36

Clemens Krug's avatar
Clemens Krug committed
37
38
39
    /**
     * The radius the target area should have. Should be set via XML config file.
     */
40
41
42
    private int defaultRadius;

    @Override
43
    public AttractionPoint getAssignment(SimLocationActuator comp) {
44
        return currentTarget.get(comp);
45
    }   
46
47
48

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

50
51
        if(!assignments.containsKey(ms))
        {
52
            AttractionPoint aPoint = IAttractionGenerator.attractionPoints.iterator().next();
53
54
            assignments.put(ms, aPoint);
            currentTarget.put(ms, nextRandomPosition(aPoint, defaultRadius));
Clemens Krug's avatar
Clemens Krug committed
55
            currentSearchRadius.put(ms, defaultRadius);
56
        }
57
        notifyListenersOfAssignmentUpdate(ms,  currentTarget.get(ms));
58
59
60
61
    }

    @Override
    public void reachedAttractionPoint(SimLocationActuator ms) {
Clemens Krug's avatar
Clemens Krug committed
62
        currentTarget.put(ms, nextRandomPosition(assignments.get(ms), currentSearchRadius.get(ms)));
63
        notifyListenersOfAssignmentUpdate(ms,  currentTarget.get(ms));
64
65
66
67
68
    }

    @Override
    public void updateTargetAttractionPoint(SimLocationActuator comp, AttractionPoint attractionPoint) {
        assignments.put(comp, attractionPoint);
Clemens Krug's avatar
Clemens Krug committed
69
        currentTarget.put(comp, nextRandomPosition(attractionPoint, currentSearchRadius.get(comp)));
70
        notifyListenersOfAssignmentUpdate(comp, attractionPoint);
71
72
    }

73
    public void setSearchRadiusForComponent(SimLocationActuator ms, int radius) {
Clemens Krug's avatar
Clemens Krug committed
74
75
76
        currentSearchRadius.put(ms, radius);
    }

77
    public void setDefaultRadius(int radius) {
78
79
80
        this.defaultRadius = radius;
    }

81
82
83
84
85
86
    /**
     * 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.
     */
87
    private BasicAttractionPoint nextRandomPosition(Location center, int radius) {
Clemens Krug's avatar
Clemens Krug committed
88
        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
89

90
91
        double x = center.getLongitudeOrX();
        double y = center.getLatitudeOrY();
92
93
94
95
96
97
98
99
100

        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())
        {
101
102
            double calcRadius = rnd.nextDouble() * radius;
            double calcAngle = rnd.nextDouble() * 360;
103
104
105
106
107
108
109
110
111
112
113
114
115

            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));
    }
}