Fixed conflict with FiveIRows.

[fiveicondock.git] / Tweak.mm

#include <substrate.h>

#import <Foundation/NSArray.h>
#import <Foundation/NSNull.h>
#import <CoreGraphics/CGGeometry.h>
#import <SpringBoard/SBButtonBar.h>
#import <SpringBoard/SBIcon.h>

MSHook(float, SBButtonBar$leftMarginForIconRowArray$, SBButtonBar *self, SEL sel, NSArray *row) {
    return [row indexOfObject:[NSNull null]] == NSNotFound ? 4 : _SBButtonBar$leftMarginForIconRowArray$(self, sel, row);
}

static int $SBButtonBar$maxIconColumns(SBButtonBar *self, SEL sel) {
    return 5;
}

static unsigned $SBButtonBar$count(SBButtonBar *self) {
    NSArray *&_iconMatrix(MSHookIvar<NSArray *>(self, "_iconMatrix"));
    NSArray *row([_iconMatrix objectAtIndex:0]);
    unsigned count(0);
    for (id icon in row)
        if (icon != [NSNull null])
            ++count;
    return count;
}

static float $SBButtonBar$topIconPadding(SBButtonBar *self, SEL sel) {
    return $SBButtonBar$count(self);
}

MSHook(void, SBIcon$setOrigin$, SBIcon *self, SEL sel, CGPoint origin) {
    unsigned count(origin.y);
    //NSLog(@"C:%u=(%f,%f)", count, origin.x, origin.y);

    if (count < 70 && count > 40)
        origin.x = 256;
    else if (count > 10 || count < 0)
        goto set;
    else if (count == 5)
        origin.x = (origin.x - 4) / 76 * 63 + 4;

    origin.y = 11;
  set:
    _SBIcon$setOrigin$(self, sel, origin);
}

MSHook(CGPoint, SBButtonBar$originForIconAtX$Y$, SBButtonBar *self, SEL sel, int x, int y) {
    CGPoint origin;//(_SBButtonBar$originForIconAtX$Y$(self, sel, x, y));
    //NSLog(@"f:%u,%u=(%f,%f)", x, y, origin.x, origin.y);
    origin.y = 11;

    unsigned count($SBButtonBar$count(self)), gap;
    unsigned space(320 - 60 * count);

    if (count >= 4)
        gap = space / (count + 1);
    else // I hate people who love Apple.
        gap = 16;

    origin.x = (space - gap * (count - 1)) / 2 + (60 + gap) * x;

    return origin;
}

extern "C" void TweakInitialize() {
    Class $SBButtonBar(objc_getClass("SBButtonBar"));
    _SBButtonBar$leftMarginForIconRowArray$ = MSHookMessage($SBButtonBar, @selector(leftMarginForIconRowArray:), &$SBButtonBar$leftMarginForIconRowArray$);
    MSHookMessage($SBButtonBar, @selector(maxIconColumns), &$SBButtonBar$maxIconColumns);
    MSHookMessage($SBButtonBar, @selector(topIconPadding), &$SBButtonBar$topIconPadding);
    _SBButtonBar$originForIconAtX$Y$ = MSHookMessage($SBButtonBar, @selector(originForIconAtX:Y:), &$SBButtonBar$originForIconAtX$Y$);

    Class $SBIcon(objc_getClass("SBIcon"));
    _SBIcon$setOrigin$ = MSHookMessage($SBIcon, @selector(setOrigin:), &$SBIcon$setOrigin$);
}