#include "oled.h"
#include "oled_font.h"

#ifdef __VTOR_OLED__

#ifdef __OLED_GRAM__
unsigned char OLED_GRAM[8][128] = {0};
#endif

struct OLED_PAINT *oledPaint[1] = {0};
struct OLED_PAINT defaultOledPaint = {8, OLED_PAINT_MODE_NORMAL};
#ifdef __OLED_MULTI_SIZE__
struct OLED_PAINT invertOledPaint = {8, OLED_PAINT_MODE_INVERT};
struct OLED_PAINT bigOledPaint = {16, OLED_PAINT_MODE_NORMAL};
struct OLED_PAINT bigInvertOledPaint = {16, OLED_PAINT_MODE_INVERT};
#endif

struct OLED_PAINT* pOledPaint = NULL;

#ifdef __OLED_LOG__
#define OLED_LOG_LINE_CNT	8
#define OLED_LOG_CHAR_CNT	21
#define OLED_LOG_LINE_LEN	22

unsigned char curLogLine = OLED_LOG_LINE_CNT;
unsigned char OledLogBuf[OLED_LOG_LINE_CNT][OLED_LOG_LINE_LEN] = {0};
#endif

//反显函数
void OLED_ColorTurn(unsigned char i)
{
	if(i==0)
		{
			OLED_WR_Byte(0xA6,OLED_CMD);//正常显示
		}
	if(i==1)
		{
			OLED_WR_Byte(0xA7,OLED_CMD);//反色显示
		}
}

//屏幕旋转180度
void OLED_DisplayTurn(unsigned char i)
{
	if(i==0)
		{
			OLED_WR_Byte(0xC8,OLED_CMD);//正常显示
			OLED_WR_Byte(0xA1,OLED_CMD);
		}
	if(i==1)
		{
			OLED_WR_Byte(0xC0,OLED_CMD);//反转显示
			OLED_WR_Byte(0xA0,OLED_CMD);
		}
}



//开启OLED显示 
void OLED_DisPlay_On(void)
{
	OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
	OLED_WR_Byte(0x14,OLED_CMD);//开启电荷泵
	OLED_WR_Byte(0xAF,OLED_CMD);//点亮屏幕
}

//关闭OLED显示 
void OLED_DisPlay_Off(void)
{
	OLED_WR_Byte(0x8D,OLED_CMD);//电荷泵使能
	OLED_WR_Byte(0x10,OLED_CMD);//关闭电荷泵
	OLED_WR_Byte(0xAE,OLED_CMD);//关闭屏幕
}

#ifdef __OLED_GRAM__
// 更新显存到OLED，预留性能较弱的弱函数
// stm32f103，diaan，硬件iic，100kHz，耗时约300ms
__attribute__((weak)) void OLED_Refresh(void)
{
	unsigned char i,n;
	for(i=0;i<8;i++)
	{
		OLED_WR_Byte(0xb0+i,OLED_CMD); //设置行起始地址
//		OLED_WR_Byte(0x00,OLED_CMD);   //设置低列起始地址
//		OLED_WR_Byte(0x10,OLED_CMD);   //设置高列起始地址
		for(n=0;n<128;n++)
		{
			OLED_WR_Byte(OLED_GRAM[i][n],OLED_DATA);
		}
	}
}
//清屏函数
void OLED_ClearGRam(void)
{
	unsigned char i,n;
	for(i=0;i<8;i++)
	{
		for(n=0;n<128;n++)
		{
			OLED_GRAM[i][n]=0;//清除所有数据
		}
	}
}

void OLED_Clear(void)
{
	OLED_ClearGRam();
	OLED_Refresh();//更新显示
}
#else
void OLED_Clear(void)
{
	char i;
	// 非cache情况，清空屏幕
	for (i = 0; i < 8; ++i)
	{
		OLED_ShowString(0, i * 8, (unsigned char*)"                      ");
	}
}
#endif
#ifdef __OLED_GRAM__
//画点 
//x:0~127
//y:0~63
//t:1 填充 0,清空	
void OLED_DrawPoint(unsigned char x,unsigned char y,unsigned char t)
{
	unsigned char i,m,n;
	i=y/8;
	m=y%8;
	n=1<<m;
	if(t)
	{
		OLED_GRAM[i][x]|=n;
	}
	else
	{
		OLED_GRAM[i][x]&=~n;
	}
}
#endif

