#include "main.h"

__C_task void main(void)
{
  uint8_t i;
  //uint8_t tmp8;

  /* Disable comaparator */
  ACSR = 1<<ACD;

  /* Init LED */
  LED_DDR = LED_1 | LED_2;
  LED_OFF(LED_1);
  LED_OFF(LED_2);

  /* Init 7 segment LED indicators */
  INDCTR_SEGMENT_DDR = 0xFF; // segment pins to output
  INDCTR_SEGMENT_PORT = 0x00; // off all segment

  INDCTR_COMMON_DDR = INDCTR_COMMON_ALL; // common pins to output
  INDCTR_COMMON_PORT = ~INDCTR_COMMON_ALL; // off all indikators

  for(i=0; i<INDCTR_NUMS; i++) {
    Display[i] = Sym_minus;
  }

  Flag.needDot = 0;

  /* Timer0 every 4 ms switch showing LED */
  TCCR0 = TIM0_PRESCALER;
  TCNT0 = TIM0_CNT; // load timer
  TIMSK |= 1<<TOIE0; // enable TIM0_OVF interrupt

  /* Timer1 ticks with 1 MHz frequency for DS18B20 delays */
  TCCR1B = TIM1_PRESCALER;
//  TCNT1 = 64536;
//  TIMSK |= 1<<TOIE1; // enable TIM1_OVF interrupt

  /* ADC init */
  resultADC = 0;
  ADMUX = (1<<REFS0) | (1<<ADLAR); // Vref = AVcc, channel ADC0, Left adjusted result
  ADCSR = 1<<ADEN | 1<<ADPS2 | 1<<ADPS1; // enable ADC, prescaler = 64

  /* I2C init */
  TWI_MasterInit(100);

  __enable_interrupt();

  /* Init Scheduler and TDelay */
  RTOS_Init();

  // Tasks for sheduler
  RTOS_SetTask(startADC,0,50);
  RTOS_SetTask(getTime,4,500);
  RTOS_SetTask(showHHMM,8,500);
  RTOS_SetTask(ds18b20_StartMeasure,12,10000);
  
  State = show_HHMM;

  while(1) {

    /* check time */
    if  (1 == Flag.newTime) {
      Flag.newTime = 0;
      
      if (show_HHMM == State) {
        // every 25 and 55 seconds shoe temperature for 5 seconds
        if ((25 == Clock.seconds) || (55 == Clock.seconds)) {
          RTOS_SetTask(showHHMM,5000,500);
          RTOS_SetTask(showTEMP,0,1000);
        }
      }
    }

    /* check buttons */
    if (1 == Flag.newBTN) {
      uint8_t mm, hh;
      
      Flag.newBTN = 0;
      
      btn_t btn = getBTN();
      switch (btn) {
      case btn_1:
        // show HH:MM
        State = show_HHMM;
        RTOS_SetTask(showTEMP,0,0);
        RTOS_SetTask(showMMSS,0,0);
        RTOS_SetTask(showHHMM,0,500);
        break;

      case btn_2:
        // show MM SS
        State = show_MMSS;
        Flag.needDot = 0;
        RTOS_SetTask(showHHMM,0,0);
        RTOS_SetTask(showTEMP,0,0);
        RTOS_SetTask(showMMSS,0,500);
        break;

      case btn_3:
        // show temperature
        State = show_TEMP;
        Flag.needDot = 0;
        RTOS_SetTask(showMMSS,0,0);
        RTOS_SetTask(showHHMM,0,0);
        RTOS_SetTask(showTEMP,0,1000);
        break;

      case btn_4:
        // time --
        mm = bcd2bin(Clock.minutes);
        hh = bcd2bin(Clock.hours);
        if (mm > 0) {
          mm --;
        } else {
          mm = 59;
          if (hh > 0) {
            hh --;
          } else {
            hh = 23;
          }
        }
        Clock.minutes = bin2bcd(mm);
        Clock.hours = bin2bcd(hh);
        break;

      case btn_5:
        // time ++
        mm = bcd2bin(Clock.minutes);
        hh = bcd2bin(Clock.hours);
        mm ++;
        if (mm > 59) {
          mm = 0;
          hh ++;
          if (hh > 23) {
            hh = 0;
          }
        }
        Clock.minutes = bin2bcd(mm);
        Clock.hours = bin2bcd(hh);
        break;

      case btn_6:
        // save new time
        Flag.blankIndktr = 0;
        setTime();
        RTOS_SetTask(getTime,4,500);
        break;

      case btn_7:
        // time set
        Flag.blankIndktr = 1;
        RTOS_SetTask(getTime,0,0);
        break;

      default:
        break;
      }
      
    } // end of new BTN

    tdelay_ms(10); // sleep and rotate dispatcher
  } // end of while(1)
} // end of main()

