Tweak.mm

   1 #include <substrate.h>
   2
   3 #import <Foundation/NSArray.h>
   4 #import <Foundation/NSNull.h>
   5 #import <CoreGraphics/CGGeometry.h>
   6 #import <SpringBoard/SBButtonBar.h>
   7 #import <SpringBoard/SBIcon.h>
   8
   9 MSHook(float, SBButtonBar$leftMarginForIconRowArray$, SBButtonBar *self, SEL sel, NSArray *row) {
  10     return [row indexOfObject:[NSNull null]] == NSNotFound ? 4 : _SBButtonBar$leftMarginForIconRowArray$(self, sel, row);
  11 }
  12
  13 static int $SBButtonBar$maxIconColumns(SBButtonBar *self, SEL sel) {
  14     return 5;
  15 }
  16
  17 static unsigned $SBButtonBar$count(SBButtonBar *self) {
  18     NSArray *&_iconMatrix(MSHookIvar<NSArray *>(self, "_iconMatrix"));
  19     NSArray *row([_iconMatrix objectAtIndex:0]);
  20     unsigned count(0);
  21     for (id icon in row)
  22         if (icon != [NSNull null])
  23             ++count;
  24     return count;
  25 }
  26
  27 static float $SBButtonBar$topIconPadding(SBButtonBar *self, SEL sel) {
  28     return 313370 + $SBButtonBar$count(self);
  29 }
  30
  31 MSHook(void, SBIcon$setOrigin$, SBIcon *self, SEL sel, CGPoint origin) {
  32     unsigned count(origin.y);
  33     if (count < 313370)
  34         goto set;
  35     count -= 313370;
  36
  37     if (count > 40)
  38         origin.x = 256;
  39     else if (count > 10 || count < 0)
  40         goto set;
  41     else if (count == 5)
  42         origin.x = (origin.x - 4) / 76 * 63 + 4;
  43
  44     origin.y = 11;
  45   set:
  46     _SBIcon$setOrigin$(self, sel, origin);
  47 }
  48
  49 MSHook(CGPoint, SBButtonBar$originForIconAtX$Y$, SBButtonBar *self, SEL sel, int x, int y) {
  50     CGPoint origin;//(_SBButtonBar$originForIconAtX$Y$(self, sel, x, y));
  51     //NSLog(@"f:%u,%u=(%f,%f)", x, y, origin.x, origin.y);
  52     origin.y = 11;
  53
  54     unsigned count($SBButtonBar$count(self)), gap;
  55     unsigned space(320 - 60 * count);
  56
  57     if (count >= 4)
  58         gap = space / (count + 1);
  59     else // I hate people who love Apple.
  60         gap = 16;
  61
  62     origin.x = (space - gap * (count - 1)) / 2 + (60 + gap) * x;
  63
  64     return origin;
  65 }
  66
  67 extern "C" void TweakInitialize() {
  68     Class $SBButtonBar(objc_getClass("SBButtonBar"));
  69     _SBButtonBar$leftMarginForIconRowArray$ = MSHookMessage($SBButtonBar, @selector(leftMarginForIconRowArray:), &$SBButtonBar$leftMarginForIconRowArray$);
  70     MSHookMessage($SBButtonBar, @selector(maxIconColumns), &$SBButtonBar$maxIconColumns);
  71     MSHookMessage($SBButtonBar, @selector(topIconPadding), &$SBButtonBar$topIconPadding);
  72     _SBButtonBar$originForIconAtX$Y$ = MSHookMessage($SBButtonBar, @selector(originForIconAtX:Y:), &$SBButtonBar$originForIconAtX$Y$);
  73
  74     Class $SBIcon(objc_getClass("SBIcon"));
  75     _SBIcon$setOrigin$ = MSHookMessage($SBIcon, @selector(setOrigin:), &$SBIcon$setOrigin$);
  76 }