/* * Copyright (C) 2019 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.google.android.material.badge; import com.google.android.material.R; import static androidx.annotation.RestrictTo.Scope.LIBRARY_GROUP; import static com.google.android.material.badge.BadgeUtils.updateBadgeBounds; import android.content.Context; import android.content.res.ColorStateList; import android.graphics.Canvas; import android.graphics.ColorFilter; import android.graphics.Paint; import android.graphics.PixelFormat; import android.graphics.Rect; import android.graphics.drawable.Drawable; import android.view.Gravity; import android.view.View; import android.view.ViewGroup; import android.widget.FrameLayout; import android.widget.FrameLayout.LayoutParams; import androidx.annotation.AttrRes; import androidx.annotation.ColorInt; import androidx.annotation.IntDef; import androidx.annotation.NonNull; import androidx.annotation.Nullable; import androidx.annotation.PluralsRes; import androidx.annotation.Px; import androidx.annotation.RestrictTo; import androidx.annotation.StringRes; import androidx.annotation.StyleRes; import androidx.annotation.XmlRes; import androidx.core.view.ViewCompat; import com.google.android.material.internal.TextDrawableHelper; import com.google.android.material.internal.TextDrawableHelper.TextDrawableDelegate; import com.google.android.material.internal.ThemeEnforcement; import com.google.android.material.resources.TextAppearance; import com.google.android.material.shape.MaterialShapeDrawable; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.ref.WeakReference; import java.text.NumberFormat; import java.util.Locale; /** * {@code BadgeDrawable} contains all the layout and draw logic for a badge. * *

You can use {@code BadgeDrawable} to display dynamic information such as a number of pending * requests in a {@link com.google.android.material.bottomnavigation.BottomNavigationView}. To * create an instance of {@code BadgeDrawable}, use {@link #create(Context)} or {@link * #createFromResource(Context, int)}. How to add and display a {@code BadgeDrawable} on top of its * anchor view depends on the API level: * *

For API 18+ (APIs supported by {@link android.view.ViewOverlay}) * *

* *
 * BadgeDrawable badgeDrawable = BadgeDrawable.create(context);
 * badgeDrawable.setVisible(true);
 * BadgeUtils.attachBadgeDrawable(badgeDrawable, anchor);
 * 
* *

For Pre API-18 * *

* * Option 1: {@code BadgeDrawable} will dynamically create and wrap the anchor view in a {@code * FrameLayout}, then insert the {@code FrameLayout} into the anchor view original position in the * view hierarchy. Same syntax as API 18+ * *
 * BadgeDrawable badgeDrawable = BadgeDrawable.create(context);
 * badgeDrawable.setVisible(true);
 * BadgeUtils.attachBadgeDrawable(badgeDrawable, anchor);
 * 
* * Option 2: If you do not want {@code BadgeDrawable} to modify your view hierarchy, you can specify * a {@code FrameLayout} to display the badge instead. * *
 * BadgeDrawable badgeDrawable = BadgeDrawable.create(context);
 * BadgeUtils.attachBadgeDrawable(badgeDrawable, anchor, anchorFrameLayoutParent);
 * 
* *

By default, {@code BadgeDrawable} is aligned to the top and end edges of its anchor view (with * some offsets). Call {@link #setBadgeGravity(int)} to change it to one of the other supported * modes. To adjust the badge's offsets w.r.t. the anchor's center, use {@link * BadgeDrawable#setHorizontalOffset(int)}, {@link BadgeDrawable#setVerticalOffset(int)} * *

Note: This is still under development and may not support the full range of customization * Material Android components generally support (e.g. themed attributes). */ public class BadgeDrawable extends Drawable implements TextDrawableDelegate { /** Position the badge can be set to. */ @IntDef({ TOP_END, TOP_START, BOTTOM_END, BOTTOM_START, }) @Retention(RetentionPolicy.SOURCE) public @interface BadgeGravity {} /** The badge is positioned along the top and end edges of its anchor view */ public static final int TOP_END = Gravity.TOP | Gravity.END; /** The badge is positioned along the top and start edges of its anchor view */ public static final int TOP_START = Gravity.TOP | Gravity.START; /** The badge is positioned along the bottom and end edges of its anchor view */ public static final int BOTTOM_END = Gravity.BOTTOM | Gravity.END; /** The badge is positioned along the bottom and start edges of its anchor view */ public static final int BOTTOM_START = Gravity.BOTTOM | Gravity.START; /** Maximum value of number that can be displayed in a circular badge. */ private static final int MAX_CIRCULAR_BADGE_NUMBER_COUNT = 9; @StyleRes private static final int DEFAULT_STYLE = R.style.Widget_MaterialComponents_Badge; @AttrRes private static final int DEFAULT_THEME_ATTR = R.attr.badgeStyle; /** * If the badge number exceeds the maximum allowed number, append this suffix to the max badge * number and display is as the badge text instead. */ static final String DEFAULT_EXCEED_MAX_BADGE_NUMBER_SUFFIX = "+"; @NonNull private final WeakReference contextRef; @NonNull private final MaterialShapeDrawable shapeDrawable; @NonNull private final TextDrawableHelper textDrawableHelper; @NonNull private final Rect badgeBounds; @NonNull private final BadgeState state; private float badgeCenterX; private float badgeCenterY; private int maxBadgeNumber; private float cornerRadius; private float halfBadgeWidth; private float halfBadgeHeight; // Need to keep a local reference in order to support updating badge gravity. @Nullable private WeakReference anchorViewRef; @Nullable private WeakReference customBadgeParentRef; @NonNull BadgeState.State getSavedState() { return state.getOverridingState(); } /** Creates an instance of {@code BadgeDrawable} with the provided {@link BadgeState.State}. */ @NonNull static BadgeDrawable createFromSavedState( @NonNull Context context, @NonNull BadgeState.State savedState) { return new BadgeDrawable(context, 0, DEFAULT_THEME_ATTR, DEFAULT_STYLE, savedState); } /** Creates an instance of {@code BadgeDrawable} with default values. */ @NonNull public static BadgeDrawable create(@NonNull Context context) { return new BadgeDrawable(context, 0, DEFAULT_THEME_ATTR, DEFAULT_STYLE, null); } /** * Returns a {@code BadgeDrawable} from the given XML resource. All attributes from {@link * R.styleable#Badge} and a custom style attribute are supported. A badge resource * may look like: * *

{@code
   * 
   * }
*/ @NonNull public static BadgeDrawable createFromResource(@NonNull Context context, @XmlRes int id) { return new BadgeDrawable(context, id, DEFAULT_THEME_ATTR, DEFAULT_STYLE, null); } /** * Convenience wrapper method for {@link Drawable#setVisible(boolean, boolean)} with the {@code * restart} parameter hardcoded to false. */ public void setVisible(boolean visible) { state.setVisible(visible); onVisibilityUpdated(); } private void onVisibilityUpdated() { boolean visible = state.isVisible(); setVisible(visible, /* restart= */ false); // When hiding a badge in pre-API 18, invalidate the custom parent in order to trigger a draw // pass to remove this badge from its foreground. if (BadgeUtils.USE_COMPAT_PARENT && getCustomBadgeParent() != null && !visible) { ((ViewGroup) getCustomBadgeParent().getParent()).invalidate(); } } private void restoreState() { onMaxCharacterCountUpdated(); onNumberUpdated(); onAlphaUpdated(); onBackgroundColorUpdated(); onBadgeTextColorUpdated(); onBadgeGravityUpdated(); updateCenterAndBounds(); onVisibilityUpdated(); } private BadgeDrawable( @NonNull Context context, @XmlRes int badgeResId, @AttrRes int defStyleAttr, @StyleRes int defStyleRes, @Nullable BadgeState.State savedState) { this.contextRef = new WeakReference<>(context); ThemeEnforcement.checkMaterialTheme(context); badgeBounds = new Rect(); shapeDrawable = new MaterialShapeDrawable(); textDrawableHelper = new TextDrawableHelper(/* delegate= */ this); textDrawableHelper.getTextPaint().setTextAlign(Paint.Align.CENTER); // TODO(b/209973014): make sure this is right setTextAppearanceResource(R.style.TextAppearance_MaterialComponents_Badge); this.state = new BadgeState(context, badgeResId, defStyleAttr, defStyleRes, savedState); restoreState(); } /** * Calculates and updates this badge's center coordinates based on its anchor's bounds. Internally * also updates this {@code BadgeDrawable BadgeDrawable's} bounds, because they are dependent on * the center coordinates. For pre API-18, coordinates will be calculated relative to {@code * customBadgeParent} because the {@code BadgeDrawable} will be set as the parent's foreground. * * @param anchorView This badge's anchor. * @param customBadgeParent An optional parent view that will set this {@code BadgeDrawable} as * its foreground. * @deprecated use {@link BadgeDrawable#updateBadgeCoordinates(View, FrameLayout)} instead. */ @Deprecated public void updateBadgeCoordinates( @NonNull View anchorView, @Nullable ViewGroup customBadgeParent) { if (customBadgeParent instanceof FrameLayout == false) { throw new IllegalArgumentException("customBadgeParent must be a FrameLayout"); } updateBadgeCoordinates(anchorView, (FrameLayout) customBadgeParent); } /** * Calculates and updates this badge's center coordinates based on its anchor's bounds. Internally * also updates this {@code BadgeDrawable BadgeDrawable's} bounds, because they are dependent on * the center coordinates. * *

For pre API-18, optionally wrap the anchor in a {@code FrameLayout} (if it's not done * already) that will be inserted into the anchor's view hierarchy and calculate the badge's * coordinates the parent {@code FrameLayout} because the {@code BadgeDrawable} will be set as the * parent's foreground. * * @param anchorView This badge's anchor. */ public void updateBadgeCoordinates(@NonNull View anchorView) { updateBadgeCoordinates(anchorView, null); } /** * Calculates and updates this badge's center coordinates based on its anchor's bounds. Internally * also updates this {@code BadgeDrawable BadgeDrawable's} bounds, because they are dependent on * the center coordinates. * *

For pre API-18, if no {@code customBadgeParent} is specified, optionally wrap the anchor in * a {@code FrameLayout} (if it's not done already) that will be inserted into the anchor's view * hierarchy and calculate the badge's coordinates the parent {@code FrameLayout} because the * {@code BadgeDrawable} will be set as the parent's foreground. * * @param anchorView This badge's anchor. * @param customBadgeParent An optional parent view that will set this {@code BadgeDrawable} as * its foreground. */ public void updateBadgeCoordinates( @NonNull View anchorView, @Nullable FrameLayout customBadgeParent) { this.anchorViewRef = new WeakReference<>(anchorView); if (BadgeUtils.USE_COMPAT_PARENT && customBadgeParent == null) { tryWrapAnchorInCompatParent(anchorView); } else { this.customBadgeParentRef = new WeakReference<>(customBadgeParent); } if (!BadgeUtils.USE_COMPAT_PARENT) { updateAnchorParentToNotClip(anchorView); } updateCenterAndBounds(); invalidateSelf(); } /** Returns a {@link FrameLayout} that will set this {@code BadgeDrawable} as its foreground. */ @Nullable public FrameLayout getCustomBadgeParent() { return customBadgeParentRef != null ? customBadgeParentRef.get() : null; } /** * ViewOverlay is not supported below api 18, wrap the anchor view in a {@code FrameLayout} in * order to support scrolling. */ private void tryWrapAnchorInCompatParent(final View anchorView) { ViewGroup anchorViewParent = (ViewGroup) anchorView.getParent(); if ((anchorViewParent != null && anchorViewParent.getId() == R.id.mtrl_anchor_parent) || (customBadgeParentRef != null && customBadgeParentRef.get() == anchorViewParent)) { return; } // Must call this before wrapping the anchor in a FrameLayout. updateAnchorParentToNotClip(anchorView); // Create FrameLayout and configure it to wrap the anchor. final FrameLayout frameLayout = new FrameLayout(anchorView.getContext()); frameLayout.setId(R.id.mtrl_anchor_parent); frameLayout.setClipChildren(false); frameLayout.setClipToPadding(false); frameLayout.setLayoutParams(anchorView.getLayoutParams()); frameLayout.setMinimumWidth(anchorView.getWidth()); frameLayout.setMinimumHeight(anchorView.getHeight()); int anchorIndex = anchorViewParent.indexOfChild(anchorView); anchorViewParent.removeViewAt(anchorIndex); anchorView.setLayoutParams( new LayoutParams(ViewGroup.LayoutParams.MATCH_PARENT, ViewGroup.LayoutParams.MATCH_PARENT)); frameLayout.addView(anchorView); anchorViewParent.addView(frameLayout, anchorIndex); customBadgeParentRef = new WeakReference<>(frameLayout); // Update the badge's coordinates after the FrameLayout has been added to the view hierarchy and // has a size. frameLayout.post( new Runnable() { @Override public void run() { updateBadgeCoordinates(anchorView, frameLayout); } }); } private static void updateAnchorParentToNotClip(View anchorView) { ViewGroup anchorViewParent = (ViewGroup) anchorView.getParent(); anchorViewParent.setClipChildren(false); anchorViewParent.setClipToPadding(false); } /** * Returns this badge's background color. * * @see #setBackgroundColor(int) * @attr ref com.google.android.material.R.styleable#Badge_backgroundColor */ @ColorInt public int getBackgroundColor() { return shapeDrawable.getFillColor().getDefaultColor(); } /** * Sets this badge's background color. * * @param backgroundColor This badge's background color. * @attr ref com.google.android.material.R.styleable#Badge_backgroundColor */ public void setBackgroundColor(@ColorInt int backgroundColor) { state.setBackgroundColor(backgroundColor); onBackgroundColorUpdated(); } private void onBackgroundColorUpdated() { ColorStateList backgroundColorStateList = ColorStateList.valueOf(state.getBackgroundColor()); if (shapeDrawable.getFillColor() != backgroundColorStateList) { shapeDrawable.setFillColor(backgroundColorStateList); invalidateSelf(); } } /** * Returns this badge's text color. * * @see #setBadgeTextColor(int) * @attr ref com.google.android.material.R.styleable#Badge_badgeTextColor */ @ColorInt public int getBadgeTextColor() { return textDrawableHelper.getTextPaint().getColor(); } /** * Sets this badge's text color. * * @param badgeTextColor This badge's text color. * @attr ref com.google.android.material.R.styleable#Badge_badgeTextColor */ public void setBadgeTextColor(@ColorInt int badgeTextColor) { if (textDrawableHelper.getTextPaint().getColor() != badgeTextColor) { state.setBadgeTextColor(badgeTextColor); onBadgeTextColorUpdated(); } } private void onBadgeTextColorUpdated() { textDrawableHelper.getTextPaint().setColor(state.getBadgeTextColor()); invalidateSelf(); } /** Returns the {@link Locale} used to show badge numbers. */ @NonNull public Locale getBadgeNumberLocale() { return state.getNumberLocale(); } /** Sets the {@link Locale} used to show badge numbers. */ public void setBadgeNumberLocale(@NonNull Locale locale) { if (!locale.equals(state.getNumberLocale())) { state.setNumberLocale(locale); invalidateSelf(); } } /** Returns whether this badge will display a number. */ public boolean hasNumber() { return state.hasNumber(); } /** * Returns this badge's number. Only non-negative integer numbers will be returned because the * setter clamps negative values to 0. * *

WARNING: Do not call this method if you are planning to compare to BADGE_NUMBER_NONE * * @see #setNumber(int) * @attr ref com.google.android.material.R.styleable#Badge_number */ public int getNumber() { return hasNumber() ? state.getNumber() : 0; } /** * Sets this badge's number. Only non-negative integer numbers are supported. If the number is * negative, it will be clamped to 0. The specified value will be displayed, unless its number of * digits exceeds {@code maxCharacterCount} in which case a truncated version will be shown. * * @param number This badge's number. * @attr ref com.google.android.material.R.styleable#Badge_number */ public void setNumber(int number) { number = Math.max(0, number); if (this.state.getNumber() != number) { state.setNumber(number); onNumberUpdated(); } } /** Resets any badge number so that a numberless badge will be displayed. */ public void clearNumber() { if (hasNumber()) { state.clearNumber(); onNumberUpdated(); } } private void onNumberUpdated() { textDrawableHelper.setTextWidthDirty(true); updateCenterAndBounds(); invalidateSelf(); } /** * Returns this badge's max character count. * * @see #setMaxCharacterCount(int) * @attr ref com.google.android.material.R.styleable#Badge_maxCharacterCount */ public int getMaxCharacterCount() { return state.getMaxCharacterCount(); } /** * Sets this badge's max character count. * * @param maxCharacterCount This badge's max character count. * @attr ref com.google.android.material.R.styleable#Badge_maxCharacterCount */ public void setMaxCharacterCount(int maxCharacterCount) { if (this.state.getMaxCharacterCount() != maxCharacterCount) { this.state.setMaxCharacterCount(maxCharacterCount); onMaxCharacterCountUpdated(); } } private void onMaxCharacterCountUpdated() { updateMaxBadgeNumber(); textDrawableHelper.setTextWidthDirty(true); updateCenterAndBounds(); invalidateSelf(); } @BadgeGravity public int getBadgeGravity() { return state.getBadgeGravity(); } /** * Sets this badge's gravity with respect to its anchor view. * * @param gravity Constant representing one of 4 possible {@link BadgeGravity} values. */ public void setBadgeGravity(@BadgeGravity int gravity) { if (state.getBadgeGravity() != gravity) { state.setBadgeGravity(gravity); onBadgeGravityUpdated(); } } private void onBadgeGravityUpdated() { if (anchorViewRef != null && anchorViewRef.get() != null) { updateBadgeCoordinates( anchorViewRef.get(), customBadgeParentRef != null ? customBadgeParentRef.get() : null); } } @Override public boolean isStateful() { return false; } @Override public void setColorFilter(ColorFilter colorFilter) { // Intentionally empty. } @Override public int getAlpha() { return state.getAlpha(); } @Override public void setAlpha(int alpha) { state.setAlpha(alpha); onAlphaUpdated(); } private void onAlphaUpdated() { textDrawableHelper.getTextPaint().setAlpha(getAlpha()); invalidateSelf(); } @Override public int getOpacity() { return PixelFormat.TRANSLUCENT; } /** Returns the height at which the badge would like to be laid out. */ @Override public int getIntrinsicHeight() { return badgeBounds.height(); } /** Returns the width at which the badge would like to be laid out. */ @Override public int getIntrinsicWidth() { return badgeBounds.width(); } @Override public void draw(@NonNull Canvas canvas) { Rect bounds = getBounds(); if (bounds.isEmpty() || getAlpha() == 0 || !isVisible()) { return; } shapeDrawable.draw(canvas); if (hasNumber()) { drawText(canvas); } } /** * Implements the TextDrawableHelper.TextDrawableDelegate interface. * * @hide */ @RestrictTo(LIBRARY_GROUP) @Override public void onTextSizeChange() { invalidateSelf(); } @Override public boolean onStateChange(int[] state) { return super.onStateChange(state); } public void setContentDescriptionNumberless(CharSequence charSequence) { state.setContentDescriptionNumberless(charSequence); } public void setContentDescriptionQuantityStringsResource(@PluralsRes int stringsResource) { state.setContentDescriptionQuantityStringsResource(stringsResource); } public void setContentDescriptionExceedsMaxBadgeNumberStringResource( @StringRes int stringsResource) { state.setContentDescriptionExceedsMaxBadgeNumberStringResource(stringsResource); } @Nullable public CharSequence getContentDescription() { if (!isVisible()) { return null; } if (hasNumber()) { if (state.getContentDescriptionQuantityStrings() != 0) { Context context = contextRef.get(); if (context == null) { return null; } if (getNumber() <= maxBadgeNumber) { return context .getResources() .getQuantityString( state.getContentDescriptionQuantityStrings(), getNumber(), getNumber()); } else { return context.getString( state.getContentDescriptionExceedsMaxBadgeNumberStringResource(), maxBadgeNumber); } } else { return null; } } else { return state.getContentDescriptionNumberless(); } } /** * Sets how much (in pixels) to horizontally move this badge towards the center of its anchor. * *

This sets the horizontal offset for badges without text (dots) and with text. * * @param px badge's horizontal offset */ public void setHorizontalOffset(int px) { setHorizontalOffsetWithoutText(px); setHorizontalOffsetWithText(px); } /** * Returns how much (in pixels) this badge is being horizontally offset towards the center of its * anchor. * *

This returns the horizontal offset for badges without text. If offset for badges with text * and without text are different consider using {@link #getHorizontalOffsetWithoutText} or {@link * #getHorizontalOffsetWithText}. */ public int getHorizontalOffset() { return state.getHorizontalOffsetWithoutText(); } /** * Sets how much (in pixels) to horizontally move this badge towards the center of its anchor when * this badge does not have text (is a dot). * * @param px badge's horizontal offset when the badge does not have text */ public void setHorizontalOffsetWithoutText(@Px int px) { state.setHorizontalOffsetWithoutText(px); updateCenterAndBounds(); } /** * Returns how much (in pixels) this badge is being horizontally offset towards the center of its * anchor when this badge does not have text (is a dot). */ @Px public int getHorizontalOffsetWithoutText() { return state.getHorizontalOffsetWithoutText(); } /** * Sets how much (in pixels) to horizontally move this badge towards the center of its anchor when * this badge has text. * * @param px badge's horizontal offset when the badge has text. */ public void setHorizontalOffsetWithText(@Px int px) { state.setHorizontalOffsetWithText(px); updateCenterAndBounds(); } /** * Returns how much (in pixels) this badge is being horizontally offset towards the center of its * anchor when this badge has text. */ @Px public int getHorizontalOffsetWithText() { return state.getHorizontalOffsetWithText(); } /** * Sets how much (in pixels) more (in addition to {@code savedState.horizontalOffset}) to * horizontally move this badge towards the center of its anchor. Currently used to adjust the * placement of badges on toolbar items. */ void setAdditionalHorizontalOffset(int px) { state.setAdditionalHorizontalOffset(px); updateCenterAndBounds(); } int getAdditionalHorizontalOffset() { return state.getAdditionalHorizontalOffset(); } /** * Sets how much (in pixels) to vertically move this badge towards the center of its anchor. * *

This sets the vertical offset for badges both without text (dots) and with text. * * @param px badge's vertical offset */ public void setVerticalOffset(int px) { setVerticalOffsetWithoutText(px); setVerticalOffsetWithText(px); } /** * Returns how much (in pixels) this badge is being vertically moved towards the center of its * anchor. * *

This returns the vertical offset for badges without text. If offset for badges with text and * without text are different consider using {@link #getVerticalOffsetWithoutText} or {@link * #getVerticalOffsetWithText}. */ public int getVerticalOffset() { return state.getVerticalOffsetWithoutText(); } /** * Sets how much (in pixels) to vertically move this badge towards the center of its anchor when * this badge does not have text (is a dot). * * @param px badge's vertical offset when the badge does not have text */ public void setVerticalOffsetWithoutText(@Px int px) { state.setVerticalOffsetWithoutText(px); updateCenterAndBounds(); } /** * Returns how much (in pixels) this badge is being vertically offset towards the center of its * anchor when this badge does not have text (is a dot). */ @Px public int getVerticalOffsetWithoutText() { return state.getVerticalOffsetWithoutText(); } /** * Sets how much (in pixels) to vertically move this badge towards the center of its anchor when * this badge has text. * * @param px badge's vertical offset when the badge has text. */ public void setVerticalOffsetWithText(@Px int px) { state.setVerticalOffsetWithText(px); updateCenterAndBounds(); } /** * Returns how much (in pixels) this badge is being vertically moved towards the center of its * anchor when the badge has text. */ @Px public int getVerticalOffsetWithText() { return state.getVerticalOffsetWithText(); } /** * Sets how much (in pixels) more (in addition to {@code savedState.verticalOffset}) to vertically * move this badge towards the center of its anchor. Currently used to adjust the placement of * badges on toolbar items. */ void setAdditionalVerticalOffset(@Px int px) { state.setAdditionalVerticalOffset(px); updateCenterAndBounds(); } @Px int getAdditionalVerticalOffset() { return state.getAdditionalVerticalOffset(); } private void setTextAppearanceResource(@StyleRes int id) { Context context = contextRef.get(); if (context == null) { return; } setTextAppearance(new TextAppearance(context, id)); } private void setTextAppearance(@Nullable TextAppearance textAppearance) { if (textDrawableHelper.getTextAppearance() == textAppearance) { return; } Context context = contextRef.get(); if (context == null) { return; } textDrawableHelper.setTextAppearance(textAppearance, context); updateCenterAndBounds(); } private void updateCenterAndBounds() { Context context = contextRef.get(); View anchorView = anchorViewRef != null ? anchorViewRef.get() : null; if (context == null || anchorView == null) { return; } Rect tmpRect = new Rect(); tmpRect.set(badgeBounds); Rect anchorRect = new Rect(); // Retrieves the visible bounds of the anchor view. anchorView.getDrawingRect(anchorRect); ViewGroup customBadgeParent = customBadgeParentRef != null ? customBadgeParentRef.get() : null; if (customBadgeParent != null || BadgeUtils.USE_COMPAT_PARENT) { // Calculates coordinates relative to the parent. ViewGroup viewGroup = customBadgeParent == null ? (ViewGroup) anchorView.getParent() : customBadgeParent; viewGroup.offsetDescendantRectToMyCoords(anchorView, anchorRect); } calculateCenterAndBounds(context, anchorRect, anchorView); updateBadgeBounds(badgeBounds, badgeCenterX, badgeCenterY, halfBadgeWidth, halfBadgeHeight); shapeDrawable.setCornerSize(cornerRadius); if (!tmpRect.equals(badgeBounds)) { shapeDrawable.setBounds(badgeBounds); } } private int getTotalVerticalOffsetForState() { int vOffset = hasNumber() ? state.getVerticalOffsetWithText() : state.getVerticalOffsetWithoutText(); return vOffset + state.getAdditionalVerticalOffset(); } private int getTotalHorizontalOffsetForState() { int hOffset = hasNumber() ? state.getHorizontalOffsetWithText() : state.getHorizontalOffsetWithoutText(); return hOffset + state.getAdditionalHorizontalOffset(); } private void calculateCenterAndBounds( @NonNull Context context, @NonNull Rect anchorRect, @NonNull View anchorView) { int totalVerticalOffset = getTotalVerticalOffsetForState(); switch (state.getBadgeGravity()) { case BOTTOM_END: case BOTTOM_START: badgeCenterY = anchorRect.bottom - totalVerticalOffset; break; case TOP_END: case TOP_START: default: badgeCenterY = anchorRect.top + totalVerticalOffset; break; } if (getNumber() <= MAX_CIRCULAR_BADGE_NUMBER_COUNT) { cornerRadius = !hasNumber() ? state.badgeRadius : state.badgeWithTextRadius; halfBadgeHeight = cornerRadius; halfBadgeWidth = cornerRadius; } else { cornerRadius = state.badgeWithTextRadius; halfBadgeHeight = cornerRadius; String badgeText = getBadgeText(); halfBadgeWidth = textDrawableHelper.getTextWidth(badgeText) / 2f + state.badgeWidePadding; } int inset = context .getResources() .getDimensionPixelSize( hasNumber() ? R.dimen.mtrl_badge_text_horizontal_edge_offset : R.dimen.mtrl_badge_horizontal_edge_offset); int totalHorizontalOffset = getTotalHorizontalOffsetForState(); // Update the centerX based on the badge width and 'inset' from start or end boundary of anchor. switch (state.getBadgeGravity()) { case BOTTOM_START: case TOP_START: badgeCenterX = ViewCompat.getLayoutDirection(anchorView) == View.LAYOUT_DIRECTION_LTR ? anchorRect.left - halfBadgeWidth + inset + totalHorizontalOffset : anchorRect.right + halfBadgeWidth - inset - totalHorizontalOffset; break; case BOTTOM_END: case TOP_END: default: badgeCenterX = ViewCompat.getLayoutDirection(anchorView) == View.LAYOUT_DIRECTION_LTR ? anchorRect.right + halfBadgeWidth - inset - totalHorizontalOffset : anchorRect.left - halfBadgeWidth + inset + totalHorizontalOffset; break; } } private void drawText(Canvas canvas) { Rect textBounds = new Rect(); String badgeText = getBadgeText(); textDrawableHelper.getTextPaint().getTextBounds(badgeText, 0, badgeText.length(), textBounds); canvas.drawText( badgeText, badgeCenterX, badgeCenterY + textBounds.height() / 2, textDrawableHelper.getTextPaint()); } @NonNull private String getBadgeText() { // If number exceeds max count, show badgeMaxCount+ instead of the number. if (getNumber() <= maxBadgeNumber) { return NumberFormat.getInstance(state.getNumberLocale()).format(getNumber()); } else { Context context = contextRef.get(); if (context == null) { return ""; } return String.format( state.getNumberLocale(), context.getString(R.string.mtrl_exceed_max_badge_number_suffix), maxBadgeNumber, DEFAULT_EXCEED_MAX_BADGE_NUMBER_SUFFIX); } } private void updateMaxBadgeNumber() { maxBadgeNumber = (int) Math.pow(10.0d, (double) getMaxCharacterCount() - 1) - 1; } }