Add options for ideal state creator to QMT test

Rich/RichFutureRecSys/tests/options/rich-dst-reco.py

@@ -15,9 +15,12 @@ from Gaudi.Configuration import *
 from GaudiConfig.ControlFlow import seq
 from GaudiKernel.SystemOfUnits import GeV
 from Configurables import LHCbApp, GaudiSequencer, DDDBConf
-from Configurables import FPEAuditor
+from Configurables import FPEAuditor, MCParticle2MCHitAlg
 from Configurables import UnpackMCParticle, UnpackMCVertex
 from Configurables import MCRichSegmentUnpacker, MCRichTrackUnpacker
+from Configurables import MCFTHitUnpacker, MCUTHitUnpacker, MCVPHitUnpacker
+from Configurables import MCRichDigitSummaryUnpacker, MCRichOpticalPhotonUnpacker
+from Configurables import MCRichHitUnpacker
 import os
 
 # --------------------------------------------------------------------------------------
@@ -186,7 +189,31 @@ else:
         'pSim/Rich/Segments', 'pSim/Rich/Tracks', 'pSim/MCVertices',
         'pSim/MCParticles'
     ]
-    preUnpackSeq.Members += [MCRichSegmentUnpacker(), MCRichTrackUnpacker()]
+    preUnpackSeq.Members += [
+        MCRichHitUnpacker(),
+        MCRichOpticalPhotonUnpacker(),
+        MCRichSegmentUnpacker(),
+        MCRichTrackUnpacker()
+    ]
+    # Needed for ideal state creator
+    unpVPHits = MCVPHitUnpacker()
+    unpFTHits = MCFTHitUnpacker()
+    unpUTHits = MCUTHitUnpacker()
+    linkVP = MCParticle2MCHitAlg(
+        "LinkVPMCHits",
+        MCHitPath=unpVPHits.OutputName,
+        OutputData="Link/MC/VP/Hits")
+    linkFF = MCParticle2MCHitAlg(
+        "LinkFTMCHits",
+        MCHitPath=unpFTHits.OutputName,
+        OutputData="Link/MC/FT/Hits")
+    linkUT = MCParticle2MCHitAlg(
+        "LinkUTMCHits",
+        MCHitPath=unpUTHits.OutputName,
+        OutputData="Link/MC/UT/Hits")
+    preUnpackSeq.Members += [
+        unpVPHits, unpFTHits, unpUTHits, linkVP, linkFF, linkUT
+    ]
     # Input tracks
     tkLocs = {"MC": "Rec/Track/FromRichMCTracks"}
     # Output PIDs
@@ -229,8 +256,6 @@ all.Members += [RichMoni]
 
 # --------------------------------------------------------------------------------------
 # Post reco Unpack sequence
-from Configurables import MCRichDigitSummaryUnpacker
-
 postUnpackSeq = GaudiSequencer("PostUnpackSeq", MeasureTime=True)
 all.Members += [postUnpackSeq]
 fetcher.DataKeys += ['pMC/Vertices', 'pMC/Particles']