Geant4  10.03.p03
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4VRML2FileSceneHandler.cc
Go to the documentation of this file.
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 // $Id: G4VRML2FileSceneHandler.cc 105186 2017-07-14 12:02:06Z gcosmo $
28 //
29 // G4VRML2FileSceneHandler.cc
30 // Satoshi Tanaka & Yasuhide Sawada
31 
32 
33 //#define DEBUG_FR_SCENE
34 
35 #include <fstream>
36 #include <stdio.h>
37 #include <string.h>
38 #include <stdlib.h>
39 #include <cmath>
40 #include <sstream>
41 #include <iomanip>
42 
43 #include "globals.hh"
44 #include "G4VPhysicalVolume.hh"
45 #include "G4LogicalVolume.hh"
46 #include "G4VisManager.hh"
47 #include "G4Point3D.hh"
48 #include "G4VisAttributes.hh"
49 #include "G4VModel.hh"
50 #include "G4Scene.hh"
51 #include "G4Polyhedron.hh"
52 #include "G4Box.hh"
53 #include "G4Cons.hh"
54 #include "G4Polyline.hh"
55 #include "G4Trd.hh"
56 #include "G4Tubs.hh"
57 #include "G4Text.hh"
58 #include "G4Circle.hh"
59 #include "G4Square.hh"
60 
62 #include "G4VRML2FileViewer.hh"
63 #include "G4VRML2File.hh"
64 
65 // CONST
66 
67 const char WRL_FILE_HEADER [] = "g4_";
68 const char DEFAULT_WRL_FILE_NAME[] = "g4.wrl";
69 const char ENV_VRML_VIEWER [] = "G4VRMLFILE_VIEWER";
70 const char NO_VRML_VIEWER [] = "NONE";
71 const char VRMLFILE_DEST_DIR [] = "G4VRMLFILE_DEST_DIR";
72 const int DEFAULT_MAX_WRL_FILE_NUM = 100 ;
73 
74 
76  G4VSceneHandler(system, fSceneIdCount++, name),
77  fSystem(system),
78  fFlagDestOpen( false ),
79  fPVPickable ( false ),
80  fDest()
81 {
82  // output file name
83  strcpy(fVRMLFileName, "");
84 
85  // destination directory
86  if ( getenv( VRMLFILE_DEST_DIR ) == NULL ) {
87  strcpy( fVRMLFileDestDir, "" );
88  } else {
89  strcpy( fVRMLFileDestDir, getenv( VRMLFILE_DEST_DIR ) );
90  }
91 
92 
93  // maximum number of g4.prim files in the dest directory
94  fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM ; // initialization
95  if ( getenv( "G4VRMLFILE_MAX_FILE_NUM" ) != NULL ) {
96 
97  sscanf( getenv("G4VRMLFILE_MAX_FILE_NUM"), "%d", &fMaxFileNum ) ;
98 
99  } else {
100  fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM ;
101  }
102  if( fMaxFileNum < 1 ) { fMaxFileNum = 1; }
103 
104 
105  // PV name pickability
106  if( getenv( "G4VRML_PV_PICKABLE" ) != NULL ) {
107 
108  int is_pickable ;
109  sscanf( getenv("G4VRML_PV_PICKABLE"), "%d", &is_pickable ) ;
110 
111  if ( is_pickable ) { SetPVPickability ( true ) ; }
112  }
113 
114  // PV Transparency
115  SetPVTransparency ();
116 
117 }
118 
119 
121 {
122 #if defined DEBUG_FR_SCENE
124  G4cout << "***** ~G4VRML2FileSceneHandler" << G4endl;
125 #endif
126  VRMLEndModeling();
127 }
128 
129 
130 #define G4VRML2SCENEHANDLER G4VRML2FileSceneHandler
131 #define IS_CONNECTED this->isConnected()
132 #include "G4VRML2SceneHandlerFunc.icc"
133 #undef IS_CONNECTED
134 #undef G4VRML2SCENEHANDLER
135 
136 
138 {
139  // g4_00.wrl, g4_01.wrl, ..., g4_MAX_FILE_INDEX.wrl
140  const int MAX_FILE_INDEX = fMaxFileNum - 1 ;
141 
142  // dest directory (null if no environmental variables is set)
143  strcpy ( fVRMLFileName, fVRMLFileDestDir) ;
144 
145  // create full path name (default)
146  strcat ( fVRMLFileName, DEFAULT_WRL_FILE_NAME );
147 
148  // Determine VRML file name
149  for( int i = 0 ; i < fMaxFileNum ; i++) {
150 
151  // Message in the final execution
152  if( i == MAX_FILE_INDEX )
153  {
155  G4cout << "===========================================" << G4endl;
156  G4cout << "WARNING MESSAGE from VRML2FILE driver: " << G4endl;
157  G4cout << " This file name is the final one in the " << G4endl;
158  G4cout << " automatic updation of the output file name." << G4endl;
159  G4cout << " You may overwrite existing files, i.e. " << G4endl;
160  G4cout << " g4_XX.wrl. " << G4endl;
161  G4cout << "===========================================" << G4endl;
162  }
163  }
164 
165  // re-determine file name as G4VRMLFILE_DEST_DIR/g4_XX.wrl
166  std::ostringstream filename;
167  filename
168  << fVRMLFileDestDir << WRL_FILE_HEADER
169  << std::setw(2) << std::setfill('0') << i << ".wrl";
170  strncpy(fVRMLFileName,filename.str().c_str(),sizeof(fVRMLFileName));
171 
172  // check validity of the file name
173  std::ifstream fin ;
174  fin.open(fVRMLFileName) ;
175  if(!fin) {
176  // new file
177  fin.close();
178  break;
179  } else {
180  // already exists (try next)
181  fin.close();
182  }
183 
184  } // for
185 
186  // open a VRML 2.0 file with determined file name
188  G4cout << "===========================================" << G4endl;
189  G4cout << "Output VRML 2.0 file: " << fVRMLFileName << G4endl;
190  G4cout << "Maximum number of files in the destination directory: " << fMaxFileNum << G4endl;
191  G4cout << " (Customizable with the environment variable: G4VRMLFILE_MAX_FILE_NUM) " << G4endl;
192  G4cout << "===========================================" << G4endl;
193  }
194  fDest.open(fVRMLFileName) ; fFlagDestOpen = true ;
195 }
196 
197 
199 {
200  char command[256] ;
201  char viewer [256] ;
202  strcpy( viewer, NO_VRML_VIEWER ); // initialization
203  if( getenv( ENV_VRML_VIEWER ) ) {
204  strcpy( viewer, getenv( ENV_VRML_VIEWER ) ) ;
205  }
206 
207  // close VRML file
208  fDest.close(); fFlagDestOpen = false ;
210  G4cout << "*** VRML 2.0 File " << fVRMLFileName << " is generated." << G4endl;
211 
212 
213  // Invoke viewer
214 
215  if ( !strcmp(viewer, NO_VRML_VIEWER )) {
217  G4cout << "MESSAGE from VRML2FILE driver:" << G4endl;
218  G4cout << " Set an environmental variable " ;
220  G4cout << " if you want to visualize the generated VRML file" << G4endl;
221  G4cout << " automatically. For example, " << G4endl;
222  G4cout << " setenv " << ENV_VRML_VIEWER << " vrwave " << G4endl;
223  }
224  } else {
225  std::ostringstream ossCommand;
226  ossCommand << viewer << ' ' << fVRMLFileName;
227  strncpy(command,ossCommand.str().c_str(),sizeof(command));
228  (void) system( command );
229  }
230 }
231 
232 G4int G4VRML2FileSceneHandler::fSceneIdCount = 0;
const XML_Char * name
Definition: expat.h:151
const int DEFAULT_MAX_WRL_FILE_NUM
int G4int
Definition: G4Types.hh:78
const char ENV_VRML_VIEWER[]
const char VRMLFILE_DEST_DIR[]
G4GLOB_DLL std::ostream G4cout
G4VRML2FileSceneHandler(G4VRML2File &system, const G4String &name="")
const char NO_VRML_VIEWER[]
static Verbosity GetVerbosity()
#define G4endl
Definition: G4ios.hh:61
const char WRL_FILE_HEADER[]
typedef void(XMLCALL *XML_ElementDeclHandler)(void *userData
const char DEFAULT_WRL_FILE_NAME[]