Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/visualization/VRML/src/G4VRML2FileSceneHandler.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

Diff markup

Differences between /visualization/VRML/src/G4VRML2FileSceneHandler.cc (Version 11.3.0) and /visualization/VRML/src/G4VRML2FileSceneHandler.cc (Version 11.2)


  1 //                                                  1 //
  2 // *******************************************      2 // ********************************************************************
  3 // * License and Disclaimer                         3 // * License and Disclaimer                                           *
  4 // *                                                4 // *                                                                  *
  5 // * The  Geant4 software  is  copyright of th      5 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
  6 // * the Geant4 Collaboration.  It is provided      6 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
  7 // * conditions of the Geant4 Software License      7 // * conditions of the Geant4 Software License,  included in the file *
  8 // * LICENSE and available at  http://cern.ch/      8 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
  9 // * include a list of copyright holders.           9 // * include a list of copyright holders.                             *
 10 // *                                               10 // *                                                                  *
 11 // * Neither the authors of this software syst     11 // * Neither the authors of this software system, nor their employing *
 12 // * institutes,nor the agencies providing fin     12 // * institutes,nor the agencies providing financial support for this *
 13 // * work  make  any representation or  warran     13 // * work  make  any representation or  warranty, express or implied, *
 14 // * regarding  this  software system or assum     14 // * regarding  this  software system or assume any liability for its *
 15 // * use.  Please see the license in the file      15 // * use.  Please see the license in the file  LICENSE  and URL above *
 16 // * for the full disclaimer and the limitatio     16 // * for the full disclaimer and the limitation of liability.         *
 17 // *                                               17 // *                                                                  *
 18 // * This  code  implementation is the result      18 // * This  code  implementation is the result of  the  scientific and *
 19 // * technical work of the GEANT4 collaboratio     19 // * technical work of the GEANT4 collaboration.                      *
 20 // * By using,  copying,  modifying or  distri     20 // * By using,  copying,  modifying or  distributing the software (or *
 21 // * any work based  on the software)  you  ag     21 // * any work based  on the software)  you  agree  to acknowledge its *
 22 // * use  in  resulting  scientific  publicati     22 // * use  in  resulting  scientific  publications,  and indicate your *
 23 // * acceptance of all terms of the Geant4 Sof     23 // * acceptance of all terms of the Geant4 Software license.          *
 24 // *******************************************     24 // ********************************************************************
 25 //                                                 25 //
 26 //                                                 26 //
 27 //                                                 27 //
 28 // G4VRML2FileSceneHandler.cc                      28 // G4VRML2FileSceneHandler.cc
 29 // Satoshi Tanaka & Yasuhide Sawada                29 // Satoshi Tanaka & Yasuhide Sawada
 30                                                    30 
 31 //#define DEBUG_FR_SCENE                           31 //#define DEBUG_FR_SCENE
 32                                                    32 
 33 #include <fstream>                                 33 #include <fstream>
 34 #include <stdio.h>                                 34 #include <stdio.h>
 35 #include <string.h>                                35 #include <string.h>
 36 #include <stdlib.h>                                36 #include <stdlib.h>
 37 #include <cmath>                                   37 #include <cmath>
 38 #include <sstream>                                 38 #include <sstream>
 39 #include <iomanip>                                 39 #include <iomanip>
 40                                                    40 
 41 #include "globals.hh"                              41 #include "globals.hh"
 42 #include "G4VPhysicalVolume.hh"                    42 #include "G4VPhysicalVolume.hh"
 43 #include "G4LogicalVolume.hh"                      43 #include "G4LogicalVolume.hh"
 44 #include "G4VisManager.hh"                         44 #include "G4VisManager.hh"
 45 #include "G4VisAttributes.hh"                      45 #include "G4VisAttributes.hh"
 46 #include "G4VModel.hh"                             46 #include "G4VModel.hh"
 47 #include "G4Scene.hh"                              47 #include "G4Scene.hh"
 48 #include "G4Polyhedron.hh"                         48 #include "G4Polyhedron.hh"
 49 #include "G4Box.hh"                                49 #include "G4Box.hh"
 50 #include "G4Cons.hh"                               50 #include "G4Cons.hh"
 51 #include "G4Polyline.hh"                           51 #include "G4Polyline.hh"
 52 #include "G4Trd.hh"                                52 #include "G4Trd.hh"
 53 #include "G4Tubs.hh"                               53 #include "G4Tubs.hh"
 54 #include "G4Text.hh"                               54 #include "G4Text.hh"
 55 #include "G4Circle.hh"                             55 #include "G4Circle.hh"
 56 #include "G4Square.hh"                             56 #include "G4Square.hh"
 57                                                    57 
 58 #include "G4VRML2FileSceneHandler.hh"              58 #include "G4VRML2FileSceneHandler.hh"
 59 #include "G4VRML2FileViewer.hh"                    59 #include "G4VRML2FileViewer.hh"
 60 #include "G4VRML2File.hh"                          60 #include "G4VRML2File.hh"
 61                                                    61 
 62 // CONST                                           62 // CONST
 63                                                    63 
 64 const char WRL_FILE_HEADER[]       = "g4_";        64 const char WRL_FILE_HEADER[]       = "g4_";
 65 const char ENV_WRL_FILE_HEADER[]   = "G4VRMLFI     65 const char ENV_WRL_FILE_HEADER[]   = "G4VRMLFILE_HEADER";
 66 const char ENV_WRL_FILE_NAME[]     = "G4VRMLFI     66 const char ENV_WRL_FILE_NAME[]     = "G4VRMLFILE_FILE_NAME";
 67 const char ENV_VRML_VIEWER[]       = "G4VRMLFI     67 const char ENV_VRML_VIEWER[]       = "G4VRMLFILE_VIEWER";
 68 const char NO_VRML_VIEWER[]        = "NONE";       68 const char NO_VRML_VIEWER[]        = "NONE";
 69 const char VRMLFILE_DEST_DIR[]     = "G4VRMLFI     69 const char VRMLFILE_DEST_DIR[]     = "G4VRMLFILE_DEST_DIR";
 70 const int DEFAULT_MAX_WRL_FILE_NUM = 100;          70 const int DEFAULT_MAX_WRL_FILE_NUM = 100;
 71                                                    71 
 72 G4VRML2FileSceneHandler::G4VRML2FileSceneHandl     72 G4VRML2FileSceneHandler::G4VRML2FileSceneHandler(G4VRML2File& system,
 73                                                    73                                                  const G4String& name)
 74   : G4VSceneHandler(system, fSceneIdCount++, n     74   : G4VSceneHandler(system, fSceneIdCount++, name)
 75   , fSystem(system)                                75   , fSystem(system)
 76   , fFlagDestOpen(false)                           76   , fFlagDestOpen(false)
 77   , fPVPickable(false)                             77   , fPVPickable(false)
 78   , fDest()                                        78   , fDest()
 79 {                                                  79 {
 80   // output file name                              80   // output file name
 81   strcpy(fVRMLFileName, "");                       81   strcpy(fVRMLFileName, "");
 82                                                    82 
 83   // destination directory                         83   // destination directory
 84   if(std::getenv(VRMLFILE_DEST_DIR) == NULL)       84   if(std::getenv(VRMLFILE_DEST_DIR) == NULL)
 85   {                                                85   {
 86     strcpy(fVRMLFileDestDir, "");                  86     strcpy(fVRMLFileDestDir, "");
 87   }                                                87   }
 88   else                                             88   else
 89   {                                                89   {
 90     strcpy(fVRMLFileDestDir, std::getenv(VRMLF     90     strcpy(fVRMLFileDestDir, std::getenv(VRMLFILE_DEST_DIR));
 91   }                                                91   }
 92                                                    92 
 93   // maximum number of g4.prim files in the de     93   // maximum number of g4.prim files in the dest directory
 94   fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;  //      94   fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;  // initialization
 95   if(std::getenv("G4VRMLFILE_MAX_FILE_NUM") !=     95   if(std::getenv("G4VRMLFILE_MAX_FILE_NUM") != NULL)
 96   {                                                96   {
 97     sscanf(std::getenv("G4VRMLFILE_MAX_FILE_NU     97     sscanf(std::getenv("G4VRMLFILE_MAX_FILE_NUM"), "%d", &fMaxFileNum);
 98   }                                                98   }
 99   else                                             99   else
100   {                                               100   {
101     fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;       101     fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;
102   }                                               102   }
103   if(fMaxFileNum < 1)                             103   if(fMaxFileNum < 1)
104   {                                               104   {
105     fMaxFileNum = 1;                              105     fMaxFileNum = 1;
106   }                                               106   }
107                                                   107 
108   // PV name pickability                          108   // PV name pickability
109   if(std::getenv("G4VRML_PV_PICKABLE") != NULL    109   if(std::getenv("G4VRML_PV_PICKABLE") != NULL)
110   {                                               110   {
111     int is_pickable;                              111     int is_pickable;
112     sscanf(std::getenv("G4VRML_PV_PICKABLE"),     112     sscanf(std::getenv("G4VRML_PV_PICKABLE"), "%d", &is_pickable);
113                                                   113 
114     if(is_pickable)                               114     if(is_pickable)
115     {                                             115     {
116       SetPVPickability(true);                     116       SetPVPickability(true);
117     }                                             117     }
118   }                                               118   }
119                                                   119 
120   // PV Transparency                              120   // PV Transparency
121   SetPVTransparency();                            121   SetPVTransparency();
122 }                                                 122 }
123                                                   123 
124 G4VRML2FileSceneHandler::~G4VRML2FileSceneHand    124 G4VRML2FileSceneHandler::~G4VRML2FileSceneHandler()
125 {                                                 125 {
126 #if defined DEBUG_FR_SCENE                        126 #if defined DEBUG_FR_SCENE
127   if(G4VisManager::GetVerbosity() >= G4VisMana    127   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
128     G4cout << "***** ~G4VRML2FileSceneHandler"    128     G4cout << "***** ~G4VRML2FileSceneHandler" << G4endl;
129 #endif                                            129 #endif
130   VRMLEndModeling();                              130   VRMLEndModeling();
131 }                                                 131 }
132                                                   132 
133 #define G4VRML2SCENEHANDLER G4VRML2FileSceneHa    133 #define G4VRML2SCENEHANDLER G4VRML2FileSceneHandler
134 #define IS_CONNECTED this->isConnected()          134 #define IS_CONNECTED this->isConnected()
135 #include "G4VRML2SceneHandlerFunc.icc"            135 #include "G4VRML2SceneHandlerFunc.icc"
136 #undef IS_CONNECTED                               136 #undef IS_CONNECTED
137 #undef G4VRML2SCENEHANDLER                        137 #undef G4VRML2SCENEHANDLER
138                                                   138 
139 void G4VRML2FileSceneHandler::connectPort()       139 void G4VRML2FileSceneHandler::connectPort()
140 {                                                 140 {
141   // g4_00.wrl, g4_01.wrl, ..., g4_MAX_FILE_IN    141   // g4_00.wrl, g4_01.wrl, ..., g4_MAX_FILE_INDEX.wrl
142   const int MAX_FILE_INDEX = fMaxFileNum - 1;     142   const int MAX_FILE_INDEX = fMaxFileNum - 1;
143                                                   143 
144   // dest directory (null if no environmental     144   // dest directory (null if no environmental variables is set)
145   strcpy(fVRMLFileName, fVRMLFileDestDir);        145   strcpy(fVRMLFileName, fVRMLFileDestDir);
146   // add filename if environment variable supp    146   // add filename if environment variable supplied
147   if(std::getenv(ENV_WRL_FILE_NAME))              147   if(std::getenv(ENV_WRL_FILE_NAME))
148   {                                               148   {
149     strcat(fVRMLFileName, std::getenv(ENV_WRL_    149     strcat(fVRMLFileName, std::getenv(ENV_WRL_FILE_NAME));
150   }                                               150   }
151   else                                            151   else
152   {                                               152   {
153     // Determine VRML file name                   153     // Determine VRML file name
154     for(int i = 0; i < fMaxFileNum; i++)          154     for(int i = 0; i < fMaxFileNum; i++)
155     {                                             155     {
156       // Message in the final execution           156       // Message in the final execution
157       if(i == MAX_FILE_INDEX)                     157       if(i == MAX_FILE_INDEX)
158       {                                           158       {
159         if(G4VisManager::GetVerbosity() >= G4V    159         if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
160         {                                         160         {
161           G4cout << "=========================    161           G4cout << "===========================================" << G4endl;
162           G4cout << "WARNING MESSAGE from VRML    162           G4cout << "WARNING MESSAGE from VRML2FILE driver:     " << G4endl;
163           G4cout << "  This file name is the f    163           G4cout << "  This file name is the final one in the   " << G4endl;
164           G4cout << "  automatic updation of t    164           G4cout << "  automatic updation of the output file name." << G4endl;
165           G4cout << "  You may overwrite exist    165           G4cout << "  You may overwrite existing files, i.e.   " << G4endl;
166           G4cout << "  g4_XX.wrl.                 166           G4cout << "  g4_XX.wrl.                               " << G4endl;
167           G4cout << "=========================    167           G4cout << "===========================================" << G4endl;
168         }                                         168         }
169       }                                           169       }
170                                                   170 
171       // re-determine file name as G4VRMLFILE_    171       // re-determine file name as G4VRMLFILE_DEST_DIR/g4_XX.wrl
172       std::ostringstream filename;                172       std::ostringstream filename;
173       filename << fVRMLFileDestDir;               173       filename << fVRMLFileDestDir;
174       if(std::getenv(ENV_WRL_FILE_HEADER) == N    174       if(std::getenv(ENV_WRL_FILE_HEADER) == NULL)
175       {                                           175       {
176         filename << WRL_FILE_HEADER;              176         filename << WRL_FILE_HEADER;
177       }                                           177       }
178       else                                        178       else
179       {                                           179       {
180         filename << std::getenv(ENV_WRL_FILE_H    180         filename << std::getenv(ENV_WRL_FILE_HEADER) << '_';
181       }                                           181       }
182       filename << std::setw(2) << std::setfill    182       filename << std::setw(2) << std::setfill('0') << i << ".wrl";
183       strncpy(fVRMLFileName, filename.str().c_    183       strncpy(fVRMLFileName, filename.str().c_str(), sizeof(fVRMLFileName) - 1);
184       fVRMLFileName[sizeof(fVRMLFileName) - 1]    184       fVRMLFileName[sizeof(fVRMLFileName) - 1] = '\0';
185                                                   185 
186       // check validity of the file name          186       // check validity of the file name
187       std::ifstream fin;                          187       std::ifstream fin;
188       fin.open(fVRMLFileName);                    188       fin.open(fVRMLFileName);
189       if(!fin)                                    189       if(!fin)
190       {                                           190       {
191         // new file                               191         // new file
192         fin.close();                              192         fin.close();
193         break;                                    193         break;
194       }                                           194       }
195       else                                        195       else
196       {                                           196       {
197         // already exists (try next)              197         // already exists (try next)
198         fin.close();                              198         fin.close();
199       }                                           199       }
200                                                   200 
201     }  // for                                     201     }  // for
202   }                                               202   }
203                                                   203 
204   // open a VRML 2.0 file with determined file    204   // open a VRML 2.0 file with determined file name
205   if(G4VisManager::GetVerbosity() >= G4VisMana    205   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
206   {                                               206   {
207     G4cout << "===============================    207     G4cout << "===========================================" << G4endl;
208     G4cout << "Output VRML 2.0 file: " << fVRM    208     G4cout << "Output VRML 2.0 file: " << fVRMLFileName << G4endl;
209     G4cout << "Maximum number of files in the     209     G4cout << "Maximum number of files in the destination directory: "
210            << fMaxFileNum << G4endl;              210            << fMaxFileNum << G4endl;
211     G4cout << "  (Customizable with the enviro    211     G4cout << "  (Customizable with the environment variable: "
212               "G4VRMLFILE_MAX_FILE_NUM) "         212               "G4VRMLFILE_MAX_FILE_NUM) "
213            << G4endl;                             213            << G4endl;
214     G4cout << "===============================    214     G4cout << "===========================================" << G4endl;
215   }                                               215   }
216   fDest.open(fVRMLFileName);                      216   fDest.open(fVRMLFileName);
217   fFlagDestOpen = true;                           217   fFlagDestOpen = true;
218 }                                                 218 }
219                                                   219 
220 void G4VRML2FileSceneHandler::closePort()         220 void G4VRML2FileSceneHandler::closePort()
221 {                                                 221 {
222   char command[256];                              222   char command[256];
223   char viewer[256];                               223   char viewer[256];
224   strcpy(viewer, NO_VRML_VIEWER);  // initiali    224   strcpy(viewer, NO_VRML_VIEWER);  // initialization
225   if(std::getenv(ENV_VRML_VIEWER))                225   if(std::getenv(ENV_VRML_VIEWER))
226   {                                               226   {
227     strcpy(viewer, std::getenv(ENV_VRML_VIEWER    227     strcpy(viewer, std::getenv(ENV_VRML_VIEWER));
228   }                                               228   }
229                                                   229 
230   // close VRML file                              230   // close VRML file
231   fDest.close();                                  231   fDest.close();
232   fFlagDestOpen = false;                          232   fFlagDestOpen = false;
233   if(G4VisManager::GetVerbosity() >= G4VisMana    233   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
234     G4cout << "*** VRML 2.0 File  " << fVRMLFi    234     G4cout << "*** VRML 2.0 File  " << fVRMLFileName << "  is generated."
235            << G4endl;                             235            << G4endl;
236                                                   236 
237   // Invoke viewer                                237   // Invoke viewer
238                                                   238 
239   if(!strcmp(viewer, NO_VRML_VIEWER))             239   if(!strcmp(viewer, NO_VRML_VIEWER))
240   {                                               240   {
241     if(G4VisManager::GetVerbosity() >= G4VisMa    241     if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
242     {                                             242     {
243       G4cout << "MESSAGE from VRML2FILE driver    243       G4cout << "MESSAGE from VRML2FILE driver:" << G4endl;
244       G4cout << "    Set an environmental vari    244       G4cout << "    Set an environmental variable  ";
245       G4cout << ENV_VRML_VIEWER << G4endl;        245       G4cout << ENV_VRML_VIEWER << G4endl;
246       G4cout << "    if you want to visualize     246       G4cout << "    if you want to visualize the generated VRML file"
247              << G4endl;                           247              << G4endl;
248       G4cout << "    automatically.  For examp    248       G4cout << "    automatically.  For example, " << G4endl;
249       G4cout << "    setenv  " << ENV_VRML_VIE    249       G4cout << "    setenv  " << ENV_VRML_VIEWER << "  vrwave " << G4endl;
250       G4cout << "ALSO you may change the file     250       G4cout << "ALSO you may change the file header with "
251              << ENV_WRL_FILE_HEADER << G4endl;    251              << ENV_WRL_FILE_HEADER << G4endl;
252       G4cout << "    or the whole filename wit    252       G4cout << "    or the whole filename with " << ENV_WRL_FILE_NAME
253              << G4endl;                           253              << G4endl;
254     }                                             254     }
255   }                                               255   }
256   else                                            256   else
257   {                                               257   {
258     std::ostringstream ossCommand;                258     std::ostringstream ossCommand;
259     ossCommand << viewer << ' ' << fVRMLFileNa    259     ossCommand << viewer << ' ' << fVRMLFileName;
260     strncpy(command, ossCommand.str().c_str(),    260     strncpy(command, ossCommand.str().c_str(), sizeof(command) - 1);
261     command[sizeof(command) - 1] = '\0';          261     command[sizeof(command) - 1] = '\0';
262     int iErr                     = system(comm    262     int iErr                     = system(command);
263     if(iErr != 0)                                 263     if(iErr != 0)
264     {                                             264     {
265       G4ExceptionDescription ed;                  265       G4ExceptionDescription ed;
266       ed << "Error " << iErr << " when calling    266       ed << "Error " << iErr << " when calling system with \"" << command
267          << "\".";                                267          << "\".";
268       G4Exception("G4VRML2FileSceneHandler::cl    268       G4Exception("G4VRML2FileSceneHandler::closePort()", "VRML-2006",
269                   JustWarning, ed);               269                   JustWarning, ed);
270     }                                             270     }
271   }                                               271   }
272 }                                                 272 }
273                                                   273 
274 G4int G4VRML2FileSceneHandler::fSceneIdCount =    274 G4int G4VRML2FileSceneHandler::fSceneIdCount = 0;
275                                                   275