261 lines
8.2 KiB
Lua
261 lines
8.2 KiB
Lua
--Spawnunit.lua
|
|
--create unit at pointer. Usage e.g. "spawnunit DWARF 0 Dwarfy"
|
|
--author Warmist, Runrusher?
|
|
--edited by expwnent
|
|
|
|
function findCasteIndex(race_id,casteName)
|
|
local cr=df.creature_raw.find(race_id)
|
|
for casteIndex,caste in ipairs(cr.caste) do
|
|
if caste.caste_id == casteName then
|
|
return casteIndex
|
|
end
|
|
end
|
|
return nil
|
|
end
|
|
function getCaste(race_id,caste_id)
|
|
local cr=df.creature_raw.find(race_id)
|
|
return cr.caste[caste_id]
|
|
end
|
|
function genBodyModifier(body_app_mod)
|
|
local a=math.random(0,#body_app_mod.ranges-2)
|
|
return math.random(body_app_mod.ranges[a],body_app_mod.ranges[a+1])
|
|
end
|
|
function getBodySize(caste,time)
|
|
--todo real body size...
|
|
return caste.body_size_1[#caste.body_size_1-1] --returns last body size
|
|
end
|
|
function genAttribute(array)
|
|
local a=math.random(0,#array-2)
|
|
return math.random(array[a],array[a+1])
|
|
end
|
|
function norm()
|
|
return math.sqrt((-2)*math.log(math.random()))*math.cos(2*math.pi*math.random())
|
|
end
|
|
function normalDistributed(mean,sigma)
|
|
return mean+sigma*norm()
|
|
end
|
|
function clampedNormal(min,median,max)
|
|
local val=normalDistributed(median,math.sqrt(max-min))
|
|
if val<min then return min end
|
|
if val>max then return max end
|
|
return val
|
|
end
|
|
function makeSoul(unit,caste)
|
|
local tmp_soul=df.unit_soul:new()
|
|
tmp_soul.unit_id=unit.id
|
|
tmp_soul.name:assign(unit.name)
|
|
tmp_soul.race=unit.race
|
|
tmp_soul.sex=unit.sex
|
|
tmp_soul.caste=unit.caste
|
|
--todo skills,preferences,traits.
|
|
local attrs=caste.attributes
|
|
for k,v in pairs(attrs.ment_att_range) do
|
|
local max_percent=attrs.ment_att_cap_perc[k]/100
|
|
local cvalue=genAttribute(v)
|
|
tmp_soul.mental_attrs[k]={value=cvalue,max_value=cvalue*max_percent}
|
|
end
|
|
for k,v in pairs(tmp_soul.traits) do
|
|
local min,mean,max
|
|
min=caste.personality.a[k]
|
|
mean=caste.personality.b[k]
|
|
max=caste.personality.c[k]
|
|
tmp_soul.traits[k]=clampedNormal(min,mean,max)
|
|
end
|
|
unit.status.souls:insert("#",tmp_soul)
|
|
unit.status.current_soul=tmp_soul
|
|
end
|
|
function CreateUnit(race_id,caste_id)
|
|
local race=df.creature_raw.find(race_id)
|
|
if race==nil then error("Invalid race_id") end
|
|
local caste=getCaste(race_id,caste_id)
|
|
local unit=df.unit:new()
|
|
unit.race=race_id
|
|
unit.caste=caste_id
|
|
unit.id=df.global.unit_next_id
|
|
df.global.unit_next_id=df.global.unit_next_id+1
|
|
if caste.misc.maxage_max==-1 then
|
|
unit.relations.old_year=-1
|
|
else
|
|
unit.relations.old_year=math.random(caste.misc.maxage_min,caste.misc.maxage_max)
|
|
end
|
|
unit.sex=caste.gender
|
|
local body=unit.body
|
|
body.body_plan=caste.body_info
|
|
local body_part_count=#body.body_plan.body_parts
|
|
local layer_count=#body.body_plan.layer_part
|
|
--components
|
|
unit.relations.birth_year=df.global.cur_year
|
|
--unit.relations.birth_time=??
|
|
|
|
--unit.relations.old_time=?? --TODO add normal age
|
|
local cp=body.components
|
|
cp.body_part_status:resize(body_part_count)
|
|
cp.numbered_masks:resize(#body.body_plan.numbered_masks)
|
|
for num,v in ipairs(body.body_plan.numbered_masks) do
|
|
cp.numbered_masks[num]=v
|
|
end
|
|
|
|
cp.layer_status:resize(layer_count)
|
|
cp.layer_wound_area:resize(layer_count)
|
|
cp.layer_cut_fraction:resize(layer_count)
|
|
cp.layer_dent_fraction:resize(layer_count)
|
|
cp.layer_effect_fraction:resize(layer_count)
|
|
local attrs=caste.attributes
|
|
for k,v in pairs(attrs.phys_att_range) do
|
|
local max_percent=attrs.phys_att_cap_perc[k]/100
|
|
local cvalue=genAttribute(v)
|
|
unit.body.physical_attrs[k]={value=cvalue,max_value=cvalue*max_percent}
|
|
--unit.body.physical_attrs:insert(k,{new=true,max_value=genMaxAttribute(v),value=genAttribute(v)})
|
|
end
|
|
|
|
body.blood_max=getBodySize(caste,0) --TODO normal values
|
|
body.blood_count=body.blood_max
|
|
body.infection_level=0
|
|
unit.status2.body_part_temperature:resize(body_part_count)
|
|
for k,v in pairs(unit.status2.body_part_temperature) do
|
|
unit.status2.body_part_temperature[k]={new=true,whole=10067,fraction=0}
|
|
end
|
|
--------------------
|
|
local stuff=unit.enemy
|
|
stuff.body_part_878:resize(body_part_count) -- all = 3
|
|
stuff.body_part_888:resize(body_part_count) -- all = 3
|
|
stuff.body_part_relsize:resize(body_part_count) -- all =0
|
|
|
|
--TODO add correct sizes. (calculate from age)
|
|
local size=caste.body_size_2[#caste.body_size_2-1]
|
|
body.size_info.size_cur=size
|
|
body.size_info.size_base=size
|
|
body.size_info.area_cur=math.pow(size,0.666)
|
|
body.size_info.area_base=math.pow(size,0.666)
|
|
body.size_info.area_cur=math.pow(size*10000,0.333)
|
|
body.size_info.area_base=math.pow(size*10000,0.333)
|
|
|
|
stuff.were_race=race_id
|
|
stuff.were_caste=caste_id
|
|
stuff.normal_race=race_id
|
|
stuff.normal_caste=caste_id
|
|
stuff.body_part_8a8:resize(body_part_count) -- all = 1
|
|
stuff.body_part_base_ins:resize(body_part_count)
|
|
stuff.body_part_clothing_ins:resize(body_part_count)
|
|
stuff.body_part_8d8:resize(body_part_count)
|
|
unit.recuperation.healing_rate:resize(layer_count)
|
|
--appearance
|
|
|
|
local app=unit.appearance
|
|
app.body_modifiers:resize(#caste.body_appearance_modifiers) --3
|
|
for k,v in pairs(app.body_modifiers) do
|
|
app.body_modifiers[k]=genBodyModifier(caste.body_appearance_modifiers[k])
|
|
end
|
|
app.bp_modifiers:resize(#caste.bp_appearance.modifier_idx) --0
|
|
for k,v in pairs(app.bp_modifiers) do
|
|
app.bp_modifiers[k]=genBodyModifier(caste.bp_appearance.modifiers[caste.bp_appearance.modifier_idx[k]])
|
|
end
|
|
--app.unk_4c8:resize(33)--33
|
|
app.tissue_style:resize(#caste.bp_appearance.style_part_idx)
|
|
app.tissue_style_civ_id:resize(#caste.bp_appearance.style_part_idx)
|
|
app.tissue_style_id:resize(#caste.bp_appearance.style_part_idx)
|
|
app.tissue_style_type:resize(#caste.bp_appearance.style_part_idx)
|
|
app.tissue_length:resize(#caste.bp_appearance.style_part_idx)
|
|
app.genes.appearance:resize(#caste.body_appearance_modifiers+#caste.bp_appearance.modifiers) --3
|
|
app.genes.colors:resize(#caste.color_modifiers*2) --???
|
|
app.colors:resize(#caste.color_modifiers)--3
|
|
|
|
makeSoul(unit,caste)
|
|
|
|
df.global.world.units.all:insert("#",unit)
|
|
df.global.world.units.active:insert("#",unit)
|
|
--todo set weapon bodypart
|
|
|
|
local num_inter=#caste.body_info.interactions
|
|
--used to be anon_5 and anon_6: I guessed at what those were before the df-structures update. It seems to work at least a bit. ~expwnent
|
|
unit.curse.own_interaction:resize(num_inter)
|
|
unit.curse.own_interaction_delay:resize(num_inter)
|
|
return unit
|
|
end
|
|
function findRace(name)
|
|
for k,v in pairs(df.global.world.raws.creatures.all) do
|
|
if v.creature_id==name then
|
|
return k
|
|
end
|
|
end
|
|
qerror("Race:"..name.." not found!")
|
|
end
|
|
function PlaceUnit(raceName,casteName,name,position)
|
|
local pos
|
|
if position.x==-30000 then
|
|
pos = copyall(df.global.cursor)
|
|
else
|
|
pos = position
|
|
end
|
|
if pos.x==-30000 then
|
|
qerror("Spawnunit: specify location or place the cursor where you want the unit to be created.")
|
|
end
|
|
local race=findRace(raceName)
|
|
local caste=findCasteIndex(race,casteName)
|
|
local u=CreateUnit(race,tonumber(caste) or 0)
|
|
u.pos:assign(pos)
|
|
if name then
|
|
u.name.first_name=name
|
|
u.name.has_name=true
|
|
end
|
|
u.civ_id=df.global.ui.civ_id
|
|
|
|
local desig,ocupan=dfhack.maps.getTileFlags(pos)
|
|
ocupan.unit=true
|
|
--createNemesis(u)
|
|
end
|
|
function createFigure(trgunit)
|
|
local hf=df.historical_figure:new()
|
|
hf.id=df.global.hist_figure_next_id
|
|
hf.race=trgunit.race
|
|
hf.caste=trgunit.caste
|
|
df.global.hist_figure_next_id=df.global.hist_figure_next_id+1
|
|
hf.name.first_name=trgunit.name.first_name
|
|
hf.name.has_name=true
|
|
df.global.world.history.figures:insert("#",hf)
|
|
return hf
|
|
end
|
|
function createNemesis(trgunit)
|
|
local id=df.global.nemesis_next_id
|
|
local nem=df.nemesis_record:new()
|
|
nem.id=id
|
|
nem.unit_id=trgunit.id
|
|
nem.unit=trgunit
|
|
nem.flags:resize(1)
|
|
nem.flags[4]=true
|
|
nem.flags[5]=true
|
|
nem.flags[6]=true
|
|
nem.flags[7]=true
|
|
nem.flags[8]=true
|
|
nem.flags[9]=true
|
|
--[[for k=4,8 do
|
|
nem.flags[k]=true
|
|
end]]
|
|
df.global.world.nemesis.all:insert("#",nem)
|
|
df.global.nemesis_next_id=id+1
|
|
trgunit.general_refs:insert("#",{new=df.general_ref_is_nemesisst,nemesis_id=id})
|
|
trgunit.flags1.important_historical_figure=true
|
|
local gen=df.global.world.worldgen
|
|
nem.save_file_id=gen.next_unit_chunk_id;
|
|
gen.next_unit_chunk_id=gen.next_unit_chunk_id+1
|
|
gen.next_unit_chunk_offset=gen.next_unit_chunk_offset+1
|
|
|
|
--[[ local gen=df.global.world.worldgen
|
|
gen.next_unit_chunk_id
|
|
gen.next_unit_chunk_offset
|
|
]]
|
|
nem.figure=createFigure(trgunit)
|
|
end
|
|
|
|
args={...}
|
|
|
|
pos = df.new(df.coord)
|
|
if #args > 3 then
|
|
pos.x = tonumber(args[4]) or -30000
|
|
pos.y = tonumber(args[5]) or -30000
|
|
pos.z = tonumber(args[6]) or -30000
|
|
end
|
|
|
|
PlaceUnit(args[1],args[2],args[3],pos)
|
|
|