/**
 *  S u b r o u t i n e s
 */
static uint8_t bcd2bin(uint8_t bcd) {
  return (10*(bcd>>4)|(bcd&0x0f));
}

static uint8_t bin2bcd(uint8_t bin) {
  return (((bin/10)<<4)|(bin%10));
}

static void showHHMM(void) {
  State = show_HHMM;
  Display[0] = IndctrNums[(0x0F & (Clock.hours>>4))];
  Display[1] = IndctrNums[(0x0F & Clock.hours)];
  Display[2] = IndctrNums[(0x0F & (Clock.minutes>>4))];
  Display[3] = IndctrNums[(0x0F & Clock.minutes)];
}

static void showMMSS(void) {
  Display[0] = IndctrNums[(0x0F & (Clock.minutes>>4))];
  Display[1] = IndctrNums[(0x0F & Clock.minutes)];
  Display[2] = IndctrNums[(0x0F & (Clock.seconds>>4))];
  Display[3] = IndctrNums[(0x0F & Clock.seconds)];
}

static void showTEMP(void) {
  int8_t t = Temperature;
  
  if (t < 0) {
    Display[0] = Sym_minus;
    t = -t;
  } else {
    Display[0] = Sym_blank;
  }

  // convert to BCD
  t = bin2bcd(t);

  Display[1] = IndctrNums[(0x0F & (t>>4))];
  Display[2] = IndctrNums[(0x0F & t)];
  Display[3] = Sym_gradus;
  
  Flag.needDot = 0;
}

static void startADC(void) {
  // enable interrupt and start conversion
  ADCSR |= ((1<<ADSC) | (1<<ADIE));
}

static btn_t getBTN(void) {
  uint8_t btn_code = resultADC;
  if (btn_code > 0xED) {
    return btn_no;
  } else if (btn_code > 0xD8) {
    return btn_1;
  } else if (btn_code > 0xD0) {
    return btn_2;
  } else if (btn_code > 0xC5) {
    return btn_3;
  } else if (btn_code > 0xB4) {
    return btn_4;
  } else if (btn_code > 0x94) {
    return btn_5;
  } else if (btn_code > 0x3F) {
    return btn_6;
  } else {
    return btn_7;
  }
}

static void getTime(void) {
  static uint8_t old_sec;

  /* Sync with indicators */
  Flag.waitIndktr = 1;
  while (Flag.waitIndktr == 1);
 
  /*устанавливаем указатель DS1307 на нулевой адрес*/
  twi_buf[0] = (DS1307_ADR<<1)|0; //адресный пакет
  twi_buf[1] = 0;                 //адрес регистра   
  TWI_SendData(twi_buf, 2);

  /*считываем время с DS1307*/
  twi_buf[0] = (DS1307_ADR<<1)|1;
  TWI_SendData(twi_buf, 5);

  /*переписываем данные буфера драйвера в свой буфер*/
  TWI_GetData(twi_buf, 5);

  Clock.seconds = twi_buf[1];
  Clock.minutes = twi_buf[2];
  Clock.hours = twi_buf[3];
  
  if (show_HHMM == State) {
    if (old_sec != Clock.seconds) {
      Flag.needDot = 1;
    } else {
      Flag.needDot = 0;
    }
  }

  old_sec = Clock.seconds;
  Flag.newTime = 1;
}

