Toplevel simulation

From Control Systems Technology Group
Revision as of 21:57, 6 December 2015 by S129520 (talk | contribs)
Jump to navigation Jump to search

Back to: PRE2015_2_Groep1

Top level behavioural algorithm

Outline of the idea for an algorithm, based on the ideas presented in [Liu B Chen P Wang G], [Qi G Song P Li K] and [Nicolis S Detrain C Demolin D Deneubourg J]:

Initially (i.e. when the blackboard is empty): all robots departing from the CZ are assigned a search task (need to implement a good search algorithm)

  • when pile is found by a robot: add the pile to the blackboard (coordinates and pheromones)

When a robot departs from CZ when the blackboard is non-empty:

  • decide (stochastically) whether to search or to forage (based on size of blackboard)
  • if decided to go forage, pick a pile from the blackboard (stochastically based on pheromones)
  • then if pile is picked, if paths are known, pick a path based on pheromones, else (if no paths are known yet): move in the direction of the pile in such a way that if obstacles are encountered a way around them it is found stochastically. Keep track of the path.

When a robot arrives at a pile: update pheromones (if path is kept track of, add path to blackboard) and go to que of pile

At every time step, some pheromone "evaporates" from the blackboard.

n.b. in the dropping the pheromones and in choosing whether to search or to forage, USE aspects come in to play:

  • how much priority should rescuing known victims get over searching for new victims?
  • how much more priority should a pile get if we know it has n victims? (Later to be added to simulation)
  • should nearby piles get priority over harder to reach piles?
  • should only path length matter, or should e.g. safety play a role in dropping path pheromones? (We may add this later, but for now we'll ignore this)

Code

This isn't functioning yet (more work needs to be done) but this is what we have so far (06-12-2015 21:55) (it's written in Python 2.7.9)

import random

