Allow multiple states when setting swipe gesture (#55).

alikaragoz · Jan 22, 2014 · a59c36c · a59c36c
1 parent 7b4bd08
commit a59c36c
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 37 deletions.
diff --git a/MCSwipeTableViewCell/MCSwipeTableViewCell.h b/MCSwipeTableViewCell/MCSwipeTableViewCell.h
@@ -11,22 +11,22 @@
 @class MCSwipeTableViewCell;
 
 /** Describes the state that has been triggered by the user. */
-typedef NS_ENUM(NSUInteger, MCSwipeTableViewCellState) {
+typedef NS_OPTIONS(NSUInteger, MCSwipeTableViewCellState) {
 
     /** No state has been triggered. */
     MCSwipeTableViewCellStateNone = 0,
 
     /** 1st state triggered during a Left -> Right swipe. */
-    MCSwipeTableViewCellState1,
+    MCSwipeTableViewCellState1 = (1 << 0),
 
     /** 2nd state triggered during a Left -> Right swipe. */
-    MCSwipeTableViewCellState2,
+    MCSwipeTableViewCellState2 = (1 << 1),
 
     /** 1st state triggered during a Right -> Left swipe. */
-    MCSwipeTableViewCellState3,
+    MCSwipeTableViewCellState3 = (1 << 2),
 
     /** 2nd state triggered during a Right -> Left swipe. */
-    MCSwipeTableViewCellState4
+    MCSwipeTableViewCellState4 = (1 << 3)
 };
 
 /** Describes the mode used during a swipe */

diff --git a/MCSwipeTableViewCell/MCSwipeTableViewCell.m b/MCSwipeTableViewCell/MCSwipeTableViewCell.m
@@ -222,40 +222,33 @@ - (void)setSwipeGestureWithView:(UIView *)view
     NSParameterAssert(color);
 
     // Depending on the state we assign the attributes
-    switch (state) {
-        case MCSwipeTableViewCellState1: {
-            _completionBlock1 = completionBlock;
-            _view1 = view;
-            _color1 = color;
-            _modeForState1 = mode;
-
-        } break;
-
-        case MCSwipeTableViewCellState2: {
-            _completionBlock2 = completionBlock;
-            _view2 = view;
-            _color2 = color;
-            _modeForState2 = mode;
-        } break;
-
-        case MCSwipeTableViewCellState3: {
-            _completionBlock3 = completionBlock;
-            _view3 = view;
-            _color3 = color;
-            _modeForState3 = mode;
-        } break;
-
-        case MCSwipeTableViewCellState4: {
-            _completionBlock4 = completionBlock;
-            _view4 = view;
-            _color4 = color;
-            _modeForState4 = mode;
-        } break;
-
-        default:
-            break;
+    if ((state & MCSwipeTableViewCellState1) == MCSwipeTableViewCellState1) {
+        _completionBlock1 = completionBlock;
+        _view1 = view;
+        _color1 = color;
+        _modeForState1 = mode;
     }
 
+    if ((state & MCSwipeTableViewCellState2) == MCSwipeTableViewCellState2) {
+        _completionBlock2 = completionBlock;
+        _view2 = view;
+        _color2 = color;
+        _modeForState2 = mode;
+    }
+
+    if ((state & MCSwipeTableViewCellState3) == MCSwipeTableViewCellState3) {
+        _completionBlock3 = completionBlock;
+        _view3 = view;
+        _color3 = color;
+        _modeForState3 = mode;
+    }
+
+    if ((state & MCSwipeTableViewCellState4) == MCSwipeTableViewCellState4) {
+        _completionBlock4 = completionBlock;
+        _view4 = view;
+        _color4 = color;
+        _modeForState4 = mode;
+    }
 }
 
 #pragma mark - Handle Gestures