Screeps2/creep.builder.js at master · UoP-screeps/Screeps2 · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
const FSM = require('creep.fsm');
const utilities = require('creep.utilities');
const repairable = utilities.repairable;
const closestRep = utilities.closestRep;
const closestSite = utilities.closestSite;
const findEnergySource = utilities.findEnergySource;
const findSource = utilities.findSource;

Creep.builder = {
    body: (room) => {
        let repetition = 1;
        let manager = room.getHarvestManager();
        if(manager.income > 0 && room.energyCapacityAvailable >= 400 && manager.getStoredEnergy() >= 2000){
            repetition = 2;
        }
        if(room.energyAvailable > 1000 && room.storage && room.find(FIND_MY_CONSTRUCTION_SITES).length > 0){
            repetition = 3;
        }
        let part = [WORK, CARRY, MOVE];
        let body = [];
        for(let i = 0; i < repetition; i++){
            body = body.concat(part);
        }
        return body;
    },
    role: 'builder',
    spawnCondition: (spawn, spawning) => {
        return spawn.room.energyAvailable >= spawning.getCost() &&
            spawn.room.countRole('builder') < 1;
    },
    fsm: new FSM(
        ['moveWithdrawEnergy', 'moveBuild', 'moveRepair', 'moveUpgrade', 'moveHarvest'],
        initScope, transitionFunction,
        [_.identity, _.identity, _.identity, _.identity, _.identity]
    ),


};

function initScope(creep) {
    return getPosObjPair(findEnergySource(creep));
}

function transitionFunction(creep){
    const full = (creep.store.getFreeCapacity() === 0);
    const empty = (creep.store[RESOURCE_ENERGY] === 0);
    const scope = creep.memory.fsm.scope;
    const obj = deserialize(scope[1], RoomObject);
    let rep = closestRep(creep);
    let site = closestSite(creep);
    switch (creep.memory.fsm.state) {
        case 0: //moveWithdrawEnergy
            if(full){
                let emergent = utilities.findEmergentRepair(creep);
                if(emergent){
                    return [2, getPosObjPair(emergent)];
                }
            }if(full && site){
                return [1, getPosObjPair(site)];
            }else if(full && rep){
                return [2, getPosObjPair(rep)];
            }else if(full){
                return [3, getPosObjPair(creep.room.controller)];
            }else if(creep.room.energyCapacityAvailable < 550 && creep.room.energyAvailable <= 500){
                return [4, getPosObjPair(findSource(creep))];
            }else if(obj.store[RESOURCE_ENERGY] === 0){
                return [0, initScope(creep)];
            } return [0, scope];
        case 1: //moveBuild
            if(empty || !site){
                return [0, initScope(creep)];
            }else if(!obj){
                return [1, getPosObjPair(site)];
            }else return [1, scope];
        case 2: //moveRepair
            if(!empty && site){
                return [1, getPosObjPair(site)];
            }
            if(empty || !rep){
                return [0, initScope(creep)];
            }else if(!obj || !repairable(obj, creep.getActiveBodyparts(WORK) * REPAIR_POWER)){
                return [2, getPosObjPair(rep)];
            }return [2, scope];
        case 3: //moveUpgrade
            if(empty){
                return [0, initScope(creep)];
            }else{
                return [3, scope];
            }
        case 4:
            if(full){
                let emergent = utilities.findEmergentRepair(creep);
                if(emergent){
                    return [2, getPosObjPair(emergent)];
                }
            }
            if(full && site){
                return [1, getPosObjPair(site)];
            }else if(full && rep){
                return [2, getPosObjPair(rep)];
            }else if(full){
                return [3, getPosObjPair(creep.room.controller)];
            }else return [4, scope];
    }

}