1 /*
2  * Copyright (C) 2021 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package com.android.car.rotaryplayground;
18 
19 import android.content.Context;
20 import android.util.AttributeSet;
21 import android.widget.Button;
22 
23 import androidx.annotation.Nullable;
24 
25 public class CustomButton extends Button {
26 
27     private static final int[] STATE_ROTARY_ENABLED = {R.attr.state_rotary_enabled};
28 
29     private boolean mRotaryEnabled = true;
30 
CustomButton(Context context)31     public CustomButton(Context context) {
32         super(context);
33     }
34 
CustomButton(Context context, @Nullable AttributeSet attrs)35     public CustomButton(Context context, @Nullable AttributeSet attrs) {
36         super(context, attrs);
37     }
38 
CustomButton(Context context, @Nullable AttributeSet attrs, int defStyleAttr)39     public CustomButton(Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
40         super(context, attrs, defStyleAttr);
41     }
42 
CustomButton(Context context, @Nullable AttributeSet attrs, int defStyleAttr, int defStyleRes)43     public CustomButton(Context context, @Nullable AttributeSet attrs, int defStyleAttr,
44             int defStyleRes) {
45         super(context, attrs, defStyleAttr, defStyleRes);
46     }
47 
setRotaryEnabled(boolean enabled)48     public void setRotaryEnabled(boolean enabled) {
49         mRotaryEnabled = enabled;
50         refreshDrawableState();
51     }
52 
isRotaryEnabled()53     public boolean isRotaryEnabled() {
54         return mRotaryEnabled;
55     }
56 
57     @Override
onCreateDrawableState(int extraSpace)58     protected int[] onCreateDrawableState(int extraSpace) {
59         final int[] drawableState = super.onCreateDrawableState(extraSpace + 1);
60         if (mRotaryEnabled) {
61             mergeDrawableStates(drawableState, STATE_ROTARY_ENABLED);
62         }
63         return drawableState;
64     }
65 }
66