谷歌为GridView添加HeaderView

/* 
 * Copyright (C) 2013 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.android.photos.views;  
import android.content.Context;  
import android.database.DataSetObservable;  
import android.database.DataSetObserver;  
import android.util.AttributeSet;  
import android.view.View;  
import android.view.ViewGroup;  
import android.widget.AdapterView;  
import android.widget.Filter;  
import android.widget.Filterable;  
import android.widget.FrameLayout;  
import android.widget.GridView;  
import android.widget.ListAdapter;  
import android.widget.WrapperListAdapter;  
import java.util.ArrayList;  
/** 
 * A {@link GridView} that supports adding header rows in a 
 * very similar way to {@link ListView}. 
 * See {@link HeaderGridView#addHeaderView(View, Object, boolean)} 
 */  
public class HeaderGridView extends GridView {  
    private static final String TAG = "HeaderGridView";  
    /** 
     * A class that represents a fixed view in a list, for example a header at the top 
     * or a footer at the bottom. 
     */  
    private static class FixedViewInfo {  
        /** The view to add to the grid */  
        public View view;  
        public ViewGroup viewContainer;  
        /** The data backing the view. This is returned from {@link ListAdapter#getItem(int)}. */  
        public Object data;  
        /** <code>true</code> if the fixed view should be selectable in the grid */  
        public boolean isSelectable;  
    }  
    private ArrayList<FixedViewInfo> mHeaderViewInfos = new ArrayList<FixedViewInfo>();  
    private void initHeaderGridView() {  
        super.setClipChildren(false);  
    }  
    public HeaderGridView(Context context) {  
        super(context);  
        initHeaderGridView();  
    }  
    public HeaderGridView(Context context, AttributeSet attrs) {  
        super(context, attrs);  
        initHeaderGridView();  
    }  
    public HeaderGridView(Context context, AttributeSet attrs, int defStyle) {  
        super(context, attrs, defStyle);  
        initHeaderGridView();  
    }  
    @Override  
    protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {  
        super.onMeasure(widthMeasureSpec, heightMeasureSpec);  
        ListAdapter adapter = getAdapter();  
        if (adapter != null && adapter instanceof HeaderViewGridAdapter) {  
            ((HeaderViewGridAdapter) adapter).setNumColumns(getNumColumns());  
        }  
    }  
    @Override  
    public void setClipChildren(boolean clipChildren) {  
       // Ignore, since the header rows depend on not being clipped  
    }  
    /** 
     * Add a fixed view to appear at the top of the grid. If addHeaderView is 
     * called more than once, the views will appear in the order they were 
     * added. Views added using this call can take focus if they want. 
     * <p> 
     * NOTE: Call this before calling setAdapter. This is so HeaderGridView can wrap 
     * the supplied cursor with one that will also account for header views. 
     * 
     * @param v The view to add. 
     * @param data Data to associate with this view 
     * @param isSelectable whether the item is selectable 
     */  
    public void addHeaderView(View v, Object data, boolean isSelectable) {  
        ListAdapter adapter = getAdapter();  
        if (adapter != null && ! (adapter instanceof HeaderViewGridAdapter)) {  
            throw new IllegalStateException(  
                    "Cannot add header view to grid -- setAdapter has already been called.");  
        }  
        FixedViewInfo info = new FixedViewInfo();  
        FrameLayout fl = new FullWidthFixedViewLayout(getContext());  
        fl.addView(v);  
        info.view = v;  
        info.viewContainer = fl;  
        info.data = data;  
        info.isSelectable = isSelectable;  
        mHeaderViewInfos.add(info);  
        // in the case of re-adding a header view, or adding one later on,  
        // we need to notify the observer  
        if (adapter != null) {  
            ((HeaderViewGridAdapter) adapter).notifyDataSetChanged();  
        }  
    }  
    /** 
     * Add a fixed view to appear at the top of the grid. If addHeaderView is 
     * called more than once, the views will appear in the order they were 
     * added. Views added using this call can take focus if they want. 
     * <p> 
     * NOTE: Call this before calling setAdapter. This is so HeaderGridView can wrap 
     * the supplied cursor with one that will also account for header views. 
     * 
     * @param v The view to add. 
     */  
    public void addHeaderView(View v) {  
        addHeaderView(v, null, true);  
    }  
    public int getHeaderViewCount() {  
        return mHeaderViewInfos.size();  
    }  
    /** 
     * Removes a previously-added header view. 
     * 
     * @param v The view to remove 
     * @return true if the view was removed, false if the view was not a header 
     *         view 
     */  
    public boolean removeHeaderView(View v) {  
        if (mHeaderViewInfos.size() > 0) {  
            boolean result = false;  
            ListAdapter adapter = getAdapter();  
            if (adapter != null && ((HeaderViewGridAdapter) adapter).removeHeader(v)) {  
                result = true;  
            }  
            removeFixedViewInfo(v, mHeaderViewInfos);  
            return result;  
        }  
        return false;  
    }  
    private void removeFixedViewInfo(View v, ArrayList<FixedViewInfo> where) {  
        int len = where.size();  
        for (int i = 0; i < len; ++i) {  
            FixedViewInfo info = where.get(i);  
            if (info.view == v) {  
                where.remove(i);  
                break;  
            }  
        }  
    }  
    @Override  
    public void setAdapter(ListAdapter adapter) {  
        if (mHeaderViewInfos.size() > 0) {  
            HeaderViewGridAdapter hadapter = new HeaderViewGridAdapter(mHeaderViewInfos, adapter);  
            int numColumns = getNumColumns();  
            if (numColumns > 1) {  
                hadapter.setNumColumns(numColumns);  
            }  
            super.setAdapter(hadapter);  
        } else {  
            super.setAdapter(adapter);  
        }  
    }  
    private class FullWidthFixedViewLayout extends FrameLayout {  
        public FullWidthFixedViewLayout(Context context) {  
            super(context);  
        }  
        @Override  
        protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {  
            int targetWidth = HeaderGridView.this.getMeasuredWidth()  
                    - HeaderGridView.this.getPaddingLeft()  
                    - HeaderGridView.this.getPaddingRight();  
            widthMeasureSpec = MeasureSpec.makeMeasureSpec(targetWidth,  
                    MeasureSpec.getMode(widthMeasureSpec));  
            super.onMeasure(widthMeasureSpec, heightMeasureSpec);  
        }  
    }  
    /** 
     * ListAdapter used when a HeaderGridView has header views. This ListAdapter 
     * wraps another one and also keeps track of the header views and their 
     * associated data objects. 
     *<p>This is intended as a base class; you will probably not need to 
     * use this class directly in your own code. 
     */  
    private static class HeaderViewGridAdapter implements WrapperListAdapter, Filterable {  
        // This is used to notify the container of updates relating to number of columns  
        // or headers changing, which changes the number of placeholders needed  
        private final DataSetObservable mDataSetObservable = new DataSetObservable();  
        private final ListAdapter mAdapter;  
        private int mNumColumns = 1;  
        // This ArrayList is assumed to NOT be null.  
        ArrayList<FixedViewInfo> mHeaderViewInfos;  
        boolean mAreAllFixedViewsSelectable;  
        private final boolean mIsFilterable;  
        public HeaderViewGridAdapter(ArrayList<FixedViewInfo> headerViewInfos, ListAdapter adapter) {  
            mAdapter = adapter;  
            mIsFilterable = adapter instanceof Filterable;  
            if (headerViewInfos == null) {  
                throw new IllegalArgumentException("headerViewInfos cannot be null");  
            }  
            mHeaderViewInfos = headerViewInfos;  
            mAreAllFixedViewsSelectable = areAllListInfosSelectable(mHeaderViewInfos);  
        }  
        public int getHeadersCount() {  
            return mHeaderViewInfos.size();  
        }  
        @Override  
        public boolean isEmpty() {  
            return (mAdapter == null || mAdapter.isEmpty()) && getHeadersCount() == 0;  
        }  
        public void setNumColumns(int numColumns) {  
            if (numColumns < 1) {  
                throw new IllegalArgumentException("Number of columns must be 1 or more");  
            }  
            if (mNumColumns != numColumns) {  
                mNumColumns = numColumns;  
                notifyDataSetChanged();  
            }  
        }  
        private boolean areAllListInfosSelectable(ArrayList<FixedViewInfo> infos) {  
            if (infos != null) {  
                for (FixedViewInfo info : infos) {  
                    if (!info.isSelectable) {  
                        return false;  
                    }  
                }  
            }  
            return true;  
        }  
        public boolean removeHeader(View v) {  
            for (int i = 0; i < mHeaderViewInfos.size(); i++) {  
                FixedViewInfo info = mHeaderViewInfos.get(i);  
                if (info.view == v) {  
                    mHeaderViewInfos.remove(i);  
                    mAreAllFixedViewsSelectable = areAllListInfosSelectable(mHeaderViewInfos);  
                    mDataSetObservable.notifyChanged();  
                    return true;  
                }  
            }  
            return false;  
        }  
        @Override  
        public int getCount() {  
            if (mAdapter != null) {  
                return getHeadersCount() * mNumColumns + mAdapter.getCount();  
            } else {  
                return getHeadersCount() * mNumColumns;  
            }  
        }  
        @Override  
        public boolean areAllItemsEnabled() {  
            if (mAdapter != null) {  
                return mAreAllFixedViewsSelectable && mAdapter.areAllItemsEnabled();  
            } else {  
                return true;  
            }  
        }  
        @Override  
        public boolean isEnabled(int position) {  
            // Header (negative positions will throw an ArrayIndexOutOfBoundsException)  
            int numHeadersAndPlaceholders = getHeadersCount() * mNumColumns;  
            if (position < numHeadersAndPlaceholders) {  
                return (position % mNumColumns == 0)  
                        && mHeaderViewInfos.get(position / mNumColumns).isSelectable;  
            }  
            // Adapter  
            final int adjPosition = position - numHeadersAndPlaceholders;  
            int adapterCount = 0;  
            if (mAdapter != null) {  
                adapterCount = mAdapter.getCount();  
                if (adjPosition < adapterCount) {  
                    return mAdapter.isEnabled(adjPosition);  
                }  
            }  
            throw new ArrayIndexOutOfBoundsException(position);  
        }  
        @Override  
        public Object getItem(int position) {  
            // Header (negative positions will throw an ArrayIndexOutOfBoundsException)  
            int numHeadersAndPlaceholders = getHeadersCount() * mNumColumns;  
            if (position < numHeadersAndPlaceholders) {  
                if (position % mNumColumns == 0) {  
                    return mHeaderViewInfos.get(position / mNumColumns).data;  
                }  
                return null;  
            }  
            // Adapter  
            final int adjPosition = position - numHeadersAndPlaceholders;  
            int adapterCount = 0;  
            if (mAdapter != null) {  
                adapterCount = mAdapter.getCount();  
                if (adjPosition < adapterCount) {  
                    return mAdapter.getItem(adjPosition);  
                }  
            }  
            throw new ArrayIndexOutOfBoundsException(position);  
        }  
        @Override  
        public long getItemId(int position) {  
            int numHeadersAndPlaceholders = getHeadersCount() * mNumColumns;  
            if (mAdapter != null && position >= numHeadersAndPlaceholders) {  
                int adjPosition = position - numHeadersAndPlaceholders;  
                int adapterCount = mAdapter.getCount();  
                if (adjPosition < adapterCount) {  
                    return mAdapter.getItemId(adjPosition);  
                }  
            }  
            return -1;  
        }  
        @Override  
        public boolean hasStableIds() {  
            if (mAdapter != null) {  
                return mAdapter.hasStableIds();  
            }  
            return false;  
        }  
        @Override  
        public View getView(int position, View convertView, ViewGroup parent) {  
            // Header (negative positions will throw an ArrayIndexOutOfBoundsException)  
            int numHeadersAndPlaceholders = getHeadersCount() * mNumColumns ;  
            if (position < numHeadersAndPlaceholders) {  
                View headerViewContainer = mHeaderViewInfos  
                        .get(position / mNumColumns).viewContainer;  
                if (position % mNumColumns == 0) {  
                    return headerViewContainer;  
                } else {  
                    if (convertView == null) {  
                        convertView = new View(parent.getContext());  
                    }  
                    // We need to do this because GridView uses the height of the last item  
                    // in a row to determine the height for the entire row.  
                    convertView.setVisibility(View.INVISIBLE);  
                    convertView.setMinimumHeight(headerViewContainer.getHeight());  
                    return convertView;  
                }  
            }  
            // Adapter  
            final int adjPosition = position - numHeadersAndPlaceholders;  
            int adapterCount = 0;  
            if (mAdapter != null) {  
                adapterCount = mAdapter.getCount();  
                if (adjPosition < adapterCount) {  
                    return mAdapter.getView(adjPosition, convertView, parent);  
                }  
            }  
            throw new ArrayIndexOutOfBoundsException(position);  
        }  
        @Override  
        public int getItemViewType(int position) {  
            int numHeadersAndPlaceholders = getHeadersCount() * mNumColumns;  
            if (position < numHeadersAndPlaceholders && (position % mNumColumns != 0)) {  
                // Placeholders get the last view type number  
                return mAdapter != null ? mAdapter.getViewTypeCount() : 1;  
            }  
            if (mAdapter != null && position >= numHeadersAndPlaceholders) {  
                int adjPosition = position - numHeadersAndPlaceholders;  
                int adapterCount = mAdapter.getCount();  
                if (adjPosition < adapterCount) {  
                    return mAdapter.getItemViewType(adjPosition);  
                }  
            }  
            return AdapterView.ITEM_VIEW_TYPE_HEADER_OR_FOOTER;  
        }  
        @Override  
        public int getViewTypeCount() {  
            if (mAdapter != null) {  
                return mAdapter.getViewTypeCount() + 1;  
            }  
            return 2;  
        }  
        @Override  
        public void registerDataSetObserver(DataSetObserver observer) {  
            mDataSetObservable.registerObserver(observer);  
            if (mAdapter != null) {  
                mAdapter.registerDataSetObserver(observer);  
            }  
        }  
        @Override  
        public void unregisterDataSetObserver(DataSetObserver observer) {  
            mDataSetObservable.unregisterObserver(observer);  
            if (mAdapter != null) {  
                mAdapter.unregisterDataSetObserver(observer);  
            }  
        }  
        @Override  
        public Filter getFilter() {  
            if (mIsFilterable) {  
                return ((Filterable) mAdapter).getFilter();  
            }  
            return null;  
        }  
        @Override  
        public ListAdapter getWrappedAdapter() {  
            return mAdapter;  
        }  
        public void notifyDataSetChanged() {  
            mDataSetObservable.notifyChanged();  
        }  
    }  
}