InDetRecAlgorithms

# --- refit the EXISTING tracks in ESD (only possible if readESD = True)
doRefitTracks = True and readESD

# --- setup InDetJobProperties
from InDetRecExample.InDetJobProperties import InDetFlags
InDetFlags.doRefit            = doRefitTracks
#added for refit
InDetFlags.doPixelClusterSplitting = False      # False for refit, True for rePattern

from InDetRecExample.InDetKeys import InDetKeys
if readESD and not redoPatternRecoAndTracking:
  InDetKeys.UnslimmedTracks              = 'Tracks'
  InDetKeys.UnslimmedTracksTruth         = 'TrackTruthCollection'
if doWriteAOD:
  InDetKeys.OutputAODFileName = "InDetRecAOD_new.root"

#--------------------------------------------------------------
# load master joboptions file
#--------------------------------------------------------------

include("InDetRecExample/InDetRec_all.py")

# ------------------------------------------------------------
    #
    # ----------- prepare output track collection for post processing
    #
    # ------------------------------------------------------------

    #
    # -- Pick one of the result collections and turn it into tracks
    #
    if InDetFlags.doNewTrackingPattern():
      InputTrackCollection = InDetKeys.UnslimmedTracks()
    else:
      # --- in case of reading from ESD, so we just set the Collection and truth
      InputTrackCollection      = InDetKeys.Tracks()
      InputTrackCollectionTruth = InDetKeys.TracksTruth()

# ---------------------------------------------------------------- 
    #
    # --- do we refit all tracks ?
    #
    # ---------------------------------------------------------------- 
    if InDetFlags.doRefit():
      from TrkRefitAlg.TrkRefitAlgConf import Trk__ReFitTrack
      InDetReFitTrack = Trk__ReFitTrack (name           = "InDetRefitTrack",
                                         FitterTool     = InDetTrackFitter,
                                         FitterToolTRT  = InDetTrackFitterTRT,
                                         SummaryTool    = InDetTrackSummaryToolSharedHits,
                                         AssoTool       = InDetPrdAssociationTool,
                                         TrackName      = InputTrackCollection,
                                         NewTrackName   = InDetKeys.RefittedTracks(),
                                         fitRIO_OnTrack = InDetFlags.refitROT(),
                                         useParticleHypothesisFromTrack = True)

      if InDetFlags.materialInteractions():
        InDetReFitTrack.matEffects = InDetFlags.materialInteractionsType()
      else:
        InDetReFitTrack.matEffects = 0

      topSequence += InDetReFitTrack
      if (InDetFlags.doPrintConfigurables()):
        print InDetReFitTrack
      # set input track name
      InputTrackCollection = InDetKeys.RefittedTracks()
      if InDetFlags.doTruth():
        InputDetailedTrackTruth   = InDetKeys.RefittedDetailedTracksTruth()
        InputTrackCollectionTruth = InDetKeys.RefittedTracksTruth()

# ------------------------------------------------------------
    #
    # ----------- now we do post-processing
    #
    # ------------------------------------------------------------
    #
    #
    # xAOD creation/conversion
    if (InDetFlags.doxAOD()):
      include("InDetRecExample/InDetxAODCreator.py")

    # Do post-processing algorithms (may depend on xAOD objects)
    include("InDetRecExample/InDetRecPostProcessing.py")

from InDetRecExample.InDetJobProperties import InDetFlags
from InDetRecExample.InDetKeys import InDetKeys

if not InDetFlags.doVertexFinding():
 from xAODTrackingCnv.xAODTrackingCnvConf import xAODMaker__VertexCnvAlg
 xAODVertexCnvAlg = xAODMaker__VertexCnvAlg()
 xAODVertexCnvAlg.xAODContainerName = InDetKeys.xAODVertexContainer()
 xAODVertexCnvAlg.AODContainerName = InDetKeys.PrimaryVertices()
 xAODVertexCnvAlg.TPContainerName = InDetKeys.xAODTrackParticleContainer()
 topSequence += xAODVertexCnvAlg

class xAODTrackParticleContainer(JobProperty):
    """xAOD TrackParticle"""
    statusOn = True
    allowedTypes = ['str']
    StoredValue = "InDetTrackParticles"

jobproperties.InDetContainerKeys.add_JobProperty(xAODTrackParticleContainer)

class doRefit(InDetFlagsJobProperty):
    """Turn running of refitting on and off"""
    statusOn     = True
    allowedTypes = ['bool']
    StoredValue  = False

# --------------------------------------------------------------------
      # ---- Loading of Tools (depends on settings above)
      # --------------------------------------------------------------------
      #
      # control if to load the tools
      self.loadTools  = self.loadTools() or self.doPattern() or self.doRefit() or self.postProcessing() or \
                        self.doNtupleCreation() or self.doD3PDCreation() or self.doMonitoring()

# --------------------------------------------------------------------
      # ---- Refit of tracks
      # --------------------------------------------------------------------
      #
      if (self.trackFitterType() is not 'KalmanFitter' and self.trackFitterType() is not 'KalmanDNAFitter') :
         self.refitROT = True
      if not self.refitROT() and not self.redoTRT_LR() :
         print 'ConfiguredInDetFlags.py       WARNING refitROT and redoTRT_LR are both False, NOT RECOMMENDED!'
      #
      # refKF needs a new method in IUpdator, where there is currently only one implementation
      if (self.trackFitterType() is 'ReferenceKalmanFitter'):
          self.kalmanUpdator = 'amg'
      #
      # check if a valid fitter has been used
      if not (    (self.trackFitterType() is 'KalmanFitter')
               or (self.trackFitterType() is 'KalmanDNAFitter')
               or (self.trackFitterType() is 'ReferenceKalmanFitter')
               or (self.trackFitterType() is 'DistributedKalmanFilter')
               or (self.trackFitterType() is 'GlobalChi2Fitter' )
               or (self.trackFitterType() is 'GaussianSumFilter') ):
         print 'InDetJobProperties.py       WARNING unregistered or invalid track fitter setup.'
         print '                                      --> re-setting to TrkKalmanFitter.'
         self.trackFitterType = 'KalmanFitter'

# --------------------------------------------------------------------
      # ---- Track slimming
      # --------------------------------------------------------------------
      #
      # Turn off slimming if the refit is enabled (since this maybe done to unlsim
      # the tracks) unless explicitly set by the user
      if self.doRefit(): self.checkThenSet(self.doSlimming, False)

# -----------------------------------------
    if self.doRefit() :
       print '*'
       print '* do a refit of all tracks'

_list_InDetJobProperties = [Enabled,
                            doRefit,
                           ]
for j in _list_InDetJobProperties:
    jobproperties.InDetJobProperties.add_JobProperty(j)