Salome HOME
52566]: TC7.5.0: Empty group of Balls at Diameter Equal to filter
[modules/smesh.git] / src / StdMeshers / StdMeshers_StartEndLength.cxx
1 // Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 //  SMESH StdMeshers_StartEndLength : implementaion of SMESH idl descriptions
24 //  File   : StdMeshers_StartEndLength.cxx
25 //  Module : SMESH
26 //
27 #include "StdMeshers_StartEndLength.hxx"
28
29 #include "SMESH_Algo.hxx"
30 #include "SMESH_Mesh.hxx"
31
32 #include <BRep_Tool.hxx>
33 #include <GCPnts_AbscissaPoint.hxx>
34 #include <GeomAdaptor_Curve.hxx>
35 #include <Geom_Curve.hxx>
36 #include <TopExp.hxx>
37 #include <TopLoc_Location.hxx>
38 #include <TopTools_IndexedMapOfShape.hxx>
39 #include <TopoDS.hxx>
40 #include <TopoDS_Edge.hxx>
41
42 using namespace std;
43
44 //=============================================================================
45 /*!
46  *  
47  */
48 //=============================================================================
49
50 StdMeshers_StartEndLength::StdMeshers_StartEndLength(int         hypId,
51                                                      int         studyId,
52                                                      SMESH_Gen * gen)
53      :SMESH_Hypothesis(hypId, studyId, gen)
54 {
55   _begLength = 1.;
56   _endLength = 10.;
57   _name = "StartEndLength";
58   _param_algo_dim = 1; // is used by SMESH_Regular_1D
59 }
60
61 //=============================================================================
62 /*!
63  *  
64  */
65 //=============================================================================
66
67 StdMeshers_StartEndLength::~StdMeshers_StartEndLength()
68 {
69 }
70
71 //=============================================================================
72 /*!
73  *  
74  */
75 //=============================================================================
76
77 void StdMeshers_StartEndLength::SetLength(double length, bool isStartLength)
78      throw(SALOME_Exception)
79 {
80   if ( (isStartLength ? _begLength : _endLength) != length ) {
81     if (length <= 0)
82       throw SALOME_Exception(LOCALIZED("length must be positive"));
83     if ( isStartLength )
84       _begLength = length;
85     else
86       _endLength = length;
87
88     NotifySubMeshesHypothesisModification();
89   }
90 }
91
92 //=============================================================================
93 /*!
94  *  
95  */
96 //=============================================================================
97
98 double StdMeshers_StartEndLength::GetLength(bool isStartLength) const
99 {
100   return isStartLength ? _begLength : _endLength;
101 }
102
103 //=============================================================================
104 /*!
105  *  
106  */
107 //=============================================================================
108
109 void StdMeshers_StartEndLength::SetReversedEdges( std::vector<int>& ids )
110 {
111   if ( ids != _edgeIDs ) {
112     _edgeIDs = ids;
113
114     NotifySubMeshesHypothesisModification();
115   }
116 }
117
118 //=============================================================================
119 /*!
120  *  
121  */
122 //=============================================================================
123
124 ostream & StdMeshers_StartEndLength::SaveTo(ostream & save)
125 {
126   int listSize = _edgeIDs.size();
127   save << _begLength << " " << _endLength << " " << listSize;
128
129   if ( listSize > 0 ) {
130     for ( int i = 0; i < listSize; i++) {
131       save << " " << _edgeIDs[i];
132     }
133     save << " " << _objEntry;
134   }
135
136   return save;
137 }
138
139 //=============================================================================
140 /*!
141  *  
142  */
143 //=============================================================================
144
145 istream & StdMeshers_StartEndLength::LoadFrom(istream & load)
146 {
147   bool isOK = true;
148   int intVal;
149   isOK = (load >> _begLength);
150   if (!isOK)
151     load.clear(ios::badbit | load.rdstate());
152   isOK = (load >> _endLength);
153
154   if (!isOK)
155     load.clear(ios::badbit | load.rdstate());
156   
157   isOK = (load >> intVal);
158   if (isOK && intVal > 0) {
159     _edgeIDs.reserve( intVal );
160     for (int i = 0; i < _edgeIDs.capacity() && isOK; i++) {
161       isOK = (load >> intVal);
162       if ( isOK ) _edgeIDs.push_back( intVal );
163     }
164     isOK = (load >> _objEntry);
165   }
166
167   return load;
168 }
169
170 //=============================================================================
171 /*!
172  *  
173  */
174 //=============================================================================
175
176 ostream & operator <<(ostream & save, StdMeshers_StartEndLength & hyp)
177 {
178   return hyp.SaveTo( save );
179 }
180
181 //=============================================================================
182 /*!
183  *  
184  */
185 //=============================================================================
186
187 istream & operator >>(istream & load, StdMeshers_StartEndLength & hyp)
188 {
189   return hyp.LoadFrom( load );
190 }
191
192 //================================================================================
193 /*!
194  * \brief Initialize start and end length by the mesh built on the geometry
195  * \param theMesh - the built mesh
196  * \param theShape - the geometry of interest
197  * \retval bool - true if parameter values have been successfully defined
198  */
199 //================================================================================
200
201 bool StdMeshers_StartEndLength::SetParametersByMesh(const SMESH_Mesh*   theMesh,
202                                                     const TopoDS_Shape& theShape)
203 {
204   if ( !theMesh || theShape.IsNull() )
205     return false;
206
207   _begLength = _endLength = 0.;
208
209   Standard_Real UMin, UMax;
210   TopLoc_Location L;
211
212   int nbEdges = 0;
213   TopTools_IndexedMapOfShape edgeMap;
214   TopExp::MapShapes( theShape, TopAbs_EDGE, edgeMap );
215   for ( int i = 1; i <= edgeMap.Extent(); ++i )
216   {
217     const TopoDS_Edge& edge = TopoDS::Edge( edgeMap( i ));
218     Handle(Geom_Curve) C = BRep_Tool::Curve(edge, L, UMin, UMax);
219     GeomAdaptor_Curve AdaptCurve(C, UMin, UMax);
220
221     vector< double > params;
222     SMESHDS_Mesh* aMeshDS = const_cast< SMESH_Mesh* >( theMesh )->GetMeshDS();
223     if ( SMESH_Algo::GetNodeParamOnEdge( aMeshDS, edge, params ))
224     {
225       nbEdges++;
226       _begLength += GCPnts_AbscissaPoint::Length( AdaptCurve, params[0], params[1]);
227       int nb = params.size();
228       _endLength += GCPnts_AbscissaPoint::Length( AdaptCurve, params[nb-2], params[nb-1]);
229     }
230   }
231   if ( nbEdges ) {
232     _begLength /= nbEdges;
233     _endLength /= nbEdges;
234   }
235   return nbEdges;
236 }
237
238 //================================================================================
239 /*!
240  * \brief Initialize my parameter values by default parameters.
241  *  \retval bool - true if parameter values have been successfully defined
242  */
243 //================================================================================
244
245 bool StdMeshers_StartEndLength::SetParametersByDefaults(const TDefaults&  dflts,
246                                                         const SMESH_Mesh* /*theMesh*/)
247 {
248   return (_begLength = _endLength = dflts._elemLength );
249 }
250