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.0)


  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 "G4Point3D.hh"
 45 #include "G4VisAttributes.hh"                      46 #include "G4VisAttributes.hh"
 46 #include "G4VModel.hh"                             47 #include "G4VModel.hh"
 47 #include "G4Scene.hh"                              48 #include "G4Scene.hh"
 48 #include "G4Polyhedron.hh"                         49 #include "G4Polyhedron.hh"
 49 #include "G4Box.hh"                                50 #include "G4Box.hh"
 50 #include "G4Cons.hh"                               51 #include "G4Cons.hh"
 51 #include "G4Polyline.hh"                           52 #include "G4Polyline.hh"
 52 #include "G4Trd.hh"                                53 #include "G4Trd.hh"
 53 #include "G4Tubs.hh"                               54 #include "G4Tubs.hh"
 54 #include "G4Text.hh"                               55 #include "G4Text.hh"
 55 #include "G4Circle.hh"                             56 #include "G4Circle.hh"
 56 #include "G4Square.hh"                             57 #include "G4Square.hh"
 57                                                    58 
 58 #include "G4VRML2FileSceneHandler.hh"              59 #include "G4VRML2FileSceneHandler.hh"
 59 #include "G4VRML2FileViewer.hh"                    60 #include "G4VRML2FileViewer.hh"
 60 #include "G4VRML2File.hh"                          61 #include "G4VRML2File.hh"
 61                                                    62 
 62 // CONST                                           63 // CONST
 63                                                    64 
 64 const char WRL_FILE_HEADER[]       = "g4_";        65 const char WRL_FILE_HEADER[]       = "g4_";
 65 const char ENV_WRL_FILE_HEADER[]   = "G4VRMLFI     66 const char ENV_WRL_FILE_HEADER[]   = "G4VRMLFILE_HEADER";
 66 const char ENV_WRL_FILE_NAME[]     = "G4VRMLFI     67 const char ENV_WRL_FILE_NAME[]     = "G4VRMLFILE_FILE_NAME";
 67 const char ENV_VRML_VIEWER[]       = "G4VRMLFI     68 const char ENV_VRML_VIEWER[]       = "G4VRMLFILE_VIEWER";
 68 const char NO_VRML_VIEWER[]        = "NONE";       69 const char NO_VRML_VIEWER[]        = "NONE";
 69 const char VRMLFILE_DEST_DIR[]     = "G4VRMLFI     70 const char VRMLFILE_DEST_DIR[]     = "G4VRMLFILE_DEST_DIR";
 70 const int DEFAULT_MAX_WRL_FILE_NUM = 100;          71 const int DEFAULT_MAX_WRL_FILE_NUM = 100;
 71                                                    72 
 72 G4VRML2FileSceneHandler::G4VRML2FileSceneHandl     73 G4VRML2FileSceneHandler::G4VRML2FileSceneHandler(G4VRML2File& system,
 73                                                    74                                                  const G4String& name)
 74   : G4VSceneHandler(system, fSceneIdCount++, n     75   : G4VSceneHandler(system, fSceneIdCount++, name)
 75   , fSystem(system)                                76   , fSystem(system)
 76   , fFlagDestOpen(false)                           77   , fFlagDestOpen(false)
 77   , fPVPickable(false)                             78   , fPVPickable(false)
 78   , fDest()                                        79   , fDest()
 79 {                                                  80 {
 80   // output file name                              81   // output file name
 81   strcpy(fVRMLFileName, "");                       82   strcpy(fVRMLFileName, "");
 82                                                    83 
 83   // destination directory                         84   // destination directory
 84   if(std::getenv(VRMLFILE_DEST_DIR) == NULL)       85   if(std::getenv(VRMLFILE_DEST_DIR) == NULL)
 85   {                                                86   {
 86     strcpy(fVRMLFileDestDir, "");                  87     strcpy(fVRMLFileDestDir, "");
 87   }                                                88   }
 88   else                                             89   else
 89   {                                                90   {
 90     strcpy(fVRMLFileDestDir, std::getenv(VRMLF     91     strcpy(fVRMLFileDestDir, std::getenv(VRMLFILE_DEST_DIR));
 91   }                                                92   }
 92                                                    93 
 93   // maximum number of g4.prim files in the de     94   // maximum number of g4.prim files in the dest directory
 94   fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;  //      95   fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;  // initialization
 95   if(std::getenv("G4VRMLFILE_MAX_FILE_NUM") !=     96   if(std::getenv("G4VRMLFILE_MAX_FILE_NUM") != NULL)
 96   {                                                97   {
 97     sscanf(std::getenv("G4VRMLFILE_MAX_FILE_NU     98     sscanf(std::getenv("G4VRMLFILE_MAX_FILE_NUM"), "%d", &fMaxFileNum);
 98   }                                                99   }
 99   else                                            100   else
100   {                                               101   {
101     fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;       102     fMaxFileNum = DEFAULT_MAX_WRL_FILE_NUM;
102   }                                               103   }
103   if(fMaxFileNum < 1)                             104   if(fMaxFileNum < 1)
104   {                                               105   {
105     fMaxFileNum = 1;                              106     fMaxFileNum = 1;
106   }                                               107   }
107                                                   108 
108   // PV name pickability                          109   // PV name pickability
109   if(std::getenv("G4VRML_PV_PICKABLE") != NULL    110   if(std::getenv("G4VRML_PV_PICKABLE") != NULL)
110   {                                               111   {
111     int is_pickable;                              112     int is_pickable;
112     sscanf(std::getenv("G4VRML_PV_PICKABLE"),     113     sscanf(std::getenv("G4VRML_PV_PICKABLE"), "%d", &is_pickable);
113                                                   114 
114     if(is_pickable)                               115     if(is_pickable)
115     {                                             116     {
116       SetPVPickability(true);                     117       SetPVPickability(true);
117     }                                             118     }
118   }                                               119   }
119                                                   120 
120   // PV Transparency                              121   // PV Transparency
121   SetPVTransparency();                            122   SetPVTransparency();
122 }                                                 123 }
123                                                   124 
124 G4VRML2FileSceneHandler::~G4VRML2FileSceneHand    125 G4VRML2FileSceneHandler::~G4VRML2FileSceneHandler()
125 {                                                 126 {
126 #if defined DEBUG_FR_SCENE                        127 #if defined DEBUG_FR_SCENE
127   if(G4VisManager::GetVerbosity() >= G4VisMana    128   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
128     G4cout << "***** ~G4VRML2FileSceneHandler"    129     G4cout << "***** ~G4VRML2FileSceneHandler" << G4endl;
129 #endif                                            130 #endif
130   VRMLEndModeling();                              131   VRMLEndModeling();
131 }                                                 132 }
132                                                   133 
133 #define G4VRML2SCENEHANDLER G4VRML2FileSceneHa    134 #define G4VRML2SCENEHANDLER G4VRML2FileSceneHandler
134 #define IS_CONNECTED this->isConnected()          135 #define IS_CONNECTED this->isConnected()
135 #include "G4VRML2SceneHandlerFunc.icc"            136 #include "G4VRML2SceneHandlerFunc.icc"
136 #undef IS_CONNECTED                               137 #undef IS_CONNECTED
137 #undef G4VRML2SCENEHANDLER                        138 #undef G4VRML2SCENEHANDLER
138                                                   139 
139 void G4VRML2FileSceneHandler::connectPort()       140 void G4VRML2FileSceneHandler::connectPort()
140 {                                                 141 {
141   // g4_00.wrl, g4_01.wrl, ..., g4_MAX_FILE_IN    142   // g4_00.wrl, g4_01.wrl, ..., g4_MAX_FILE_INDEX.wrl
142   const int MAX_FILE_INDEX = fMaxFileNum - 1;     143   const int MAX_FILE_INDEX = fMaxFileNum - 1;
143                                                   144 
144   // dest directory (null if no environmental     145   // dest directory (null if no environmental variables is set)
145   strcpy(fVRMLFileName, fVRMLFileDestDir);        146   strcpy(fVRMLFileName, fVRMLFileDestDir);
146   // add filename if environment variable supp    147   // add filename if environment variable supplied
147   if(std::getenv(ENV_WRL_FILE_NAME))              148   if(std::getenv(ENV_WRL_FILE_NAME))
148   {                                               149   {
149     strcat(fVRMLFileName, std::getenv(ENV_WRL_    150     strcat(fVRMLFileName, std::getenv(ENV_WRL_FILE_NAME));
150   }                                               151   }
151   else                                            152   else
152   {                                               153   {
153     // Determine VRML file name                   154     // Determine VRML file name
154     for(int i = 0; i < fMaxFileNum; i++)          155     for(int i = 0; i < fMaxFileNum; i++)
155     {                                             156     {
156       // Message in the final execution           157       // Message in the final execution
157       if(i == MAX_FILE_INDEX)                     158       if(i == MAX_FILE_INDEX)
158       {                                           159       {
159         if(G4VisManager::GetVerbosity() >= G4V    160         if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
160         {                                         161         {
161           G4cout << "=========================    162           G4cout << "===========================================" << G4endl;
162           G4cout << "WARNING MESSAGE from VRML    163           G4cout << "WARNING MESSAGE from VRML2FILE driver:     " << G4endl;
163           G4cout << "  This file name is the f    164           G4cout << "  This file name is the final one in the   " << G4endl;
164           G4cout << "  automatic updation of t    165           G4cout << "  automatic updation of the output file name." << G4endl;
165           G4cout << "  You may overwrite exist    166           G4cout << "  You may overwrite existing files, i.e.   " << G4endl;
166           G4cout << "  g4_XX.wrl.                 167           G4cout << "  g4_XX.wrl.                               " << G4endl;
167           G4cout << "=========================    168           G4cout << "===========================================" << G4endl;
168         }                                         169         }
169       }                                           170       }
170                                                   171 
171       // re-determine file name as G4VRMLFILE_    172       // re-determine file name as G4VRMLFILE_DEST_DIR/g4_XX.wrl
172       std::ostringstream filename;                173       std::ostringstream filename;
173       filename << fVRMLFileDestDir;               174       filename << fVRMLFileDestDir;
174       if(std::getenv(ENV_WRL_FILE_HEADER) == N    175       if(std::getenv(ENV_WRL_FILE_HEADER) == NULL)
175       {                                           176       {
176         filename << WRL_FILE_HEADER;              177         filename << WRL_FILE_HEADER;
177       }                                           178       }
178       else                                        179       else
179       {                                           180       {
180         filename << std::getenv(ENV_WRL_FILE_H    181         filename << std::getenv(ENV_WRL_FILE_HEADER) << '_';
181       }                                           182       }
182       filename << std::setw(2) << std::setfill    183       filename << std::setw(2) << std::setfill('0') << i << ".wrl";
183       strncpy(fVRMLFileName, filename.str().c_    184       strncpy(fVRMLFileName, filename.str().c_str(), sizeof(fVRMLFileName) - 1);
184       fVRMLFileName[sizeof(fVRMLFileName) - 1]    185       fVRMLFileName[sizeof(fVRMLFileName) - 1] = '\0';
185                                                   186 
186       // check validity of the file name          187       // check validity of the file name
187       std::ifstream fin;                          188       std::ifstream fin;
188       fin.open(fVRMLFileName);                    189       fin.open(fVRMLFileName);
189       if(!fin)                                    190       if(!fin)
190       {                                           191       {
191         // new file                               192         // new file
192         fin.close();                              193         fin.close();
193         break;                                    194         break;
194       }                                           195       }
195       else                                        196       else
196       {                                           197       {
197         // already exists (try next)              198         // already exists (try next)
198         fin.close();                              199         fin.close();
199       }                                           200       }
200                                                   201 
201     }  // for                                     202     }  // for
202   }                                               203   }
203                                                   204 
204   // open a VRML 2.0 file with determined file    205   // open a VRML 2.0 file with determined file name
205   if(G4VisManager::GetVerbosity() >= G4VisMana    206   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
206   {                                               207   {
207     G4cout << "===============================    208     G4cout << "===========================================" << G4endl;
208     G4cout << "Output VRML 2.0 file: " << fVRM    209     G4cout << "Output VRML 2.0 file: " << fVRMLFileName << G4endl;
209     G4cout << "Maximum number of files in the     210     G4cout << "Maximum number of files in the destination directory: "
210            << fMaxFileNum << G4endl;              211            << fMaxFileNum << G4endl;
211     G4cout << "  (Customizable with the enviro    212     G4cout << "  (Customizable with the environment variable: "
212               "G4VRMLFILE_MAX_FILE_NUM) "         213               "G4VRMLFILE_MAX_FILE_NUM) "
213            << G4endl;                             214            << G4endl;
214     G4cout << "===============================    215     G4cout << "===========================================" << G4endl;
215   }                                               216   }
216   fDest.open(fVRMLFileName);                      217   fDest.open(fVRMLFileName);
217   fFlagDestOpen = true;                           218   fFlagDestOpen = true;
218 }                                                 219 }
219                                                   220 
220 void G4VRML2FileSceneHandler::closePort()         221 void G4VRML2FileSceneHandler::closePort()
221 {                                                 222 {
222   char command[256];                              223   char command[256];
223   char viewer[256];                               224   char viewer[256];
224   strcpy(viewer, NO_VRML_VIEWER);  // initiali    225   strcpy(viewer, NO_VRML_VIEWER);  // initialization
225   if(std::getenv(ENV_VRML_VIEWER))                226   if(std::getenv(ENV_VRML_VIEWER))
226   {                                               227   {
227     strcpy(viewer, std::getenv(ENV_VRML_VIEWER    228     strcpy(viewer, std::getenv(ENV_VRML_VIEWER));
228   }                                               229   }
229                                                   230 
230   // close VRML file                              231   // close VRML file
231   fDest.close();                                  232   fDest.close();
232   fFlagDestOpen = false;                          233   fFlagDestOpen = false;
233   if(G4VisManager::GetVerbosity() >= G4VisMana    234   if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
234     G4cout << "*** VRML 2.0 File  " << fVRMLFi    235     G4cout << "*** VRML 2.0 File  " << fVRMLFileName << "  is generated."
235            << G4endl;                             236            << G4endl;
236                                                   237 
237   // Invoke viewer                                238   // Invoke viewer
238                                                   239 
239   if(!strcmp(viewer, NO_VRML_VIEWER))             240   if(!strcmp(viewer, NO_VRML_VIEWER))
240   {                                               241   {
241     if(G4VisManager::GetVerbosity() >= G4VisMa    242     if(G4VisManager::GetVerbosity() >= G4VisManager::errors)
242     {                                             243     {
243       G4cout << "MESSAGE from VRML2FILE driver    244       G4cout << "MESSAGE from VRML2FILE driver:" << G4endl;
244       G4cout << "    Set an environmental vari    245       G4cout << "    Set an environmental variable  ";
245       G4cout << ENV_VRML_VIEWER << G4endl;        246       G4cout << ENV_VRML_VIEWER << G4endl;
246       G4cout << "    if you want to visualize     247       G4cout << "    if you want to visualize the generated VRML file"
247              << G4endl;                           248              << G4endl;
248       G4cout << "    automatically.  For examp    249       G4cout << "    automatically.  For example, " << G4endl;
249       G4cout << "    setenv  " << ENV_VRML_VIE    250       G4cout << "    setenv  " << ENV_VRML_VIEWER << "  vrwave " << G4endl;
250       G4cout << "ALSO you may change the file     251       G4cout << "ALSO you may change the file header with "
251              << ENV_WRL_FILE_HEADER << G4endl;    252              << ENV_WRL_FILE_HEADER << G4endl;
252       G4cout << "    or the whole filename wit    253       G4cout << "    or the whole filename with " << ENV_WRL_FILE_NAME
253              << G4endl;                           254              << G4endl;
254     }                                             255     }
255   }                                               256   }
256   else                                            257   else
257   {                                               258   {
258     std::ostringstream ossCommand;                259     std::ostringstream ossCommand;
259     ossCommand << viewer << ' ' << fVRMLFileNa    260     ossCommand << viewer << ' ' << fVRMLFileName;
260     strncpy(command, ossCommand.str().c_str(),    261     strncpy(command, ossCommand.str().c_str(), sizeof(command) - 1);
261     command[sizeof(command) - 1] = '\0';          262     command[sizeof(command) - 1] = '\0';
262     int iErr                     = system(comm    263     int iErr                     = system(command);
263     if(iErr != 0)                                 264     if(iErr != 0)
264     {                                             265     {
265       G4ExceptionDescription ed;                  266       G4ExceptionDescription ed;
266       ed << "Error " << iErr << " when calling    267       ed << "Error " << iErr << " when calling system with \"" << command
267          << "\".";                                268          << "\".";
268       G4Exception("G4VRML2FileSceneHandler::cl    269       G4Exception("G4VRML2FileSceneHandler::closePort()", "VRML-2006",
269                   JustWarning, ed);               270                   JustWarning, ed);
270     }                                             271     }
271   }                                               272   }
272 }                                                 273 }
273                                                   274 
274 G4int G4VRML2FileSceneHandler::fSceneIdCount =    275 G4int G4VRML2FileSceneHandler::fSceneIdCount = 0;
275                                                   276