diff --git a/badgeviewlib/src/main/java/q/rorbin/badgeview/QBadgeView.java b/badgeviewlib/src/main/java/q/rorbin/badgeview/QBadgeView.java index c96a30f..ee6fb22 100644 --- a/badgeviewlib/src/main/java/q/rorbin/badgeview/QBadgeView.java +++ b/badgeviewlib/src/main/java/q/rorbin/badgeview/QBadgeView.java @@ -155,7 +155,7 @@ public Badge bindTarget(final View targetView) { ViewGroup.LayoutParams targetParams = targetView.getLayoutParams(); targetContainer.removeView(targetView); final BadgeContainer badgeContainer = new BadgeContainer(getContext()); - if(targetContainer instanceof RelativeLayout){ + if (targetContainer instanceof RelativeLayout) { badgeContainer.setId(targetView.getId()); } targetContainer.addView(badgeContainer, index, targetParams); @@ -832,7 +832,7 @@ private class BadgeContainer extends ViewGroup { @Override protected void dispatchRestoreInstanceState(SparseArray container) { - if(!(getParent() instanceof RelativeLayout)){ + if (!(getParent() instanceof RelativeLayout)) { super.dispatchRestoreInstanceState(container); } } @@ -871,5 +871,21 @@ protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) { setMeasuredDimension(targetView.getMeasuredWidth(), targetView.getMeasuredHeight()); } } + + protected View findViewTraversal(int id) { + + final int len = getChildCount(); + + for (int i = 0; i < len; i++) { + View v = getChildAt(i); + v = v.findViewById(id); + + if (v != null) { + return v; + } + } + + return null; + } } }