#ifdef __OLED_GRAPH__
//画线
//x1,y1:起点坐标
//x2,y2:结束坐标
void OLED_DrawLine(unsigned char x1,unsigned char y1,unsigned char x2,unsigned char y2)
{
	unsigned short t;
	int xerr=0,yerr=0,delta_x,delta_y,distance;
	int incx,incy,uRow,uCol;
	unsigned char mode = pOledPaint->mode;
	delta_x=x2-x1; //计算坐标增量 
	delta_y=y2-y1;
	uRow=x1;//画线起点坐标
	uCol=y1;
	if(delta_x>0)incx=1; //设置单步方向 
	else if (delta_x==0)incx=0;//垂直线 
	else {incx=-1;delta_x=-delta_x;}
	if(delta_y>0)incy=1;
	else if (delta_y==0)incy=0;//水平线 
	else {incy=-1;delta_y=-delta_x;}
	if(delta_x>delta_y)distance=delta_x; //选取基本增量坐标轴 
	else distance=delta_y;
	for(t=0;t<distance+1;t++)
	{
		OLED_DrawPoint(uRow,uCol,mode);//画点
		xerr+=delta_x;
		yerr+=delta_y;
		if(xerr>distance)
		{
			xerr-=distance;
			uRow+=incx;
		}
		if(yerr>distance)
		{
			yerr-=distance;
			uCol+=incy;
		}
	}
}
//x,y:圆心坐标
//r:圆的半径
void OLED_DrawCircle(unsigned char x,unsigned char y,unsigned char r)
{
	int a, b,num;
    a = 0;
    b = r;
    while(2 * b * b >= r * r)      
    {
        OLED_DrawPoint(x + a, y - b,1);
        OLED_DrawPoint(x - a, y - b,1);
        OLED_DrawPoint(x - a, y + b,1);
        OLED_DrawPoint(x + a, y + b,1);
 
        OLED_DrawPoint(x + b, y + a,1);
        OLED_DrawPoint(x + b, y - a,1);
        OLED_DrawPoint(x - b, y - a,1);
        OLED_DrawPoint(x - b, y + a,1);
        
        a++;
        num = (a * a + b * b) - r*r;//计算画的点离圆心的距离
        if(num > 0)
        {
            b--;
            a--;
        }
    }
}
#endif



//在指定位置显示一个字符,包括部分字符
//x:0~127
//y:0~63
//size1:选择字体 6x8/6x12/8x16/12x24
//mode:0,反色显示;1,正常显示
void OLED_ShowChar(unsigned char x,unsigned char y,unsigned char chr)
{
	unsigned char i,m,temp,size2,chr1;
	unsigned char x0=x,y0=y;
	unsigned char size1 = pOledPaint->size;
	unsigned char mode = pOledPaint->mode;
	if(size1==8)size2=6;
	else size2=(size1/8+((size1%8)?1:0))*(size1/2);  //得到字体一个字符对应点阵集所占的字节数
	chr1=chr-' ';  //计算偏移后的值
#ifdef __OLED_GRAM__
#else
	OLED_WR_Byte(0xb0 + y / 8,OLED_CMD); //设置行起始地址
	OLED_WR_Byte(0x00 + ((x) & 0x0f),OLED_CMD);   //设置低列起始地址
	OLED_WR_Byte(0x10 + ((x) >> 4),OLED_CMD);   //设置高列起始地址
#endif
	for(i=0;i<size2;i++)
	{
		if(size1==8)
			  {temp=asc2_0806[chr1][i];} //调用0806字体
#ifdef __OLED_MULTI_SIZE__
		else if(size1==12)
        {temp=asc2_1206[chr1][i];} //调用1206字体
		else if(size1==16)
        {temp=asc2_1608[chr1][i];} //调用1608字体
		else if(size1==24)
        {temp=asc2_2412[chr1][i];} //调用2412字体
#endif
		else return;
#ifdef __OLED_GRAM__
		for(m=0;m<8;m++)
		{
			if(temp&0x01)OLED_DrawPoint(x,y,mode);
			else OLED_DrawPoint(x,y,!mode);
			temp>>=1;
			y++;
		}
#else
		OLED_WR_Byte(temp,OLED_DATA);
#endif
		x++;
		if((size1!=8)&&((x-x0)==size1/2))
		{x=x0;y0=y0+8;}
		y=y0;
  }
}


//显示字符串
//x,y:起点坐标  
//size1:字体大小 
//*chr:字符串起始地址 
//mode:0,反色显示;1,正常显示
void OLED_ShowString(unsigned char x,unsigned char y,unsigned char *chr)
{
	unsigned char size1 = pOledPaint->size;
	while(' ' <= *chr && *chr<='~')//判断是不是非法字符!
	{
		if ('\n' == *chr)
		{
			chr++;
			x = 0;
			y += 8;
			if (y >= 64-8) y = 0;
			continue;
		}
		OLED_ShowChar(x,y,*chr);
		if(size1==8)x+=6;
		else x+=size1/2;
		// 自动换行，避免后面显示不全
		if (x >= 128)
		{
			x = 0;
			y+=8;
			if (y >= 64-8) y = 0;
		}
		chr++;
  }
}

