src/main/drivers/buf_writer.c

   1 /*
   2  * This file is part of Cleanflight and Betaflight.
   3  *
   4  * Cleanflight and Betaflight are free software. You can redistribute
   5  * this software and/or modify this software under the terms of the
   6  * GNU General Public License as published by the Free Software
   7  * Foundation, either version 3 of the License, or (at your option)
   8  * any later version.
   9  *
  10  * Cleanflight and Betaflight are distributed in the hope that they
  11  * will be useful, but WITHOUT ANY WARRANTY; without even the implied
  12  * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13  * See the GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this software.
  17  *
  18  * If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <stdint.h>
  22
  23 #include "platform.h"
  24
  25 #include "buf_writer.h"
  26
  27 bufWriter_t *bufWriterInit(uint8_t *b, int total_size, bufWrite_t writer, void *arg)
  28 {
  29     bufWriter_t *buf = (bufWriter_t *)b;
  30     buf->writer = writer;
  31     buf->arg = arg;
  32     buf->at = 0;
  33     buf->capacity = total_size - sizeof(*buf);
  34
  35     return buf;
  36 }
  37
  38 void bufWriterAppend(bufWriter_t *b, uint8_t ch)
  39 {
  40     b->data[b->at++] = ch;
  41     if (b->at >= b->capacity) {
  42         bufWriterFlush(b);
  43     }
  44 }
  45
  46 void bufWriterFlush(bufWriter_t *b)
  47 {
  48     if (b->at != 0) {
  49         b->writer(b->arg, b->data, b->at);
  50         b->at = 0;
  51     }
  52 }