Fixed a crash when Core Image was available.

[ITKit.git] / ITTableView.m

diff --git a/ITTableView.m b/ITTableView.m
old mode 100755 (executable)
new mode 100644 (file)
index 7cba72b..829b039
--- a/ITTableView.m
+++ b/ITTableView.m
@@ -1,47 +1,50 @@
 #import "ITTableView.h"
 #import "ITTableCornerView.h"
 
-
 @implementation ITTableView
 
 - (id)initWithFrame:(NSRect)frame {
-    self = [super initWithFrame:frame];
-    if (self) {
-        corner = [[ITTableCornerView alloc] initWithFrame:[[self cornerView] frame]];
-        [corner setPullsDown:YES];
-        [self setCornerView:corner];
-    }
-    return self;
+       if ((self = [super initWithFrame:frame])) {
+               [self setCornerView:[[ITTableCornerView alloc] initWithFrame:[[self cornerView] frame]]];
+       }
+       return self;
 }
 
-- (id)initWithCoder:(NSCoder*)coder {
-    self = [super initWithCoder:coder];
-    if (self) {
-        corner = [[ITTableCornerView alloc] initWithFrame:[[self cornerView] frame]];
-        [corner setPullsDown:YES];
-        [self setCornerView:corner];
-    }
-    return self;
+- (id)initWithCoder:(NSCoder *)coder {
+       if ((self = [super initWithCoder:coder])) {
+               [self setCornerView:[[ITTableCornerView alloc] initWithFrame:[[self cornerView] frame]]];
+       }
+       return self;
 }
 
-- (void)encodeWithCoder:(NSCoder*)coder {
-    [super encodeWithCoder:coder];
+- (void)encodeWithCoder:(NSCoder *)coder {
+       [super encodeWithCoder:coder];
 }
 
-- (void)setCornerImage:(NSImage*)image {
-    [corner setImage:image];
+- (void)setCornerImage:(NSImage *)image {
+       if ([_cornerView isKindOfClass:[ITTableCornerView class]]) {
+               [(ITTableCornerView *)_cornerView setImage:image];
+       }
 }
 
-- (NSImage*)cornerImage {
-    return [corner image];
+- (NSImage *)cornerImage {
+       if ([_cornerView isKindOfClass:[ITTableCornerView class]]) {
+               return [(ITTableCornerView *)_cornerView image];
+       }
+       return nil;
 }
 
-- (void)setCornerMenu:(NSMenu*)menu {
-    [corner setMenu:menu];
+- (void)setCornerMenu:(NSMenu *)menu {
+       if ([_cornerView isKindOfClass:[ITTableCornerView class]]) {
+               [(ITTableCornerView *)_cornerView setMenu:menu];
+       }
 }
 
-- (NSMenu*)cornerMenu {
-    return [corner menu];
+- (NSMenu *)cornerMenu {
+       if ([_cornerView isKindOfClass:[ITTableCornerView class]]) {
+               return [(ITTableCornerView *)_cornerView menu];
+       }
+       return nil;
 }
 
-@end
+@end
\ No newline at end of file