source: trunk/source/visualization/OpenInventor/src/SoCons.cc @ 954

Last change on this file since 954 was 954, checked in by garnier, 15 years ago

remise a jour
  • Property svn:mime-type set to text/cpp
File size: 15.3 KB
Line 
1//
2// ********************************************************************
3// * License and Disclaimer                                           *
4// *                                                                  *
5// * The  Geant4 software  is  copyright of the Copyright Holders  of *
6// * the Geant4 Collaboration.  It is provided  under  the terms  and *
7// * conditions of the Geant4 Software License,  included in the file *
8// * LICENSE and available at  http://cern.ch/geant4/license .  These *
9// * include a list of copyright holders.                             *
10// *                                                                  *
11// * Neither the authors of this software system, nor their employing *
12// * institutes,nor the agencies providing financial support for this *
13// * work  make  any representation or  warranty, express or implied, *
14// * regarding  this  software system or assume any liability for its *
15// * use.  Please see the license in the file  LICENSE  and URL above *
16// * for the full disclaimer and the limitation of liability.         *
17// *                                                                  *
18// * This  code  implementation is the result of  the  scientific and *
19// * technical work of the GEANT4 collaboration.                      *
20// * By using,  copying,  modifying or  distributing the software (or *
21// * any work based  on the software)  you  agree  to acknowledge its *
22// * use  in  resulting  scientific  publications,  and indicate your *
23// * acceptance of all terms of the Geant4 Software license.          *
24// ********************************************************************
25//
26//
27//
28// $Id: SoCons.cc,v 1.8 2006/06/29 21:22:38 gunter Exp $
29// GEANT4 tag $Name:  $
30//
31/*-----------------------------HEPVis---------------------------------------*/
32/*                                                                          */
33/* Node:             SoCons                                                 */
34/* Description:      Represents the G4Cons Geant Geometry entity            */
35/* Author:           Joe Boudreau Nov 11 1996                               */
36/*                                                                          */
37/*--------------------------------------------------------------------------*/
38
39#ifdef G4VIS_BUILD_OI_DRIVER
40
41// this :
42#include "HEPVis/nodes/SoCons.h"
43
44#include <assert.h>
45#include <cmath>
46#include <Inventor/SbBox.h>
47#include <Inventor/actions/SoAction.h>
48#include <Inventor/fields/SoSFFloat.h>
49#include <Inventor/misc/SoChildList.h>
50#include <Inventor/nodes/SoSeparator.h>
51#include <Inventor/nodes/SoIndexedFaceSet.h>
52#include <Inventor/nodes/SoNormal.h>
53#include <Inventor/nodes/SoCoordinate3.h>
54#include <Inventor/nodes/SoNormalBinding.h>
55#include <Inventor/SoPrimitiveVertex.h>
56#include <Inventor/elements/SoTextureCoordinateElement.h>
57
58#include "HEPVis/SbMath.h"
59
60// This statement is required
61SO_NODE_SOURCE(SoCons)
62
63// Constructor
64SoCons::SoCons() {
65  // This statement is required
66  SO_NODE_CONSTRUCTOR(SoCons);
67
68  // Data fields are initialized like this:
69  SO_NODE_ADD_FIELD(fRmin1,               (0.0));
70  SO_NODE_ADD_FIELD(fRmin2,               (0.0));
71  SO_NODE_ADD_FIELD(fRmax1,               (1.0));
72  SO_NODE_ADD_FIELD(fRmax2,               (1.0));
73  SO_NODE_ADD_FIELD(fDz,                 (10.0));
74  SO_NODE_ADD_FIELD(fSPhi,                (0.0));
75  SO_NODE_ADD_FIELD(fDPhi,             ((float)(2*M_PI)));
76  SO_NODE_ADD_FIELD(smoothDraw,          (TRUE));
77  SO_NODE_ADD_FIELD(alternateRep,        (NULL));
78  children = new SoChildList(this);
79}
80
81// Destructor
82SoCons::~SoCons() {
83  delete children;
84}
85
86
87// initClass
88void SoCons::initClass(){
89  // This statement is required.
90  SO_NODE_INIT_CLASS(SoCons,SoShape,"Shape");
91}
92
93
94// generatePrimitives
95void SoCons::generatePrimitives(SoAction *action) {
96  // This variable is used to store each vertex
97  SoPrimitiveVertex pv;
98
99  // Access the stat from the action
100  SoState *state = action->getState();
101
102  // See if we have to use a texture coordinate function,
103  // rather than generating explicit texture coordinates.
104  SbBool useTexFunction=
105    (SoTextureCoordinateElement::getType(state) ==
106     SoTextureCoordinateElement::FUNCTION);
107
108  // If we need to generate texture coordinates with a function,
109  // we'll need an SoGLTextureCoordinateElement.  Otherwise, we'll
110  // set up the coordinates directly.
111  const SoTextureCoordinateElement *tce = NULL;
112  SbVec4f texCoord;
113  if (useTexFunction) {
114    tce = SoTextureCoordinateElement::getInstance(state);
115  } else {
116    texCoord[2] = 0.0;
117    texCoord[3] = 1.0;
118  }
119  SbVec3f point, normal;
120
121  ///////////////////////////////////////////////////////
122  //-----------------------------------------------------
123#define GEN_VERTEX(pv,x,y,z,s,t,nx,ny,nz)               \
124  point.setValue((float)(x),(float)(y),(float)(z));     \
125  normal.setValue((float)(nx),(float)(ny),(float)(nz)); \
126  if (useTexFunction) {                                 \
127    texCoord=tce->get(point,normal);                    \
128  } else {                                              \
129    texCoord[0]=(float)(s);                             \
130    texCoord[1]=(float)(t);                             \
131  }                                                     \
132  pv.setPoint(point);                                   \
133  pv.setNormal(normal);                                 \
134  pv.setTextureCoords(texCoord);                        \
135  shapeVertex(&pv);
136  //-----------------------------------------------------
137  ///////////////////////////////////////////////////////
138
139
140  int NPHI = (int)(2+22*std::fabs(fDPhi.getValue()/(2.0*M_PI)));
141  double deltaPhi = fDPhi.getValue()/NPHI;
142  double phi0     = fSPhi.getValue();
143  double phi1     = phi0 + fDPhi.getValue();
144  double rMax1    = fRmax1.getValue();
145  double rMin1    = fRmin1.getValue();
146  double rMax2    = fRmax2.getValue();
147  double rMin2    = fRmin2.getValue();
148  double zMax     = fDz.getValue();
149  double zMin     = -zMax;
150  double cosPhi0  = std::cos(phi0);
151  double sinPhi0  = std::sin(phi0);
152  double cosPhi1  = std::cos(phi1);
153  double sinPhi1  = std::sin(phi1);
154  double cosDeltaPhi = std::cos(deltaPhi);
155  double sinDeltaPhi = std::sin(deltaPhi);
156  //
157  // The outer surface!
158  //
159  beginShape(action,TRIANGLE_STRIP);
160  int    i;
161  double sinPhi=sinPhi0;
162  double cosPhi=cosPhi0;
163  for (i = 0; i<=NPHI; i++) {
164    GEN_VERTEX(pv,rMax2*cosPhi,rMax2*sinPhi,zMax,0.0,0.0,cosPhi,sinPhi,0);   
165    GEN_VERTEX(pv,rMax1*cosPhi,rMax1*sinPhi,zMin,1.0,1.0,cosPhi,sinPhi,0);   
166    inc(sinPhi, cosPhi, sinDeltaPhi, cosDeltaPhi);   
167  }
168  endShape();
169  //
170  // The inner surface!
171  //
172  beginShape(action,TRIANGLE_STRIP);
173  sinPhi=sinPhi0;
174  cosPhi=cosPhi0;
175  for (i = 0; i<=NPHI; i++) {
176    GEN_VERTEX(pv,rMin2*cosPhi,rMin2*sinPhi,zMax,0.0,0.0,-cosPhi,-sinPhi,0);   
177    GEN_VERTEX(pv,rMin1*cosPhi,rMin1*sinPhi,zMin,1.0,1.0,-cosPhi,-sinPhi,0);   
178    inc(sinPhi, cosPhi, sinDeltaPhi, cosDeltaPhi);   
179  }
180  endShape();
181  if (std::fabs(deltaPhi)<2.0*M_PI) {
182    //
183    // The end
184    //
185    beginShape(action,TRIANGLE_STRIP);
186    sinPhi=sinPhi0;
187    cosPhi=cosPhi0;
188    GEN_VERTEX(pv,rMax2*cosPhi,rMax2*sinPhi,zMax,0.0,0.0,sinPhi,-cosPhi,0);   
189    GEN_VERTEX(pv,rMax1*cosPhi,rMax1*sinPhi,zMin,1.0,1.0,sinPhi,-cosPhi,0);   
190    GEN_VERTEX(pv,rMin2*cosPhi,rMin2*sinPhi,zMax,1.0,0.0,sinPhi,-cosPhi,0);   
191    GEN_VERTEX(pv,rMin1*cosPhi,rMin1*sinPhi,zMin,0.0,1.0,sinPhi,-cosPhi,0);   
192    endShape();
193    //
194    // The other end
195    //
196    beginShape(action,TRIANGLE_STRIP);
197    sinPhi=sinPhi1;
198    cosPhi=cosPhi1;
199    GEN_VERTEX(pv,rMax2*cosPhi,rMax2*sinPhi, zMax,0.0,0.0,-sinPhi,+cosPhi,0); 
200    GEN_VERTEX(pv,rMax1*cosPhi,rMax1*sinPhi, zMin,1.0,1.0,-sinPhi,+cosPhi,0); 
201    GEN_VERTEX(pv,rMin2*cosPhi,rMin2*sinPhi, zMax,1.0,0.0,-sinPhi,+cosPhi,0); 
202    GEN_VERTEX(pv,rMin1*cosPhi,rMin1*sinPhi, zMin,0.0,1.0,-sinPhi,+cosPhi,0); 
203    endShape();
204  }
205
206  //
207  // The outer surface at z=+PDZ
208  //
209  beginShape(action,TRIANGLE_STRIP);
210  sinPhi=sinPhi0;
211  cosPhi=cosPhi0;
212  for (i = 0; i<=NPHI; i++) {
213    GEN_VERTEX(pv,rMin2*cosPhi,rMin2*sinPhi,zMax,0.0,0.0,0,0,1);   
214    GEN_VERTEX(pv,rMax2*cosPhi,rMax2*sinPhi,zMax,1.0,1.0,0,0,1);   
215    inc(sinPhi, cosPhi, sinDeltaPhi, cosDeltaPhi);   
216  }
217  endShape();
218  //
219  // The outer surface at z=-PDZ
220  //
221  beginShape(action,TRIANGLE_STRIP);
222  sinPhi=sinPhi0;
223  cosPhi=cosPhi0;
224  for (i = 0; i<=NPHI; i++) {
225    GEN_VERTEX(pv,rMin1*cosPhi,rMin1*sinPhi,zMin,0.0,0.0,0,0,-1);   
226    GEN_VERTEX(pv,rMax1*cosPhi,rMax1*sinPhi,zMin,1.0,1.0,0,0,-1);   
227    inc(sinPhi, cosPhi, sinDeltaPhi, cosDeltaPhi);   
228  }
229  endShape();
230
231}
232
233// getChildren
234SoChildList *SoCons::getChildren() const {
235  return children;
236}
237
238
239// computeBBox
240void SoCons::computeBBox(SoAction *, SbBox3f &box, SbVec3f &center ){
241  float fRmax= fRmax1.getValue(); 
242  if (fRmax2.getValue() > fRmax) fRmax = fRmax2.getValue();
243
244  SbVec3f vmin(-fRmax,-fRmax,-fDz.getValue()),
245          vmax( fRmax, fRmax, fDz.getValue());
246  center.setValue(0,0,0);
247  box.setBounds(vmin,vmax);
248}
249
250
251
252
253// updateChildren
254void SoCons::updateChildren() {
255
256
257  // Redraw the G4Cons....
258
259  assert(children->getLength()==1);
260  SoSeparator       *sep                = (SoSeparator *)  ( *children)[0];
261  SoCoordinate3     *theCoordinates     = (SoCoordinate3 *)      ( sep->getChild(0));
262  SoNormal          *theNormals         = (SoNormal *)           ( sep->getChild(1));
263  SoNormalBinding   *theNormalBinding   = (SoNormalBinding *)    ( sep->getChild(2));
264  SoIndexedFaceSet  *theFaceSet         = (SoIndexedFaceSet *)   ( sep->getChild(3));
265
266  const int NPHI=24, NPOINTS=2*(2*NPHI+2), NFACES=4*NPHI+2, NINDICES = NFACES*5;
267  float points[NPOINTS][3], normals[NFACES][3];
268#ifdef INVENTOR2_0
269  static long     indices[NINDICES];
270#else
271  static int32_t  indices[NINDICES];
272#endif
273  static int init=0;
274  double phi, pp, DeltaPhi;
275
276  // Indices need to be generated once! This is here to keep it close to the point
277  // generation, since otherwise it will be confusing.
278
279  int i;
280  if (!init) {
281    init = 1;
282    // Outer face
283    for (i = 0; i< NPHI; i++) {
284      // 0 1 3 2;
285      indices[5*i+0] =  2*i+0;
286      indices[5*i+1] =  2*i+1;
287      indices[5*i+2] =  2*i+3;
288      indices[5*i+3] =  2*i+2;
289      indices[5*i+4] = SO_END_FACE_INDEX;
290    }
291    // the inner face
292    for (i=0;i<NPHI;i++) {
293      indices[5*1*NPHI + 5*i+0] = 2*NPHI+2 + 2*i+0; 
294      indices[5*1*NPHI + 5*i+1] = 2*NPHI+2 + 2*i+1;
295      indices[5*1*NPHI + 5*i+2] = 2*NPHI+2 + 2*i+3;
296      indices[5*1*NPHI + 5*i+3] = 2*NPHI+2 + 2*i+2;
297      indices[5*1*NPHI + 5*i+4] = SO_END_FACE_INDEX;
298    }
299    // the top side
300    for (i=0;i<NPHI;i++) {
301      indices[5*2*NPHI + 5*i+0] = 2*i+0;
302      indices[5*2*NPHI + 5*i+1] = 2*i+2;
303      indices[5*2*NPHI + 5*i+2] = NPOINTS - (2*i+4);
304      indices[5*2*NPHI + 5*i+3] = NPOINTS - (2*i+2);
305      indices[5*2*NPHI + 5*i+4] = SO_END_FACE_INDEX;
306    }
307    // the bottom side
308    for (i=0;i<NPHI;i++) {
309      indices[5*3*NPHI + 5*i+0] = 2*i+1;
310      indices[5*3*NPHI + 5*i+1] = NPOINTS - (2*i+1);
311      indices[5*3*NPHI + 5*i+2] = NPOINTS - (2*i+3);
312      indices[5*3*NPHI + 5*i+3] = 2*i+3;
313      indices[5*3*NPHI + 5*i+4] = SO_END_FACE_INDEX;
314    }
315    // the odd side
316    indices[5*4*NPHI +0] = 2*NPHI;
317    indices[5*4*NPHI +1] = 2*NPHI+1;
318    indices[5*4*NPHI +2] = 2*NPHI+3;
319    indices[5*4*NPHI +3] = 2*NPHI+2;
320    indices[5*4*NPHI +4] = SO_END_FACE_INDEX;
321    // aother odd side
322    indices[5*4*NPHI +5 +0] = 0;
323    indices[5*4*NPHI +5 +1] = NPOINTS-2;
324    indices[5*4*NPHI +5 +2] = NPOINTS-1;
325    indices[5*4*NPHI +5 +3] = 1;
326    indices[5*4*NPHI +5 +4] = SO_END_FACE_INDEX;
327  }
328  // Points need to be generated each time:
329  // The outer surface
330  DeltaPhi = fDPhi.getValue()/NPHI, phi = fSPhi.getValue();
331  float  t,st,ct;
332  t = FATAN((fRmax2.getValue()-fRmax1.getValue())/(2*fDz.getValue()));
333  st = FSIN(t);
334  ct = FCOS(t);
335  for (i = 0; i<=NPHI; i++) {
336    points[2*i+0][0] = fRmax2.getValue()*FCOS(phi);
337    points[2*i+0][1] = fRmax2.getValue()*FSIN(phi);
338    points[2*i+0][2] = +fDz.getValue();
339    points[2*i+1][0] = fRmax1.getValue()*FCOS(phi);
340    points[2*i+1][1] = fRmax1.getValue()*FSIN(phi);
341    points[2*i+1][2] = -fDz.getValue();
342    pp = phi+DeltaPhi/2.0;
343    if (i!=NPHI) {
344      normals[i][0] = ct * FCOS(pp);
345      normals[i][1] = ct * FSIN(pp);
346      normals[i][2] = -st;
347    }
348    phi+=DeltaPhi;
349  }
350  // The inner surface
351  phi = fSPhi.getValue() + fDPhi.getValue();
352  t = FATAN((fRmin2.getValue()-fRmin1.getValue())/(2*fDz.getValue()));
353  st = FSIN(t);
354  ct = FCOS(t);
355  for (i = 0; i<=NPHI; i++) {
356    points[2*NPHI+2+2*i+0][0] = fRmin2.getValue()*FCOS(phi);
357    points[2*NPHI+2+2*i+0][1] = fRmin2.getValue()*FSIN(phi);
358    points[2*NPHI+2+2*i+0][2] = +fDz.getValue();
359    points[2*NPHI+2+2*i+1][0] = fRmin1.getValue()*FCOS(phi);
360    points[2*NPHI+2+2*i+1][1] = fRmin1.getValue()*FSIN(phi);
361    points[2*NPHI+2+2*i+1][2] = -fDz.getValue();
362    pp = phi-DeltaPhi/2.0;
363    if (i!=NPHI) {
364      normals[NPHI+i][0] = -ct*FCOS(pp);
365      normals[NPHI+i][1] = -ct*FSIN(pp);
366      normals[NPHI+i][2] = st;
367    }
368    phi-=DeltaPhi;
369  }
370  // The top side
371  for (i=0;i<NPHI;i++) {
372    normals[2*NPHI+i][0]=normals[2*NPHI+i][1]=0;
373    normals[2*NPHI+i][2]=  1.0;
374  }
375  // The bottom side
376  for (i=0;i<NPHI;i++) {
377    normals[3*NPHI+i][0]=normals[3*NPHI+i][1]=0;
378    normals[3*NPHI+i][2]= -1.0;
379  }
380  // The odd side
381  phi = fSPhi.getValue();
382  normals[4*NPHI+0][0]=  FSIN(phi);
383  normals[4*NPHI+0][1]= -FCOS(phi);
384  normals[4*NPHI+0][2]= 0;
385
386  // Another odd side
387  phi = fSPhi.getValue()+fDPhi.getValue();
388  normals[4*NPHI+1][0]= -FSIN(phi);
389  normals[4*NPHI+1][1]= +FCOS(phi);
390  normals[4*NPHI+1][2]=0;
391
392  for (int np=0;np<NPOINTS;np++) theCoordinates->point.set1Value(np,points[np][0],points[np][1],points[np][2]);
393  theFaceSet->coordIndex.setValues(0,NINDICES,indices);
394  if (smoothDraw.getValue()) {
395    //    This Line is replaced by the next one because of an apparent Bug in Inventor (mem. leak).
396    //    theNormals->vector.deleteValues(0);
397    for (int nf=0;nf<NFACES;nf++) theNormals->vector.set1Value(nf,normals[nf][0],normals[nf][1],normals[nf][2]);
398    theNormalBinding->value=SoNormalBinding::PER_FACE;
399  }
400  else {
401    for (int nf=0;nf<NFACES;nf++) theNormals->vector.set1Value(nf,normals[nf][0],normals[nf][1],normals[nf][2]);
402    theNormalBinding->value=SoNormalBinding::PER_FACE;
403  }
404}
405
406// generateChildren
407void SoCons::generateChildren() {
408
409  // This routines creates one SoSeparator, one SoCoordinate3, and
410  // one SoLineSet, and puts it in the child list.  This is done only
411  // once, whereas redrawing the position of the coordinates occurs each
412  // time an update is necessary, in the updateChildren routine.
413
414  assert(children->getLength() ==0);
415  SoSeparator      *sep              = new SoSeparator();
416  SoCoordinate3    *theCoordinates   = new SoCoordinate3();
417  SoNormal         *theNormals       = new SoNormal();
418  SoNormalBinding  *theNormalBinding = new SoNormalBinding();
419  SoIndexedFaceSet *theFaceSet       = new SoIndexedFaceSet();
420  //
421  // This line costs some in render quality! but gives speed.
422  //
423  sep->addChild(theCoordinates);
424  sep->addChild(theNormals);
425  sep->addChild(theNormalBinding);
426  sep->addChild(theFaceSet);
427  children->append(sep);
428}
429
430// generateAlternateRep
431void SoCons::generateAlternateRep() {
432
433  // This routine sets the alternate representation to the child
434  // list of this mode. 
435
436  if (children->getLength() == 0) generateChildren();
437  updateChildren();
438  alternateRep.setValue((SoSeparator *)  ( *children)[0]);
439}
440
441// clearAlternateRep
442void SoCons::clearAlternateRep() {
443  alternateRep.setValue(NULL);
444}
445
446#endif
Note: See TracBrowser for help on using the repository browser.