func (a *aiAdapter) TakeTurn(call botapi.Ai_takeTurn) error { ib, err := call.Params.Board() if err != nil { return err } board, err := ib.Board() if err != nil { return err } gameID, err := board.GameId() if err != nil { return err } // Convert the board to the game representation b, robots, err := convertBoard(board) if err != nil { return err } // Load the AI for this game, or create a new one ai := a.games[gameID].ai if ai == nil { ai = a.factory(gameID) // Load the cells for the board cells, err := ib.Cells() if err != nil { return nil } locs := convertLocs(cells, len(b.Cells), len(b.Cells[0])) a.games[gameID] = gameState{ ai: ai, locs: locs, } } b.LType = a.games[gameID].locs turns, err := botapi.NewTurn_List(call.Results.Segment(), int32(len(robots))) if err != nil { return err } for i, r := range robots { t := ai.Act(b, r) t.ToWire(r.ID, turns.At(i)) } call.Results.SetTurns(turns) return nil }
func runMatch(gidCh chan<- gameID, ctx gocontext.Context, ds datastore, aiA, aiB *onlineAI, bc engine.BoardConfig) error { sTime := time.Now() // Create new board and store it. b := engine.EmptyBoard(bc) b.InitBoard(bc) _, seg, _ := capnp.NewMessage(capnp.SingleSegment(nil)) wb, _ := botapi.NewRootInitialBoard(seg) b.ToWireWithInitial(wb, engine.P1Faction) gid, err := ds.startGame(aiA.Info.ID, aiB.Info.ID, wb) if err != nil { return err } gidCh <- gid // Run the game for !b.IsFinished() { turnCtx, _ := gocontext.WithTimeout(ctx, 30*time.Second) chA, chB := make(chan turnResult), make(chan turnResult) go aiA.takeTurn(turnCtx, gid, b, engine.P1Faction, chA) go aiB.takeTurn(turnCtx, gid, b, engine.P2Faction, chB) ra, rb := <-chA, <-chB if ra.err.HasError() { log.Printf("Errors from AI ID %s: %v", aiA.Info.ID, ra.err) } if rb.err.HasError() { log.Printf("Errors from AI ID %s: %v", aiB.Info.ID, rb.err) } b.Update(ra.results, rb.results) _, s, err := capnp.NewMessage(capnp.SingleSegment(nil)) if err != nil { return err } r, err := botapi.NewRootReplay_Round(s) if err != nil { return err } wireBoard, err := r.NewEndBoard() if err != nil { return err } b.ToWire(wireBoard, engine.P1Faction) ral, rbl := ra.results.Len(), rb.results.Len() turns, err := botapi.NewTurn_List(r.Segment(), int32(ral+rbl)) if err != nil { return err } for i := 0; i < ral; i++ { t := ra.results.At(i) if err := turns.Set(i, t); err != nil { return err } } for i := 0; i < rbl; i++ { t := rb.results.At(i) if err := turns.Set(i+ral, t); err != nil { return err } } r.SetMoves(turns) db.addRound(gid, r) } gInfo := &gameInfo{ ID: gid, AI1: &aiA.Info, AI2: &aiB.Info, AI1Score: b.BotCount(1), AI2Score: b.BotCount(2), StartTime: sTime, EndTime: time.Now(), } return db.finishGame(gid, &aiA.Info, &aiB.Info, gInfo) }