1

/** Serial modes */

     2

static enum {

     3

    SERIAL_MODE_ASYNC   = 0b00,

     4

} serial_mode = SERIAL_MODE_ASYNC;

     5

     6

static enum {

     7

    SERIAL_BAUD_9600    = 103,

     8

} serial_baud = SERIAL_BAUD_9600;

     9

    10

static enum {

    11

    SERIAL_PARITY_N     = 0b00,

    12

} serial_parity = SERIAL_PARITY_N;

    13

    14

static enum {

    15

    SERIAL_STOPBITS_1   = 0b0,

    16

} serial_stopbits = SERIAL_STOPBITS_1;

    17

    18

static enum {

    19

    SERIAL_CHARS_8      = 0b011,

    20

} serial_chars = SERIAL_CHARS_8;

    21

    22

/** Serial state */

    23

char serial_rx = 0;

    24

char serial_tx = 0;

    25

    26

void serial_enable (char rx, char tx);

    27

    28

/*

    29

 * Setup the UART for serial mode.

    30

 */

    31

void serial_init ()

    32

{

    33

    UCSR0A = (

    34

            // Double the USART Transmission Speed

    35

            (0b0 << U2X0)   // single speed

    36

    37

            // Multi-processor Communication Mode

    38

        |   (0b0 << MPCM0)  // off

    39

    );

    40

    UCSR0B = (

    41

            // Character Size

    42

            ((serial_chars >> 2) << UCSZ02) // standard character sizes

    43

    );

    44

    UCSR0C = (

    45

            // USART Mode Select

    46

            (serial_mode << UMSEL00) // async

    47

    48

            // Parity Mode

    49

        |   (serial_parity << UPM00)

    50

    51

            // Stop Bit Select

    52

        |   (serial_stopbits << USBS0)

    53

    54

            // Character Size

    55

        |   ((serial_chars & 0b11) << UCSZ00)

    56

    57

            // Clock Polarity

    58

        |   (0b0 << UCPOL0)

    59

    );

    60

    UBRR0 = serial_baud;

    61

    62

    // active RX, idle TX

    63

    serial_enable(1, 0);

    64

}

    65

    66

void serial_enable (char rx, char tx)

    67

{

    68

    UCSR0B = (

    69

            // RX Complete Interrupt Enable

    70

            (!!rx << RXCIE0)

    71

    72

            // USART Data Register Empty Interrupt Enable

    73

        |   (!!tx << UDRIE0)

    74

    75

            // Receiver Enable

    76

        |   (0b1 << RXEN0)  // enabled

    77

    78

            // Transmitter Enable

    79

        |   (0b1 << TXEN0)  // enabled

    80

    81

            // Character Size

    82

        |   ((serial_chars >> 2) << UCSZ02) // standard character sizes

    83

    );

    84

}

    85

    86

/*

    87

 * Read incoming into buffer.

    88

 */

    89

ISR(USART_RX_vect)

    90

{

    91

    serial_rx = UDR0;

    92

}

    93

    94

/*

    95

 * Write outgoing buffer.

    96

 */

    97

ISR(USART_UDRE_vect)

    98

{

    99

    UDR0 = serial_tx;

   100

   101

    // tx buffer consumed

   102

    serial_tx = 0;

   103

   104

    // idle tx

   105

    serial_enable(1, 0);

   106

}

   107

   108

char serial_read ()

   109

{

   110

    char rx = serial_rx;

   111

   112

    // rx buffer consumed

   113

    serial_rx = 0;

   114

   115

    return rx;

   116

}

   117

   118

void serial_write (char c)

   119

{

   120

    // XXX: atomic?

   121

    if (tbi(&UCSR0A, UDRE0)) {

   122

        // direct

   123

        UDR0 = c;

   124

    } else {

   125

        // buffered

   126

        serial_tx = c;

   127

        serial_enable(1, 1);

   128

    }

   129

}