[SuwLib] Add progress bars to GLIF layouts

Change-Id: I5a32cabcc1d89cfb978fbceb3d3f4811bccaea92

1 files changed, 25 insertions, 0 deletions

diff --git a/library/main/src/com/android/setupwizardlib/GlifLayout.java b/library/main/src/com/android/setupwizardlib/GlifLayout.java
index f1ec6d4..0590a81 100644
--- a/library/main/src/com/android/setupwizardlib/GlifLayout.java
+++ b/library/main/src/com/android/setupwizardlib/GlifLayout.java
@@ -27,6 +27,7 @@ import android.util.AttributeSet;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
+import android.view.ViewStub;
 import android.widget.ImageView;
 import android.widget.ScrollView;
 import android.widget.TextView;
@@ -189,4 +190,28 @@ public class GlifLayout extends TemplateLayout {
     protected ImageView getIconView() {
         return (ImageView) findViewById(R.id.suw_layout_icon);
     }
+
+    public boolean isProgressBarShown() {
+        final View progressBar = findViewById(R.id.suw_layout_progress);
+        return progressBar != null && progressBar.getVisibility() == View.VISIBLE;
+    }
+
+    public void setProgressBarShown(boolean shown) {
+        final View progressBar = findViewById(R.id.suw_layout_progress);
+        if (shown) {
+            if (progressBar != null) {
+                progressBar.setVisibility(View.VISIBLE);
+            } else {
+                final ViewStub progressBarStub =
+                        (ViewStub) findViewById(R.id.suw_layout_progress_stub);
+                if (progressBarStub != null) {
+                    progressBarStub.inflate();
+                }
+            }
+        } else {
+            if (progressBar != null) {
+                progressBar.setVisibility(View.GONE);
+            }
+        }
+    }
 }