src/libs/StreamOutputPool.h

   1 /*
   2       This file is part of Smoothie (http://smoothieware.org/). The motion control part is heavily based on Grbl (https://github.com/simen/grbl).
   3       Smoothie 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 3 of the License, or (at your option) any later version.
   4       Smoothie 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.
   5       You should have received a copy of the GNU General Public License along with Smoothie. If not, see <http://www.gnu.org/licenses/>.
   6 */
   7
   8 #ifndef STREAMOUTPUTPOOL_H
   9 #define STREAMOUTPUTPOOL_H
  10
  11 using namespace std;
  12 #include <set>
  13 #include <string>
  14 #include <cstdio>
  15 #include <cstdarg>
  16
  17 #include "libs/StreamOutput.h"
  18
  19 class StreamOutputPool : public StreamOutput {
  20
  21 public:
  22     StreamOutputPool(){
  23     }
  24
  25     int puts(const char* s)
  26     {
  27         int r = 0;
  28         for(set<StreamOutput*>::iterator i = this->streams.begin(); i != this->streams.end(); i++)
  29         {
  30             int k = (*i)->puts(s);
  31             if (k > r)
  32                 r = k;
  33         }
  34         return r;
  35     }
  36
  37     void append_stream(StreamOutput* stream)
  38     {
  39         this->streams.insert(stream);
  40     }
  41
  42     void remove_stream(StreamOutput* stream)
  43     {
  44         this->streams.erase(stream);
  45     }
  46
  47 private:
  48     set<StreamOutput*> streams;
  49 };
  50
  51 #endif