ALL IN ONE

--[[
USW Noteskin.lua Version V2
Made for Version Stepmania 5 Preview 4

I am the bone of my noteskin
Arrows are my body, and explosions are my blood
I have created over a thousand noteskins
Unknown to death
Nor known to life
Have withstood pain to create many noteskins
Yet these hands will never hold anything
So as I pray, Unlimited Stepman Works

If you want to know the list of lua commands you can use here is the main page for help http://kki.ajworld.net/lua/sm5/Lua.xml
--]]

local ret = ... or {};

--I always had the problem were I wrote var instead of Var, which really iritated me wondering why a code wouldnt work
--You can edit it out if you want, But I'm just gonna leave it here so incase a person wants to make a noteskin
--And gets the same problem as me were you use var instead of Var, atleast this should make it still work
--But its best to use Var, And make sure to check all your code for little mistakes while you write ;)
local var = Var;

--[[
This is the general redirect table, Not the general redirect code, We use this if we are lazy ;)
Or if we cant a directon of a noteskin use the exact same files as another direction
The most re used direction in this case is "Down"
You can add and remove values if you want, It works for every game type,
If you want an easy way to know how all directions are called go in to stepmaniaoptions and change gametype
Then if you go to key configure look at the names for the keys, They are the same names that are used in the noteskin.lua
--]]
ret.RedirTable =
{
    Up = "Down",
    Down = "Down",
    Left = "Down",
    Right = "Down",
    Center = "Down",
    DownLeft = "Down",
    DownRight = "Down",
    UpLeft = "Down",
    UpRight = "Down",
};

--[[
This is the general redirect code, We use this if we want to redirect parts of noteskins to other parts
Because unlike the ret.RedirTable which redirects everything to the defined direction.
Here we can add redirects for seperate elements which means we can use separate images if we want
--]]
local OldRedir = ret.Redir;
ret.Redir = function(sButton, sElement)
    sButton, sElement = OldRedir(sButton, sElement);

    --[[
    We want to use custom hold/roll per direction, But keep global hold/roll heads and explosions.
    We also want to use later defined images for Upleft and Upright for solo and piu
    For this we added a define if its not Upleft or UpRight.
    The reason we do this is because if we dont the code later on will be ignored because at the end
    We return sButton and sElement which will result in the later code being ignored
    --]]
    if not Var "Button" == "UpLeft" then if not Var "Button" == "UpRight" then
        if not string.find(sElement, "Head") and not string.find(sElement, "Explosion") then
            if string.find(sElement, "Hold") or string.find(sElement, "Roll") then
                return sButton, sElement;
            end
        end
    end end

    --This is were we call the ret.RedirTable and define it as sButton
    --So it get called when we return sButton including the code under here
    sButton = ret.RedirTable[sButton];

    --Making Roll Head/Explosion fallback on Hold
    if  sElement == "Roll Head Inactive"
    then
        sElement = "Hold Head Inactive";
    end

    if  sElement == "Roll Head Active"
    then
        sElement = "Hold Head Active";
    end

    if  sElement == "Roll Explosion"
    then
        sElement = "Hold Explosion";
    end

    --Adding stuff for more directions for diffrent gametypes
    if GAMESTATE:GetCurrentGame():GetName() == "dance" then
        --Making the Upleft and Upright Hold body/BottomCap/TopCap and Explosions use SoloUpLeft and SoloUpRight instead
        --So we can define diffrent images for diffrent gametypes
        if Var "Button" == "UpLeft" and string.find(sElement, "Hold") and not string.find(sElement, "Explosion") then sButton = "SoloUpLeft"; end
        if Var "Button" == "UpRight" and string.find(sElement, "Hold") and not string.find(sElement, "Explosion") then sButton = "SoloUpRight"; end
        if Var "Button" == "UpLeft" and string.find(sElement, "Roll") and not string.find(sElement, "Explosion") then sButton = "SoloUpLeft"; end
        if Var "Button" == "UpRight" and string.find(sElement, "Roll") and not string.find(sElement, "Explosion") then sButton = "SoloUpRight"; end
    end
    if GAMESTATE:GetCurrentGame():GetName() == "pump" then
        --Making Pump it up be rythm color based, Only problem is that in sm player 1 and player 2 noteskins are both defined as player 1
        --Need a way to get the parent but atm it ends up as nil because of no parent
        --So Player 1 and 2 both use Player 1 images for now
        --We also let every direction aside from Center use DownLeftp1
        if GAMESTATE:GetMasterPlayerNumber() == "PlayerNumber_P1" then
            if Var "Button" == "UpLeft" then sButton = "DownLeftp1"; end
            if Var "Button" == "UpRight" then sButton = "DownLeftp1"; end
            if Var "Button" == "DownLeft" then sButton = "DownLeftp1"; end
            if Var "Button" == "DownRight" then sButton = "DownLeftp1"; end
            if Var "Button" == "Center" then sButton = "DownCenterp1"; end
        elseif GAMESTATE:GetMasterPlayerNumber() == "PlayerNumber_P2" then
            if Var "Button" == "UpLeft" then sButton = "Downleftp2"; end
            if Var "Button" == "UpRight" then sButton = "Downleftp2"; end
            if Var "Button" == "DownLeft" then sButton = "DownLeftp2"; end
            if Var "Button" == "DownRight" then sButton = "DownLeftp2"; end
            if Var "Button" == "Center" then sButton = "DownCenterp2"; end
        end
    end

    --Define that every direction uses Tap Mine from Down
    if Var "Element" == "Tap Mine" then sButton = "Down"; end

    return sButton, sElement;
