tahoma2d/toonz/sources/include/tcolumnset.h

#pragma once

#ifndef TCOLUMNSET_INCLUDED
#define TCOLUMNSET_INCLUDED

#include "tsmartpointer.h"

#undef DVAPI
#undef DVVAR
#ifdef TXSHEET_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

//=============================================================================

class DVAPI TColumnHeader : public TSmartObject
{
	DECLARE_CLASS_CODE

	int m_index; //!< The header's index in a columns set
	int m_pos;   //!< The header's screen X pos in a columns viewer
	int m_width; //!< The header's width in a columns viewer

	bool m_inColumnsSet; //!< (TO BE REMOVED ASAP) Whether the header
						 //!< belongs to a columns set. Should be
						 //!< redirected to a negative m_index.
public:
	TColumnHeader();
	virtual ~TColumnHeader() {}

	int getIndex() const { return m_index; }
	int getPos() const { return m_pos; }
	int getX0() const { return m_pos; }
	int getX1() const { return m_pos + m_width - 1; }

	void getCoords(int &x0, int &x1) const
	{
		x0 = m_pos;
		x1 = m_pos + m_width - 1;
	}

	int getWidth() const { return m_width; }

	bool inColumnsSet() const { return m_inColumnsSet; }

private:
	template <class T>
	friend class TColumnSetT;
};

//---------------------------------------------------------

#ifdef _WIN32
template class DVAPI TSmartPointerT<TColumnHeader>;
#endif

typedef TSmartPointerT<TColumnHeader> TColumnHeaderP;

//=============================================================================

template <class T>
class TColumnSetT
{
	typedef TSmartPointerT<T> ColumnP;

	std::vector<ColumnP> m_columns;
	int m_defaultWidth;

	static bool compareColumnPos(const int pos, const ColumnP &ch2)
	{
		return pos <= ch2->getX1();
	}

public:
	TColumnSetT(int defaultWidth = 100) : m_defaultWidth(defaultWidth) {}
	~TColumnSetT() {}

	int getColumnCount() const { return m_columns.size(); }

	void clear() { m_columns.clear(); }

	//---------------------------------------------------------

	void col2pos(int index, int &x0, int &x1) const
	{
		assert(index >= 0);
		int columnCount = (int)m_columns.size();
		if (index < columnCount)
			m_columns[index]->getCoords(x0, x1);
		else {
			x0 = (columnCount > 0 ? m_columns.back()->getX1() + 1 : 0) + (index - columnCount) * m_defaultWidth;
			x1 = x0 + m_defaultWidth - 1;
		}
	}

	//---------------------------------------------------------
	//versione con upper_bound

	int pos2col(int pos) const
	{
		// n.b. endPos e' la coordinata del primo pixel non occupato da colonne
		int endPos = m_columns.empty() ? 0 : m_columns.back()->getX1() + 1;
		if (pos >= endPos)
			return m_columns.size() + (pos - endPos) / m_defaultWidth;

		typename std::vector<ColumnP>::const_iterator loc;
		loc = std::upper_bound(m_columns.begin(), m_columns.end(), pos, compareColumnPos);
		return std::distance(m_columns.begin(), loc);
	}

	//---------------------------------------------------------

	const ColumnP &getColumn(int index) const
	{
		static const ColumnP empty;

		if (index >= 0 && index < (int)m_columns.size())
			return m_columns[index];

		return empty;
	}

	//---------------------------------------------------------

	const ColumnP &touchColumn(int index, int type = 0)
	{
		assert(index >= 0);

		const int count = m_columns.size();
		if (index < count)
			return m_columns[index];

		for (int i = count; i <= index; ++i) {
			int cType = (i != index) ? 0 : type;
			ColumnP col = T::createEmpty(cType);

			m_columns.push_back(col);
		}

		update(count);

		assert(index == (int)m_columns.size() - 1);
		return m_columns[index];
	}

	//---------------------------------------------------------

	const ColumnP &insertColumn(int index, const ColumnP &column)
	{
		//assert(column && column->m_index < 0);
		assert(column && !column->m_inColumnsSet);

		if (index > 0)
			touchColumn(index - 1);

		m_columns.insert(m_columns.begin() + index, column);
		update(index);

		return column;
	}

	//---------------------------------------------------------

	const ColumnP removeColumn(int index)
	{
		assert(index >= 0);

		int columnCount = m_columns.size();
		if (index >= columnCount) // Shouldn't be asserted instead ?
			return ColumnP();

		ColumnP column = m_columns[index];
		//column->m_index = -1;                           // We should enforce this. Unfortunately, must be tested extensively.
		column->m_inColumnsSet = false;

		m_columns.erase(m_columns.begin() + index);
		update(index);

		return column;
	}

	//---------------------------------------------------------

	void rollLeft(int index, int count)
	{
		assert(index >= 0);
		assert(count > 1);

		int columnCount = m_columns.size();
		if (index + count > columnCount)
			count = columnCount - index;

		if (count < 2)
			return;

		assert(0 <= index && index + count - 1 < columnCount);
		assert(count > 0);

		int i = index, j = index + count - 1;
		ColumnP tmp = m_columns[i];

		for (int k = i; k < j; ++k)
			m_columns[k] = m_columns[k + 1];
		m_columns[j] = tmp;

		update(0);
	}

	//---------------------------------------------------------

	void rollRight(int index, int count)
	{
		assert(index >= 0);
		assert(count > 1);

		int columnCount = m_columns.size();
		if (index + count > columnCount)
			count = columnCount - index;

		if (count < 2)
			return;

		assert(0 <= index && index + count - 1 < columnCount);
		assert(count > 0);

		int i = index, j = index + count - 1;
		ColumnP tmp = m_columns[j];

		for (int k = j; k > i; --k)
			m_columns[k] = m_columns[k - 1];
		m_columns[i] = tmp;

		update(0);
	}

	//---------------------------------------------------------

	void update(int fromIdx)
	{
		int pos = 0, index = 0;
		if (fromIdx > 0) {
			assert(fromIdx <= (int)m_columns.size());

			const ColumnP &left = m_columns[fromIdx - 1];

			pos = left->getX1() + 1;
			index = left->m_index + 1;
		}

		int c, cCount = m_columns.size();
		for (c = fromIdx; c != cCount; ++c) {
			const ColumnP &col = m_columns[c];

			col->m_pos = pos;
			col->m_index = index++;
			pos += col->m_width;

			col->m_inColumnsSet = true;
		}
	}

private:
	// Not copyable
	TColumnSetT(const TColumnSetT &);
	TColumnSetT &operator=(const TColumnSetT &);
};

#endif // TCOLUMNSET_INCLUDED