Colony

/**

 * Copyright (c) 2008 Colony Project

 * 

 * Permission is hereby granted, free of charge, to any person

 * obtaining a copy of this software and associated documentation

 * files (the "Software"), to deal in the Software without

 * restriction, including without limitation the rights to use,

 * copy, modify, merge, publish, distribute, sublicense, and/or sell

 * copies of the Software, and to permit persons to whom the

 * Software is furnished to do so, subject to the following

 * conditions:

 * 

 * The above copyright notice and this permission notice shall be

 * included in all copies or substantial portions of the Software.

 * 

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,

 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

 * OTHER DEALINGS IN THE SOFTWARE.

 **/

/**

 * @file orb.c

 * @brief Orbs

 *

 * Implemenation for the orbs (tri-colored LEDs)

 *

 * @author Colony Project, CMU Robotics Club

 **/ 

#include <avr/io.h>

#include "orb.h"

int orb_init (void);

int set_orb (int red, int green, int blue);

/**

 * @defgroup orbs Orbs

 * @brief Functions for controlling the color of the orbs.

 * 

 * Functions for controlling the color and lighting of the orbs.

 *

 * @{

 **/

/**

 * Initializes the PWM for Orb control. This must be called before 

 * the orbs are used for them to function.

 **/

int orb_init (void)

{

        /*data direction registers, blue, green (bit 7) and red (bit 5)*/

        DDRB = _BV(DDB3);

        DDRD = _BV(DDD7)|_BV(DDD5);

        /*blue is here. it goes on timer 0*/

        /*timer counter control registers for timer 0*/

        TCCR0A = _BV(COM0A1)|_BV(COM0A0)|_BV(WGM01)|_BV(WGM00);

        TCCR0B = _BV(CS00);

        OCR0A = 0;

        /*green is here. it goes on timer 2*/

        TCCR2A = _BV(COM2A1)|_BV(COM2A0)|_BV(WGM21)|_BV(WGM20);

        TCCR2B = _BV(CS20);

        OCR2A = 0;

        /*red is here. it goes on timer 1 which counts to the value in ICR1 instead of 8 bits*/

        TCCR1A = _BV(COM1A1)|_BV(COM1A0)|_BV(WGM11);

        TCCR1B = _BV(WGM13)|_BV(WGM12)|_BV(CS10);

        ICR1 = 0x9C40;

        OCR1A = 0;

        return 0;

}

/**

 * Set both orbs to the color specified. orb_init must

 * be called before this function may be used.

 *

 * @param red the red component of the color

 * @param green the green component of the color

 * @param blue the blue component of the color

 *

 * @see orb_init

 **/

int set_orb(int red, int green, int blue){

        /*blue is here. it goes on timer 0*/

        blue = blue%256;

        OCR0A = blue;

        /*green is here. it goes on timer 2*/

        green = green%256;

        OCR2A = green;

        /*red is here. it goes on timer 1*/

        red = red%256;

        red *= (int)(ICR1/256);

        OCR1A = red;

        return 0;

}

/** @} **/ //end group