diff --git a/app/src/main/java/protect/card_locker/LoyaltyCardViewActivity.java b/app/src/main/java/protect/card_locker/LoyaltyCardViewActivity.java index 49166d7b9..677abd902 100644 --- a/app/src/main/java/protect/card_locker/LoyaltyCardViewActivity.java +++ b/app/src/main/java/protect/card_locker/LoyaltyCardViewActivity.java @@ -227,15 +227,11 @@ public class LoyaltyCardViewActivity extends AppCompatActivity case R.id.action_lock_unlock: if(rotationEnabled) { - item.setIcon(R.drawable.ic_lock_outline_white_24dp); - item.setTitle(R.string.unlockScreen); - setRequestedOrientation(ActivityInfo.SCREEN_ORIENTATION_NOSENSOR); + setOrientatonLock(item, true); } else { - item.setIcon(R.drawable.ic_lock_open_white_24dp); - item.setTitle(R.string.lockScreen); - setRequestedOrientation(ActivityInfo.SCREEN_ORIENTATION_SENSOR); + setOrientatonLock(item, false); } rotationEnabled = !rotationEnabled; return true; @@ -243,4 +239,20 @@ public class LoyaltyCardViewActivity extends AppCompatActivity return super.onOptionsItemSelected(item); } + + private void setOrientatonLock(MenuItem item, boolean lock) + { + if(lock) + { + item.setIcon(R.drawable.ic_lock_outline_white_24dp); + item.setTitle(R.string.unlockScreen); + setRequestedOrientation(ActivityInfo.SCREEN_ORIENTATION_NOSENSOR); + } + else + { + item.setIcon(R.drawable.ic_lock_open_white_24dp); + item.setTitle(R.string.lockScreen); + setRequestedOrientation(ActivityInfo.SCREEN_ORIENTATION_SENSOR); + } + } } \ No newline at end of file