class AO(object):

   """Represents area of operation"""
  1. todo: implement something for keeping track of interesting statistics
   __size = None
   __obstacles = None
   __piles = None
   robotsInAO = None
   blackboard =None
   CZ=None
   l=2#the "sensitivity of the choice process"
   speed=1
   def __init__(self, size, cz, nRobots, piles, obstacles, evaporatePile = lambda x:x, evaporatePath = lambda x:x, dropPile = lambda x,y,z:y++, dropPath = lambda x,y:x++, l=2, speed=1):
       """
       needs as input:
       -the size of the AO
       -the coordinates for the CZ (just one point for now)
       -the number of robots
       -a list of piles
           two points per pile giving two corners of the pile which is assumed to be rectangular (possibly degenerated)
           a list of objects as ints indicating the number of robots needed to move the object
           e.g. [ ((1,2), (3,4), [1,1,6,3,1]), ((6,7), (8,12), [3,1,1,3,4])]
       -a list of obstacles
           two points per obstacle giving two corners of the obstacle which is assumed to be rectangular (possibly degenerated)
           e.g. [((1,1),(2,2)), ((3,1),(4,3))]
       -a function that determines the speed of evaporation of the pheromones for the piles (mapping floats to floats in a suitable manner)
       -a function that determines the speed of evaporation of the pheromones for the paths (mapping floats to floats in a suitable manner)
       -a function called by a robot to increase the pheromone concentration at a pile (three arguments: pilesize, pheromone concentration, treshold)
       -a function called by a robot to increase the pheromone concentration at a path (to floats)
       """
       self.l=l#the "sensitivity of the choice process"
       self.speed=speed
       self.__size = (int(size[0]),int(size[1]))
       #automatically checks for right type this way
       self.CZ=(int(cz[0]),int(cz[1]))
       self.__evaporatePile=evaporatePile
       self.__dropPile=dropPile
       self.__evaporatePath=evaporatePath
       self.__dropPath=dropPath
       #for i in range(nRobots):
           #self.robotsInAO.append(Robot(self.CZ,self.__drop))#fill the list with robots
       #want to fill the list of robots one at a time so they don't all live at the same time
       self.numrobots=nRobots
       self.__piles=[]
       for pile in piles:#create all the piles according to input
           self.__piles.append(Pile(
               (int(pile[0][0]),int(pile[0][1])),
               (int(pile[1][0]),int(pile[1][1])),
               [int(obj) for obj in pile[2]]))
       self.__obstacles = []
       for obstacle in obstacles:
           #all obstacles are assumed to be rectangles
           self.__obstacles.append((
               (int(obstacle[0][0]),int(obstacle[0][1])),
               (int(obstacle[1][0]),int(obstacle[1][1]))))
       self.robotsInAO = []
       self.blackboard = []
   def update(self):
       if self.numrobots>0:#release robots one at a time
           self.robotsInAO.append(Robot(self, self.CZ, self.__dropPile, self.__dropPath))
           self.numrobots--
       for robot in self.robotsInAO:
           robot.update()
           for pile in self.__piles:
               if min(pile.coordinates[0][0],pile.coordinates[1][0])<=robot.coordinates[0]<=max(pile.coordinates[0][0],pile.coordinates[1][0]) and \
                  min(pile.coordinates[0][1],pile.coordinates[1][1])<=robot.coordinates[1]<=max(pile.coordinates[0][1],pile.coordinates[1][1]):
                   #the robot has arrived at the pile
                   pile.update(robot, self)
       #self.pheromones= map(self.__evaporate,self.pheromones)#negative feedback on the pheromones
  1. todo: implement evaporation
       return len(self.__piles)#this will be the check for the main loop
   def removePile(self, pile):
       self.__piles.remove(pile)
  1. todo: remove pile from blackboard
   def totalRobots(self):
       return len(self.robotsInAO)+sum([pile.totalRobots() for pile in self.__piles)])
   #def allowedMove(self, coordinates1, coordinates2):
   #    """Decides wether moving from coordinates1 to coordinates2 is possible"""
   #    #a move is possible if there is no obstacle intersecting with [coordinates1, coordinates2] and also no other pile than that at coordinates2 is intersecting with [coordinates1, coordinates2]
   #    length = ((coordinates2[0]-coordinates1[0])**2+(coordinates2[1]-coordinates1[1])**2)**.5
   #    direction = (float(coordinates2[0]-coordinates1[0])/length,
   #                 float(coordinates2[1]-coordinates1[1])/length)
   #    line=[(round(coordinates1[0] + .1*n*direction[0]),round(coordinates1[1]+.1*n*direction[1])) for n in range(round(10*length)+1)]
   #isn't practical
   def getMovements(self,coordinates):
       freeSpace=[(coordinates[0]+n,coordinates[1]+m) for n in range(-speed,speed+1) for m in range(-speed, speed+1) if n*n+m*m<=speed*speed] #circle around the point
       pilepoints=[]
       for coord in freeSpace:
           #remove all points where there are obstacles
           for obstacle in self.__obstacles:
               if min(obstacle[0][0],obstacle[1][0])<=coord[0]<=max(obstacle[0][0],obstacle[1][0]) and  \
                  min(obstacle[0][1],obstacle[1][1])<=coord[1]<=max(obstacle[0][1],obstacle[1][1]):
                   freeSpace.remove(coord)
           for pile in self.__piles:
               if min(pile.coordinates[0][0],pile.coordinates[1][0])<=coord[0]<=max(pile.coordinates[0][0],pile.coordinates[1][0]) and \
                  min(pile.coordinates[0][1],pile.coordinates[1][1])<=coord[1]<=max(pile.coordinates[0][1],pile.coordinates[1][1]):
                   pilepoints.append(coord)
       #now remove all coordinates that are blocked indirectly
  1. todo implement this
       return (freeSpace, pilepoints) #n.b. this way a robot might go to a pile unintendedly
   def getPileSizeAt(self, coordinates):
       """returns the pile size of the pile at the coordinates"""
       for pile in self.__piles:
               if min(pile.coordinates[0][0],pile.coordinates[1][0])<=coordinates[0]<=max(pile.coordinates[0][0],pile.coordinates[1][0]) and \
                  min(pile.coordinates[0][1],pile.coordinates[1][1])<=coordinates[1]<=max(pile.coordinates[0][1],pile.coordinates[1][1]):
                   return pile.pileSize()
       return None
               
       
       
               
   

class Pile(object):

   __objects=None
   coordinates = None
   __que = None
   def __init__(self,coordinate1, coordinate2, *objects):
       #all piles are assumed to be rectangles
       self.__objects=[obj for obj in objects]
       self.__que=[]
       self.coordinates = (coordinate1, coordinate2)
   def update(self, robot, ao):
       self.__que.append(ao.robotsInAo.pop(ao.robotsInAo.index(robot)))
       try:
           if len(self.__que)>self.__objects[0]:
               obj=self.__objects.pop(0)#remove the object and get the number of needed robots
               for i in range(obj):
                   rob = self.__que.pop()#remove a robot from the que
                   rob.carrying = True #tell that robot it is carrying an object
                   rob.sendToCZ()#send that robot to the clear zone
               ao.robotsInAO.append(rob)#add that robot to the list of the AO
           if len(self.__objects)==0:
               ao.removePile(self)#if the pile is empty, remove it from the list of piles
       except IndexError:
           ao.removePile(self)
   def totalRobots(self):
       return len(self.__que)
   def pileSize(self):
       return sum(self.__objects)#should change this so we also have  victims

