java/com/android/dialer/util/DrawableConverter.java

/*
 * Copyright (C) 2012 The Android Open Source Project
 * Copyright (C) 2023 The LineageOS 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.dialer.util;

import android.content.Context;
import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.drawable.BitmapDrawable;
import android.graphics.drawable.Drawable;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.core.graphics.drawable.RoundedBitmapDrawable;
import androidx.core.graphics.drawable.RoundedBitmapDrawableFactory;

import com.android.dialer.common.LogUtil;

/** Provides utilities for bitmaps and drawables. */
public class DrawableConverter {

  private DrawableConverter() {}

  /** Converts the provided drawable to a bitmap using the drawable's intrinsic width and height. */
  @Nullable
  public static Bitmap drawableToBitmap(@Nullable Drawable drawable) {
    return drawableToBitmap(drawable, 0, 0);
  }

  /**
   * Converts the provided drawable to a bitmap with the specified width and height.
   *
   * <p>If both width and height are 0, the drawable's intrinsic width and height are used (but in
   * that case {@link #drawableToBitmap(Drawable)} should be used).
   */
  @Nullable
  public static Bitmap drawableToBitmap(@Nullable Drawable drawable, int width, int height) {
    if (drawable == null) {
      return null;
    }

    Bitmap bitmap;
    if (drawable instanceof BitmapDrawable) {
      bitmap = ((BitmapDrawable) drawable).getBitmap();
    } else {
      if (width > 0 || height > 0) {
        bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
      } else if (drawable.getIntrinsicWidth() <= 0 || drawable.getIntrinsicHeight() <= 0) {
        // Needed for drawables that are just a colour.
        bitmap = Bitmap.createBitmap(1, 1, Bitmap.Config.ARGB_8888);
      } else {
        bitmap =
            Bitmap.createBitmap(
                drawable.getIntrinsicWidth(),
                drawable.getIntrinsicHeight(),
                Bitmap.Config.ARGB_8888);
      }

      LogUtil.i(
          "DrawableConverter.drawableToBitmap",
          "created bitmap with width: %d, height: %d",
          bitmap.getWidth(),
          bitmap.getHeight());

      Canvas canvas = new Canvas(bitmap);
      drawable.setBounds(0, 0, canvas.getWidth(), canvas.getHeight());
      drawable.draw(canvas);
    }
    return bitmap;
  }

  @Nullable
  public static Drawable getRoundedDrawable(
          @NonNull Context context, @Nullable Drawable photo, int width, int height) {
    Bitmap bitmap = drawableToBitmap(photo);
    if (bitmap != null) {
      Bitmap scaledBitmap = Bitmap.createScaledBitmap(bitmap, width, height, false);
      RoundedBitmapDrawable drawable =
          RoundedBitmapDrawableFactory.create(context.getResources(), scaledBitmap);
      drawable.setAntiAlias(true);
      drawable.setCornerRadius(drawable.getIntrinsicHeight() / 2);
      return drawable;
    }
    return null;
  }
}