Repository navigation
Expand file tree
/
Copy pathemigration.lua
More file actions
188 lines (157 loc) · 5.57 KB
/
Copy pathemigration.lua
File metadata and controls
188 lines (157 loc) · 5.57 KB
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
--@module = true
--@enable = true
local utils = require('utils')
local nobles = reqscript('internal/emigration/emigrate-nobles')
local unit_link_utils = reqscript('internal/emigration/unit-link-utils')
local GLOBAL_KEY = 'emigration' -- used for state change hooks and persistence
local function get_default_state()
return {
enabled=false,
last_cycle_tick=0
}
end
state = state or get_default_state()
function isEnabled()
return state.enabled
end
local function persist_state()
dfhack.persistent.saveSiteData(GLOBAL_KEY, state)
end
local TICKS_PER_MONTH = 33600
local TICKS_PER_YEAR = 12 * TICKS_PER_MONTH
function desireToStay(unit,method,civ_id)
-- on a percentage scale
local value = 100 - unit.status.current_soul.personality.stress / 5000
if method == 'merchant' then
if civ_id ~= unit.civ_id then value = value*2 end end
if method == 'wild' then
value = value*5 end
return value
end
function desert(u,method,civ)
u.following = nil
local line = dfhack.units.getReadableName(u) .. " has "
if method == 'merchant' then
line = line.."joined the merchants"
unit_link_utils.markUnitForEmigration(u, civ, false)
else
line = line.."abandoned the settlement in search of a better life."
unit_link_utils.markUnitForEmigration(u, civ, true)
end
local hf = df.historical_figure.find(u.hist_figure_id)
local fort_ent = df.global.plotinfo.main.fortress_entity
local civ_ent = df.historical_entity.find(hf.civ_id)
local newent_id = -1
local newsite_id = -1
unit_link_utils.removeUnitAssociations(u)
unit_link_utils.removeHistFigFromEntity(hf, fort_ent)
-- try to find a new entity for the unit to join
for _,entity_link in ipairs(civ_ent.entity_links) do
if entity_link.type == df.entity_entity_link_type.CHILD and entity_link.target ~= fort_ent.id then
newent_id = entity_link.target
break
end
end
if newent_id > -1 then
-- try to find a new site for the unit to join
for _,site_link in ipairs(df.global.world.entities.all[hf.civ_id].site_links) do
local site_id = df.global.plotinfo.site_id
if site_link.type == df.entity_site_link_type.Claim and site_link.target ~= site_id then
newsite_id = site_link.target
break
end
end
local newent = df.historical_entity.find(newent_id)
unit_link_utils.addHistFigToSite(hf, newsite_id, newent)
end
print(dfhack.df2console(line))
dfhack.gui.showAnnouncement(line, COLOR_WHITE)
end
function canLeave(unit)
if not unit.status.current_soul then
return false
end
return dfhack.units.isActive(unit) and
dfhack.units.isCitizen(unit) and
not dfhack.units.getNoblePositions(unit) and
not unit.flags1.chained and
unit.military.squad_id == -1 and
not dfhack.units.isBaby(unit) and
not dfhack.units.isChild(unit)
end
function checkForDeserters(method,civ_id)
local allUnits = df.global.world.units.active
for i=#allUnits-1,0,-1 do -- search list in reverse
local u = allUnits[i]
if canLeave(u) and math.random(100) > desireToStay(u,method,civ_id) then
desert(u,method,civ_id)
end
end
end
function checkmigrationnow()
local merchant_civ_ids = {} --as:number[]
local allUnits = df.global.world.units.active
for i=0, #allUnits-1 do
local unit = allUnits[i]
if dfhack.units.isSane(unit)
and dfhack.units.isActive(unit)
and not dfhack.units.isOpposedToLife(unit)
and not unit.flags1.tame
then
if unit.flags1.merchant then table.insert(merchant_civ_ids, unit.civ_id) end
end
end
if #merchant_civ_ids == 0 then
checkForDeserters('wild', df.global.plotinfo.main.fortress_entity.entity_links[0].target)
else
for _, civ_id in pairs(merchant_civ_ids) do checkForDeserters('merchant', civ_id) end
end
state.last_cycle_tick = dfhack.world.ReadCurrentTick() + TICKS_PER_YEAR * dfhack.world.ReadCurrentYear()
end
local function event_loop()
if not state.enabled then return end
local current_tick = dfhack.world.ReadCurrentTick() + TICKS_PER_YEAR * dfhack.world.ReadCurrentYear()
if current_tick - state.last_cycle_tick < TICKS_PER_MONTH then
local timeout_ticks = state.last_cycle_tick - current_tick + TICKS_PER_MONTH
dfhack.timeout(timeout_ticks, 'ticks', event_loop)
else
checkmigrationnow()
dfhack.timeout(1, 'months', event_loop)
end
end
dfhack.onStateChange[GLOBAL_KEY] = function(sc)
if sc == SC_MAP_UNLOADED then
state.enabled = false
return
end
if sc ~= SC_MAP_LOADED or df.global.gamemode ~= df.game_mode.DWARF then
return
end
state = get_default_state()
utils.assign(state, dfhack.persistent.getSiteData(GLOBAL_KEY, state))
event_loop()
end
if dfhack_flags.module then
return
end
if df.global.gamemode ~= df.game_mode.DWARF or not dfhack.isMapLoaded() then
dfhack.printerr('emigration needs a loaded fortress map to work')
return
end
local args = {...}
if dfhack_flags and dfhack_flags.enable then
args = {dfhack_flags.enable_state and 'enable' or 'disable'}
end
if args[1] == "enable" then
state.enabled = true
elseif args[1] == "disable" then
state.enabled = false
elseif args[1] == "nobles" then
table.remove(args, 1)
nobles.run(args)
else
print('emigration is ' .. (state.enabled and 'enabled' or 'not enabled'))
return
end
event_loop()
persist_state()