#ifdef __OLED_LOG__
void OLED_PushLog(unsigned char *log)
{
	int x = OLED_LOG_CHAR_CNT;

	while(' ' <= *log && *log <= 'z' + 1)
	{
		if(OLED_LOG_CHAR_CNT == x) // 如果遇到当前行最后一个字符，跳过
		{
			x = 0;
			curLogLine = (curLogLine + 1) % OLED_LOG_LINE_CNT;
		}
		OledLogBuf[curLogLine][x++] = *log++;
	}
	OledLogBuf[curLogLine][x] = '\0';
}

void OLED_RefreshLog()
{
	char y = 0;;
#ifdef __OLED_GRAM__
	// 先清空gram
	OLED_ClearGRam();
#else
	OLED_Clear();
#endif
	//OLED_ShowString(0, 0, OledLogBuf);
	for (y = 0; y < OLED_LOG_LINE_CNT; y++)
	{
		OLED_ShowString(0,y*8,OledLogBuf[(y + curLogLine + 1) % OLED_LOG_LINE_CNT]);
	}
}
#endif

#ifdef __OLED_PRINTF__
//m^n
unsigned int OLED_Pow(unsigned char m,unsigned char n)
{
	unsigned int result=1;
	while(n--)
	{
	  result*=m;
	}
	return result;
}

//显示数字
//x,y :起点坐标
//num :要显示的数字
//len :数字的位数
//size:字体大小
//mode:0,反色显示;1,正常显示
void OLED_ShowNum(unsigned char x,unsigned char y,unsigned int num,unsigned char len)
{
	unsigned char size1 = pOledPaint->size;
	unsigned char t,temp,m=0;
	if(size1==8)m=2;
	for(t=0;t<len;t++)
	{
		temp=(num/OLED_Pow(10,len-t-1))%10;
			if(temp==0)
			{
				OLED_ShowChar(x+(size1/2+m)*t,y,'0');
      }
			else 
			{
			  OLED_ShowChar(x+(size1/2+m)*t,y,temp+'0');
			}
  }
}
#endif

#ifdef __OLED_HZK__
//显示汉字
//x,y:起点坐标
//num:汉字对应的序号
//mode:0,反色显示;1,正常显示
void OLED_ShowChinese(unsigned char x,unsigned char y,unsigned char num)
{
	unsigned char m,temp;
	unsigned char x0=x,y0=y;
	unsigned char size1 = pOledPaint->size;
	unsigned char mode = pOledPaint->mode;
	unsigned short i,size3=(size1/8+((size1%8)?1:0))*size1;  //得到字体一个字符对应点阵集所占的字节数
	for(i=0;i<size3;i++)
	{
		if(size1==16)
				{temp=Hzk1[num][i];}//调用16*16字体
		else if(size1==24)
				{temp=Hzk2[num][i];}//调用24*24字体
		else if(size1==32)       
				{temp=Hzk3[num][i];}//调用32*32字体
		else if(size1==64)
				{temp=Hzk4[num][i];}//调用64*64字体
		else return;
		for(m=0;m<8;m++)
		{
			if(temp&0x01)OLED_DrawPoint(x,y,mode);
			else OLED_DrawPoint(x,y,!mode);
			temp>>=1;
			y++;
		}
		x++;
		if((x-x0)==size1)
		{x=x0;y0=y0+8;}
		y=y0;
	}
}
#endif // __OLED_HZK__

#ifdef __OLED_GRAM__
#ifdef __OLED_HZK__
//num 显示汉字的个数
//space 每一遍显示的间隔
//mode:0,反色显示;1,正常显示
void OLED_ScrollDisplay(unsigned char num,unsigned char space)
{
	unsigned char i,n,t=0,m=0,r;
	while(1)
	{
		if(m==0)
		{
			OLED_ShowChinese(128,24,t); //写入一个汉字保存在OLED_GRAM[][]数组中
			t++;
		}
		if(t==num)
		{
			for(r=0;r<16*space;r++)      //显示间隔
			{
				for(n=0;n<8;n++)
				{
					for(i=1;i<128;i++)
					{
						OLED_GRAM[n][i - 1]=OLED_GRAM[n][i];
					}
				}
				OLED_Refresh();
			}
			t=0;
		}
		m++;
		if(m==16){m=0;}
		for(n=0;n<8;n++)
		{
			for(i=1;i<128;i++)   //实现左移
			{
				OLED_GRAM[n][i - 1]=OLED_GRAM[n][i];
			}
		}
		OLED_Refresh();
	}
}
#endif // __OLED_HZK__

