- Fix initial expand.

iammert · Aug 22, 2017 · 5a7b1ed · 5a7b1ed
1 parent a5b2f05
commit 5a7b1ed
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/app/src/main/java/iammert/com/expandablelayout/MainActivity.java b/app/src/main/java/iammert/com/expandablelayout/MainActivity.java
@@ -66,6 +66,7 @@ public Section<FruitCategory, Fruit> getSection() {
         section.children.add(fruit3);
         section.children.add(fruit4);
         section.children.add(fruit5);
+        section.expanded = true;
         return section;
     }
 }
diff --git a/expandablelib/src/main/java/iammert/com/expandablelib/ExpandableLayout.java b/expandablelib/src/main/java/iammert/com/expandablelib/ExpandableLayout.java
@@ -114,6 +114,9 @@ public <P, C> void addChild(P parent, C child) {
         }
         if (parentIndex != NO_INDEX) {
             notifyItemAdded(parentIndex, child);
+            if (sections.get(parentIndex).expanded) {
+                expand(parent);
+            }
         }
     }
 
@@ -129,6 +132,9 @@ public <P, C> void addChildren(P parent, List<C> children) {
         }
         if (parentIndex != NO_INDEX) {
             notifyItemAdded(parentIndex, children);
+            if (sections.get(parentIndex).expanded) {
+                expand(parent);
+            }
         }
     }
 
@@ -187,7 +193,7 @@ public void onClick(View view) {
         addView(sectionLayout);
     }
 
-    public <P> void expand(@NonNull P parent) {
+    private <P> void expand(@NonNull P parent) {
         for (int i = 0; i < sections.size(); i++) {
             if (parent.equals(sections.get(i).parent)) {
                 ViewGroup sectionView = ((ViewGroup) getChildAt(i));
@@ -201,7 +207,7 @@ public <P> void expand(@NonNull P parent) {
         }
     }
 
-    public <P> void collapse(@NonNull P parent) {
+    private <P> void collapse(@NonNull P parent) {
         for (int i = 0; i < sections.size(); i++) {
             if (parent.equals(sections.get(i).parent)) {
                 ViewGroup sectionView = ((ViewGroup) getChildAt(i));