#ifndef CBUFFER_H
#define CBUFFER_H

#include <complex.h>

#define COMPLEX complex

#include "buffer.h" // 包含實數(shù)緩沖區(qū)說明文件

// 定義復數(shù)緩存區(qū)的各類

// 復數(shù)緩存區(qū)抽象類
class cbuffer {
 public:
	size_t refnum; // 引用數(shù)，目的與buffer類一致
	cbuffer():refnum(1){};
	virtual void alloc(size_t num)=0; // 將緩存區(qū)的尺寸重新定位為num個復數(shù)
	virtual void release()=0; // 釋放緩存區(qū)
	virtual COMPLEX& retrieve(size_t n)=0; // 獲得第n個復數(shù)的引用
	virtual cbuffer* clone()=0; // 克隆一個與自己的內(nèi)容一樣的緩存區(qū)
	virtual cbuffer* newbuf(size_t n=0)=0; // 產(chǎn)生一個與自己類型一樣的緩存區(qū)并返回指針
	COMPLEX&  operator[](size_t n){ // 重載[]運算符但只能取值不能賦值
		return retrieve(n);};
	virtual size_t len()=0; // 緩存區(qū)長度，以復數(shù)個數(shù)計
	cbuffer* set(size_t n, COMPLEX d); // 設第n個復數(shù)的值為d，如果引用數(shù)超過1
					// 則先克隆出一個新的緩存后再行操作
	friend ostream& operator<<(ostream& o, cbuffer& b);
};

ostream& operator<<(ostream& o, cbuffer& b);
cbuffer * getnewcbuffer(size_t n=0); // 根據(jù)缺省的復數(shù)緩存區(qū)類型產(chǎn)生一個新的緩存區(qū)
extern cbuffer *defcbuffer; // 指向缺省的復數(shù)緩存區(qū)變量

class memcbuffer : public cbuffer { // 復數(shù)內(nèi)存緩存區(qū)類
 public:
	COMPLEX * buf; // 指向復數(shù)緩存的指針
	size_t length; // 緩存區(qū)的長度，以復數(shù)個數(shù)計
	memcbuffer():cbuffer(),buf(0),length(0){}; // 缺省構造函數(shù)
	memcbuffer(size_t lth):cbuffer(),length(lth){ // 構造長度為lth的復數(shù)緩存區(qū)
		if(lth > 0)
			buf = new COMPLEX[lth];
		else buf = 0; };
	~memcbuffer(){delete buf;} // 析構函數(shù)，釋放內(nèi)存
	void alloc(size_t num){ // 將緩存區(qū)容量重新定為長度為num個復數(shù)
		delete []buf;
		length = num;
		buf = new COMPLEX[num]; };
	COMPLEX& retrieve(size_t n){ // 獲得第n個復數(shù)的引用
		return buf[n]; };
	void release() { delete []buf; length = 0; }; // 釋放內(nèi)存
	cbuffer* clone(); // 克隆一個內(nèi)容一樣的新緩存
	cbuffer* newbuf(size_t n=0){ // 產(chǎn)生一個類型同自己一樣的具有長度n的新緩存
		return new memcbuffer(n);};
	size_t len(){return length;}; // 返回緩存的長度
};

class diskcbuffer : public cbuffer { // 復數(shù)磁盤緩存區(qū)子類
 public:
	COMPLEX buf;	// 當前的復數(shù)值
	size_t length;	// 緩存區(qū)長度
	size_t n;	// 當前復數(shù)值的序號
	FILE * tempfp; // 臨時文件指針
	diskcbuffer():cbuffer(),buf(0.0),length(0),tempfp(0),n(0){}; // 缺省構造函數(shù)
	diskcbuffer(size_t lth); // 構造長度為lth個復數(shù)的緩存
	~diskcbuffer(); // 析構函數(shù)
	void alloc(size_t num); // 將緩存區(qū)尺寸重新定為num
	COMPLEX& retrieve(size_t n); // 取得第n個復數(shù)的引用
	void release(); // 關閉臨時文件
	cbuffer* clone(); // 克隆一個內(nèi)容相同的緩存
	cbuffer* newbuf(size_t n=0){return new diskcbuffer(n);}; // 產(chǎn)生一個子類相同的
						// 長度為n的緩存
	size_t len(){return length;}; // 返回緩存長度
};

#endif // CBUFFER_H