libs/ardour/ardour/chan_count.h

   1 /*
   2     Copyright (C) 2006 Paul Davis
   3         Author: Dave Robillard
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     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 program; if not, write to the Free Software
  17     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18 */
  19
  20 #ifndef __ardour_chan_count_h__
  21 #define __ardour_chan_count_h__
  22
  23 #include <ardour/data_type.h>
  24 #include <cassert>
  25
  26 namespace ARDOUR {
  27
  28
  29 /** A count of channels, possibly with many types.
  30  *
  31  * Operators are defined so this may safely be used as if it were a simple
  32  * (single-typed) integer count of channels.
  33  */
  34 class ChanCount {
  35 public:
  36         ChanCount() { reset(); }
  37
  38         // Convenience constructor for making single-typed streams (stereo, mono, etc)
  39         ChanCount(DataType type, uint32_t channels)
  40         {
  41                 reset();
  42                 set(type, channels);
  43         }
  44
  45         void reset()
  46         {
  47                 for (DataType::iterator t = DataType::begin(); t != DataType::end(); ++t) {
  48                         _counts[*t] = 0;
  49                 }
  50         }
  51
  52         void     set(DataType t, uint32_t count) { assert(t != DataType::NIL); _counts[t] = count; }
  53         uint32_t get(DataType t) const { assert(t != DataType::NIL); return _counts[t]; }
  54
  55         inline uint32_t n_audio() const { return _counts[DataType::AUDIO]; }
  56         inline void set_audio(uint32_t a) { _counts[DataType::AUDIO] = a; }
  57
  58         inline uint32_t n_midi()  const { return _counts[DataType::MIDI]; }
  59         inline void set_midi(uint32_t m) { _counts[DataType::MIDI] = m; }
  60
  61         uint32_t n_total() const
  62         {
  63                 uint32_t ret = 0;
  64                 for (uint32_t i=0; i < DataType::num_types; ++i)
  65                         ret += _counts[i];
  66
  67                 return ret;
  68         }
  69
  70         bool operator==(const ChanCount& other) const
  71         {
  72                 for (uint32_t i=0; i < DataType::num_types; ++i)
  73                         if (_counts[i] != other._counts[i])
  74                                 return false;
  75
  76                 return true;
  77         }
  78
  79         bool operator!=(const ChanCount& other) const
  80         {
  81                 return ! (*this == other);
  82         }
  83
  84         bool operator<(const ChanCount& other) const
  85         {
  86                 for (DataType::iterator t = DataType::begin(); t != DataType::end(); ++t) {
  87                         if (_counts[*t] > other._counts[*t]) {
  88                                 return false;
  89                         }
  90                 }
  91                 return (*this != other);
  92         }
  93
  94         bool operator<=(const ChanCount& other) const
  95         {
  96                 return ( (*this < other) || (*this == other) );
  97         }
  98
  99         bool operator>(const ChanCount& other) const
 100         {
 101                 for (DataType::iterator t = DataType::begin(); t != DataType::end(); ++t) {
 102                         if (_counts[*t] < other._counts[*t]) {
 103                                 return false;
 104                         }
 105                 }
 106                 return (*this != other);
 107         }
 108
 109         bool operator>=(const ChanCount& other) const
 110         {
 111                 return ( (*this > other) || (*this == other) );
 112         }
 113
 114         static const ChanCount INFINITE;
 115         static const ChanCount ZERO;
 116
 117 private:
 118
 119         uint32_t _counts[DataType::num_types];
 120 };
 121
 122
 123 } // namespace ARDOUR
 124
 125 #endif // __ardour_chan_count_h__
 126