treepainter.h

来自「组播密钥的批次更新算法」· C头文件 代码 · 共 40 行

#pragma once

class CTreePainter
{
public:
	CTreePainter(int horizontalSpace = CTreePainter::defaultHorizontalSpace,
				 int verticalSpace = CTreePainter::defaultVerticalSpace,
				 int nodeLength = CTreePainter::defaultLength,
				 int nodeHeight = CTreePainter::defaultHeight);
	~CTreePainter(void);

	enum size {defaultLength=80,defaultHeight=40};
	enum distance {defaultHorizontalSpace=(defaultLength+(defaultLength>>1)),defaultVerticalSpace=defaultLength};
	enum color {defaultLeafColor=RGB(192,192,192),defaultCommonColor=RGB(255,255,255)};
	void PaintTree(CBinaryTree& btree,CDC* dc);
	void SetOriginPoint(const CPoint &point);
	CPoint GetOriginPoint(void);
	CSize GetNodeSize(void);
	virtual void PaintAppendants(CDC* pdc, int posX, int posY, CNode* node);
	virtual void PaintNode(CDC* pdc, int posX, int posY, CNode* node);
	CPoint GetViewSize(CBinaryTree& btree,CSize &size);

protected:
	void PaintSubTree(CDC* dc, int posx, int posy, int childdistance, CNode* subroot);
	void PaintRow(CDC* pdc, const CRect& rect, CString& label, int index);
	
	int m_hSpace;
	int m_vSpace;
	int m_nodeLength;
	int m_nodeHeight;
	int m_rowHeight;
	CPoint m_origin;
	int m_gapInNodeAndStack;
	int m_rowAmount;


	//以下是定义颜色和画刷
	COLORREF m_leafColor;
	COLORREF m_commonColor;
};