Replacing a function

[armacham01]

How would I go about changing a particular function in a lua file, for a Sim mod?

I've made a mod that changes a blueprint by adding to it/overwriting part of it, without overwriting the entire file. but I'm curious what the code looks like for overwriting a single function. OR, can I not do that?

Can I replace just one function from units.lua or would I need to overwrite the entire "units.lua" file?

I would prefer to overwrite just a single function, if that is possible. It would make my mod probably easier to maintain going forward. Unless you think there is an advantage to overwriting the entire file.

Also, if you could point me to an example of a mod that does this, I could download that and look at it.

Thanks.

[Uveso]

Hello armacham01,

here is a simple example how to hook a function inside a defined class:
https://github.com/Uveso/AI-Uveso/blob/ ... anager.lua

First you need to make a copy of the original class and save it. I use the name "TheOldEngineerManager" for the copy of the original class.

Code: Select all

TheOldEngineerManager = EngineerManager

Then you need to create a new cass and use the orginal class as base.
Because i use the original name "EngineerManager", i will overwrite the original class.
The original class is still present inside "TheOldEngineerManage", so i use this as base for the new one.

Code: Select all

EngineerManager = Class(TheOldEngineerManager)

Now you can overwrite a function inside the "EngineerManager" class
In this case the function is called "LowMass"

Code: Select all

    LowMass = function(self)
        -- Only use this with AI-Uveso
        if not self.Brain.Uveso then
            return TheOldEngineerManager.LowMass(self)
        end
    end,


This function checks if AI-Uveso is active and in case its not, the original function LowMass will be called.
Remember we stored the original class inside "TheOldEngineerManager", so we can call
the original LowMass function like this:

Code: Select all

TheOldEngineerManager.LowMass(self)

I hope this helps a bit.

[armacham01]

Thank you. With your advice, I think it is working now.

[Uveso]

You are welcome.

[BlackYps]

This seems to work only if the hooked funktion is not defined in the same class.
I want to get rid of the additions that the StructureUnit class makes to OnDestroy

When using your method and writing

Code: Select all

local oldStructureUnit = StructureUnit
StructureUnit = Class(oldStructureUnit) {
   OnDestroy = function(self)
      Unit.OnDestroy(self)
   end,
}

I get the error:

Code: Select all

field 'OnDestroy' is ambiguous in class definition

It seems that it doesn't overwrite the new one with the old one but uses both and then the game is confused abot which one to take.
What am I doing wrong?

[Uveso]

Code: Select all

local oldStructureUnit = StructureUnit
StructureUnit = Class(oldStructureUnit) {
   OnDestroy = function(self)
      Unit.OnDestroy(self)
   end,
}


Looks good so far; you copied the original function to "oldStructureUnit".
But now you try to access "oldStructureUnit" only with "unit" to get the OnDestroy function.
You need to use the copy of the original:

Code: Select all

oldStructureUnit.OnDestroy(self)

so the function should look like this:

Code: Select all

local oldStructureUnit = StructureUnit
StructureUnit = Class(oldStructureUnit) {
   OnDestroy = function(self)
      oldStructureUnit.OnDestroy(self)
   end,
}


[Uveso]

Well i checked hooking the OnDestroy StructureUnit class and its working.
Just to be sure;
create a file named:

Code: Select all

\Mods\YOURMODNAME\hook\lua\defaultunits.lua

and put this into the empty file:

Code: Select all

local OLDStructureUnit = StructureUnit
StructureUnit = Class(OLDStructureUnit) {
    OnDestroy = function(self)
        LOG('***---*** StructureUnit OnDestroy HOOK')
        Unit.OnDestroy(self)
        local orient = self.TarmacBag.Orientation
        local currentBP = self.TarmacBag.CurrentBP
        self:DestroyTarmac()
        self:CreateTarmac(true, true, true, orient, currentBP, currentBP.DeathLifetime or 300)
    end,
}


Does this work on your side ?

[BlackYps]

Yes, it works!

There must be a reason why it doesn't work in the other mod...
I will try to find it. But I don't really have an idea what the reason could be.

[Uveso]

In case you got stucked there, upload the mod and i will take a look if you want.

[BlackYps]

I found the root of the problem:
Hooking the structureunit class works well. The problem occurs when I try to propagate the change down to the subclasses.

Apparently this is not the way to go:

Code: Select all

local oldFactoryUnit = FactoryUnit
FactoryUnit = Class(StructureUnit, oldFactoryUnit) {}