static void setTime(void) {
  /* Sync with indicators */
  Flag.waitIndktr = 1;
  while (Flag.waitIndktr == 1);
 
   /* prepare new time */
  twi_buf[0] = (DS1307_ADR<<1)|0; //адресный пакет
  twi_buf[1] = 0;                 //адрес регистра
  twi_buf[2] = 0;                 //значение секунд 
  twi_buf[3] = Clock.minutes;     //значение минут 
  twi_buf[4] = Clock.hours;       //значение часов

  /* senr to rtc */
  TWI_SendData(twi_buf, 5);
}

/* DS18B20 functions */
static uint8_t ds18b20_Reset(void) {
  uint8_t i;

  /* Pull line low and wait for 480uS */
  DS18B20_LOW;
  DS18B20_OUTPUT_MODE;

  //ds18b20_delay(us(480));
  TCNT1 = 0;
  while (TCNT1 <= 500) {
  }

  /* Release line and wait for 60uS */
  DS18B20_INPUT_MODE;
  //ds18b20_delay(us(60));
  TCNT1 = 0;
  while (TCNT1 <= 70) {
  }

  /* Store line value and wait until the completion of 480uS period */
  i = DS18B20_VALUE;
  //ds18b20_delay(us(420));
  //OCR1B = 420;
  TCNT1 = 0;
  while (TCNT1 <= 430) {
  }

  /* Return the value read from the presence pulse (0=OK, else=WRONG) */
  return i;
}

static void ds18b20_WriteBit(uint8_t bit) {
  /* Pull line low for 1uS */
  DS18B20_LOW;
  DS18B20_OUTPUT_MODE;
  //ds18b20_delay(us(1));
  TCNT1 = 0;
  while (TCNT1 <= 1);

  /* If we want to write 1, release the line (if not will keep low) */
  if(1 == bit) {
    DS18B20_INPUT_MODE;
  }

  /* Wait for 60uS and release the line */
  //ds18b20_delay(us(60));
  TCNT1 = 0;
  while (TCNT1 <= 60) {
  }

  DS18B20_INPUT_MODE;
}

static uint8_t ds18b20_ReadBit(void) {
  uint8_t bit=0;

  /* Pull line low for 1uS */
  DS18B20_LOW;
  DS18B20_OUTPUT_MODE;
  //ds18b20_delay(us(1));
  TCNT1 = 0;
  while (TCNT1 <= 1);

  /* Release line and wait for 14uS */
  DS18B20_INPUT_MODE;
  //ds18b20_delay(us(14));
  TCNT1 = 0;
  while (TCNT1 <= 14) {
  }

  /* Read line value */
  if (DS18B20_VALUE != 0) {
    bit = 0x80;
  }

  /* Wait for 45uS to end and return read value */
  //ds18b20_delay(us(45));
  TCNT1 = 0;
  while (TCNT1 <= 45) {
  }

  return bit;
}

static uint8_t ds18b20_ReadByte(void) {
  uint8_t i=8, n=0;

  while (i--) {
    /* Shift one position right and store read value */
    n >>= 1;
    n |= ds18b20_ReadBit();
  }
  return n;
}

static void ds18b20_WriteByte(uint8_t byte) {
  uint8_t i=8;

  while (i--) {
    /* Write actual bit and shift one position right to make the next bit ready */
    ds18b20_WriteBit(byte & 0x01);
    byte >>= 1;
  }
}