end

--[[
This is the general function code
In here we can define how we want the stuff to act
Which basicly means instead of using a load of lua files for just some effect
We can just use code in here so we need to use less files
Which is nice if you want to save up space ;)
Only problem is that the Hold/Roll parts are written down in the code as sprite files
Which is the original old 3.9 code, Which basicly means that their code doesnt work in here

Also unlike the general redirect code which has sElement and sButton defined in the common noteskin
They need to be defined here manualy which can be done with
local sElement = Var "Element";
local sButton = Var "Button";
To make it easier for everyone I already added them
The reason we use local to define the stuff is for when a code doesnt accept the full code
Like for example if we did string.find(Var "Element", "Down") it wouldnt work
--]]
local OldFunc = ret.Load;
function ret.Load()
    local t = OldFunc();
    local sElement = Var "Element";
    local sButton = Var "Button";

    --Explosion should not be rotated; it calls other actors.
    if Var "Element" == "Explosion" then
        t.BaseRotationZ = nil;
    end


    if GAMESTATE:GetCurrentGame():GetName() == "dance" then
    --We define that we dont want the hold heads for UpLeft and UpRight to be rotated because we rotate them in a lua file
        if Var "Button" == "UpLeft" and string.find(sElement, "Head") then t.BaseRotationZ = nil; end
        if Var "Button" == "UpRight" and string.find(sElement, "Head") then t.BaseRotationZ = nil; end
    end
    if GAMESTATE:GetCurrentGame():GetName() == "pump" then
    --Because the images for Pump it up are using the DownLeft for every direction aside from center, We let it be rotated here
    --Because the rotate table is set up for dance and these are for PIU
        if Var "Button" == "UpLeft" then t.BaseRotationZ = 90; end
        if Var "Button" == "UpRight" then t.BaseRotationZ = 180; end
        if Var "Button" == "DownLeft" then t.BaseRotationZ = nil; end
        if Var "Button" == "DownRight" then t.BaseRotationZ = -90; end
    end
    return t;
end
-- >


--Define which parts of noteskins which we want to rotate
ret.PartsToRotate =
{
    ["Receptor"] = true,
    ["Tap Explosion Bright"] = true,
    ["Tap Explosion Dim"] = true,
    ["Tap Note"] = true,
    ["Tap Fake"] = true,
    ["Tap Lift"] = true,
    ["Tap Addition"] = true,
    ["Hold Explosion"] = true,
    ["Hold Head Active"] = true,
    ["Hold Head Inactive"] = true,
    ["Roll Explosion"] = true,
    ["Roll Head Active"] = true,
    ["Roll Head Inactive"] = true,
};
--Defined the parts to be rotated at which degree
ret.Rotate =
{
    Up = 180,
    Down = 0,
    Left = 90,
    Right = -90,
    UpLeft = 135,
    UpRight = -135,
    Center = 0,
    DownLeft = 0,
    DownRight = -0,
};

--Parts that should be Redirected to _Blank.png
--you can add/remove stuff if you want
ret.Blank =
{
    ["Hold Tail Active"] = true,
    ["Hold Tail Inactive"] = true,
    ["Roll Tail Active"] = true,
    ["Roll Tail Inactive"] = true,
};

--dont forget to close the ret cuz else it wont work ;>
return ret;