#include <TVout.h>
#include "a.h"
#include "u.h"
#include "o.h"
#include "m.h"
#include "e.h"
#include "dance1.h"
#include "dance2.h"
#include "dance3.h"
#include "dance4.h"
#include "disturbance1.h"
#include "disturbance2.h"
#include "disturbance3.h"
#include "font6x8.h"
#include <IRremote.h>

TVout tv;
IRrecv receiver(2);

int tv_state = 0, last_tv_state = 0;  //  initially off

void starting_animation(){
  for (int i = 0; i < 5; i++){
    tv.bitmap(20,16,d1);
    tv.delay_frame(3);  //  i frame = 20ms (PAL tv)
    tv.bitmap(20,16,d2);
    tv.delay_frame(3);
    tv.bitmap(20,16,d3);
    tv.delay_frame(3);
  }
  tv.clear_screen();
  tv.delay(100);
}

void disable_receiver(){
  receiver.disableIRIn();  //  disable receiver, enable tv
  tv.begin(PAL);
}

void enable_receiver(){
  tv.end();  //  disable tv, enable receiver
  receiver.enableIRIn();
}

void speaking_animation(int x, int y){
  char message [] = "Robot World News";
  tv.print(30,tv.vres()-10,message);
  for (int i = 0; i < 20; i++){
    int frame = random(0,5);
    if (frame  ==  0)tv.bitmap(x,y,a);
    else if (frame  ==  1)tv.bitmap(x,y,u);
    else if (frame  ==  2)tv.bitmap(x,y,o);
    else if (frame  ==  3)tv.bitmap(x,y,m);
    else if (frame  ==  4)tv.bitmap(x,y,e);
    tv.delay_frame(5);
  }
  tv.clear_screen();
  tv.delay(100);
}

void setup(){
  Serial.begin(9600);
  tv.select_font(font6x8);  //  selecting font

  enable_receiver();

  // tv.begin(PAL);
  // tv.clear_screen();
  // tv.delay(100);
  // starting_animation();
  // tv.clear_screen();
  // tv.delay(100);
  // speaking_animation(35,0);
  // tv.clear_screen();

}

void loop(){

  if (receiver.decode()){
    int response = receiver.decodedIRData.command;
    if (response  ==  162)tv_state = !tv_state;  //  flip the state
    television(response);
    receiver.resume();
  }
}

void television(int code){
   disable_receiver();
  if (tv_state){  //  tv is on
    if (last_tv_state  ==  0){  //  tv is just turned on
      starting_animation();
      select_channel();
      last_tv_state = tv_state;
    }
    else if (last_tv_state  ==  1){
      if (code  ==  104)news_channel();
      else if (code  ==  48)dance_channel(30,0);
    }
  }
  else{  //  tv is off
    tv.clear_screen();
    tv.delay(100);
    last_tv_state = tv_state;
  }
  enable_receiver();
}

void select_channel(){
  tv.println(30,40, "SELECT A CHANNEL");
  tv.println(45,50,"0 : NEWS");
  tv.println(45,60,"1 : DANCE");
  tv.delay(100);
}

void news_channel(){
  speaking_animation(30,0);
  select_channel();
}

void dance_channel(int x, int y){
   for (int i = 0; i < 20; i++){
    int frame = random(0,5);
    if (frame  ==  0)tv.bitmap(x,y,dance1);
    else if (frame  ==  1)tv.bitmap(x,y,dance2);
    else if (frame  ==  2)tv.bitmap(x,y,dance3);
    else if (frame  ==  3)tv.bitmap(x,y,dance4);
  
    tv.delay_frame(5);
  }
  tv.clear_screen();
  tv.delay(100);
  select_channel();
}