// Programmer : Ts. Abd Rahim Bin Kasiman
// Date       : 28 January 2025
// This is a template program for Arduino Multi Learning Sheild

#define dataS(x)	digitalWrite(8, x? HIGH:LOW)
#define shCLK(x) 	digitalWrite(7, x? HIGH:LOW)
#define stCLK(x) 	digitalWrite(4, x? HIGH:LOW)

#define BUZ(x) 		digitalWrite(3, x? LOW:HIGH)

#define LED1(x) 	digitalWrite(13, x? LOW:HIGH)
#define LED2(x) 	digitalWrite(12, x? LOW:HIGH)
#define LED3(x) 	digitalWrite(11, x? LOW:HIGH)
#define LED4(x) 	digitalWrite(10, x? LOW:HIGH)

#define S1 			digitalRead(A1)
#define S2 			digitalRead(A2)
#define S3 			digitalRead(A3)

// Array data & variables used for 7-segment Common Anode - numeric value only
//                  INDEX:  0    1    2    3    4    5    6    7    8    9   10,  11,  12,  13   14   15   16   17   18   19   20   21   22  23   24   25
//                    NUM:  0    1    2    3    4    5    6    7    8    9 BLNK,   E    L    C   G,   o,   d,   E,   r,   c,   -,   U,   P,   n,   t,   F
const uint8_t num[30] = {0xC0,0XF9,0xA4,0xB0,0x99,0x92,0x82,0xF8,0x80,0x90,0xFF,0x86,0xC7,0xC6,0xC2,0xA3,0xA1,0x86,0xAF,0xA7,0xBF,0xC1,0x8C,0xAB,0x87,0x8E};
const uint8_t seg[4]	= {0xF1,0xF2,0xF4,0xF8};
uint16_t ptr2disp;
uint16_t buff[4];

int ledSequence1[]={1,2,4,8,9,10,12,13,14,15,14,12,8,0,8,4,2,1,9,5,3,11,7,15,7,3,1,0};
int ledSequence2[]={1,3,6,12,6,3,6,12,8,0,8,12,6,3,6,12,6,3,1,0};

uint8_t getPB(void)
{	uint8_t PBvalue=0;
	if(S1==LOW)	PBvalue |= 1<<0;	//1
	if(S2==LOW)	PBvalue |= 1<<1;	//2
	if(S3==LOW)	PBvalue |= 1<<2;	//4
	return PBvalue;
}
void LEDs(uint8_t decimal)
{	LED1(decimal & 1<<0);
	LED2(decimal & 1<<1);
	LED3(decimal & 1<<2);
	LED4(decimal & 1<<3);
}
void displaySegment(void)
{	static uint8_t seg=0;

	stCLK(LOW);
	for(int8_t i=15;i>-1;i--)
	{	shCLK(LOW);
		if(buff[seg] & (1<<i))	dataS(HIGH);
		else					dataS(LOW);
		shCLK(HIGH);
	}
	stCLK(HIGH);
	if(++seg>3) seg=0;
}
void multiplexSegment(uint16_t num4digit)
{	for(uint8_t i=0;i<4;i++)
	{	switch(i)
		{	case 0:	// DIGIT 1
				ptr2disp = num4digit / 1000;
				break;
			case 1: // DIGIT 2
				ptr2disp = (num4digit % 1000) / 100;
				break;
			case 2: // DIGIT 3
				ptr2disp = (num4digit % 100) / 10;
				break;
			case 3: // DIGIT 4
				ptr2disp = num4digit % 10;
				break;
		}
		buff[i] = (num[ptr2disp]<<8) | seg[i];
	}
}
void beep(uint8_t bil, uint16_t tempoh)
{ for(uint8_t i=0;i<bil;i++)
  { BUZ(1); delay(tempoh);
    BUZ(0); delay(tempoh);
  }
}

void pinSetup()
{ pinMode(8,OUTPUT);  //DS
  pinMode(7,OUTPUT);  //SH
  pinMode(4,OUTPUT);  //ST

  pinMode(3,OUTPUT);  //BUZZ
  
  pinMode(13,OUTPUT); //LED1
  pinMode(12,OUTPUT); //LED2
  pinMode(11,OUTPUT); //LED3
  pinMode(10,OUTPUT); //LED4
  
  pinMode(A1,INPUT); //PB1
  pinMode(A2,INPUT); //PB2
  pinMode(A3,INPUT); //PB3

	timerSetup();
}

void timerSetup()
{ cli(); 
	TCCR1A = 0; 
	TCCR1B = 0; 
	TCNT1  = 0; 
	OCR1A = 15999; 
	TCCR1B |= (1 << WGM12);
	TCCR1B |= (0 << CS12) | (0 << CS11) | (1 << CS10);
	TIMSK1 |= (1 << OCIE1A);
	sei(); 
}

void setup()
{ pinSetup();
  beep(3,50);

  Serial.begin(9600);

}
uint8_t a=0;

int an;
void loop()
{ LEDs(10);
	delay(500);
	multiplexSegment(a++);
	displaySegment();
	Serial.println(getPB());
}

ISR(TIMER1_COMPA_vect)
{ displaySegment();
}