/***************************************************************************
 *   Copyright (c) 2014-2015 Nathan Miller    <Nathan.A.Mill[at]gmail.com> *
 *                           Balázs Bámer                                  *
 *                                                                         *
 *   This file is part of the FreeCAD CAx development system.              *
 *                                                                         *
 *   This library is free software; you can redistribute it and/or         *
 *   modify it under the terms of the GNU Library General Public           *
 *   License as published by the Free Software Foundation; either          *
 *   version 2 of the License, or (at your option) any later version.      *
 *                                                                         *
 *   This library  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 Library General Public License for more details.                  *
 *                                                                         *
 *   You should have received a copy of the GNU Library General Public     *
 *   License along with this library; see the file COPYING.LIB. If not,    *
 *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
 *   Suite 330, Boston, MA  02111-1307, USA                                *
 *                                                                         *
 ***************************************************************************/

#include "PreCompiled.h"
#ifndef _PreComp_
#include <TopoDS.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Edge.hxx>
#include <TopoDS_Wire.hxx>
#include <Geom_BSplineCurve.hxx>
#include <Geom_BSplineSurface.hxx>
#include <Precision.hxx>
#include <gp_Trsf.hxx>
#include <BRep_Tool.hxx>
#include <TopExp_Explorer.hxx>
#include <Standard_ConstructionError.hxx>
#include <GeomFill_BSplineCurves.hxx>
#include <GeomFill.hxx>
#include <BRepBuilderAPI_NurbsConvert.hxx>
#include <StdFail_NotDone.hxx>
#endif

#include <Base/Tools.h>
#include <Base/Exception.h>

#include "FeatureBSplineSurface.h"


using namespace Surface;

PROPERTY_SOURCE(Surface::BSplineSurface, Surface::SurfaceFeature)


BSplineSurface::BSplineSurface() : SurfaceFeature()
{
}

App::DocumentObjectExecReturn *BSplineSurface::execute(void)
{
    correcteInvalidFillType();

    try{
        std::vector<Handle_Geom_BSplineCurve> crvs;
        crvs.reserve(4);
        TopoDS_Wire aWire;

        //Gets the healed wire
        getWire(aWire);

        Standard_Real u1, u2; // contains output
        TopExp_Explorer anExp (aWire, TopAbs_EDGE);
        for (; anExp.More(); anExp.Next()) {
            const TopoDS_Edge& edge = TopoDS::Edge (anExp.Current());
            TopLoc_Location heloc; // this will be output
            Handle_Geom_Curve c_geom = BRep_Tool::Curve(edge, heloc, u1, u2); //The geometric curve
            Handle_Geom_BSplineCurve b_geom = Handle_Geom_BSplineCurve::DownCast(c_geom); //Try to get BSpline curve

            if (!b_geom.IsNull()) {
                gp_Trsf transf = heloc.Transformation();
                b_geom->Transform(transf); // apply original transformation to control points
                //Store Underlying Geometry
                crvs.push_back(b_geom);
            }
            else {
                // try to convert it into a b-spline
                BRepBuilderAPI_NurbsConvert mkNurbs(edge);
                TopoDS_Edge nurbs = TopoDS::Edge(mkNurbs.Shape());
                // avoid copying
                TopLoc_Location heloc2; // this will be output
                Handle_Geom_Curve c_geom2 = BRep_Tool::Curve(nurbs, heloc2, u1, u2); //The geometric curve
                Handle_Geom_BSplineCurve b_geom2 = Handle_Geom_BSplineCurve::DownCast(c_geom2); //Try to get BSpline curve

                if (!b_geom2.IsNull()) {
                    gp_Trsf transf = heloc2.Transformation();
                    b_geom2->Transform(transf); // apply original transformation to control points
                    //Store Underlying Geometry
                    crvs.push_back(b_geom2);
                }
                else {
                    Standard_Failure::Raise("A curve was not a b-spline and could not be converted into one.");
                }
            }
        }

        GeomFill_FillingStyle fstyle = getFillingStyle();
        GeomFill_BSplineCurves aSurfBuilder; //Create Surface Builder

        std::size_t edgeCount = crvs.size();
        if (edgeCount==2) {
            aSurfBuilder.Init(crvs[0], crvs[1], fstyle);
        }
        else if (edgeCount==3) {
            aSurfBuilder.Init(crvs[0], crvs[1], crvs[2], fstyle);
        }
        else if (edgeCount==4) {
            aSurfBuilder.Init(crvs[0], crvs[1], crvs[2], crvs[3], fstyle);
        }

        createFace(aSurfBuilder.Surface());

        return App::DocumentObject::StdReturn;
    }
    catch (Standard_ConstructionError) {
        // message is in a Latin language, show a normal one
        return new App::DocumentObjectExecReturn("Curves are disjoint.");
    }
    catch (StdFail_NotDone) {
        return new App::DocumentObjectExecReturn("A curve was not a b-spline and could not be converted into one.");
    }
    catch (Standard_Failure) {
        Handle_Standard_Failure e = Standard_Failure::Caught();
        return new App::DocumentObjectExecReturn(e->GetMessageString());
    }
}