OCCT

// Created on: 1994-02-18
// Created by: Remi LEQUETTE
// Copyright (c) 1994-1999 Matra Datavision
// Copyright (c) 1999-2014 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <Standard_Stream.hxx>
#include <BRepTest.hxx>
#include <Draw_Interpretor.hxx>
#include <Draw_Appli.hxx>
#include <DBRep.hxx>
#include <BRepGProp.hxx>
#include <TopoDS_Shape.hxx>
#include <GProp_PrincipalProps.hxx>

#include <Draw_Axis3D.hxx>
#include <Precision.hxx>

#ifdef _WIN32
Standard_IMPORT Draw_Viewer dout;
#endif


Standard_Integer props(Draw_Interpretor& di, Standard_Integer n, const char** a)
{
  if (n < 2) {
    di << "Use: " << a[0] << " shape [epsilon] [c[losed]] [x y z] [-skip] [-full] [-tri]\n";
    di << "Compute properties of the shape, exact geometry (curves, surfaces) or\n";
    di << "some discrete data (polygons, triangulations) can be used for calculations\n";
    di << "The epsilon, if given, defines relative precision of computation\n";
    di << "The \"closed\" flag, if present, do computation only closed shells of the shape\n";
    di << "The centroid coordinates will be put to DRAW variables x y z (if given)\n";
    di << "Shared entities will be take in account only one time in the skip mode\n";
    di << "All values are outputted with the full precision in the full mode.\n";
    di << "Preferable source of geometry data are triangulations in case if it exists, if the -tri key is used.\n";
    di << "If epsilon is given, exact geometry (curves, surfaces) are used for calculations independently of using key -tri\n\n";
    return 1;
  }

  Standard_Boolean UseTriangulation = Standard_False;
  if (n >= 2 && strcmp(a[n - 1], "-tri") == 0)
  {
    UseTriangulation = Standard_True;
    --n;
  }
  Standard_Boolean isFullMode = Standard_False;
  if (n >= 2 && strcmp(a[n-1], "-full") == 0)
  {
    isFullMode = Standard_True;
    --n;
  }
  Standard_Boolean SkipShared = Standard_False;
  if (n >= 2 && strcmp(a[n-1], "-skip") == 0)
  {
    SkipShared = Standard_True;
    --n;
  }

  TopoDS_Shape S = DBRep::Get(a[1]);
  if (S.IsNull()) return 0;

  GProp_GProps G;

  Standard_Boolean onlyClosed = Standard_False;
  Standard_Real eps = 1.0;
  Standard_Boolean witheps = Standard_False;
  if((n > 2 && *a[2]=='c') || (n > 3 && *a[3]=='c')) onlyClosed = Standard_True;
  if(n > 2 && *a[2]!='c' && n != 5) {eps = Draw::Atof (a[2]); witheps = Standard_True;}

  if (witheps){
    if (Abs(eps) < Precision::Angular()) return 2;
    if (*a[0] == 'l')
      BRepGProp::LinearProperties(S,G,SkipShared);
    else if (*a[0] == 's')
      eps = BRepGProp::SurfaceProperties(S,G,eps,SkipShared);
    else 
      eps = BRepGProp::VolumeProperties(S,G,eps,onlyClosed,SkipShared);
  }
  else {
    if (*a[0] == 'l')
      BRepGProp::LinearProperties(S, G, SkipShared, UseTriangulation);
    else if (*a[0] == 's')
      BRepGProp::SurfaceProperties(S, G, SkipShared, UseTriangulation);
    else 
      BRepGProp::VolumeProperties(S,G,onlyClosed,SkipShared, UseTriangulation);
  }
  
  gp_Pnt P = G.CentreOfMass();
  gp_Mat I = G.MatrixOfInertia();

  if (n >= 5) {
    Standard_Integer shift =  n - 5;
    Draw::Set(a[shift+2],P.X());
    Draw::Set(a[shift+3],P.Y());
    Draw::Set(a[shift+4],P.Z());
  }

  GProp_PrincipalProps Pr = G.PrincipalProperties();
  Standard_Real Ix,Iy,Iz;
  Pr.Moments(Ix,Iy,Iz);

  if (!isFullMode)
  {
    Standard_SStream aSStream1;
    aSStream1 << "\n\n";
    aSStream1 << "Mass : " << std::setw(15) << G.Mass() << "\n\n";
    if(witheps && *a[0] != 'l') aSStream1 << "Relative error of mass computation : " <<  std::setw(15) << eps <<  "\n\n";
  
    aSStream1 << "Center of gravity : \n";
    aSStream1 << "X = " << std::setw(15) << P.X() << "\n";
    aSStream1 << "Y = " << std::setw(15) << P.Y() << "\n";
    aSStream1 << "Z = " << std::setw(15) << P.Z() << "\n";
    aSStream1 << "\n";
  
    aSStream1 << "Matrix of Inertia : \n";
    aSStream1 << std::setw(15) << I(1,1);
    aSStream1 << " " << std::setw(15) << I(1,2);
    aSStream1 << " " << std::setw(15) << I(1,3) << "\n";
    aSStream1 << std::setw(15) << I(2,1);
    aSStream1 << " " << std::setw(15) << I(2,2);
    aSStream1 << " " << std::setw(15) << I(2,3) << "\n";
    aSStream1 << std::setw(15) << I(3,1);
    aSStream1 << " " << std::setw(15) << I(3,2);
    aSStream1 << " " << std::setw(15) << I(3,3) << "\n";
    aSStream1 << "\n";
    aSStream1 << std::ends;
    di << aSStream1;

    Standard_SStream aSStream2;
    aSStream2 << "Moments : \n";
    aSStream2 << "IX = " << std::setw(15) << Ix << "\n";
    aSStream2 << "IY = " << std::setw(15) << Iy << "\n";
    aSStream2 << "IZ = " << std::setw(15) << Iz << "\n";
    aSStream2 << "\n";
    aSStream2 << std::ends;
    di << aSStream2;
  }
  else
  {
    di << "\n\nMass : " << G.Mass() << "\n\n";
    if (witheps && *a[0] != 'l')
    {
      di << "Relative error of mass computation : " << eps <<  "\n\n";
    }

    di << "Center of gravity : \n";
    di << "X = " << P.X() << "\n";
    di << "Y = " << P.Y() << "\n";
    di << "Z = " << P.Z() << "\n\n";

    di << "Matrix of Inertia :\n";
    di << I(1,1) << "    " << I(1,2) << "    " << I(1,3) << "\n";
    di << I(2,1) << "    " << I(2,2) << "    " << I(2,3) << "\n";
    di << I(3,1) << "    " << I(3,2) << "    " << I(3,3) << "\n\n";

    di << "Moments :\n";
    di << "IX = " << Ix << "\n";
    di << "IY = " << Iy << "\n";
    di << "IZ = " << Iz << "\n\n";
  }

  //if (n == 2) {  
    gp_Ax2 axes(P,Pr.ThirdAxisOfInertia(),Pr.FirstAxisOfInertia());
    
    Handle(Draw_Axis3D) Dax = new Draw_Axis3D(axes,Draw_orange,30);
    dout << Dax;
  //}

  return 0;
}


