examples/spi-flash-client/main.c

   1 /*
   2  * Copyright 2015, Olivier MATZ <zer0@droids-corp.org>
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of the University of California, Berkeley nor the
  13  *       names of its contributors may be used to endorse or promote products
  14  *       derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <avr/io.h>
  29 #include <avr/interrupt.h>
  30 #include <avr/pgmspace.h>
  31
  32 #include <stdio.h>
  33 #include <ctype.h>
  34
  35 #include <ucg_irq.h>
  36 #include <ucg_delay.h>
  37
  38 #include "uart.h"
  39
  40 static void ss_high(void)
  41 {
  42         printf("ss_high\r\n");
  43         PORTB |= (1 << 2);
  44 }
  45
  46 static void ss_low(void)
  47 {
  48         printf("ss_low\r\n");
  49         PORTB &= (~(1 << 2));
  50 }
  51
  52 static uint8_t spi_sendrecv(uint8_t tx)
  53 {
  54         uint8_t rx;
  55
  56         SPDR = tx;
  57
  58         /* Wait for transmission complete */
  59         while ((SPSR & (1 << SPIF)) == 0)
  60                 ;
  61
  62         ucg_delay_ms(100);
  63         rx = SPDR;
  64
  65         printf("sent 0x%2.2x, recvd 0x%2.2x '%c'\r\n",
  66                 tx, rx, isprint(rx) ? rx : '.');
  67         ucg_delay_ms(1000);
  68
  69         return rx;
  70 }
  71
  72 int main(void)
  73 {
  74         uint8_t i;
  75
  76         /* spi: SS (PB2), MOSI (PB3), SCK (PB5) */
  77         DDRB = (1 << 2) | (1 << 3) | (1 << 5);
  78
  79         ss_high();
  80
  81         /* blink led before start (unfortunatly it's on SCK, we can't
  82          * use it during spi transfer) */
  83         for (i = 0; i < 3; i++) {
  84                 PORTB |= (1 << 5);
  85                 ucg_delay_ms(500);
  86                 PORTB &= (~(1 << 5));
  87                 ucg_delay_ms(500);
  88         }
  89
  90         uart_init();
  91
  92         ucg_irq_unlock();
  93
  94         printf("hello\r\n");
  95
  96 #if 0 /* test serial (echo) */
  97         {
  98                 char c;
  99                 int ret;
 100
 101                 while (1) {
 102                         ret = fread(&c, 1, 1, stdin);
 103                         if (ret == 1)
 104                                 fwrite(&c, 1, 1, stdout);
 105                 }
 106         }
 107 #endif
 108
 109         /* remove power reduction on spi */
 110         PRR &= ~(1 << PRSPI);
 111
 112         /* Enable SPI, Master, set clock rate fck/16 */
 113         SPCR = (1 << SPE) | (1 << MSTR) | (1 << SPR0);
 114
 115         ss_low();
 116         ucg_delay_ms(1000);
 117
 118         /* read 5 bytes at address 0x1000 */
 119         spi_sendrecv(0x03);
 120         spi_sendrecv(0x00);
 121         spi_sendrecv(0x10);
 122         spi_sendrecv(0x00);
 123         for (i = 0; i < 12; i++)
 124                 spi_sendrecv(0x00); /* data 0 to 12 */
 125         ss_high();
 126
 127         while (1);
 128         return 0;
 129 }