/* ----------------------------------------------------------------------------

 * GTSAM Copyright 2010, Georgia Tech Research Corporation, 
 * Atlanta, Georgia 30332-0415
 * All Rights Reserved
 * Authors: Frank Dellaert, et al. (see THANKS for the full author list)

 * See LICENSE for the license information

 * -------------------------------------------------------------------------- */

/**
 * @file Class.cpp
 * @author Frank Dellaert
 * @author Andrew Melim
 * @author Richard Roberts
 **/

#include "TypeAttributesTable.h"
#include "Class.h"
#include "utilities.h"

#include <boost/foreach.hpp>
#include <boost/range/adaptor/map.hpp>
#include <boost/range/algorithm/copy.hpp>

#include <iterator>     // std::ostream_iterator
using namespace std;

namespace wrap {

/* ************************************************************************* */
const TypeAttributes& TypeAttributesTable::attributes(const string& key) const {
  try {
    return table_.at(key);
  } catch (const out_of_range& oor) {
    cerr << "Class::method: key not found: " << oor.what()
        << ", methods are:\n";
    using boost::adaptors::map_keys;
    ostream_iterator<string> out_it(cerr, "\n");
    boost::copy(table_ | map_keys, out_it);
    throw runtime_error("Internal error in wrap");
  }
}

/* ************************************************************************* */
void TypeAttributesTable::addClasses(const vector<Class>& classes) {
  BOOST_FOREACH(const Class& cls, classes) {
    if (!table_.insert(
        make_pair(cls.qualifiedName("::"), TypeAttributes(cls.isVirtual))).second)
      throw DuplicateDefinition("class " + cls.qualifiedName("::"));
  }
}

/* ************************************************************************* */
void TypeAttributesTable::addForwardDeclarations(
    const vector<ForwardDeclaration>& forwardDecls) {
  BOOST_FOREACH(const ForwardDeclaration& fwDec, forwardDecls) {
    if (!table_.insert(make_pair(fwDec.name, TypeAttributes(fwDec.isVirtual))).second)
      throw DuplicateDefinition("class " + fwDec.name);
  }
}

/* ************************************************************************* */
void TypeAttributesTable::checkValidity(const vector<Class>& classes) const {
  BOOST_FOREACH(const Class& cls, classes) {
    // Check that class is virtual if it has a parent
    if (!cls.qualifiedParent.empty() && !cls.isVirtual)
      throw AttributeError(cls.qualifiedName("::"),
          "Has a base class so needs to be declared virtual, change to 'virtual class "
              + cls.name + " ...'");
    // Check that parent is virtual as well
    Qualified parent = cls.qualifiedParent;
    if (!parent.empty() && !table_.at(parent.qualifiedName("::")).isVirtual)
      throw AttributeError(parent.qualifiedName("::"),
          "Is the base class of " + cls.qualifiedName("::")
              + ", so needs to be declared virtual");
  }
}

}