Files
geant4/examples/advanced/ChargeExchangeMC/include/CexmcHistoManager.hh
2017-12-08 12:52:30 +01:00

344 lines
10 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: CexmcHistoManager.hh
*
* Description: histograming manager (singleton)
*
* Version: 1.0
* Created: 26.11.2009 20:55:16
* Revision: none
* Compiler: gcc
*
* Author: Alexey Radkov (),
* Company: PNPI
*
* =============================================================================
*/
#ifndef CEXMC_HISTO_MANAGER_HH
#define CEXMC_HISTO_MANAGER_HH
#ifdef CEXMC_USE_ROOT
#include <vector>
#include <map>
#include <Rtypes.h>
#include <G4String.hh>
#include "CexmcAngularRange.hh"
#include "CexmcCommon.hh"
class TDirectoryFile;
class TH1;
#ifdef CEXMC_USE_ROOTQT
class TQtWidget;
class TList;
class G4UIsession;
class G4UIQt;
#endif
class CexmcHistoManagerMessenger;
enum CexmcHistoType
{
CexmcMomentumBP_TPT_Histo,
CexmcMomentumBP_RT_Histo,
CexmcMomentumIP_TPT_Histo,
CexmcTPInMonitor_TPT_Histo,
CexmcTPInTarget_TPT_Histo,
CexmcTPInTarget_RT_Histo,
CexmcRecMasses_EDT_Histo,
CexmcRecMasses_RT_Histo,
CexmcAbsorbedEnergy_EDT_Histo,
CexmcAbsorbedEnergy_RT_Histo,
CexmcHistoType_ARReal_START,
CexmcRecMassOP_ARReal_RT_Histo = CexmcHistoType_ARReal_START,
CexmcRecMassNOP_ARReal_RT_Histo,
CexmcOPDPAtLeftCalorimeter_ARReal_EDT_Histo,
CexmcOPDPAtRightCalorimeter_ARReal_EDT_Histo,
CexmcOPDPAtLeftCalorimeter_ARReal_RT_Histo,
CexmcOPDPAtRightCalorimeter_ARReal_RT_Histo,
CexmcRecOPDPAtLeftCalorimeter_ARReal_EDT_Histo,
CexmcRecOPDPAtRightCalorimeter_ARReal_EDT_Histo,
CexmcRecOPDPAtLeftCalorimeter_ARReal_RT_Histo,
CexmcRecOPDPAtRightCalorimeter_ARReal_RT_Histo,
CexmcKinEnAtLeftCalorimeter_ARReal_TPT_Histo,
CexmcKinEnAtRightCalorimeter_ARReal_TPT_Histo,
CexmcKinEnAtLeftCalorimeter_ARReal_RT_Histo,
CexmcKinEnAtRightCalorimeter_ARReal_RT_Histo,
CexmcAbsEnInLeftCalorimeter_ARReal_EDT_Histo,
CexmcAbsEnInRightCalorimeter_ARReal_EDT_Histo,
CexmcAbsEnInLeftCalorimeter_ARReal_RT_Histo,
CexmcAbsEnInRightCalorimeter_ARReal_RT_Histo,
CexmcMissEnFromLeftCalorimeter_ARReal_RT_Histo,
CexmcMissEnFromRightCalorimeter_ARReal_RT_Histo,
CexmcKinEnOP_LAB_ARReal_TPT_Histo,
CexmcKinEnOP_LAB_ARReal_RT_Histo,
CexmcAngleOP_SCM_ARReal_TPT_Histo,
CexmcAngleOP_SCM_ARReal_RT_Histo,
CexmcRecAngleOP_SCM_ARReal_RT_Histo,
CexmcDiffAngleOP_SCM_ARReal_RT_Histo,
CexmcOpenAngle_ARReal_TPT_Histo,
CexmcOpenAngle_ARReal_RT_Histo,
CexmcRecOpenAngle_ARReal_RT_Histo,
CexmcDiffOpenAngle_ARReal_RT_Histo,
CexmcTPInTarget_ARReal_TPT_Histo,
CexmcTPInTarget_ARReal_RT_Histo,
CexmcHistoType_ARReal_END = CexmcTPInTarget_ARReal_RT_Histo,
CexmcHistoType_SIZE
};
class CexmcHistoManager
{
private:
typedef std::vector< TH1 * > CexmcHistoVector;
typedef std::map< CexmcHistoType, CexmcHistoVector > CexmcHistosMap;
typedef CexmcHistosMap::value_type CexmcHistoPair;
struct CexmcHistoAxisData
{
CexmcHistoAxisData() : nBins( 0 ), nBinsMin( 0 ), nBinsMax( 0 )
{}
CexmcHistoAxisData( Int_t nBins_, Double_t nBinsMin_,
Double_t nBinsMax_ ) :
nBins( nBins_ ), nBinsMin( nBinsMin_ ), nBinsMax( nBinsMax_ )
{}
Int_t nBins;
Double_t nBinsMin;
Double_t nBinsMax;
};
typedef std::vector< CexmcHistoAxisData > CexmcHistoAxes;
enum CexmcHistoImpl
{
Cexmc_TH1F,
Cexmc_TH2F,
Cexmc_TH3F
};
struct CexmcHistoData
{
CexmcHistoData() :
type( CexmcHistoType_SIZE ), impl( Cexmc_TH1F ),
isARHisto( false ), isARRec( false ), triggerType( CexmcTPT )
{}
CexmcHistoData( CexmcHistoType type_, CexmcHistoImpl impl_,
bool isARHisto_, bool isARRec_,
CexmcTriggerType triggerType_,
const G4String & name_, const G4String & title_,
const CexmcHistoAxes & axes_ ) :
type( type_ ), impl( impl_ ), isARHisto( isARHisto_ ),
isARRec( isARRec_ ), triggerType( triggerType_ ), name( name_ ),
title( title_ ), axes( axes_ )
{}
CexmcHistoType type;
CexmcHistoImpl impl;
bool isARHisto;
bool isARRec;
CexmcTriggerType triggerType;
G4String name;
G4String title;
CexmcHistoAxes axes;
};
public:
static CexmcHistoManager * Instance( void );
static void Destroy( void );
private:
CexmcHistoManager();
~CexmcHistoManager();
public:
void Initialize( void );
void SetupARHistos( const CexmcAngularRangeList & aRanges );
void AddARHistos( const CexmcAngularRange & aRange );
void Add( CexmcHistoType histoType, unsigned int index,
G4double x );
void Add( CexmcHistoType histoType, unsigned int index, G4double x,
G4double y );
void Add( CexmcHistoType histoType, unsigned int index, G4double x,
G4double y, G4double z );
void Add( CexmcHistoType histoType, unsigned int index, G4int binX,
G4int binY, G4double value );
void List( void ) const;
void Print( const G4String & value );
#ifdef CEXMC_USE_ROOTQT
void Draw( const G4String & histoName,
const G4String & histoDrawOptions = "" );
void EnableLiveHistograms( G4UIsession * session, G4bool on = true );
void AddHistoMenu( const G4String & handle,
const G4String & label = "Histograms" );
void SetDrawOptions1D( const G4String & value );
void SetDrawOptions2D( const G4String & value );
void SetDrawOptions3D( const G4String & value );
#endif
public:
void SetVerboseLevel( G4int value );
G4int GetVerboseLevel( void ) const;
private:
void AddHisto( const CexmcHistoData & data,
const CexmcAngularRange & aRange = CexmcAngularRange() );
void CreateHisto( CexmcHistoVector & histoVector,
CexmcHistoImpl histoImpl, const G4String & name,
const G4String & title,
const CexmcHistoAxes & axes );
#ifdef CEXMC_USE_ROOTQT
void BuildMenuTree( G4UIQt * session, const G4String & menu,
TList * ls );
void AddSubmenu( G4UIQt * session, const G4String & parent,
const G4String & name, const G4String & label );
#endif
private:
TDirectoryFile * outFile;
private:
CexmcHistosMap histos;
bool isInitialized;
G4String opName;
G4String nopName;
G4double opMass;
G4double nopMass;
G4int verboseLevel;
#ifdef CEXMC_USE_ROOTQT
private:
TQtWidget * rootCanvas;
G4bool areLiveHistogramsEnabled;
G4bool isHistoMenuInitialized;
G4String drawOptions1D;
G4String drawOptions2D;
G4String drawOptions3D;
G4String histoMenuHandle;
G4String histoMenuLabel;
#endif
private:
CexmcHistoManagerMessenger * messenger;
private:
static CexmcHistoManager * instance;
};
inline void CexmcHistoManager::SetVerboseLevel( G4int value )
{
verboseLevel = value;
}
inline G4int CexmcHistoManager::GetVerboseLevel( void ) const
{
return verboseLevel;
}
#ifdef CEXMC_USE_ROOTQT
inline void CexmcHistoManager::AddHistoMenu( const G4String & handle,
const G4String & label )
{
histoMenuHandle = handle;
histoMenuLabel = label;
}
inline void CexmcHistoManager::SetDrawOptions1D( const G4String & value )
{
drawOptions1D = value;
}
inline void CexmcHistoManager::SetDrawOptions2D( const G4String & value )
{
drawOptions2D = value;
}
inline void CexmcHistoManager::SetDrawOptions3D( const G4String & value )
{
drawOptions3D = value;
}
#endif //CEXMC_USE_ROOTQT
#endif
#endif