packages/SystemUI/src/com/android/systemui/statusbar/policy/KeyButtonDrawable.java

/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License
 */

package com.android.systemui.statusbar.policy;

import android.annotation.Nullable;
import android.content.Context;
import android.content.res.Resources;
import android.graphics.Color;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.LayerDrawable;
import android.util.FloatProperty;
import android.view.Gravity;

import com.android.systemui.R;
import com.android.systemui.statusbar.phone.ShadowKeyDrawable;

/**
 * Drawable for {@link KeyButtonView}s which contains an asset for both normal mode and light
 * navigation bar mode.
 */
public class KeyButtonDrawable extends LayerDrawable {

    public static final FloatProperty<KeyButtonDrawable> KEY_DRAWABLE_ROTATE =
        new FloatProperty<KeyButtonDrawable>("KeyButtonRotation") {
            @Override
            public void setValue(KeyButtonDrawable drawable, float degree) {
                drawable.setRotation(degree);
            }

            @Override
            public Float get(KeyButtonDrawable drawable) {
                return drawable.getRotation();
            }
        };

    public static final FloatProperty<KeyButtonDrawable> KEY_DRAWABLE_TRANSLATE_Y =
        new FloatProperty<KeyButtonDrawable>("KeyButtonTranslateY") {
            @Override
            public void setValue(KeyButtonDrawable drawable, float y) {
                drawable.setTranslationY(y);
            }

            @Override
            public Float get(KeyButtonDrawable drawable) {
                return drawable.getTranslationY();
            }
        };

    private final boolean mHasDarkDrawable;

    public static KeyButtonDrawable create(Context lightContext, Drawable lightDrawable,
            @Nullable Drawable darkDrawable, boolean hasShadow) {
        if (darkDrawable != null) {
            ShadowKeyDrawable light = new ShadowKeyDrawable(lightDrawable.mutate());
            ShadowKeyDrawable dark = new ShadowKeyDrawable(darkDrawable.mutate());
            if (hasShadow) {
                // Only apply the shadow on the light drawable
                Resources res = lightContext.getResources();
                int offsetX = res.getDimensionPixelSize(R.dimen.nav_key_button_shadow_offset_x);
                int offsetY = res.getDimensionPixelSize(R.dimen.nav_key_button_shadow_offset_y);
                int radius = res.getDimensionPixelSize(R.dimen.nav_key_button_shadow_radius);
                int color = lightContext.getColor(R.color.nav_key_button_shadow_color);
                light.setShadowProperties(offsetX, offsetY, radius, color);
            }
            return new KeyButtonDrawable(new Drawable[] { light, dark });
        } else {
            return new KeyButtonDrawable(new Drawable[] {
                    new ShadowKeyDrawable(lightDrawable.mutate()) });
        }
    }

    protected KeyButtonDrawable(Drawable[] drawables) {
        super(drawables);
        for (int i = 0; i < drawables.length; i++) {
            setLayerGravity(i, Gravity.CENTER);
        }
        mutate();
        mHasDarkDrawable = drawables.length > 1;
        setDarkIntensity(0f);
    }

    public void setDarkIntensity(float intensity) {
        if (!mHasDarkDrawable) {
            return;
        }
        getDrawable(0).setAlpha((int) ((1 - intensity) * 255f));
        getDrawable(1).setAlpha((int) (intensity * 255f));
        invalidateSelf();
    }

    public void setRotation(float degrees) {
        if (getDrawable(0) instanceof ShadowKeyDrawable) {
            ((ShadowKeyDrawable) getDrawable(0)).setRotation(degrees);
        }
        if (mHasDarkDrawable && getDrawable(1) instanceof ShadowKeyDrawable) {
            ((ShadowKeyDrawable) getDrawable(1)).setRotation(degrees);
        }
    }

    public void setTranslationY(float y) {
        if (getDrawable(0) instanceof ShadowKeyDrawable) {
            ((ShadowKeyDrawable) getDrawable(0)).setTranslationY(y);
        }
        if (mHasDarkDrawable && getDrawable(1) instanceof ShadowKeyDrawable) {
            ((ShadowKeyDrawable) getDrawable(1)).setTranslationY(y);
        }
    }

    public float getRotation() {
        if (getDrawable(0) instanceof ShadowKeyDrawable) {
            return ((ShadowKeyDrawable) getDrawable(0)).getRotation();
        }
        if (mHasDarkDrawable && getDrawable(1) instanceof ShadowKeyDrawable) {
            return ((ShadowKeyDrawable) getDrawable(1)).getRotation();
        }
        return 0;
    }

    public float getTranslationY() {
        if (getDrawable(0) instanceof ShadowKeyDrawable) {
            return ((ShadowKeyDrawable) getDrawable(0)).getTranslationY();
        }
        if (mHasDarkDrawable && getDrawable(1) instanceof ShadowKeyDrawable) {
            return ((ShadowKeyDrawable) getDrawable(1)).getTranslationY();
        }
        return 0;
    }
}