-
Notifications
You must be signed in to change notification settings - Fork 0
/
aliens.go
49 lines (40 loc) · 1.01 KB
/
aliens.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package main
import (
"math/rand"
randomdata "github.com/Pallinder/go-randomdata"
)
type alien struct {
name string
atCity *city
dead bool
}
// generateAliens creates aliens and place them in random cities
func generateAliens(numAliens int, listCities []*city, rng *rand.Rand) []*alien {
aliens := make([]*alien, 0, numAliens)
for i := 0; i < numAliens; i++ {
// give alien a name
alienToAdd := alien{name: randomdata.SillyName()}
for {
// find random city
randCity := rng.Intn(len(listCities))
atCity := listCities[randCity]
// check if city has no alien
if len(atCity.aliens) != 0 {
continue
}
alienToAdd.atCity = atCity
// add to list of aliens
aliens = append(aliens, &alienToAdd)
// add to the city as well
atCity.aliens = append(atCity.aliens, &alienToAdd)
//
break
}
}
return aliens
}
func (a *alien) moveAlien(from, to *city) {
from.aliens = from.aliens[:0] // there can only be one alien in a city
to.aliens = append(to.aliens, a)
a.atCity = to
}