/*

   Separated List Adapter

   Taken from http://jsharkey.org/blog/2008/08/18/separating-lists-with-headers-in-android-09/

   Copyright Jeff Sharkey

   This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.

   If a copy of the MPL was not distributed with this file, You can obtain one at

   http://mozilla.org/MPL/2.0/.

*/

package com.freerdp.freerdpcore.utils;

import android.content.Context;

import android.view.View;

import android.view.ViewGroup;

import android.widget.Adapter;

import android.widget.ArrayAdapter;

import android.widget.BaseAdapter;

import com.freerdp.freerdpcore.R;

import java.util.LinkedHashMap;

import java.util.Map;

public class SeparatedListAdapter extends BaseAdapter

{

	public final static int TYPE_SECTION_HEADER = 0;

	public final Map<String, Adapter> sections = new LinkedHashMap<String, Adapter>();

	public final ArrayAdapter<String> headers;

	public SeparatedListAdapter(Context context)

	{

		headers = new ArrayAdapter<String>(context, R.layout.list_header);

	}

	public void addSection(String section, Adapter adapter)

	{

		this.headers.add(section);

		this.sections.put(section, adapter);

	}

	public void setSectionTitle(int section, String title)

	{

		String oldTitle = this.headers.getItem(section);

		// remove/add to headers array

		this.headers.remove(oldTitle);

		this.headers.insert(title, section);

		// remove/add to section map

		Adapter adapter = this.sections.get(oldTitle);

		this.sections.remove(oldTitle);

		this.sections.put(title, adapter);

	}

	public Object getItem(int position)

	{

		for (int i = 0; i < headers.getCount(); i++)

		{

			String section = headers.getItem(i);

			Adapter adapter = sections.get(section);

			// ignore empty sections

			if (adapter.getCount() > 0)

			{

				int size = adapter.getCount() + 1;

				// check if position inside this section

				if (position == 0)

					return section;

				if (position < size)

					return adapter.getItem(position - 1);

				// otherwise jump into next section

				position -= size;

			}

		}

		return null;

	}

	public int getCount()

	{

		// total together all sections, plus one for each section header (except if the section is

		// empty)

		int total = 0;

		for (Adapter adapter : this.sections.values())

			total += ((adapter.getCount() > 0) ? adapter.getCount() + 1 : 0);

		return total;

	}

	public int getViewTypeCount()

	{

		// assume that headers count as one, then total all sections

		int total = 1;

		for (Adapter adapter : this.sections.values())

			total += adapter.getViewTypeCount();

		return total;

	}

	public int getItemViewType(int position)

	{

		int type = 1;

		for (int i = 0; i < headers.getCount(); i++)

		{

			String section = headers.getItem(i);

			Adapter adapter = sections.get(section);

			// skip empty sections

			if (adapter.getCount() > 0)

			{

				int size = adapter.getCount() + 1;

				// check if position inside this section

				if (position == 0)

					return TYPE_SECTION_HEADER;

				if (position < size)

					return type + adapter.getItemViewType(position - 1);

				// otherwise jump into next section

				position -= size;

				type += adapter.getViewTypeCount();

			}

		}

		return -1;

	}

	public boolean areAllItemsSelectable()

	{

		return false;

	}

	public boolean isEnabled(int position)

	{

		return (getItemViewType(position) != TYPE_SECTION_HEADER);

	}

	@Override public View getView(int position, View convertView, ViewGroup parent)

	{

		int sectionnum = 0;

		for (int i = 0; i < headers.getCount(); i++)

		{

			String section = headers.getItem(i);

			Adapter adapter = sections.get(section);

			// skip empty sections

			if (adapter.getCount() > 0)

			{

				int size = adapter.getCount() + 1;

				// check if position inside this section

				if (position == 0)

					return headers.getView(sectionnum, convertView, parent);

				if (position < size)

					return adapter.getView(position - 1, null, parent);

				// otherwise jump into next section

				position -= size;

			}

			sectionnum++;

		}

		return null;

	}

	@Override public long getItemId(int position)

	{

		for (int i = 0; i < headers.getCount(); i++)

		{

			String section = headers.getItem(i);

			Adapter adapter = sections.get(section);

			if (adapter.getCount() > 0)

			{

				int size = adapter.getCount() + 1;

				// check if position inside this section

				if (position < size)

					return adapter.getItemId(position - 1);

				// otherwise jump into next section

				position -= size;

			}

		}

		return -1;

	}

	public String getSectionForPosition(int position)

	{

		int curPos = 0;

		for (int i = 0; i < headers.getCount(); i++)

		{

			String section = headers.getItem(i);

			Adapter adapter = sections.get(section);

			if (adapter.getCount() > 0)

			{

				int size = adapter.getCount() + 1;

				// check if position inside this section

				if (position >= curPos && position < (curPos + size))

					return section.toString();

				// otherwise jump into next section

				curPos += size;

			}

		}

		return null;

	}

}