class Robot(object):

   coordinates=None
   #Robots are assumed to be small enough to be well represented by a single coordinate
   __goal=None
   __task=None
   carrying=False
   __path=None
   __newpath=None
   ao=None
   __pathlength=0
   def __init__(self,ao, coordinates, dropPile, dropPath):
       self.coordinates=(int(coordinates[0]),int(coordinates[1]))
       self.__dropPile = dropPile
       self.__dropPath = dropPath
       self.ao= ao
   def sendToCZ(self):
       self.__goal = self.ao.CZ
       self.__task="Return"
       #choose the fastest known path to the CZ
       #maybe wan't to optimize on the path back too, but don't really feel like reversed path changing yet so maybe later
       #last coordinate was at the pile, so for simplicity assume it is still there
       pile=None
       for elem in self.ao.blackboard:
           if self.coordinates == elem[0]:
               pile=elem
               break
       minlen = min([path[0] for path in pile[3]])
       minlenpaths = [path for path in pile[3] if path[0]<= minlen+0.1]
       #choose a path of minimal length
       self.__path=minlenpaths[0]#all robots need to take the same one because they need to stay together
       self.pathindex=-1#since we're following backwards
       
  1. todo complete this
   def update(self):
       if self.coordinates == self.ao.CZ:
           self.carrying = False
           self.__pathlength=0
           #assign a task and if needed a pile and a path
           rnum = random.random()
           if rnum>self.searchChoice():
               self.__task = "Search"
           else:
               self.__task = "Forage"
           #now, if it is assigned a forage task, pick a pile from the blackboard
           if self.__task=="Forage":
               #example of what a blackboard element may look like:
               #(coordinates, pheromone, treshold (k in the articles), [list of paths])
               #function from articles
               rnum = random.random()
               denominator = sum([(pile[1]+pile[2])**ao.l for pile in ao.blackboard])
               treshold = 0
               for pile in ao.blackboard:
                   treshold += (pile[1]+pile[2])**ao.l
                   if rnum <treshold:
                       self.__goal=pile[0]
                       #this is the pile it gets assigned too
                       #so now choose a path if available
                       if len(pile[3])>0:
                           rnum =random.random()#we don't need the previous rnum anymore since we'll now break from the loop
                           #now choosing from the known paths
                           #in the article it is done by pheromones but I'm now wondering why we shouldn't just do this by shortest known path
  1. todo: reconsider all of this
                           #OK, so instead of doing the pheromones for pathfinding, we'll let every robot pick either the shortest rout from the blackboard
                           #or find a new rout
                           #furthermore, to make short routs, instead of following the rout, we'll have the robot attempt to shorten it
                           #actually I now understand why you need the pheromone thing:
                           #the first found path gets optimized first so most newfound paths, when they're found, will not be shorter even though they leave more room for improvement
  1. todo: change all to pheromone based approach
                           if rnum >(0.25+(0.5*len(pile[3]))/(len(pile[3])+1)): # want to keep looking for new paths, but intensity should depend on number of known paths
  1. todo: think of a better distribution
                               #a path looks like (length (or pheromones), [coordinates])
                               minlen = min([path[0] for path in pile[3]])
                               minlenpaths = [path for path in pile[3] if path[0]<= minlen+0.1]#just in case more paths of minimal length exist (taking into account propperties of floats
                               self.__path=random.choice(minlenpaths)[-1]# want to keep using both since one might have more potential for updating than the other
                               #n.b. stochastics only yield good results for large number of robots
                               self.__newpath=[self.ao.CZ]
                               self.pathindex =0
                           else:
                               #seek for new path
                               self.__path=[]
                               self.__newpath=[self.ao.CZ]
  1. note to self: since now we're coming up with new paths for every robot, garbage collection should be performed on the blackboard
                       else:
                           self.__newpath=[self.ao.CZ]
                           self.__path=[]
                       break #we've found our pile and decided how to get there so break from the loop
               #basically, self.__task, self.__goal, self.__path, self.__pathlength and self.__newpath have been updated
       elif self.__task==None:
           #if robots start at a point different from the cz
           #probably need something better than this
  1. todo implement something better
           self.__task = "Search"
           
       movements= self.ao.getMovements(self.coordinates)#get the list of possible movements looks like [possiblemovements, reachablepiles]
       if self.__task == "Search":
           pass
  1. todo implement a good swarm-search algorithm
       elif self.__task == "Return":
           #simply follow the choosen path (choosen in sendToCZ()) backwards. No backwards optimization of the path at this point
  1. n.b. at this moment we assume no new obstacles are introduced during the simulation
  2. need to extend this part for if new obstacles are introduced
  3. basically need to do the same path finding algorithm as for the Forage when no path is selected except we need some way to keep the robots together (since they're pushing the same piece)
           if self.__path[self.pathindex-1] in movements[0]:
               coordinates = self.__path[self.pathindex-1]
               self.pathindex-=1
           else:
               pass
           
       elif self.__task == "Forage":
           if len(self.__path)>0:
               #find the next point to go to
               currentLength = len(self.__newpath)
               if self.__goal in movements[0]:
                   #we're there
                   self.__newpath.append(self.__goal)
                   self.__pathlength+=((self.coordinates[0]-self.__goal[0])**2 + (self.coordinates[1]-self.__goal[1])**2 )**.5
                   self.coordinates = self.__goal
                   #since self.__goal was a pile, after the update, the update() function of the ao will add this robot to the pile it's at
                   bbElement = None
                   for element in self.ao.blackboard:#find the corresponding pile in the blackboard
                       if element[0]==self.coordinates:
                           bbElement = element
                           break
  1. todo implement something to get rid of loops in the path (if there are any)
                   bbElement[-1].append((self.__pathlength, self.__newpath))
                   #do garbage collection over the blackboard element to prevent the blackboard from getting too large
                   minlen = min([path[0] for path in bbElement[-1]])
                   bbElement[-1] = [path for path in bbElement[-1] if path[0]<=1.2*minlen] #need to change this to pheromone based approach
                   #drop pheromones to pile
                   bbElement[1] = self.dropPile(self.ao.getPileSizeAt(self.__goal),*bbElement[1:3])
  1. todo change this to pheromone based approach
  2. todo consider implementing an algorithm that optimizes over the old path points
               else:
                   try:
                       proposedScale = ((self.coordinates[0]-self.__path[self.pathindex+2][0])**2 +(self.coordinates[1]-self.__path[self.pathindex+2][1])**2)**.5
                       proposedDirection = ((self.__path[self.pathindex+2][0]-self.coordinates[0])/proposedScale,(self.__path[self.pathindex+2][1]-self.coordinates[1])/proposedScale)
                       proposedCoordinate = (round(self.coordinates[0] + self.ao.speed*proposedDirection[0]),round(self.coordinates[1] + self.ao.speed*proposedDirection[1]))
                   except ZeroDivisionError:
                       proposedCoordinate = self.__path[self.pathindex+3]
                       self.pathindex+=1
                   self.pathindex+=1
                   if proposedCoordinate in movements[0]:#at this moment, robots might get into the wrong pile accidentally. Not sure if this is bad
                       self.__newpath.append(proposedCoordinate)
                       self.coordinates=proposedCoordinate
                       if self.__path[self.pathindex+2] in self.ao.getMovements(self.coordinates)[0]:#might want to reduce computation time by moving the getMovements
  1. todo: consider moving the getMovements step to other iteration since now we're computing twice
                           self.pathindex+=1
                   else:#the proposed move can't be executed
                       if self.__path[self.pathindex] in movements[0]:
                           self.__newpath.append(self.__path[self.pathindex])
                           self.coordinates = self.__path[self.pathindex]
                       elif self.path[self.pathindex-1] in movements[0]:#might need to think of something better than this (so that all the optimizations at parts of the path where
                           #everything went right don't get dismissed because of one point where it went wrong)
                           self.__newpath.append(self.__path[self.pathindex-1])
                           self.coordinates = self.__path[self.pathindex-1]
                           self.pathindex-=1
                       else:
                           raise Exception # might want to assume a new  obstacle is simply introduced and hence the old path has become invalid
                       #in this case we want to continue to searching for a new path
  1. todo: come up with an appropriate solution
                   self.pathlength+= ((self.__newpath[-1][0]-self.__newpath[-2][0])**2 + (self.__newpath[-1][1]-self.__newpath[-2][1])**2)**.5
           else:
               #the algorithm comes down to this:
               #try to move towards the goal
               #if that something is blocking that move goal:
               #  there is one minimal deflection to the left and one minimal deflection to the right, flip a coin between them
               #  except one of them means going back the same path, that's forbidden unless it's the only option
  1. todo: implement this
               pass
  1. todo: implement this
  2. todo implement loop cutting at the moment a path is stored
   def searchChoice():
       """probability of a robot going for foraging"""
       #depending on the number of robots, size of the ao and number of known piles
       #this is just a place holder. The final choice will be based on USE decisions
       #for now we assume that there should be approximately 10 robots for every pile
       return min((10.*len(ao.blackboard))/ao.totalRobots,1) #although the min(...,1) doesn't really have any effect due to implementation