/*
Copyright_License {

  XCSoar Glide Computer - http://www.xcsoar.org/
  Copyright (C) 2000-2015 The XCSoar Project
  A detailed list of copyright holders can be found in the file "AUTHORS".

  This program 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 2
  of the License, or (at your option) any later version.

  This program 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.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
}
*/

#include "FLARM/Friends.hpp"
#include "FLARM/FlarmId.hpp"
#include "Global.hpp"
#include "TrafficDatabases.hpp"

FlarmColor
FlarmFriends::GetFriendColor(FlarmId id)
{
  if (traffic_databases == nullptr)
    return FlarmColor::NONE;

  return traffic_databases->GetColor(id);
}

void
FlarmFriends::SetFriendColor(FlarmId id, FlarmColor color)
{
  assert(traffic_databases != nullptr);

  if (color == FlarmColor::NONE)
    traffic_databases->flarm_colors.Remove(id);
  else
    traffic_databases->flarm_colors.Set(id, color);
}
