Ticket #44: rumble.c

/* Copyright (C) 2007 L. Donnie Smith <cwiidabstrakraft.org>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 *  ChangeLog:
 *  2007-08-26 L. Nick Wilbanks <nickishappy@gmail.com>
 *  * Initial Release
 */

#include "wmplugin.h"
#include <stdio.h>
#include <math.h>

static int Enabled = 1.0;

cwiid_wiimote_t *wiimote;

static struct wmplugin_info info;
static struct wmplugin_data data;

wmplugin_info_t wmplugin_info;
wmplugin_init_t wmplugin_init;
wmplugin_exec_t wmplugin_exec;

struct wmplugin_info *wmplugin_info() {
        static unsigned char info_init = 0;

        if (!info_init) {
                info.button_count = 0;
                info.axis_count = 0;
                info.param_count = 1;
                info.param_info[0].name = "Enabled";
                info.param_info[0].type = WMPLUGIN_PARAM_INT;
                info.param_info[0].ptr = &Enabled;
                info_init = 1;
        }
        return &info;
}

int wmplugin_init(int id, cwiid_wiimote_t *arg_wiimote)
{
        wiimote = arg_wiimote;

        data.buttons = 0;

        if (wmplugin_set_rpt_mode(id, CWIID_RPT_BTN)) {
                return -1;
        }

        return 0;
}

struct wmplugin_data *wmplugin_exec(int mesg_count, union cwiid_mesg mesg[])
{
        int i;
        struct cwiid_btn_mesg *btn_mesg;
        struct cwiid_state state;

        btn_mesg = NULL;
        for (i=0; i < mesg_count; i++) {
                if (mesg[i].type == CWIID_MESG_BTN) {
                        btn_mesg = &mesg[i].btn_mesg;
                }
        }

        if (!btn_mesg) {
                return NULL;
        }
        
        if(Enabled)
        {
                cwiid_get_state(wiimote, &state);
                if(state.rumble == 0){
                        cwiid_set_rumble(wiimote, 1);
                } else {
                        cwiid_set_rumble(wiimote, 0);
                }
        }

        return &data;
}