static void ds18b20_StartMeasure(void) {
  /* Sync with indicators */
  Flag.waitIndktr = 1;
  while (Flag.waitIndktr == 1);
 
  /* Reset, skip ROM and start temperature conversion */
  if (ds18b20_Reset() != 0) {
    Temperature = -99;
    Flag.newTemp = 1;
  } else {
    ds18b20_WriteByte(DS18B20_CMD_SKIPROM);
    ds18b20_WriteByte(DS18B20_CMD_CONVERTTEMP);
    RTOS_SetTask(ds18b20_ReadTemperature, 800, 0);
  }
}

static void ds18b20_ReadTemperature(void) {
  uint8_t temperature[2];
  uint8_t decimal;

  /* Sync with indicators */
  Flag.waitIndktr = 1;
  while (Flag.waitIndktr == 1);
 
  /* Reset, skip ROM and send command to read Scratchpad */
  if (ds18b20_Reset() != 0) {
    Temperature = -98;
  } else {
    ds18b20_WriteByte(DS18B20_CMD_SKIPROM);
    ds18b20_WriteByte(DS18B20_CMD_RSCRATCHPAD);

    /* Read Scratchpad (only 2 first bytes) */
    temperature[0]=ds18b20_ReadByte();
    temperature[1]=ds18b20_ReadByte();

    //ds18b20_reset();

    /* Store temperature integer digits */
    Temperature = ((temperature[1]&0x7)<<4) | (temperature[0]>>4);

    // get decimal part
    decimal = (temperature[0] & 15);
    
    // convert to binary
    //decimal *= 10; //(tempDecimals << 1) + (tempDecimals << 3);// Умножаем на 10
    //decimal >>= 4; //(tempDecimals >> 4);//делим на 16 или умножаем на 0.0625
    
    // round integer part
    if (decimal > 7) {
      Temperature ++;
    }
  }
  
  Flag.newTemp = 1;
}


/**
 *  I n t e r r u p t   h a n d l e r s
 */

#pragma vector=TIMER0_OVF_vect
__interrupt void TIMER0_OVF_ISR(void) {
  static uint8_t indicator = 0;
  static uint8_t blank_time = 0; // blanking timer
  uint8_t flag = 1;

  TCNT0 = TIM0_CNT; // reload timer

  INDCTR_COMMON_PORT &= ~INDCTR_COMMON_ALL; // off all indikators

  Flag.waitIndktr = 0;

  if (1 == Flag.blankIndktr) {
    if (blank_time <= 200) {
      blank_time ++;
      flag = 1;
    } else if (blank_time <= 250) {
      blank_time ++;
      flag = 0;
    } else {
      blank_time = 0;
      flag = 1;
    }
  }

  if (1 == flag) {
    switch (indicator) {
    case 0:
      INDCTR_SEGMENT_PORT = Display[0];
      INDCTR_COMMON_PORT |= 0x01;
      indicator = 1;
      break;

    case 1:
      if (1 == Flag.needDot) {
        INDCTR_SEGMENT_PORT = Display[1] | Sym_dot;
      } else {
        INDCTR_SEGMENT_PORT = Display[1];
      }
      INDCTR_COMMON_PORT |= 0x02;
      indicator = 2;
      break;

    case 2:
      INDCTR_SEGMENT_PORT = Display[2];
      INDCTR_COMMON_PORT |= 0x04;
      indicator = 3;
      break;

    case 3:
      INDCTR_SEGMENT_PORT = Display[3];
      INDCTR_COMMON_PORT |= 0x08;

    default:
      indicator = 0;
      break;
    } // end of switch
  }
}

#pragma vector=TIMER1_COMPA_vect
__interrupt void TIMER1_COMPA_ISR(void) {
}

#pragma vector=TIMER1_COMPB_vect
__interrupt void TIMER1_COMPB_ISR(void) {
}

#pragma vector=TIMER1_OVF_vect
__interrupt void TIMER1_OVF_ISR(void) {
  Flag.T1OC = 1;
}

#pragma vector=ADC_vect
__interrupt void ADC_ISR(void) {
  resultADC = ADCH;
  ADCSR &= ~(1<<ADIE); // disable interrupt
  Flag.newBTN = 1;
}