Putnam3145

local df_date={} --lol I really should put this in its own file

df_date.__eq=function(date1,date2)
    return date1.year==date2.year and date1.year_tick==date2.year_tick
end

df_date.__lt=function(date1,date2)
    if date1.year<date2.year then return true end
    if date1.year>date2.year then return false end
    if date1.year==date2.year then

Putnam3145

local gui=require('gui')

local widgets=require('gui.widgets')

local ClickableLabel=defclass(ClickableLabel,widgets.Label)

ClickableLabel.ATTRS.on_click=DEFAULT_NIL

ClickableLabel.ATTRS.on_rclick=DEFAULT_NIL

Putnam3145

FamilyNode=defclass(FamilyNode)

FamilyNode.ATTRS {
    histfig_id = DEFAULT_NIL,
    mother = DEFAULT_NIL,
    father = DEFAULT_NIL,
    spouse = DEFAULT_NIL,
    progenitor = false,
    children = {}
}

Putnam3145

autoShineActivated=false
autoFireActivated=false
autoDamageBoostActivated=false

assistantUI = forms.newform(200, 120, "SMAssist")
autoShineActivated = forms.checkbox(assistantUI,'Auto Shinespark Crouch',1,1)
autoFireActivated = nil --not in there yet
autoDamageBoostActivated = forms.checkbox(assistantUI,'Auto Damage Boost',1,23)
homingMissileActivated = forms.checkbox(assistantUI,'Homing Missile',1,45)

Putnam3145

local function getRaceName()
	return df.creature_raw.find(df.global.ui.race_id).name[1]
end
local function emotionIsNegative(thought)
	return thought.type~=-1 and df.emotion_type.attrs[thought.type].divider>0
end
local function write_gamelog_and_announce(msg,color)
	dfhack.gui.showAnnouncement(msg,color)
	dfhack.gui.writeToGamelog(msg)
end

Putnam3145

local creatures = assert(io.open(dfhack.getDFPath()..'/generated_creatures.txt', 'w'))

local items = assert(io.open(dfhack.getDFPath()..'/generated_items.txt', 'w'))

local materials = assert(io.open(dfhack.getDFPath()..'/generated_materials.txt', 'w'))

local entities = assert(io.open(dfhack.getDFPath()..'/generated_entities.txt', 'w'))
    
local interactions = assert(io.open(dfhack.getDFPath()..'/generated_interactions.txt', 'w'))

Putnam3145

local utils=require('utils')

validArgs = validArgs or utils.invert({
 'pets'
})

local args = utils.processArgs({...}, validArgs)

for k,v in ipairs(df.global.world.buildings.other.COFFIN) do
    if v.owner_id==-1 then

Putnam3145

-- lets babies be adopted by couples with no infants.

local function getSpouseOrLover(unit)
    local lover_unit=df.unit.find(unit.relations.lover_id) or df.unit.find(unit.relations.spouse_id)
    if lover_unit then
        return lover_unit.hist_figure_id
    else
        local hist_fig=df.historical_figure.find(unit.hist_figure_id)
        for k,v in ipairs(hist_fig.histfig_links) do
            if df.histfig_hf_link_spousest:is_instance(v) or df.histfig_hf_link_loverst:is_instance(v) then

Putnam3145

-- Adds emotions to creatures.

local utils=require('utils')

local function addEmotionToUnit(emotions,thought,emotion,subthought,severity)
    if not (type(emotion)=='number') then emotion=df.emotion_type[emotion] end
    if not (type(thought)=='number') then thought=df.unit_thought_type[thought] end
    emotions:insert('#',{new=df.unit_personality.T_emotions,
    type=emotion,
    unk2=1,

Putnam3145

local utils = require('utils')
 
validArgs = utils.invert({
  'all',
  'citizens',
  'named',
  'notStraight',
  'gayOnly',
  'biOnly',
  'straightOnly',