0027038: Visualization - add new presentation for object manipulation
[occt.git] / src / StdPrs / StdPrs_ToolQuadric.cxx
1 // Created on: 2016-02-04
2 // Created by: Anastasia BORISOVA
3 // Copyright (c) 2016 OPEN CASCADE SAS
4 //
5 // This file is part of Open CASCADE Technology software library.
6 //
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
12 //
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
15
16 #include <StdPrs_ToolQuadric.hxx>
17
18 #include <gp_Quaternion.hxx>
19 #include <Graphic3d_ArrayOfTriangles.hxx>
20 #include <Poly_Array1OfTriangle.hxx>
21 #include <TColgp_Array1OfPnt.hxx>
22
23 //=======================================================================
24 //function : fillArrays
25 //purpose  :
26 //=======================================================================
27 void StdPrs_ToolQuadric::fillArrays (const gp_Trsf& theTrsf, TColgp_Array1OfPnt& theArray, NCollection_Array1<gp_Dir>& theNormals)
28 {
29   Standard_ShortReal aStepU = 1.0f / mySlicesNb;
30   Standard_ShortReal aStepV = 1.0f / myStacksNb;
31
32   for (Standard_Integer aU = 0; aU <= mySlicesNb; aU++)
33   {
34     const Standard_Real aParamU = aU * aStepU;
35     for (Standard_Integer aV = 0; aV <= myStacksNb; aV++)
36     {
37       const Standard_ShortReal aParamV = aV * aStepV;
38       const Standard_Integer   aVertId = aU * (myStacksNb + 1) + aV + 1;
39       gp_Pnt aVertex = Vertex(aParamU, aParamV);
40       gp_Dir aNormal = Normal(aParamU, aParamV);
41
42       aVertex.Transform (theTrsf);
43       aNormal.Transform (theTrsf);
44
45       theArray.SetValue (aVertId, aVertex);
46       theNormals.SetValue (aVertId, aNormal);
47     }
48   }
49 }
50
51 //=======================================================================
52 //function : FIllArray
53 //purpose  :
54 //=======================================================================
55 void StdPrs_ToolQuadric::FillArray (Handle(Graphic3d_ArrayOfTriangles)& theArray, const gp_Trsf& theTrsf)
56 {
57   const Standard_Integer aTrianglesNb = TrianglesNb();
58   theArray = new Graphic3d_ArrayOfTriangles (aTrianglesNb * 3, 0, Standard_True);
59
60   Poly_Array1OfTriangle aPolyTriangles (1, aTrianglesNb);
61   TColgp_Array1OfPnt anArray (1, aTrianglesNb * 3);
62   NCollection_Array1<gp_Dir> aNormals (1, aTrianglesNb * 3);
63   fillArrays (theTrsf, anArray, aNormals);
64
65   // Fill primitives
66   for (Standard_Integer aU = 0; aU < mySlicesNb; ++aU)
67   {
68     for (Standard_Integer aV = 1; aV <= myStacksNb; ++aV)
69     {
70       theArray->AddVertex (anArray.Value (aU * (myStacksNb + 1) + aV), aNormals.Value (aU * (myStacksNb + 1) + aV));
71       theArray->AddVertex (anArray.Value ((aU + 1) * (myStacksNb + 1) + aV), aNormals.Value ((aU + 1) * (myStacksNb + 1) + aV));
72       theArray->AddVertex (anArray.Value ((aU + 1) * (myStacksNb + 1) + (aV + 1)), aNormals.Value ((aU + 1) * (myStacksNb + 1) + (aV + 1)));
73       theArray->AddVertex (anArray.Value ((aU + 1) * (myStacksNb + 1) + (aV + 1)), aNormals.Value ((aU + 1) * (myStacksNb + 1) + (aV + 1)));
74       theArray->AddVertex (anArray.Value (aU * (myStacksNb + 1) + (aV + 1)), aNormals.Value (aU * (myStacksNb + 1) + (aV + 1)));
75       theArray->AddVertex (anArray.Value (aU * (myStacksNb + 1) + aV), aNormals.Value (aU * (myStacksNb + 1) + aV));
76     }
77   }
78 }
79
80 //=======================================================================
81 //function : FillTriangulation
82 //purpose  :
83 //=======================================================================
84 void StdPrs_ToolQuadric::FillArray (Handle(Graphic3d_ArrayOfTriangles)& theArray,
85                                     Handle(Poly_Triangulation)& theTriangulation,
86                                     const gp_Trsf& theTrsf)
87 {
88   const Standard_Integer aTrianglesNb = TrianglesNb();
89   theArray = new Graphic3d_ArrayOfTriangles(aTrianglesNb * 3, 0, Standard_True);
90
91   Poly_Array1OfTriangle aPolyTriangles(1, aTrianglesNb);
92   TColgp_Array1OfPnt anArray(1, aTrianglesNb * 3);
93   NCollection_Array1<gp_Dir> aNormals(1, aTrianglesNb * 3);
94   fillArrays(theTrsf, anArray, aNormals);
95
96   // Fill triangles
97   for (Standard_Integer aU = 0, anIndex = 0; aU < mySlicesNb; ++aU)
98   {
99     for (Standard_Integer aV = 1; aV <= myStacksNb; ++aV)
100     {
101       theArray->AddVertex(anArray.Value(aU * (myStacksNb + 1) + aV), aNormals.Value(aU * (myStacksNb + 1) + aV));
102       theArray->AddVertex(anArray.Value((aU + 1) * (myStacksNb + 1) + aV), aNormals.Value((aU + 1) * (myStacksNb + 1) + aV));
103       theArray->AddVertex(anArray.Value((aU + 1) * (myStacksNb + 1) + (aV + 1)), aNormals.Value((aU + 1) * (myStacksNb + 1) + (aV + 1)));
104       theArray->AddVertex(anArray.Value((aU + 1) * (myStacksNb + 1) + (aV + 1)), aNormals.Value((aU + 1) * (myStacksNb + 1) + (aV + 1)));
105       theArray->AddVertex(anArray.Value(aU * (myStacksNb + 1) + (aV + 1)), aNormals.Value(aU * (myStacksNb + 1) + (aV + 1)));
106       theArray->AddVertex(anArray.Value(aU * (myStacksNb + 1) + aV), aNormals.Value(aU * (myStacksNb + 1) + aV));
107
108       aPolyTriangles.SetValue (++anIndex, Poly_Triangle(aU * (myStacksNb + 1) + aV,
109                                (aU + 1) * (myStacksNb + 1) + aV,
110                                (aU + 1) * (myStacksNb + 1) + (aV + 1)));
111       aPolyTriangles.SetValue (++anIndex, Poly_Triangle((aU + 1) * (myStacksNb + 1) + (aV + 1),
112                                aU * (myStacksNb + 1) + (aV + 1),
113                                aU * (myStacksNb + 1) + aV));
114     }
115   }
116
117   theTriangulation = new Poly_Triangulation (anArray, aPolyTriangles);
118 }