src/FLARM/ColorDatabase.hpp

   1 /*
   2 Copyright_License {
   3
   4   XCSoar Glide Computer - http://www.xcsoar.org/
   5   Copyright (C) 2000-2013 The XCSoar Project
   6   A detailed list of copyright holders can be found in the file "AUTHORS".
   7
   8   This program is free software; you can redistribute it and/or
   9   modify it under the terms of the GNU General Public License
  10   as published by the Free Software Foundation; either version 2
  11   of the License, or (at your option) any later version.
  12
  13   This program is distributed in the hope that it will be useful,
  14   but WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16   GNU General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with this program; if not, write to the Free Software
  20   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  21 }
  22 */
  23
  24 #ifndef XCSOAR_FLARM_COLOR_DATABASE_HPP
  25 #define XCSOAR_FLARM_COLOR_DATABASE_HPP
  26
  27 #include "FlarmId.hpp"
  28 #include "Color.hpp"
  29 #include "Compiler.h"
  30
  31 #include <map>
  32
  33 #include <assert.h>
  34
  35 class FlarmColorDatabase {
  36   typedef std::map<FlarmId, FlarmColor> Map;
  37   Map data;
  38
  39 public:
  40   typedef Map::const_iterator const_iterator;
  41
  42   gcc_pure
  43   const_iterator begin() const {
  44     return data.begin();
  45   }
  46
  47   gcc_pure
  48   const_iterator end() const {
  49     return data.end();
  50   }
  51
  52   gcc_pure
  53   FlarmColor Get(FlarmId id) const {
  54     auto i = data.find(id);
  55     if (i == data.end())
  56       return FlarmColor::NONE;
  57
  58     return i->second;
  59   }
  60
  61   void Set(FlarmId id, FlarmColor color) {
  62     assert(color != FlarmColor::NONE);
  63     assert(color != FlarmColor::COUNT);
  64
  65     auto i = data.insert(std::make_pair(id, color));
  66     if (!i.second)
  67       /* the id already exists in the map: overwrite the old color */
  68       i.first->second = color;
  69   }
  70
  71   void Remove(FlarmId id) {
  72     data.erase(id);
  73   }
  74 };
  75
  76 #endif