// Copyright 2018, University of Freiburg,
// Chair of Algorithms and Data Structures.
// Authors: Patrick Brosi <brosi@informatik.uni-freiburg.de>

#include <string>
#include <unordered_map>
#include "pfaedle/trgraph/NodePL.h"
#include "pfaedle/trgraph/StatGroup.h"
#include "pfaedle/trgraph/StatInfo.h"
#include "util/String.h"

using pfaedle::trgraph::StatInfo;
using pfaedle::trgraph::NodePL;
using pfaedle::trgraph::Component;

// we use the adress of this dummy station info as a special value
// of this node, meaning "is a station block". Re-using the _si field here
// saves some memory
StatInfo NodePL::_blockerSI = StatInfo();

std::unordered_map<const Component*, size_t> NodePL::_comps;

// _____________________________________________________________________________
NodePL::NodePL()
    : _geom(0, 0),
      _si(0),
      _component(0)
#ifdef PFAEDLE_DBG
      ,
      _vis(0)
#endif
{
}

// _____________________________________________________________________________
NodePL::NodePL(const NodePL& pl)
    : _geom(pl._geom),
      _si(0),
      _component(pl._component)
#ifdef PFAEDLE_DBG
      ,
      _vis(pl._vis)
#endif
{
  if (pl._si) setSI(*(pl._si));
}

// _____________________________________________________________________________
NodePL::NodePL(const DPoint& geom)
    : _geom(geom),
      _si(0),
      _component(0)
#ifdef PFAEDLE_DBG
      ,
      _vis(0)
#endif
{
}

// _____________________________________________________________________________
NodePL::NodePL(const DPoint& geom, const StatInfo& si)
    : _geom(geom),
      _si(0),
      _component(0)
#ifdef PFAEDLE_DBG
      ,
      _vis(0)
#endif
{
  setSI(si);
}

// _____________________________________________________________________________
NodePL::~NodePL() {
  if (getSI()) delete _si;
  if (_component) {
    _comps[_component]--;
    if (_comps[_component] == 0) {
      delete _component;
      _comps.erase(_comps.find(_component));
    }
  }
}

// _____________________________________________________________________________
void NodePL::setVisited() const {
#ifdef PFAEDLE_DBG
  _vis = true;
#endif
}

// _____________________________________________________________________________
void NodePL::setNoStat() { _si = 0; }

// _____________________________________________________________________________
const Component* NodePL::getComp() const { return _component; }

// _____________________________________________________________________________
void NodePL::setComp(const Component* c) {
  if (_component == c) return;
  _component = c;

  // NOT thread safe!
  if (!_comps.count(c))
    _comps[c] = 1;
  else
    _comps[c]++;
}

// _____________________________________________________________________________
const DPoint* NodePL::getGeom() const { return &_geom; }

// _____________________________________________________________________________
void NodePL::setGeom(const DPoint& geom) { _geom = geom; }

// _____________________________________________________________________________
util::json::Dict NodePL::getAttrs() const {
  util::json::Dict obj;
  obj["component"] = std::to_string(reinterpret_cast<size_t>(_component));
#ifdef PFAEDLE_DBG
  obj["dijkstra_vis"] = _vis ? "yes" : "no";
#endif
  if (getSI()) {
    obj["station_info_ptr"] = util::toString(_si);
    obj["station_name"] = _si->getName();
    obj["station_alt_names"] = util::implode(_si->getAltNames(), ",");
    obj["from_osm"] = _si->isFromOsm() ? "yes" : "no";
    obj["station_platform"] = _si->getTrack();
    obj["station_group"] =
        std::to_string(reinterpret_cast<size_t>(_si->getGroup()));

    std::stringstream gtfsIds;
    if (_si->getGroup()) {
      for (auto* s : _si->getGroup()->getStops()) {
        gtfsIds << s->getId() << " (" << s->getName() << "),";
      }
    }

    obj["station_group_stops"] = gtfsIds.str();
  }
  return obj;
}

// _____________________________________________________________________________
void NodePL::setSI(const StatInfo& si) { _si = new StatInfo(si); }

// _____________________________________________________________________________
const StatInfo* NodePL::getSI() const {
  if (isBlocker()) return 0;
  return _si;
}

// _____________________________________________________________________________
StatInfo* NodePL::getSI() {
  if (isBlocker()) return 0;
  return _si;
}

// _____________________________________________________________________________
void NodePL::setBlocker() { _si = &_blockerSI; }

// _____________________________________________________________________________
bool NodePL::isBlocker() const { return _si == &_blockerSI; }