0ad/source/tools/atlas/AtlasUI/CustomControls/EditableListCtrl/EditableListCtrl.h

#ifndef EDITABLELISTCTRL_H__
#define EDITABLELISTCTRL_H__

#include "wx/listctrl.h"

#include "IAtlasSerialiser.h"

#include <vector>

class FieldEditCtrl;
class AtObj;
class AtIter;

class EditableListCtrl : public wxListCtrl, public IAtlasSerialiser
{
	friend class DeleteCommand;
	friend class DragCommand;
	friend class ImportCommand;

public:
	EditableListCtrl(wxWindow *parent,
		wxWindowID id = wxID_ANY,
		const wxPoint& pos = wxDefaultPosition,
		const wxSize& size = wxDefaultSize,
		long style = wxLC_ICON,
		const wxValidator& validator = wxDefaultValidator,
		const wxString& name = wxListCtrlNameStr);

	~EditableListCtrl();

	void OnMouseEvent(wxMouseEvent& event);

	void MakeSizeAtLeast(int n);

	long GetSelection();
	void SetSelection(long item);

	void UpdateDisplay();

	wxString GetCellString(long item, long column) const;
	AtObj GetCellObject(long item, long column) const;
	void SetCellString(long item, long column, wxString& str);
	void SetCellObject(long item, long column, AtObj& obj);

	struct ColumnData
	{
		ColumnData(const char* k, const FieldEditCtrl* c) : key(k), ctrl(c) {}
		const char* key;
		const FieldEditCtrl* ctrl;
	};
	std::vector<ColumnData> m_ColumnTypes;

	void CloneListData(std::vector<AtObj>& out);
	void SetListData(std::vector<AtObj>& in);

	void DeleteData();


	AtObj FreezeData();
	void ThawData(AtObj& in);

	AtObj ExportData();
	void ImportData(AtObj& in);

private:
	int GetColumnAtPosition(wxPoint& pos);
	void GetCellRect(long row, int col, wxRect& rect);

	void TrimBlankEnds();

	wxString OnGetItemText(long item, long column) const;

protected:

	wxListItemAttr* OnGetItemAttr(long item) const;

	virtual void DoImport(AtObj&)=0;
	virtual AtObj DoExport()=0;

	std::vector<AtObj> m_ListData;

	// objectkey must remain in existence for as long as this list control
	// exists (so you really don't want to be dynamically allocating it;
	// just use static constant strings)
	void AddColumnType(const wxString& title, int width, const char* objectkey, FieldEditCtrl* ctrl);

	void AddRow(AtObj& obj);
	void AddRow(AtIter& iter);

	bool IsRowBlank(int n);

	wxListItemAttr m_ListItemAttr[2]; // standard+alternate colours

	DECLARE_EVENT_TABLE();
};

#endif // EDITABLELISTCTRL_H__