selections.c

#include <stdlib.h>
#include <avr/pgmspace.h>
#include "config.h"
#include <util/delay.h>
#include "TFT.h"
#include "TouchScreen.h"
#include "twicam.h"
#include "camregdef.h"
#include "MT9D111_regs.h"
#include "captureimage.h"
#include "twicam.h"
#include "gammaedit.h"
#include "regedit.h"
#ifdef haveSDcard
#include "mmc.h"
#endif
#include "filebrowser.h"
#include "exiticon.h"
#ifndef MT9D111
static void setmatrix(uint8_t id){
	switch (id) {
		case 0:
			#ifdef ov7670
				wrReg(MTX1,0x80);        
				wrReg(MTX2,0x80);      
				wrReg(MTX3,0x00);        
				wrReg(MTX4,0x22);        
				wrReg(MTX5,0x5e);        
				wrReg(MTX6,0x80);        
				wrReg(MTXS,0x9e); 
			#elif defined ov7740
				wrReg(ISP_CTRL01,rdReg(ISP_CTRL01)&(~ISP_CTRL01_CMX_enable));
			#endif
		break;
		case 1:
			#ifdef ov7670
				wrReg(MTX1,0x40);        
				wrReg(MTX2,0x34);      
				wrReg(MTX3,0x0c);        
				wrReg(MTX4,0x17);
				wrReg(MTX5,0x29);        
				wrReg(MTX6,0x40);
				wrReg(MTXS,0x1A);
			#elif defined ov7740
				wrReg(ISP_CTRL01,rdReg(ISP_CTRL01)|ISP_CTRL01_CMX_enable);
			#endif
		break;
		#ifdef ov7670
		case 2:
			wrReg(MTX1,0xB3);        
			wrReg(MTX2,0xB3);      
			wrReg(MTX3,0x0);        
			wrReg(MTX4,0x3D);        
			wrReg(MTX5,0xA7);        
			wrReg(MTX6,0xE4);
			wrReg(MTXS,0x9E);
		break;
		#endif
	}
}
#endif
const char menu0[] PROGMEM = "Register edit";
#ifdef MT9D111
const char menu1[] PROGMEM = "Reg edit micro";
#else
const char menu1[] PROGMEM = "Pick a matrix";
#endif
const char menu2[] PROGMEM = "Reset Reg";
const char menu3[] PROGMEM = "Compare Matrix";
const char menu4[] PROGMEM = "QQVGA preview";
#ifndef MT9D111
const char menu5[] PROGMEM = "gamma edit";
#else
const char menu5[] PROGMEM = "Fixme";
#endif
const char menu6[] PROGMEM = "File Browser";
#ifdef ov7670
const char menu7[] PROGMEM = "White balance";
#else
const char menu7[] PROGMEM = "FIXME!";
#endif
const char menu8[] PROGMEM = "Image to PC";
const char menu9[] PROGMEM = "Next Page";

