Salome HOME
Merge with branch V2_2_0_VISU_improvement
[modules/visu.git] / src / PIPELINE / VISU_CutLinesPL.cxx
1 //
2 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
3 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
4 //
5 //  This library is free software; you can redistribute it and/or
6 //  modify it under the terms of the GNU Lesser General Public
7 //  License as published by the Free Software Foundation; either
8 //  version 2.1 of the License.
9 //
10 //  This library is distributed in the hope that it will be useful,
11 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
12 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 //  Lesser General Public License for more details.
14 //
15 //  You should have received a copy of the GNU Lesser General Public
16 //  License along with this library; if not, write to the Free Software
17 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
18 //
19 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
20 //
21 //
22 // File:    VISU_PipeLine.cxx
23 // Author:  Alexey PETROV
24 // Module : VISU
25
26
27 #include "VISU_CutLinesPL.hxx"
28 #include "VISU_PipeLineUtils.hxx"
29 #include "VTKViewer_GeometryFilter.h"
30
31 #include <vtkAppendPolyData.h>
32
33 using namespace std;
34
35 vtkStandardNewMacro(VISU_CutLinesPL);
36
37 VISU_CutLinesPL::VISU_CutLinesPL(){}
38
39 void VISU_CutLinesPL::ShallowCopy(VISU_PipeLine *thePipeLine){
40   VISU_CutPlanesPL::ShallowCopy(thePipeLine);
41   if(VISU_CutLinesPL *aPipeLine = dynamic_cast<VISU_CutLinesPL*>(thePipeLine)){
42     SetOrientation(aPipeLine->GetPlaneOrientation(1),
43                    aPipeLine->GetRotateX(1),aPipeLine->GetRotateY(1),1);
44     SetDisplacement(aPipeLine->GetDisplacement(1),1);
45     SetDefault();
46     if (!aPipeLine->IsDefault()) SetPosition(aPipeLine->GetPosition());
47   }
48 }
49
50 void VISU_CutLinesPL::Init(){
51   VISU_CutPlanesPL::Init();
52
53   myBasePlane[0] = XY;
54   myBasePlane[1] = YZ;
55   myDisplacement[0] = myDisplacement[1] = 0.5;
56   myAng[0][0] = myAng[0][1] = myAng[0][2] = 0.0;
57   myAng[1][0] = myAng[1][1] = myAng[1][2] = 0.0;
58
59   myCondition = 1;
60   myPosition = 0;
61 }
62
63
64 void VISU_CutLinesPL::SetPosition(float thePosition){
65   myPosition = thePosition;
66   myCondition = 0;
67   Modified();
68 }
69 float VISU_CutLinesPL::GetPosition(){
70   float aPosition = myPosition;
71   if(myCondition){
72       float aDir[3], aBounds[6], aBoundPrj[3];
73       GetInput2()->GetBounds(aBounds);
74       GetDir(aDir,myAng[0],myBasePlane[0]);
75       GetBoundProject(aBoundPrj,aBounds,aDir);
76       aPosition = aBoundPrj[0] + aBoundPrj[2]*myDisplacement[0];
77   }
78   return aPosition;
79 }
80
81 void VISU_CutLinesPL::SetDefault(){
82   myCondition = 1;
83   Modified();
84 }
85 int VISU_CutLinesPL::IsDefault(){
86   return myCondition;
87 }
88
89
90 void VISU_CutLinesPL::Update(){
91   ClearAppendPolyData(myAppendPolyData);
92   SetPartPosition(1);
93   vtkAppendPolyData *anAppendPolyData = vtkAppendPolyData::New();
94   //Build base plane
95   float aDir[2][3], aBaseBounds[6];
96   GetInput2()->GetBounds(aBaseBounds);
97   GetDir(aDir[0],myAng[0],myBasePlane[0]);
98   vtkUnstructuredGrid* anUnstructuredGrid =
99     myFieldTransform->GetUnstructuredGridOutput();
100   CutWithPlanes(anAppendPolyData,anUnstructuredGrid,1,aDir[0],aBaseBounds,
101                 myPosition,myCondition,myDisplacement[0]);
102   //Build lines
103   float aBounds[6];
104   vtkDataSet *aDataSet = anAppendPolyData->GetOutput();
105   aDataSet->Update();
106   if(aDataSet->GetNumberOfCells() == 0)
107     aDataSet = anUnstructuredGrid;
108   aDataSet->GetBounds(aBounds);
109   GetDir(aDir[1],myAng[1],myBasePlane[1]);
110   VISU_CutPlanesPL::CutWithPlanes(myAppendPolyData,aDataSet,GetNbParts(),aDir[1],aBounds,
111                                   myPartPosition,myPartCondition,myDisplacement[1]);
112   anAppendPolyData->Register(myAppendPolyData);
113   anAppendPolyData->Delete();
114   //Calculate values for building of table
115   vtkMath::Cross(aDir[0],aDir[1],myDirLn);
116   GetBoundProject(myBoundPrjLn, aBaseBounds, myDirLn);
117   VISU::Mul(myDirLn,myBoundPrjLn[0],myBasePnt);
118   CorrectPnt(myBasePnt,aBaseBounds);
119
120   VISU_ScalarMapPL::Update();
121 }
122
123
124 void VISU_CutLinesPL::CutWithPlanes(vtkAppendPolyData* theAppendPolyData, vtkDataSet* theDataSet,
125                                     int theNbPlanes, float theDir[3], float theBounds[6],
126                                     float thePartPosition, int thePartCondition,
127                                     float theDisplacement)
128 {
129   vector<float> aPartPosition(1,thePartPosition);
130   vector<int> aPartCondition(1,thePartCondition);
131   VISU_CutPlanesPL::CutWithPlanes(theAppendPolyData,theDataSet,theNbPlanes,theDir,theBounds,
132                                   aPartPosition,aPartCondition,theDisplacement);
133 }