Standard_Integer vpropsgk(Draw_Interpretor& di, Standard_Integer n, const char** a)
{
  if (n < 2) {
    di << "Use: " << a[0] << " shape epsilon closed span mode [x y z] [-skip]\n";
    di << "Compute properties of the shape\n";
    di << "The epsilon defines relative precision of computation\n";
    di << "The \"closed\" flag, if equal 1, causes computation only closed shells of the shape\n";
    di << "The \"span\" flag, if equal 1, says that computation is performed on spans\n";
    di << "      This option makes effect only for BSpline surfaces.\n";
    di << "mode can be 0 - only volume calculations\n";
    di << "            1 - volume and gravity center\n";
    di << "            2 - volume, gravity center and matrix of inertia\n";
    di << "The centroid coordinates will be put to DRAW variables x y z (if given)\n\n";
    return 1;
  }

  if ( n > 2 && n < 6) {
    di << "Wrong arguments\n";
    return 1;
  }

  TopoDS_Shape S = DBRep::Get(a[1]);
  if (S.IsNull()) return 0;

  GProp_GProps G;
  Standard_Boolean SkipShared = Standard_False;
  if (n >= 2 && strcmp(a[n-1], "-skip") == 0)
  {
    SkipShared = Standard_True;
    --n;
  }

  Standard_Boolean onlyClosed  = Standard_False;
  Standard_Boolean isUseSpan   = Standard_False;
  Standard_Boolean CGFlag = Standard_False;
  Standard_Boolean IFlag = Standard_False;
  Standard_Real    eps         = 1.e-3;
//Standard_Real    aDefaultTol = 1.e-3;
  Standard_Integer mode = 0;

  eps = Draw::Atof(a[2]);
  mode = Draw::Atoi(a[3]);
  if(mode > 0) onlyClosed = Standard_True;
  mode = Draw::Atoi(a[4]);
  if(mode > 0) isUseSpan = Standard_True;

  mode = Draw::Atoi(a[5]);
  if(mode == 1 || mode == 3) CGFlag = Standard_True;
  if(mode == 2 || mode == 3) IFlag = Standard_True;

  //OSD_Chronometer aChrono;

  //aChrono.Reset();
  //aChrono.Start();
  eps = BRepGProp::VolumePropertiesGK(S, G, eps, onlyClosed, isUseSpan, CGFlag, IFlag, SkipShared);
  //aChrono.Stop();

  Standard_SStream aSStream0;
  Standard_Integer anOutWidth = 24;

  aSStream0.precision(15);
  aSStream0 << "\n\n";
  aSStream0 << "Mass : " << std::setw(anOutWidth) << G.Mass() << "\n\n";
  aSStream0 << "Relative error of mass computation : " <<  std::setw(anOutWidth) << eps <<  "\n\n";
  aSStream0 << std::ends;
  di << aSStream0;

  if(CGFlag || IFlag) {
    Standard_SStream aSStream1;
    gp_Pnt P = G.CentreOfMass();
    if (n > 6) {
      Draw::Set(a[6],P.X());
    }
    if (n > 7) {
      Draw::Set(a[7],P.Y());
    }
    if (n > 8) {
      Draw::Set(a[8],P.Z());
    }
  
    aSStream1.precision(15);
    aSStream1 << "Center of gravity : \n";
    aSStream1 << "X = " << std::setw(anOutWidth) << P.X() << "\n";
    aSStream1 << "Y = " << std::setw(anOutWidth) << P.Y() << "\n";
    aSStream1 << "Z = " << std::setw(anOutWidth) << P.Z() << "\n";
    aSStream1 << "\n";
 
    if(IFlag) {
      gp_Mat I = G.MatrixOfInertia();

      aSStream1 << "Matrix of Inertia : \n";
      aSStream1 << std::setw(anOutWidth) << I(1,1);
      aSStream1 << " " << std::setw(anOutWidth) << I(1,2);
      aSStream1 << " " << std::setw(anOutWidth) << I(1,3) << "\n";
      aSStream1 << std::setw(anOutWidth) << I(2,1);
      aSStream1 << " " << std::setw(anOutWidth) << I(2,2);
      aSStream1 << " " << std::setw(anOutWidth) << I(2,3) << "\n";
      aSStream1 << std::setw(anOutWidth) << I(3,1);
      aSStream1 << " " << std::setw(anOutWidth) << I(3,2);
      aSStream1 << " " << std::setw(anOutWidth) << I(3,3) << "\n";
      aSStream1 << "\n";
    }
    aSStream1 << std::ends;
    di << aSStream1;
  }

  if(IFlag) {

    GProp_PrincipalProps Pr = G.PrincipalProperties();

    Standard_Real Ix,Iy,Iz;
    Pr.Moments(Ix,Iy,Iz);
    gp_Pnt P = G.CentreOfMass();
  
    Standard_SStream aSStream2;

    aSStream2.precision(15);
    aSStream2 << "Moments : \n";
    aSStream2 << "IX = " << std::setw(anOutWidth) << Ix << "\n";
    aSStream2 << "IY = " << std::setw(anOutWidth) << Iy << "\n";
    aSStream2 << "IZ = " << std::setw(anOutWidth) << Iz << "\n";
    aSStream2 << "\n";
    aSStream2 << "\n";
    aSStream2 << std::ends;
    di << aSStream2;

    gp_Ax2 axes(P,Pr.ThirdAxisOfInertia(),Pr.FirstAxisOfInertia());
    
    Handle(Draw_Axis3D) Dax = new Draw_Axis3D(axes,Draw_orange,30);
    dout << Dax;
  }
  return 0;
}



//=======================================================================
//function : GPropCommands
//purpose  : 
//=======================================================================

void  BRepTest::GPropCommands(Draw_Interpretor& theCommands)
{
  static Standard_Boolean done = Standard_False;
  if (done) return;
  done = Standard_True;

  DBRep::BasicCommands(theCommands);

  const char* g = "Global properties";
  theCommands.Add("lprops",
    "lprops name [x y z] [-skip] [-full] [-tri]: compute linear properties",
    __FILE__, props, g);
  theCommands.Add("sprops", "sprops name [epsilon] [x y z] [-skip] [-full] [-tri]:\n"
"  compute surfacic properties", __FILE__, props, g);
  theCommands.Add("vprops", "vprops name [epsilon] [c[losed]] [x y z] [-skip] [-full] [-tri]:\n"
"  compute volumic properties", __FILE__, props, g);

  theCommands.Add("vpropsgk",
		  "vpropsgk name epsilon closed span mode [x y z] [-skip] : compute volumic properties",
		  __FILE__,
		  vpropsgk,
		  g);
}