@@ -53,16 +53,18 @@ - (instancetype)init {
5353 return self;
5454}
5555
56- - (MMTabBarButton *)controlView {
57- return (MMTabBarButton *)[super controlView ];
56+ - (nullable MMTabBarButton *)controlView {
57+ // return (MMTabBarButton *)[super controlView];
58+ return _tabBarButton;
5859}
5960
60- - (void )setControlView : (MMTabBarButton *)aView {
61- [super setControlView: aView];
61+ - (void )setControlView : (nullable NSView *)aView {
62+ _tabBarButton = (MMTabBarButton*) aView;
63+ [super setControlView: aView];
6264}
6365
64- - (MMTabBarView *)tabBarView {
65- return self. controlView .tabBarView ;
66+ - (nullable MMTabBarView *)tabBarView {
67+ return _tabBarButton .tabBarView ;
6668}
6769
6870- (void )calcDrawInfo : (NSRect )aRect {
@@ -183,15 +185,15 @@ - (void)setState:(NSInteger)value {
183185#pragma mark -
184186#pragma mark Progress Indicator Support
185187
186- - (MMProgressIndicator *)indicator {
187- return self.controlView .indicator ;
188+ - (nullable MMProgressIndicator *)indicator {
189+ return self.tabBarButton .indicator ;
188190}
189191
190192#pragma mark -
191193#pragma mark Close Button Support
192194
193- - (MMRolloverButton *)closeButton {
194- return self.controlView .closeButton ;
195+ - (nullable MMRolloverButton *)closeButton {
196+ return self.tabBarButton .closeButton ;
195197}
196198
197199- (NSImage *)closeButtonImageOfType : (MMCloseButtonImageType)type {
@@ -599,7 +601,7 @@ - (NSImage *)_closeButtonImageOfType:(MMCloseButtonImageType)type {
599601- (void )_updateCloseButton {
600602
601603 MMTabBarView *tabBarView = self.tabBarView ;
602- MMTabBarButton *button = self.controlView ;
604+ MMTabBarButton *button = self.tabBarButton ;
603605 MMRolloverButton *closeButton = button.closeButton ;
604606
605607 [self _updateCloseButtonImages ];
@@ -630,7 +632,7 @@ - (void)_updateCloseButton {
630632- (void )_updateIndicator {
631633
632634 MMTabBarView *tabBarView = self.tabBarView ;
633- MMTabBarButton *button = self.controlView ;
635+ MMTabBarButton *button = self.tabBarButton ;
634636 MMProgressIndicator *indicator = button.indicator ;
635637
636638 // adjust visibility and position of process indicator
0 commit comments