examples/mraa-gpio.c

   1 /*
   2  * Author: Brendan Le Foll <brendan.le.foll@intel.com>
   3  * Copyright (c) 2014 Intel Corporation.
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining
   6  * a copy of this software and associated documentation files (the
   7  * "Software"), to deal in the Software without restriction, including
   8  * without limitation the rights to use, copy, modify, merge, publish,
   9  * distribute, sublicense, and/or sell copies of the Software, and to
  10  * permit persons to whom the Software is furnished to do so, subject to
  11  * the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be
  14  * included in all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  17  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  18  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  19  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  20  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  21  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  22  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  23  */
  24
  25 #include <stdio.h>
  26 #include <string.h>
  27 #include <unistd.h>
  28
  29 #include "mraa/gpio.h"
  30
  31 struct gpio_source {
  32     int pin;
  33     mraa_gpio_context context;
  34 };
  35
  36 void
  37 print_version()
  38 {
  39     fprintf(stdout, "Version %s on %s\n", mraa_get_version(), mraa_get_platform_name());
  40 }
  41
  42 void
  43 print_help()
  44 {
  45     fprintf(stdout, "list              List pins\n");
  46     fprintf(stdout, "set pin level     Set pin to level (0/1)\n");
  47     fprintf(stdout, "setraw pin level  Set pin to level (0/1) via mmap (if available)\n");
  48     fprintf(stdout, "get pin           Get pin level\n");
  49     fprintf(stdout, "getraw pin        Get pin level via mmap (if available)\n");
  50     fprintf(stdout, "monitor pin       Monitor pin level changes\n");
  51     fprintf(stdout, "version           Get mraa version and board name\n");
  52 }
  53
  54 void
  55 print_command_error()
  56 {
  57     fprintf(stdout, "Invalid command, options are:\n");
  58     print_help();
  59 }
  60
  61 void
  62 list_pins()
  63 {
  64     int pin_count = mraa_get_pin_count();
  65     if (pin_count == 0) {
  66         fprintf(stdout, "No Pins\n");
  67         return;
  68     }
  69     int i;
  70     for (i = 0; i < pin_count; ++i) {
  71         if (strcmp(mraa_get_pin_name(i), "INVALID") != 0) {
  72             fprintf(stdout, "%02d ", i);
  73             fprintf(stdout, "%*s: ", (MRAA_PIN_NAME_SIZE - 1), mraa_get_pin_name(i));
  74             if (mraa_pin_mode_test(i, MRAA_PIN_GPIO))
  75                 fprintf(stdout, "GPIO ");
  76             if (mraa_pin_mode_test(i, MRAA_PIN_I2C))
  77                 fprintf(stdout, "I2C  ");
  78             if (mraa_pin_mode_test(i, MRAA_PIN_SPI))
  79                 fprintf(stdout, "SPI  ");
  80             if (mraa_pin_mode_test(i, MRAA_PIN_PWM))
  81                 fprintf(stdout, "PWM  ");
  82             if (mraa_pin_mode_test(i, MRAA_PIN_AIO))
  83                 fprintf(stdout, "AIO  ");
  84             if (mraa_pin_mode_test(i, MRAA_PIN_UART))
  85                 fprintf(stdout, "UART ");
  86             fprintf(stdout, "\n");
  87         }
  88     }
  89 }
  90
  91 mraa_result_t
  92 gpio_set(int pin, int level, mraa_boolean_t raw)
  93 {
  94     mraa_gpio_context gpio = mraa_gpio_init(pin);
  95     if (gpio != NULL) {
  96         mraa_gpio_dir(gpio, MRAA_GPIO_OUT);
  97         if (raw != 0) {
  98             if (mraa_gpio_use_mmaped(gpio, 1) != MRAA_SUCCESS) {
  99                 fprintf(stdout,
 100                         "mmapped access to gpio not supported, falling back to normal mode\n", pin);
 101             }
 102         }
 103         mraa_gpio_write(gpio, level);
 104         return MRAA_SUCCESS;
 105     }
 106     return MRAA_ERROR_INVALID_RESOURCE;
 107 }
 108
 109 mraa_result_t
 110 gpio_get(int pin, int* level, mraa_boolean_t raw)
 111 {
 112     mraa_gpio_context gpio = mraa_gpio_init(pin);
 113     if (gpio != NULL) {
 114         mraa_gpio_dir(gpio, MRAA_GPIO_IN);
 115         if (raw != 0) {
 116             if (mraa_gpio_use_mmaped(gpio, 1) != MRAA_SUCCESS) {
 117                 fprintf(stdout,
 118                         "mmapped access to gpio not supported, falling back to normal mode\n", pin);
 119             }
 120         }
 121         *level = mraa_gpio_read(gpio);
 122         return MRAA_SUCCESS;
 123     }
 124     return MRAA_ERROR_INVALID_RESOURCE;
 125 }
 126
 127
 128 void
 129 gpio_isr_handler(void* args)
 130 {
 131     struct gpio_source* gpio_info = (struct gpio_source*) args;
 132     int level = mraa_gpio_read(gpio_info->context);
 133     fprintf(stdout, "Pin %d = %d\n", gpio_info->pin, level);
 134 }
 135
 136 mraa_result_t
 137 gpio_isr_start(struct gpio_source* gpio_info)
 138 {
 139     gpio_info->context = mraa_gpio_init(gpio_info->pin);
 140     if (gpio_info->context != NULL) {
 141         mraa_result_t status = mraa_gpio_dir(gpio_info->context, MRAA_GPIO_IN);
 142         if (status == MRAA_SUCCESS) {
 143             status = mraa_gpio_isr(gpio_info->context, MRAA_GPIO_EDGE_BOTH, &gpio_isr_handler, gpio_info);
 144         }
 145         return status;
 146     } else {
 147         return MRAA_ERROR_INVALID_RESOURCE;
 148     }
 149 }
 150
 151
 152 mraa_result_t
 153 gpio_isr_stop(struct gpio_source* gpio_info)
 154 {
 155     mraa_gpio_isr_exit(gpio_info->context);
 156     mraa_gpio_close(gpio_info->context);
 157     return MRAA_SUCCESS;
 158 }
 159
 160
 161 int
 162 main(int argc, char** argv)
 163 {
 164     if (argc == 1) {
 165         print_command_error();
 166     }
 167
 168     if (argc > 1) {
 169         if (strcmp(argv[1], "list") == 0) {
 170             list_pins();
 171         } else if (strcmp(argv[1], "help") == 0) {
 172             print_help();
 173         } else if (strcmp(argv[1], "version") == 0) {
 174             print_version();
 175         } else if ((strcmp(argv[1], "set") == 0) || (strcmp(argv[1], "setraw") == 0)) {
 176             if (argc == 4) {
 177                 int pin = atoi(argv[2]);
 178                 mraa_boolean_t rawmode = strcmp(argv[1], "setraw") == 0;
 179                 if (gpio_set(pin, atoi(argv[3]), rawmode) != MRAA_SUCCESS)
 180                     fprintf(stdout, "Could not initialize gpio %d\n", pin);
 181             } else {
 182                 print_command_error();
 183             }
 184         } else if ((strcmp(argv[1], "get") == 0) || (strcmp(argv[1], "getraw") == 0)) {
 185             if (argc == 3) {
 186                 int pin = atoi(argv[2]);
 187                 int level;
 188                 mraa_boolean_t rawmode = strcmp(argv[1], "getraw") == 0;
 189                 if (gpio_get(pin, &level, rawmode) == MRAA_SUCCESS) {
 190                     fprintf(stdout, "Pin %d = %d\n", pin, level);
 191                 } else {
 192                     fprintf(stdout, "Could not initialize gpio %d\n", pin);
 193                 }
 194             } else {
 195                 print_command_error();
 196             }
 197         } else if (strcmp(argv[1], "monitor") == 0) {
 198             if (argc == 3) {
 199                 int pin = atoi(argv[2]);
 200                 struct gpio_source gpio_info;
 201                 gpio_info.pin = pin;
 202                 if (gpio_isr_start(&gpio_info) == MRAA_SUCCESS) {
 203                     fprintf(stdout, "Monitoring level changes to pin %d. Press RETURN to exit.\n", pin);
 204                     gpio_isr_handler(&gpio_info);
 205                     while (getchar() != '\n')
 206                         ;
 207                     gpio_isr_stop(&gpio_info);
 208                 } else {
 209                     fprintf(stdout, "Failed to register ISR for pin %d\n", pin);
 210                 }
 211             } else {
 212                 print_command_error();
 213             }
 214         } else {
 215             print_command_error();
 216         }
 217     }
 218     return 0;
 219 }