const char *const menu_table[] PROGMEM = {
	menu0,menu1,menu2,menu3,
	menu4,menu5,menu6,menu7,
	menu8,menu9
};
const char menup20[] PROGMEM = "Touch Test";
const char menup21[] PROGMEM = "Time Lapse";
const char menup22[] PROGMEM = "Previous Page";
const char *const menu_tablep2[] PROGMEM = {
	menup20,menup21,menup22
};
#ifdef ov7670
const char maxtrix0[] PROGMEM = "Matrix YUV422";
const char maxtrix1[] PROGMEM = "Matrix 2";
const char maxtrix2[] PROGMEM = "Matrix RGB565";
const char *const maxtrix_table[] PROGMEM = {maxtrix0,maxtrix1,maxtrix2};
#elif defined ov7740
const char maxtrix0[] PROGMEM = "Matrix off";
const char maxtrix1[] PROGMEM = "OVT Matrix";
const char *const maxtrix_table[] PROGMEM = {maxtrix0,maxtrix1};
#endif
const char wb0[] PROGMEM="No AWB";
const char wb1[] PROGMEM="Advanced AWB";
const char wb2[] PROGMEM="Simple AWB";
const char wb3[] PROGMEM="Sunny";
const char wb4[] PROGMEM="Cloudy";
const char wb5[] PROGMEM="Office";
const char wb6[] PROGMEM="Home";
const char *const wb_table[] PROGMEM={wb0,wb1,wb2,wb3,wb4,wb5,wb6};
#ifdef MT9D111
const char res0[] PROGMEM="UXGA jpeg";
const char res1[] PROGMEM="SVGA";
const char res2[] PROGMEM="QVGA";
const char res3[] PROGMEM="Main Menu";
const char *const res_tab[] PROGMEM={res0,res1,res2,res3};
#else
const char res0[] PROGMEM="VGA";
const char res1[] PROGMEM="QVGA";
const char res2[] PROGMEM="Main Menu";
const char *const res_tab[] PROGMEM={res0,res1,res2};
#endif
uint8_t selection(const char ** table,uint8_t maxitems){
	uint8_t item;
	uint16_t x,y,z;
	tft_setOrientation(1);
	tft_setXY(0,0);
	CS_LOW;
    RS_HIGH;
    RD_HIGH;
    DDRA=0xFF;
    z=31<<8;
	item=240;
	while(item--){
		x=320;
		while(x--){
			WR_HIGH;
			PORTA=0;
			WR_LOW;
			WR_HIGH;
			PORTA=z>>8;
			WR_LOW;
		}
		z-=33;
	}
	tft_setDisplayDirect(DOWN2UP);
	item=0;
	for(x=0;x<maxitems*24;x+=24){
		tft_drawStringP((PGM_P)pgm_read_word(&(table[item])),x,320,3,WHITE);
		++item;
	}
	while(1){
		do{
			getPoint(&x,&y,&z);
		}while(z<10);
		item=x/24;
		if(item<maxitems)
			break;
	}
	tft_drawStringP((PGM_P)pgm_read_word(&(table[item])),item*24,320,3,RED);
	_delay_ms(100);
	tft_paintScreenBlack();
	return item;
}
#ifdef ov7670
const char config0[] PROGMEM = "Linux driver";
const char config1[] PROGMEM = "suwa-koubou";
const char config2[] PROGMEM = "Arducam";
const char *const config_tab[] PROGMEM = {config0,config1,config2};
void configSel(void){
	initCam(selection((const char**)config_tab,3));
}
#endif

