Skip to content

Commit

Permalink
import MagneticField from g4SimHits
Browse files Browse the repository at this point in the history
  • Loading branch information
VinInn committed Jul 27, 2022
1 parent 264a5ff commit c8c63c9
Show file tree
Hide file tree
Showing 3 changed files with 36 additions and 128 deletions.
48 changes: 10 additions & 38 deletions TrackPropagation/Geant4e/python/geantRefit_cff.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,48 +6,20 @@

from TrackingTools.TrackRefitter.TracksToTrajectories_cff import *

from SimG4Core.Application.g4SimHits_cfi import g4SimHits as _g4SimHits


## Set up geometry
geopro = cms.EDProducer("GeometryProducer",
GeoFromDD4hep = cms.bool(True),
GeoFromDD4hep = cms.bool(False),
UseMagneticField = cms.bool(True),
UseSensitiveDetectors = cms.bool(False),
MagneticField = cms.PSet(
UseLocalMagFieldManager = cms.bool(False),
Verbosity = cms.bool(False),
ConfGlobalMFM = cms.PSet(
Volume = cms.string('OCMS'),
OCMS = cms.PSet(
Stepper = cms.string('G4TDormandPrince45'),
Type = cms.string('CMSIMField'),
StepperParam = cms.PSet(
VacRegions = cms.vstring(),
# VacRegions = cms.vstring('DefaultRegionForTheWorld','BeamPipeVacuum','BeamPipeOutside'),
EnergyThTracker = cms.double(0.2), ## in GeV
RmaxTracker = cms.double(8000), ## in mm
ZmaxTracker = cms.double(11000), ## in mm
MaximumEpsilonStep = cms.untracked.double(0.01),
DeltaOneStep = cms.double(0.001), ## in mm
DeltaOneStepTracker = cms.double(1e-4),## in mm
MaximumLoopCounts = cms.untracked.double(1000.0),
DeltaChord = cms.double(0.002), ## in mm
DeltaChordTracker = cms.double(0.001), ## in mm
MinStep = cms.double(0.1), ## in mm
DeltaIntersectionAndOneStep = cms.untracked.double(-1.0),
DeltaIntersection = cms.double(0.0001), ## in mm
DeltaIntersectionTracker = cms.double(1e-6),## in mm
MaxStep = cms.double(150.), ## in cm
MinimumEpsilonStep = cms.untracked.double(1e-05),
EnergyThSimple = cms.double(0.015), ## in GeV
DeltaChordSimple = cms.double(0.1), ## in mm
DeltaOneStepSimple = cms.double(0.1), ## in mm
DeltaIntersectionSimple = cms.double(0.01), ## in mm
MaxStepSimple = cms.double(50.), ## in cm
)
)
),
delta = cms.double(1.0)
),
)
MagneticField = _g4SimHits.MagneticField.clone()
)

from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
dd4hep.toModify(geopro, GeoFromDD4hep = True )


# load this to do a track refit
from RecoTracker.TrackProducer.TrackRefitters_cff import *
Expand Down
51 changes: 8 additions & 43 deletions TrackPropagation/Geant4e/test/simpleTestPropagator_cfg.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,61 +44,26 @@
# Then modify them
setattr(process.MessageLogger.files, category, messageLogger[main_key])


process.source = cms.Source("EmptySource")
process.maxEvents = cms.untracked.PSet(
input = cms.untracked.int32(1)
)

## Set up geometry
from SimG4Core.Application.g4SimHits_cfi import g4SimHits as _g4SimHits
process.geopro = cms.EDProducer("GeometryProducer",
GeoFromDD4hep= cms.bool(True),
# GeoFromDD4hep= cms.bool(False),
GeoFromDD4hep = cms.bool(False),
UseMagneticField = cms.bool(True),
UseSensitiveDetectors = cms.bool(False),
MagneticField = cms.PSet(
UseLocalMagFieldManager = cms.bool(False),
Verbosity = cms.bool(False),
ConfGlobalMFM = cms.PSet(
Volume = cms.string('OCMS'),
OCMS = cms.PSet(
Stepper = cms.string('G4TDormandPrince45'),
Type = cms.string('CMSIMField'),
StepperParam = cms.PSet(
VacRegions = cms.vstring(),
# VacRegions = cms.vstring('DefaultRegionForTheWorld','BeamPipeVacuum','BeamPipeOutside'),
EnergyThTracker = cms.double(0.2), ## in GeV
RmaxTracker = cms.double(8000), ## in mm
ZmaxTracker = cms.double(11000), ## in mm
MaximumEpsilonStep = cms.untracked.double(0.01),
DeltaOneStep = cms.double(0.001), ## in mm
DeltaOneStepTracker = cms.double(1e-4),## in mm
MaximumLoopCounts = cms.untracked.double(1000.0),
DeltaChord = cms.double(0.002), ## in mm
DeltaChordTracker = cms.double(0.001), ## in mm
MinStep = cms.double(0.1), ## in mm
DeltaIntersectionAndOneStep = cms.untracked.double(-1.0),
DeltaIntersection = cms.double(0.0001), ## in mm
DeltaIntersectionTracker = cms.double(1e-6),## in mm
MaxStep = cms.double(150.), ## in cm
MinimumEpsilonStep = cms.untracked.double(1e-05),
EnergyThSimple = cms.double(0.015), ## in GeV
DeltaChordSimple = cms.double(0.1), ## in mm
DeltaOneStepSimple = cms.double(0.1), ## in mm
DeltaIntersectionSimple = cms.double(0.01), ## in mm
MaxStepSimple = cms.double(50.), ## in cm
)
)
),
delta = cms.double(1.0)
),
)