//x,y：起点坐标
//sizex,sizey,图片长宽
//BMP[]：要写入的图片数组
//mode:0,反色显示;1,正常显示
void OLED_ShowPicture(unsigned char x,unsigned char y,unsigned char sizex,unsigned char sizey,unsigned char BMP[])
{
	unsigned short j=0;
	unsigned char i,n,temp,m;
	unsigned char x0=x,y0=y;
	unsigned char mode = pOledPaint->mode;
	sizey=sizey/8+((sizey%8)?1:0);
	for(n=0;n<sizey;n++)
	{
		 for(i=0;i<sizex;i++)
		 {
				temp=BMP[j];
				j++;
				for(m=0;m<8;m++)
				{
					if(temp&0x01)OLED_DrawPoint(x,y,mode);
					else OLED_DrawPoint(x,y,!mode);
					temp>>=1;
					y++;
				}
				x++;
				if((x-x0)==sizex)
				{
					x=x0;
					y0=y0+8;
				}
				y=y0;
     }
	 }
}
#endif
//OLED的初始化
void OLED_Init(void)
{
	oledPaint[0] = &defaultOledPaint;
#ifdef __OLED_MULTI_SIZE__
	oledPaint[1] = &invertOledPaint;
	oledPaint[2] = &bigOledPaint;
	oledPaint[3] = &bigInvertOledPaint;
#endif
	pOledPaint = oledPaint[0];

	OLED_WR_Byte(0xAE,OLED_CMD);//--turn off oled panel
	OLED_WR_Byte(0x00,OLED_CMD);//---set low column address
	OLED_WR_Byte(0x10,OLED_CMD);//---set high column address
	OLED_WR_Byte(0x40,OLED_CMD);//--set start line address  Set Mapping RAM Display Start Line (0x00~0x3F)
	OLED_WR_Byte(0x81,OLED_CMD);//--set contrast control register
	OLED_WR_Byte(0xCF,OLED_CMD);// Set SEG Output Current Brightness
	OLED_WR_Byte(0xA1,OLED_CMD);//--Set SEG/Column Mapping     0xa0���ҷ��� 0xa1����
	OLED_WR_Byte(0xC8,OLED_CMD);//Set COM/Row Scan Direction   0xc0���·��� 0xc8����
	OLED_WR_Byte(0xA6,OLED_CMD);//--set normal display
	OLED_WR_Byte(0xA8,OLED_CMD);//--set multiplex ratio(1 to 64)
	OLED_WR_Byte(0x3f,OLED_CMD);//--1/64 duty
	OLED_WR_Byte(0xD3,OLED_CMD);//-set display offset	Shift Mapping RAM Counter (0x00~0x3F)
	OLED_WR_Byte(0x00,OLED_CMD);//-not offset
	OLED_WR_Byte(0xd5,OLED_CMD);//--set display clock divide ratio/oscillator frequency
	OLED_WR_Byte(0x80,OLED_CMD);//--set divide ratio, Set Clock as 100 Frames/Sec
	OLED_WR_Byte(0xD9,OLED_CMD);//--set pre-charge period
	OLED_WR_Byte(0xF1,OLED_CMD);//Set Pre-Charge as 15 Clocks & Discharge as 1 Clock
	OLED_WR_Byte(0xDA,OLED_CMD);//--set com pins hardware configuration
	OLED_WR_Byte(0x12,OLED_CMD);
	OLED_WR_Byte(0xDB,OLED_CMD);//--set vcomh
	OLED_WR_Byte(0x40,OLED_CMD);//Set VCOM Deselect Level
	OLED_WR_Byte(0x20,OLED_CMD);//-Set Page Addressing Mode (0x00/0x01/0x02)
	OLED_WR_Byte(0x00,OLED_CMD);// 00horizontal 01vertical 02page
	OLED_WR_Byte(0x8D,OLED_CMD);//--set Charge Pump enable/disable
	OLED_WR_Byte(0x14,OLED_CMD);//--set(0x10) disable
	OLED_WR_Byte(0xA4,OLED_CMD);// Disable Entire Display On (0xa4/0xa5)
	OLED_WR_Byte(0xA6,OLED_CMD);// Disable Inverse Display On (0xa6/a7) 
	OLED_Clear();
	OLED_WR_Byte(0xAF,OLED_CMD);
}

#endif // __VTOR_OLED__