git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/lib/io/gr_message_source.cc

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright 2005 Free Software Foundation, Inc.
   4  *
   5  * This file is part of GNU Radio
   6  *
   7  * GNU Radio is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 3, or (at your option)
  10  * any later version.
  11  *
  12  * GNU Radio is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with GNU Radio; see the file COPYING.  If not, write to
  19  * the Free Software Foundation, Inc., 51 Franklin Street,
  20  * Boston, MA 02110-1301, USA.
  21  */
  22
  23 #ifdef HAVE_CONFIG_H
  24 #include "config.h"
  25 #endif
  26
  27 #include <gr_message_source.h>
  28 #include <gr_io_signature.h>
  29 #include <cstdio>
  30 #include <errno.h>
  31 #include <sys/types.h>
  32 #include <sys/stat.h>
  33 #include <fcntl.h>
  34 #include <stdexcept>
  35 #include <string.h>
  36
  37
  38 // public constructor that returns a shared_ptr
  39
  40 gr_message_source_sptr
  41 gr_make_message_source(size_t itemsize, int msgq_limit)
  42 {
  43   return gr_message_source_sptr(new gr_message_source(itemsize, msgq_limit));
  44 }
  45
  46 gr_message_source::gr_message_source (size_t itemsize, int msgq_limit)
  47   : gr_sync_block("message_source",
  48                   gr_make_io_signature(0, 0, 0),
  49                   gr_make_io_signature(1, 1, itemsize)),
  50     d_itemsize(itemsize), d_msgq(gr_make_msg_queue(msgq_limit)), d_msg_offset(0), d_eof(false)
  51 {
  52 }
  53
  54 gr_message_source::~gr_message_source()
  55 {
  56 }
  57
  58 int
  59 gr_message_source::work(int noutput_items,
  60                         gr_vector_const_void_star &input_items,
  61                         gr_vector_void_star &output_items)
  62 {
  63   char *out = (char *) output_items[0];
  64   int nn = 0;
  65
  66   while (nn < noutput_items){
  67     if (d_msg){
  68       //
  69       // Consume whatever we can from the current message
  70       //
  71       int mm = std::min(noutput_items - nn, (int)((d_msg->length() - d_msg_offset) / d_itemsize));
  72       memcpy (out, &(d_msg->msg()[d_msg_offset]), mm * d_itemsize);
  73
  74       nn += mm;
  75       out += mm * d_itemsize;
  76       d_msg_offset += mm * d_itemsize;
  77       assert(d_msg_offset <= d_msg->length());
  78
  79       if (d_msg_offset == d_msg->length()){
  80         if (d_msg->type() == 1)            // type == 1 sets EOF
  81           d_eof = true;
  82         d_msg.reset();
  83       }
  84     }
  85     else {
  86       //
  87       // No current message
  88       //
  89       if (d_msgq->empty_p() && nn > 0){    // no more messages in the queue, return what we've got
  90         break;
  91       }
  92
  93       if (d_eof)
  94         return -1;
  95
  96       d_msg = d_msgq->delete_head();       // block, waiting for a message
  97       d_msg_offset = 0;
  98
  99       if ((d_msg->length() % d_itemsize) != 0)
 100         throw std::runtime_error("msg length is not a multiple of d_itemsize");
 101     }
 102   }
 103
 104   return nn;
 105 }