// Filename: egg_parametrics.cxx
// Created by:  drose (13Oct03)
//
////////////////////////////////////////////////////////////////////
//
// PANDA 3D SOFTWARE
// Copyright (c) Carnegie Mellon University.  All rights reserved.
//
// All use of this software is subject to the terms of the revised BSD
// license.  You should have received a copy of this license along
// with this source code in a file named "LICENSE."
//
////////////////////////////////////////////////////////////////////

#include "egg_parametrics.h"
#include "config_egg2pg.h"

////////////////////////////////////////////////////////////////////
//     Function: make_nurbs_surface
//  Description: Returns a new NurbsSurfaceEvaluator that's filled in
//               with the values from the given EggSurface (and
//               transformed by the indicated matrix), or NULL if the
//               object is invalid.  If there is vertex color, it will
//               be applied to values 0 - 3 of the extended vertex
//               values.
////////////////////////////////////////////////////////////////////
PT(NurbsSurfaceEvaluator)
make_nurbs_surface(EggNurbsSurface *egg_surface, const LMatrix4d &mat) {
  if (egg_surface->get_u_order() < 1 || egg_surface->get_u_order() > 4) {
    egg2pg_cat.error()
      << "Invalid NURBSSurface U order for " << egg_surface->get_name() << ": "
      << egg_surface->get_u_order() << "\n";
    return NULL;
  }

  if (egg_surface->get_v_order() < 1 || egg_surface->get_v_order() > 4) {
    egg2pg_cat.error()
      << "Invalid NURBSSurface V order for " << egg_surface->get_name() << ": "
      << egg_surface->get_v_order() << "\n";
    return NULL;
  }

  PT(NurbsSurfaceEvaluator) nurbs = new NurbsSurfaceEvaluator;

  nurbs->set_u_order(egg_surface->get_u_order());
  nurbs->set_v_order(egg_surface->get_v_order());

  int num_u_vertices = egg_surface->get_num_u_cvs();
  int num_v_vertices = egg_surface->get_num_v_cvs();
  nurbs->reset(num_u_vertices, num_v_vertices);
  for (int ui = 0; ui < num_u_vertices; ui++) {
    for (int vi = 0; vi < num_v_vertices; vi++) {
      int i = egg_surface->get_vertex_index(ui, vi);
      EggVertex *egg_vertex = egg_surface->get_vertex(i);
      nurbs->set_vertex(ui, vi, LCAST(PN_stdfloat, egg_vertex->get_pos4() * mat));

      LColor color = egg_vertex->get_color();
      nurbs->set_extended_vertices(ui, vi, 0, color.get_data(), 4);
    }
  }

  int num_u_knots = egg_surface->get_num_u_knots();
  if (num_u_knots != nurbs->get_num_u_knots()) {
    egg2pg_cat.error()
      << "Invalid NURBSSurface number of U knots for "
      << egg_surface->get_name() << ": got " << num_u_knots
      << " knots, expected " << nurbs->get_num_u_knots() << "\n";
    return NULL;
  }

  int num_v_knots = egg_surface->get_num_v_knots();
  if (num_v_knots != nurbs->get_num_v_knots()) {
    egg2pg_cat.error()
      << "Invalid NURBSSurface number of U knots for "
      << egg_surface->get_name() << ": got " << num_v_knots
      << " knots, expected " << nurbs->get_num_v_knots() << "\n";
    return NULL;
  }

  int i;
  for (i = 0; i < num_u_knots; i++) {
    nurbs->set_u_knot(i, egg_surface->get_u_knot(i));
  }
  for (i = 0; i < num_v_knots; i++) {
    nurbs->set_v_knot(i, egg_surface->get_v_knot(i));
  }

  return nurbs;
}

////////////////////////////////////////////////////////////////////
//     Function: make_nurbs_curve
//  Description: Returns a new NurbsCurveEvaluator that's filled in
//               with the values from the given EggCurve (and
//               transformed by the indicated matrix), or NULL if the
//               object is invalid.  If there is vertex color, it will
//               be applied to values 0 - 3 of the extended vertex
//               values.
////////////////////////////////////////////////////////////////////
PT(NurbsCurveEvaluator)
make_nurbs_curve(EggNurbsCurve *egg_curve, const LMatrix4d &mat) {
  if (egg_curve->get_order() < 1 || egg_curve->get_order() > 4) {
    egg2pg_cat.error()
      << "Invalid NURBSCurve order for " << egg_curve->get_name() << ": "
      << egg_curve->get_order() << "\n";
    return NULL;
  }

  PT(NurbsCurveEvaluator) nurbs = new NurbsCurveEvaluator;
  nurbs->set_order(egg_curve->get_order());

  nurbs->reset(egg_curve->size());
  EggPrimitive::const_iterator pi;
  int vi = 0;
  for (pi = egg_curve->begin(); pi != egg_curve->end(); ++pi) {
    EggVertex *egg_vertex = (*pi);
    nurbs->set_vertex(vi, LCAST(PN_stdfloat, egg_vertex->get_pos4() * mat));
    LColor color = egg_vertex->get_color();
    nurbs->set_extended_vertices(vi, 0, color.get_data(), 4);
    vi++;
  }

  int num_knots = egg_curve->get_num_knots();
  if (num_knots != nurbs->get_num_knots()) {
    egg2pg_cat.error()
      << "Invalid NURBSCurve number of knots for "
      << egg_curve->get_name() << ": got " << num_knots
      << " knots, expected " << nurbs->get_num_knots() << "\n";
    return NULL;
  }

  for (int i = 0; i < num_knots; i++) {
    nurbs->set_knot(i, egg_curve->get_knot(i));
  }

  return nurbs;
}