#ifdef MT9D111
static const uint8_t jpegHeader[] PROGMEM ={//619 bytes
255,216,255,224,0,16,74,70,73,70,0,1,2,0,0,1,0,1,0,0,255,219,0,132,0,28,19,21,25,21,18,28,
25,23,25,32,30,28,33,42,70,46,42,39,39,42,86,61,65,51,70,102,89,107,105,100,89,98,96,112,126,161,137,112,
119,152,121,96,98,140,191,142,152,166,172,180,182,180,109,135,198,212,196,175,210,161,177,180,173,1,30,32,32,42,37,42,
82,46,46,82,173,116,98,116,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,
173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,173,255,192,0,17,8,4,
176,6,64,3,0,33,0,1,17,1,2,17,1,255,196,0,31,0,0,1,5,1,1,1,1,1,1,0,0,0,0,0,
0,0,0,1,2,3,4,5,6,7,8,9,10,11,255,196,0,181,16,0,2,1,3,3,2,4,3,5,5,4,4,0,
0,1,125,1,2,3,0,4,17,5,18,33,49,65,6,19,81,97,7,34,113,20,50,129,145,161,8,35,66,177,193,21,
82,209,240,36,51,98,114,130,9,10,22,23,24,25,26,37,38,39,40,41,42,52,53,54,55,56,57,58,67,68,69,70,
71,72,73,74,83,84,85,86,87,88,89,90,99,100,101,102,103,104,105,106,115,116,117,118,119,120,121,122,131,132,133,134,
135,136,137,138,146,147,148,149,150,151,152,153,154,162,163,164,165,166,167,168,169,170,178,179,180,181,182,183,184,185,186,194,
195,196,197,198,199,200,201,202,210,211,212,213,214,215,216,217,218,225,226,227,228,229,230,231,232,233,234,241,242,243,244,245,
246,247,248,249,250,255,196,0,31,1,0,3,1,1,1,1,1,1,1,1,1,0,0,0,0,0,0,1,2,3,4,5,
6,7,8,9,10,11,255,196,0,181,17,0,2,1,2,4,4,3,4,7,5,4,4,0,1,2,119,0,1,2,3,17,
4,5,33,49,6,18,65,81,7,97,113,19,34,50,129,8,20,66,145,161,177,193,9,35,51,82,240,21,98,114,209,10,
22,36,52,225,37,241,23,24,25,26,38,39,40,41,42,53,54,55,56,57,58,67,68,69,70,71,72,73,74,83,84,85,
86,87,88,89,90,99,100,101,102,103,104,105,106,115,116,117,118,119,120,121,122,130,131,132,133,134,135,136,137,138,146,147,
148,149,150,151,152,153,154,162,163,164,165,166,167,168,169,170,178,179,180,181,182,183,184,185,186,194,195,196,197,198,199,200,
201,202,210,211,212,213,214,215,216,217,218,226,227,228,229,230,231,232,233,234,242,243,244,245,246,247,248,249,250,255,218,0,
12,3,0,0,1,17,2,17,0,63,0};
#endif
static inline void serialWrB(uint8_t dat){
	UDR0=dat;
	while (!( UCSR0A & (1<<UDRE0))); //wait for byte to transmit
}
static void BSend(void){
	uint16_t res=tft_readRegister(0x22);
	serialWrB(res>>8);
	serialWrB(res&255);
}
void menu(void){
	uint16_t x,y,z;
	while (1){
		switch (selection((const char**)menu_table,10)){
			case 0:
			#ifdef MT9D111
				setRes(QVGA);
				setColor(RGB565);
				MT9D111Refresh();
				editRegs(0);
			#else
				setColor(RGB565);
				setRes(QVGA);
				editRegs();
			#endif
		break;
		case 1:
			#ifdef MT9D111
				setRes(QVGA);
				setColor(RGB565);
				MT9D111Refresh();
				editRegs(1);
			#else
				#ifdef ov7740
					setmatrix(selection((const char**)maxtrix_table,2));
				#else
					setmatrix(selection((const char**)maxtrix_table,3));
				#endif
				tft_setOrientation(1);
				capImg();
				tft_setDisplayDirect(DOWN2UP);
			#endif
		break;
		case 2:
			#ifdef ov7670
				initCam(0);
			#else
				initCam();
			#endif
		break;
		#ifndef MT9D111
		case 3:
			//compare matrices
			tft_setOrientation(1);
			do{
				getPoint(&x,&y,&z);
				uint8_t a;
				#ifdef ov7670
					for (a=0;a<3;a++){
				#elif defined ov7740
					for (a=0;a<2;a++){
				#endif
					setmatrix(a);
					capImg();
				}
			}while (z < 10);
			tft_setDisplayDirect(DOWN2UP);
		break;
		#endif
			case 4:
				setColor(RGB565);
				setRes(QQVGA);
				#ifdef MT9D111
					MT9D111Refresh();
				#endif
				do{
					getPoint(&x,&y,&z);
					tft_setOrientation(1);
					capImgqqvga(160);
					tft_setDisplayDirect(DOWN2UP);
				}while(z<10);
				setRes(QVGA);
			break;
			case 5:
				#ifdef ov7670
					setColor(RGB565);
				#endif
				gammaEdit();
			break;
			case 6:
				//File browser
				//start by listing files
				#ifdef haveSDcard
					browserSD();
				#else
					tft_drawStringP(PSTR("No SD card"),16,320,3,WHITE);
					_delay_ms(666);
				#endif
			break;
			case 7:
				#ifdef ov7670
				{uint8_t pick=selection((const char**)wb_table,7);//registers from http://thinksmallthings.wordpress.com/2012/10/25/white-balance-control-with-ov7670/
				if(pick==1||pick==2){
					wrReg(0x13, 0xE7);
					wrReg(0x6F, 0x9E|(pick&1));
				}else{
					wrReg(0x13, 0xE5);
					switch(pick){
						case 2:
							wrReg(0x01, 0x5A);
							wrReg(0x02, 0x5C);
						break;
						case 3:
							wrReg(0x01, 0x58);
							wrReg(0x02, 0x60);
						break;
						case 4:
							wrReg(0x01, 0x84);
							wrReg(0x02, 0x4C);
						break;
						case 5:
							wrReg(0x01, 0x96);
							wrReg(0x02, 0x40);
						break;
					}
				}
				tft_setOrientation(1);
				setRes(QVGA);
				setColor(RGB565);
				capImg();
				tft_setDisplayDirect(DOWN2UP);}
				#endif
			break;
			case 8:
				#ifdef ov7670
				wrReg(0x1e,rdReg(0x1e)&(~(1<<5))&(~(1<<4)));
				#endif
				{
				#ifdef MT9D111
					uint8_t reso=selection((const char**)res_tab,4);
				#else
					uint8_t reso=selection((const char**)res_tab,3);
				#endif
				#ifdef MT9D111
					wrReg16(0xF0,2);//page 2
					wrReg16(0x0D,0);
					setColor(YUV422);
				#endif
				switch(reso){
					case 0:
						#ifdef ov7670
							wrReg(REG_COM7, COM7_BAYER); // BGBGBG... GRGRGR...
						#elif defined MT9D111
							//setup jpeg
							MT9D111JPegCapture();
						#endif
					break;
					case 1:
						#ifdef MT9D111
							setRes(SVGA);
						#else
							setRes(QVGA);
							setColor(YUV422);
						#endif
					break;
					#ifdef MT9D111
					case 2:
						setRes(QVGA);
					break;
					#endif
					default:
						goto theEnd;
					break;
				}
				#ifdef ov7670
					_delay_ms(200);
					if(reso)
						wrReg(0x11,1);
					else
						wrReg(0x11,2);
				#endif
				tft_setOrientation(1);
				do{
					#ifdef MT9D111
						switch(reso){
							case 0:
								{uint32_t jpgSize=capJpeg();
								serialWrB('R');
								serialWrB('D');
								serialWrB('Y');
								uint16_t w;
								uint8_t h=0;
								serialWrB(jpgSize&255);
								serialWrB(jpgSize>>8);
								serialWrB(jpgSize>>16);
								serialWrB(jpgSize>>24);
								while(jpgSize){
									if(jpgSize>=640){
										for (w=0;w<320;++w){
											tft_setXY(h,w);
											BSend();
										}
										++h;
										jpgSize-=640;
									}else{
										for(w=0;w<jpgSize/2;++w){
											tft_setXY(h,w);
											BSend();
										}
										if(jpgSize&1){
											tft_setXY(h,w);
											uint16_t res=tft_readRegister(0x22);
											serialWrB(res>>8);
										}
										jpgSize=0;
									}
								}
								}
							break;
							case 1:
								capImgPC();
							break;
							case 2:
								capImgPCqvga();
							break;
						}
					#else
						if(reso)
							capImgPCqvga();
						else
							capImgPC();
					#endif
					getPoint(&x,&y,&z);
				}while(z<10);
theEnd:
				tft_setDisplayDirect(DOWN2UP);
				}
			break;
			case 9:
				switch(selection((const char**)menu_tablep2,3)){
					case 0:
						{
							tft_drawImage_P(exit_icon,32,32,0,0);
							uint16_t x1,y1;
							do{
								getPoint(&x,&y,&z);
							}while(z<10);
							if((y<=32)&&(x<=32))
									break;
							tft_fillCircle(x,y,4,WHITE);
							while(1){
								x1=x;
								y1=y;
								do{
									getPoint(&x,&y,&z);
								}while(z<10);
								tft_fillRectangle(224,320,16,36,BLACK);
								tft_fillCircle(x1,y1,4,BLACK);
								tft_fillCircle(x,y,4,WHITE);
								if((y<=32)&&(x<=32))
									break;
								char temp[6];
								utoa(x,temp,10);
								tft_drawString(temp,224,320,1,WHITE);
								utoa(y,temp,10);
								tft_drawString(temp,232,320,1,WHITE);
							}
						}
					break;
					case 1:
						//time lapse
						#ifdef MT9D111
							//MT9D111Refresh();
							//Since This is a time lapse we want to be in "video" mode
							MT9D111JPegCapture();
							/*do{
								_delay_ms(10);
								wrReg16(0xC6,(1<<15)|(1<<13)|(1<<8)|4);
							}while(rdReg16(0xC8)<4);
							waitStateMT9D111(3);*/
						#endif
						#ifdef haveSDcard
						{
							char buf[24];
							uint16_t imgc=0;
							tft_setOrientation(1);
							do{
								FIL Fo;
								#ifdef MT9D111
									uint32_t jpgSize=capJpeg();
								#else
									capImg();
								#endif
								utoa(imgc,buf,10);
								#ifdef MT9D111
								strcat(buf,".JPG");
								#else
								strcat(buf,".RAW");
								#endif
								f_open(&Fo,buf,FA_WRITE|FA_CREATE_ALWAYS);
								++imgc;
								UINT written;
								uint16_t w;
								uint8_t h;
								uint16_t cpybuf[320];
								#ifdef MT9D111
									h=0;
									uint8_t * cpyptr=cpybuf;
									for(w=0;w<619;++w)
										*cpyptr++=pgm_read_byte_near(jpegHeader+w);
									f_write(&Fo,cpybuf,619,&written);
									while(jpgSize){
										if(jpgSize>=640){
											for (w=0;w<320;++w){
												tft_setXY(h,w);
												cpybuf[w]=__builtin_bswap16(tft_readRegister(0x22));//Either bytes need to be swapped or a byte is being missed
											}
											f_write(&Fo,cpybuf,640,&written);
											++h;
											jpgSize-=640;
										}else{
											for(w=0;w<jpgSize/2;++w){
												tft_setXY(h,w);
												cpybuf[w]=__builtin_bswap16(tft_readRegister(0x22));
											}
											f_write(&Fo,cpybuf,jpgSize,&written);
											if(jpgSize&1){
												tft_setXY(h,w);
												cpybuf[w]=tft_readRegister(0x22);
												f_write(&Fo,&cpybuf[w],1,&written);
											}
											jpgSize=0;
										}
									}
									cpybuf[0]=0xFFD9;
									f_write(&Fo,cpybuf,2,&written);
								#else
								for (h=0;h<240;++h){
									for (w=0;w<320;++w){
										tft_setXY(h,w);
										cpybuf[w]=tft_readRegister(0x22);
									}
									f_write(&Fo,cpybuf,640,&written);
								}
								#endif
								f_close(&Fo);
								getPoint(&x,&y,&z);
							}while(z<10);
							tft_setDisplayDirect(DOWN2UP);
							#ifdef MT9D111
								MT9D111DoPreview();
							#endif
						}
						#else
							tft_drawStringP(PSTR("No SD card"),16,320,3,WHITE);
							_delay_ms(666);
						#endif
					break;
					case 2:
						//previous page
					break;
				}
			break;
		}
	}
}