/* Copyright (C) 2006 MySQL AB

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

// EncodeTransform.h: interface for the EncodeTransform class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_ENCODETRANSFORM_H__AB15A637_E683_4B7C_B606_E3B4F4C930E0__INCLUDED_)
#define AFX_ENCODETRANSFORM_H__AB15A637_E683_4B7C_B606_E3B4F4C930E0__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "Transform.h"

template <class Source, class Encode>
class EncodeTransform : public Transform  
{
public:
	EncodeTransform(const char *string) 
		: source(string), encode(true, &source)
		{}

	EncodeTransform(int length, const UCHAR *data) 
		: source(length, data), encode(true, &source)
		{}

	virtual unsigned int get(unsigned int bufferLength, UCHAR *buffer)
		{
		return encode.get(bufferLength, buffer);
		}

	virtual unsigned int getLength()
		{
		return encode.getLength();
		}

	virtual void reset()
		{
		encode.reset();
		}

	virtual void setString(const char *string, bool flag = false)
		{
		reset();
		source.setString(string, flag);
		}

	virtual void setString(unsigned int length, const UCHAR *data, bool flag = false)
		{
		reset();
		source.setString(length, data, flag);
		}

	Source	source;
	Encode	encode;
};

#endif // !defined(AFX_ENCODETRANSFORM_H__AB15A637_E683_4B7C_B606_E3B4F4C930E0__INCLUDED_)
