141 lines
5.0 KiB
C++
141 lines
5.0 KiB
C++
//
|
|
// ********************************************************************
|
|
// * DISCLAIMER *
|
|
// * *
|
|
// * The following disclaimer summarizes all the specific disclaimers *
|
|
// * of contributors to this software. The specific disclaimers,which *
|
|
// * govern, are listed with their locations in: *
|
|
// * http://cern.ch/geant4/license *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. *
|
|
// * *
|
|
// * This code implementation is the intellectual property of the *
|
|
// * GEANT4 collaboration. *
|
|
// * By copying, distributing or modifying the Program (or any work *
|
|
// * based on the Program) you indicate your acceptance of this *
|
|
// * statement, and all its terms. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
// $Id: G4HepRepFileSceneHandler.hh,v 1.7 2002/02/02 04:00:14 perl Exp $
|
|
// GEANT4 tag $Name: geant4-04-01 $
|
|
//
|
|
//
|
|
// Joseph Perl 27th January 2002
|
|
// A base class for a scene handler to export geometry and trajectories
|
|
// to the HepRep xml file format.
|
|
|
|
#ifndef G4HepRepFileSCENEHANDLER_HH
|
|
#define G4HepRepFileSCENEHANDLER_HH
|
|
|
|
// #define G4HEPREPFILEDEBUG // Comment this out to suppress debug code.
|
|
|
|
#include "G4VSceneHandler.hh"
|
|
#include "G4Box.hh"
|
|
#include "G4Cons.hh"
|
|
#include "G4Tubs.hh"
|
|
#include "G4Trd.hh"
|
|
#include "G4Trap.hh"
|
|
#include "G4Sphere.hh"
|
|
#include "G4Para.hh"
|
|
#include "G4Torus.hh"
|
|
#include "G4Polycone.hh"
|
|
#include "G4Polyhedra.hh"
|
|
|
|
// HepRep
|
|
#include "HepRepXMLWriter.hh"
|
|
|
|
class G4VPhysicalVolume;
|
|
class G4LogicalVolume;
|
|
class G4ModelingParameters;
|
|
|
|
class G4HepRepFileSceneHandler: public G4VSceneHandler {
|
|
|
|
public:
|
|
G4HepRepFileSceneHandler(G4VGraphicsSystem& system,
|
|
const G4String& name);
|
|
virtual ~G4HepRepFileSceneHandler();
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// No need to implement these, but if you do...
|
|
void AddThis(const G4Box&);
|
|
void AddThis(const G4Cons&);
|
|
void AddThis(const G4Tubs&);
|
|
void AddThis(const G4Trd&);
|
|
void AddThis(const G4Trap&);
|
|
void AddThis(const G4Sphere&);
|
|
void AddThis(const G4Para&);
|
|
void AddThis(const G4Torus&);
|
|
void AddThis(const G4Polycone&);
|
|
void AddThis(const G4Polyhedra&);
|
|
void AddThis(const G4VSolid&);
|
|
// void PreAddThis(const G4Transform3D& objectTransformation,
|
|
// const G4VisAttributes&);
|
|
// void PostAddThis();
|
|
|
|
// Allows G4PhysicalVolumeModel to provide extra information...
|
|
void EstablishSpecials(G4PhysicalVolumeModel&);
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Required implementation of pure virtual functions...
|
|
|
|
void AddPrimitive(const G4Polyline&);
|
|
void AddPrimitive(const G4Text&);
|
|
void AddPrimitive(const G4Circle&);
|
|
void AddPrimitive(const G4Square&);
|
|
void AddPrimitive(const G4Polyhedron&);
|
|
void AddPrimitive(const G4NURBS&);
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////
|
|
// Further optional AddPrimtive methods. Explicitly invoke base
|
|
// class methods if not otherwise defined to avoid warnings about
|
|
// hiding of base class methods.
|
|
void AddPrimitive(const G4Polymarker&);
|
|
void AddPrimitive(const G4Scale& scale) {
|
|
G4VSceneHandler::AddPrimitive (scale);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Further optional virtual functions...
|
|
|
|
// void BeginPrimitives(const G4Transform3D& objectTransformation);
|
|
// void EndPrimitives();
|
|
|
|
void BeginModeling();
|
|
void EndModeling();
|
|
|
|
////////////////////////////////////////////////////////////////
|
|
// Required...
|
|
|
|
static G4int GetSceneCount() {return fSceneCount;}
|
|
|
|
HepRepXMLWriter *GetHepRepXMLWriter();
|
|
|
|
protected:
|
|
static G4int fSceneIdCount; // Counter for HepRep scene handlers.
|
|
static G4int fSceneCount; // No. of extanct scene handlers.
|
|
G4int fCurrentDepth; // Current depth of geom. hierarchy.
|
|
G4VPhysicalVolume* fpCurrentPV; // Current physical volume.
|
|
G4LogicalVolume* fpCurrentLV; // Current logical volume.
|
|
const G4ModelingParameters* fpOriginalMP; // Keeps pointer to original.
|
|
G4ModelingParameters* fpNonCullingMP; // For temporary non-culling.
|
|
|
|
private:
|
|
HepRepXMLWriter *hepRepXMLWriter;
|
|
void AddHepRepInstance(const char* primName,
|
|
const G4Visible visible);
|
|
int fileCounter;
|
|
|
|
#ifdef G4HEPREPFILEDEBUG
|
|
void PrintThings();
|
|
#endif
|
|
|
|
};
|
|
|
|
#endif
|