MagneticField = _g4SimHits.MagneticField.clone()
)

from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
dd4hep.toModify(process.geopro, GeoFromDD4hep = True )


#####################################################################
# Extrapolator ######################################################
#
# Extrapolator ######################################################
process.propTest = cms.EDAnalyzer("SimpleGeant4ePropagatorTest",
)

Expand Down
65 changes: 18 additions & 47 deletions TrackPropagation/Geant4e/test/testPropagatorAnalyzer_cfg.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
import FWCore.ParameterSet.Config as cms

process = cms.Process("PROPAGATORTEST")

from Configuration.Eras.Era_Run3_cff import Run3

process = cms.Process("PROPAGATORTEST",Run3)



Expand All @@ -17,8 +20,8 @@
process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")

from Configuration.AlCa.GlobalTag import GlobalTag
process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run1_mc', '')
# process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase1_2022_realistic', '')
#process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run1_mc', '')
process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase1_2022_realistic', '')

process.load("FWCore.MessageService.MessageLogger_cfi")

Expand Down Expand Up @@ -46,8 +49,9 @@
# Pool Source #######################################################
#
process.source = cms.Source("PoolSource",
fileNames = cms.untracked.vstring("file:step1.root")
# '/store/relval/CMSSW_12_5_0_pre3/RelValSingleMuPt10/GEN-SIM-RECO/124X_mcRun3_2022_realistic_v8-v2/10000/6a6528c0-9d66-4358-bacc-158c40b439cf.root'
fileNames = cms.untracked.vstring(
'/store/relval/CMSSW_12_5_0_pre3/RelValSingleMuPt10/GEN-SIM-RECO/124X_mcRun3_2022_realistic_v8-v2/10000/6a6528c0-9d66-4358-bacc-158c40b439cf.root'
)
)

process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(20) )
Expand All @@ -59,53 +63,20 @@
from TrackingTools.TrackRefitter.TracksToTrajectories_cff import *

## Set up geometry
from SimG4Core.Application.g4SimHits_cfi import g4SimHits as _g4SimHits
process.geopro = cms.EDProducer("GeometryProducer",
GeoFromDD4hep = cms.bool(False),
UseMagneticField = cms.bool(True),
UseSensitiveDetectors = cms.bool(False),
MagneticField = cms.PSet(
UseLocalMagFieldManager = cms.bool(False),
Verbosity = cms.bool(False),
ConfGlobalMFM = cms.PSet(
Volume = cms.string('OCMS'),
OCMS = cms.PSet(
Stepper = cms.string('G4TDormandPrince45'),
Type = cms.string('CMSIMField'),
StepperParam = cms.PSet(
VacRegions = cms.vstring(),
# VacRegions = cms.vstring('DefaultRegionForTheWorld','BeamPipeVacuum','BeamPipeOutside'),
EnergyThTracker = cms.double(0.2), ## in GeV
RmaxTracker = cms.double(8000), ## in mm
ZmaxTracker = cms.double(11000), ## in mm
MaximumEpsilonStep = cms.untracked.double(0.01),
DeltaOneStep = cms.double(0.001), ## in mm
DeltaOneStepTracker = cms.double(1e-4),## in mm
MaximumLoopCounts = cms.untracked.double(1000.0),
DeltaChord = cms.double(0.002), ## in mm
DeltaChordTracker = cms.double(0.001), ## in mm
MinStep = cms.double(0.1), ## in mm
DeltaIntersectionAndOneStep = cms.untracked.double(-1.0),
DeltaIntersection = cms.double(0.0001), ## in mm
DeltaIntersectionTracker = cms.double(1e-6),## in mm
MaxStep = cms.double(150.), ## in cm
MinimumEpsilonStep = cms.untracked.double(1e-05),
EnergyThSimple = cms.double(0.015), ## in GeV
DeltaChordSimple = cms.double(0.1), ## in mm
DeltaOneStepSimple = cms.double(0.1), ## in mm
DeltaIntersectionSimple = cms.double(0.01), ## in mm
MaxStepSimple = cms.double(50.), ## in cm
)
)
),
delta = cms.double(1.0)
),
)


MagneticField = _g4SimHits.MagneticField.clone()
)

from Configuration.ProcessModifiers.dd4hep_cff import dd4hep
dd4hep.toModify(process.geopro, GeoFromDD4hep = True )

#####################################################################
# Extrapolator ######################################################
#
#####################################################################
# Extrapolator ######################################################
#
process.propAna = cms.EDAnalyzer("Geant4ePropagatorAnalyzer",
G4VtxSrc = cms.InputTag("g4SimHits"),
G4TrkSrc = cms.InputTag("g4SimHits"),
Expand Down

0 comments on commit c8c63c9

Please sign in to comment.