190 lines
7.3 KiB
C++
190 lines
7.3 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * 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. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
/*
|
|
* =============================================================================
|
|
*
|
|
* Filename: CexmcScenePrimitives.cc
|
|
*
|
|
* Description: auxiliary scene primitives (radial lines etc.)
|
|
*
|
|
* Version: 1.0
|
|
* Created: 03.01.2011 11:45:33
|
|
* Revision: none
|
|
* Compiler: gcc
|
|
*
|
|
* Author: Alexey Radkov (),
|
|
* Company: PNPI
|
|
*
|
|
* =============================================================================
|
|
*/
|
|
|
|
#include <cmath>
|
|
#include <G4Polyline.hh>
|
|
#include <G4Circle.hh>
|
|
#include <G4Polyhedron.hh>
|
|
#include <G4ThreeVector.hh>
|
|
#include <G4VisAttributes.hh>
|
|
#include <G4VGraphicsScene.hh>
|
|
#include <G4AffineTransform.hh>
|
|
#include <G4Transform3D.hh>
|
|
#include <G4Point3D.hh>
|
|
#include <G4SystemOfUnits.hh>
|
|
#include "CexmcScenePrimitives.hh"
|
|
#include "CexmcScenePrimitivesMessenger.hh"
|
|
#include "CexmcSetup.hh"
|
|
#include "CexmcCommon.hh"
|
|
|
|
|
|
namespace
|
|
{
|
|
G4double CexmcRadialLineWidth( 2.0 );
|
|
G4double CexmcRadialLineCapScreenSize( 4.0 );
|
|
G4double CexmcMarkerScreenSize( 2.0 );
|
|
G4double CexmcICHlLineLineWidth( 1.0 );
|
|
G4Colour CexmcDefaultSPColour( 1.0, 1.0, 1.0 );
|
|
}
|
|
|
|
|
|
CexmcScenePrimitives::CexmcScenePrimitives( CexmcSetup * setup_ ) :
|
|
setup( setup_ ), markTargetCenter( false ), highlightInnerCrystals( false ),
|
|
messenger( NULL )
|
|
{
|
|
messenger = new CexmcScenePrimitivesMessenger( this );
|
|
SetGlobalDescription( CexmcScenePrimitivesDescription );
|
|
spColours[ CexmcTargetCenterMark_SP ] = CexmcDefaultSPColour;
|
|
spColours[ CexmcRadialLine_SP ] = CexmcDefaultSPColour;
|
|
spColours[ CexmcInnerCrystalsHl_SP ] = CexmcDefaultSPColour;
|
|
}
|
|
|
|
|
|
CexmcScenePrimitives::~CexmcScenePrimitives()
|
|
{
|
|
delete messenger;
|
|
}
|
|
|
|
|
|
void CexmcScenePrimitives::DescribeYourselfTo( G4VGraphicsScene & scene )
|
|
{
|
|
if ( markTargetCenter )
|
|
MarkTargetCenter( scene );
|
|
if ( highlightInnerCrystals )
|
|
HighlightInnerCrystals( scene );
|
|
for ( CexmcRadialLines::const_iterator k( radialLines.begin() );
|
|
k != radialLines.end(); ++k )
|
|
{
|
|
DrawRadialLine( scene, &*k );
|
|
}
|
|
}
|
|
|
|
|
|
void CexmcScenePrimitives::MarkTargetCenter( G4VGraphicsScene & scene )
|
|
{
|
|
G4Circle circle;
|
|
circle.SetScreenSize( CexmcMarkerScreenSize );
|
|
circle.SetFillStyle( G4Circle::filled );
|
|
circle.SetVisAttributes( spColours[ CexmcTargetCenterMark_SP ] );
|
|
|
|
const G4AffineTransform & transform( setup->GetTargetTransform() );
|
|
G4Transform3D transform3D( G4RotationMatrix(),
|
|
transform.NetTranslation() );
|
|
|
|
scene.BeginPrimitives( transform3D );
|
|
scene.AddPrimitive( circle );
|
|
scene.EndPrimitives();
|
|
}
|
|
|
|
|
|
void CexmcScenePrimitives::DrawRadialLine( G4VGraphicsScene & scene,
|
|
const CexmcRadialLine * rLine )
|
|
{
|
|
G4double theta( rLine->theta * deg );
|
|
G4double phi( rLine->phi * deg );
|
|
G4double length( rLine->length * cm );
|
|
G4Point3D radialLineEnd( - std::sin( theta ) * std::cos( phi ) * length,
|
|
std::sin( theta ) * std::sin( phi ) * length,
|
|
std::cos( theta ) * length );
|
|
|
|
G4Polyline line;
|
|
line.push_back( G4ThreeVector() );
|
|
line.push_back( radialLineEnd );
|
|
|
|
G4VisAttributes visAttributes( spColours[ CexmcRadialLine_SP ] );
|
|
visAttributes.SetLineWidth( CexmcRadialLineWidth );
|
|
line.SetVisAttributes( visAttributes );
|
|
|
|
G4Circle circle;
|
|
circle.SetScreenSize( CexmcRadialLineCapScreenSize );
|
|
circle.SetFillStyle( G4Circle::filled );
|
|
circle.SetVisAttributes( spColours[ CexmcRadialLine_SP ] );
|
|
|
|
const G4AffineTransform & transform( setup->GetTargetTransform() );
|
|
G4Transform3D transform3D( G4RotationMatrix(),
|
|
transform.NetTranslation() );
|
|
|
|
scene.BeginPrimitives( transform3D );
|
|
scene.AddPrimitive( circle );
|
|
scene.AddPrimitive( line );
|
|
scene.EndPrimitives();
|
|
}
|
|
|
|
|
|
void CexmcScenePrimitives::HighlightInnerCrystals( G4VGraphicsScene & scene )
|
|
{
|
|
const CexmcSetup::CalorimeterGeometryData & calorimeterGeometry(
|
|
setup->GetCalorimeterGeometry() );
|
|
G4double icWidth( calorimeterGeometry.crystalWidth *
|
|
( calorimeterGeometry.nCrystalsInRow - 2 ) / 2 );
|
|
G4double icHeight( calorimeterGeometry.crystalHeight *
|
|
( calorimeterGeometry.nCrystalsInColumn - 2 ) / 2 );
|
|
G4double icLength( calorimeterGeometry.crystalLength / 2 );
|
|
icWidth = icWidth < 0 ? 0 : icWidth;
|
|
icHeight = icHeight < 0 ? 0 : icHeight;
|
|
|
|
G4PolyhedronBox innerCrystals( icWidth, icHeight, icLength );
|
|
G4VisAttributes visAttributes( spColours[ CexmcInnerCrystalsHl_SP ] );
|
|
visAttributes.SetLineWidth( CexmcICHlLineLineWidth );
|
|
innerCrystals.SetVisAttributes( visAttributes );
|
|
|
|
const G4AffineTransform & transformLeft(
|
|
setup->GetCalorimeterLeftTransform() );
|
|
G4Transform3D transform3DLeft(
|
|
transformLeft.NetRotation().inverse(),
|
|
transformLeft.NetTranslation() );
|
|
const G4AffineTransform & transformRight(
|
|
setup->GetCalorimeterRightTransform() );
|
|
G4Transform3D transform3DRight(
|
|
transformRight.NetRotation().inverse(),
|
|
transformRight.NetTranslation() );
|
|
|
|
scene.BeginPrimitives( transform3DLeft );
|
|
scene.AddPrimitive( innerCrystals );
|
|
scene.EndPrimitives();
|
|
scene.BeginPrimitives( transform3DRight );
|
|
scene.AddPrimitive( innerCrystals );
|
|
scene.EndPrimitives();
|
|
}
|
|
|