WIP: New tuple tools

SUMMARY.md

@@ -1,10 +1,9 @@
 # Summary
 
 * [The LHCb Starterkit](README.md)
-  * [Contributing](CONTRIBUTING.md)
-
-----
+    * [Contributing](CONTRIBUTING.md)
 
+-----
 * [First analysis steps](first-analysis-steps/README.md)
     * [Pre-workshop checklist](first-analysis-steps/prerequisites.md)
     * [Goals of the course](first-analysis-steps/introduction-to-course.md)
@@ -18,6 +17,7 @@
     * [Running a minimal DaVinci job locally](first-analysis-steps/minimal-dv-job.md)
     * [Running DaVinci on the grid](first-analysis-steps/davinci-grid.md)
     * [TupleTools and branches](first-analysis-steps/add-tupletools.md)
+    * [TupleTools](first-analysis-steps/TupleTools.md)
     * [How do I use DecayTreeFitter?](first-analysis-steps/decay-tree-fitter.md)
     * [More Ganga](first-analysis-steps/ganga-data.md)
     * [Storing large files on EOS](first-analysis-steps/eos-storage.md)
@@ -27,31 +27,30 @@
     * [Early career, gender and diversity](first-analysis-steps/ecgd.md)
     * [Contribute to this lesson](first-analysis-steps/contributing-lesson.md)
 
-----
-
+-----
 * [Second analysis steps](second-analysis-steps/README.md)
     * [Using git to develop LHCb software](second-analysis-steps/lb-git.md)
     * [Building your own decay](second-analysis-steps/building-decays.md)
-      * [The Selection Framework](second-analysis-steps/building-decays-part0.md)
-      * [A Historical Approach](second-analysis-steps/building-decays-part1.md)
-      * [Modern Selection Framework](second-analysis-steps/building-decays-part2.md)
+        * [The Selection Framework](second-analysis-steps/building-decays-part0.md)
+        * [A Historical Approach](second-analysis-steps/building-decays-part1.md)
+        * [Modern Selection Framework](second-analysis-steps/building-decays-part2.md)
     * [What to do when something fails](second-analysis-steps/fixing-errors.md)
     * [Run a different stripping line on simulated data](second-analysis-steps/rerun-stripping.md)
     * [Replace a mass hypothesis](second-analysis-steps/switch-mass-hypo.md)
     * [Reuse particles from a decay tree](second-analysis-steps/filter-in-trees.md)
     * [The simulation framework](second-analysis-steps/simulation.md)
-      * [Introduction: The basics of Gauss](second-analysis-steps/simulation-intro.md)
-      * [Getting the correct option files](second-analysis-steps/simulation-running-gauss.md)
-      * [Controlling the decay](second-analysis-steps/simulation-dkfiles.md)
-      * [Advanced: Controlling the decay](second-analysis-steps/simulation-advanced-dkfiles.md)
-      * [Modifying generator cuts](second-analysis-steps/simulation-gencuts.md)
-      * [Fast simulation options](second-analysis-steps/simulation-fastsim.md)
+        * [Introduction: The basics of Gauss](second-analysis-steps/simulation-intro.md)
+        * [Getting the correct option files](second-analysis-steps/simulation-running-gauss.md)
+        * [Controlling the decay](second-analysis-steps/simulation-dkfiles.md)
+        * [Advanced: Controlling the decay](second-analysis-steps/simulation-advanced-dkfiles.md)
+        * [Modifying generator cuts](second-analysis-steps/simulation-gencuts.md)
+        * [Fast simulation options](second-analysis-steps/simulation-fastsim.md)
     * [HLT intro](second-analysis-steps/hlt-intro.md)
     * [TisTos DIY](second-analysis-steps/tistos-diy.md)
     * [Ganga Scripting](second-analysis-steps/ganga-scripting.md)
     * [Managing files in Ganga](second-analysis-steps/managing-files-with-ganga.md)
     * [Analysis automation: snakemake](second-analysis-steps/analysis-automation-snakemake.md)
 
-----
+-----
+* [Download PDF](ref://starterkit-lessons.pdf)
 
- * [Download PDF](ref://starterkit-lessons.pdf)

first-analysis-steps/DecayTrees/MCTupleToolAngles.cpp

@@ -0,0 +1,65 @@
+// Include files
+#include "gsl/gsl_sys.h"
+
+// from Gaudi
+#include "GaudiKernel/PhysicalConstants.h"
+#include "GaudiKernel/Vector3DTypes.h"
+
+// local
+#include "MCTupleToolAngles.h"
+
+#include "GaudiAlg/Tuple.h"
+#include "GaudiAlg/TupleObj.h"
+
+#include "Event/MCParticle.h"
+
+using namespace LHCb;
+//-----------------------------------------------------------------------------
+// Implementation file for class : MCTupleToolAngles
+//
+// 2009-01-19 : Patrick Koppenburg
+//-----------------------------------------------------------------------------
+
+// Declaration of the Tool Factory
+DECLARE_COMPONENT( MCTupleToolAngles )
+
+//=============================================================================
+// Standard constructor, initializes variables
+//=============================================================================
+  MCTupleToolAngles::MCTupleToolAngles( const std::string& type,
+                                        const std::string& name,
+                                        const IInterface* parent )
+    : TupleToolBase ( type, name , parent )
+{
+  declareInterface<IMCParticleTupleTool>(this);
+}
+
+//=============================================================================
+// Destructor
+//=============================================================================
+MCTupleToolAngles::~MCTupleToolAngles() {}
+
+//=============================================================================
+// Fill
+//=============================================================================
+StatusCode MCTupleToolAngles::fill( const LHCb::MCParticle* mother
+                                    , const LHCb::MCParticle* mcp
+                                    , const std::string& head
+                                    , Tuples::Tuple& tuple )
+{
+  const std::string prefix=fullName(head);
+
+  bool test = true;
+
+  double cosT =-999.;
+
+  if ( 0!=mcp && 0!=mother && mcp!=mother ) cosT = cosTheta(mother->momentum(), mcp->momentum() );
+  // fill the tuple:
+  test &= tuple->column( prefix+"_TRUECosTheta", cosT );
+  if(isVerbose()) test &= tuple->column( prefix+"_TRUETheta", acos(cosT) );
+  if ( msgLevel(MSG::DEBUG) && 0!=mcp && 0!=mother)
+    debug() << mother->particleID().pid() << " " << mother->momentum() << " "
+            << mcp->particleID().pid() << " " << mcp->momentum() << endmsg ;
+
+  return StatusCode(test);
+}

first-analysis-steps/DecayTrees/MCTupleToolAngles.h

@@ -0,0 +1,60 @@
+// $Id: MCTupleToolAngles.h,v 1.3 2010-01-26 15:39:25 rlambert Exp $
+#ifndef MCTUPLETOOLANGLES_H
+#define MCTUPLETOOLANGLES_H 1
+
+// Include files
+// from Gaudi
+#include "DecayTreeTupleBase/TupleToolBase.h"
+#include "Kernel/IMCParticleTupleTool.h"            // Interface
+#include "Math/Boost.h"
+#include "GaudiKernel/Vector4DTypes.h"
+
+/** @class MCTupleToolAngles MCTupleToolAngles.h
+ *
+ *  Fill MC Particle with decay angle in mother frame
+ *
+ * - head_TRUECosTheta : angle in mother's frame
+ *
+ * \sa TupleToolAngles, DecayTreeTuple, MCDecayTreeTuple
+ *
+ *  @author Patrick Koppenburg
+ *  @date   2009-01-19
+ */
+
+class MCTupleToolAngles : public TupleToolBase,
+                          virtual public IMCParticleTupleTool
+{
+
+public:
+
+  /// Standard constructor
+  MCTupleToolAngles( const std::string& type,
+                     const std::string& name,
+                     const IInterface* parent);
+
+  virtual ~MCTupleToolAngles( ); ///< Destructor
+
+  StatusCode fill( const LHCb::MCParticle*
+                   , const LHCb::MCParticle*
+                   , const std::string&
+                   , Tuples::Tuple& ) override;
+
+private:
+
+  inline double cosTheta( const Gaudi::LorentzVector& mother,
+                          const Gaudi::LorentzVector& mcp )
+  {
+    ROOT::Math::Boost boost( mother.BoostToCM() );
+    const Gaudi::XYZVector boostedParticle = (boost( mcp )).Vect().unit();
+    const Gaudi::XYZVector boostedMother = mother.Vect().unit();
+    double cosT = boostedParticle.Dot(boostedMother) ;
+    if (msgLevel(MSG::VERBOSE)) verbose() <<  mother << " "
+                                          <<  mcp << " " << boostedMother
+                                          << " " << boostedParticle
+                                          << " " << cosT << endmsg ;
+    return cosT;
+  }
+
+};
+
+#endif // MCTUPLETOOLANGLES_H

first-analysis-steps/DecayTrees/MCTupleToolDalitz.cpp

@@ -0,0 +1,68 @@
+// Include files
+
+#include <set>
+#include <string>
+
+// local
+#include "MCTupleToolDalitz.h"
+
+//-----------------------------------------------------------------------------
+// Implementation file for class : MCTupleToolDalitz
+//
+// 2009-02-17 : Patrick Koppenburg
+//-----------------------------------------------------------------------------
+
+// Declaration of the Tool Factory
+DECLARE_COMPONENT( MCTupleToolDalitz )
+
+//=============================================================================
+// Standard constructor, initializes variables
+//=============================================================================
+  MCTupleToolDalitz::MCTupleToolDalitz( const std::string& type,
+                                        const std::string& name,
+                                        const IInterface* parent )
+    : TupleToolBase ( type, name , parent )
+    , m_ppSvc(0)
+{
+  declareInterface<IMCParticleTupleTool>(this);
+
+}
+//=============================================================================
+// Destructor
+//=============================================================================
+MCTupleToolDalitz::~MCTupleToolDalitz() {}
+
+//=============================================================================
+//=============================================================================
+// Fill
+//=============================================================================
+StatusCode MCTupleToolDalitz::fill( const LHCb::MCParticle* mother
+                                    , const LHCb::MCParticle* part
+                                    , const std::string& head
+                                    , Tuples::Tuple& tuple )
+{
+  const std::string prefix=fullName(head);
+
+  if (0==part) return StatusCode::FAILURE ;
+  if (msgLevel(MSG::DEBUG)) debug() << "Decay of " << part->particleID().pid() << " with mother " << mother << endmsg ;
+  LHCb::MCParticle::ConstVector dauts ;
+  for ( SmartRefVector<LHCb::MCVertex>::const_iterator iv = part->endVertices().begin() ;
+        iv != part->endVertices().end() ; ++iv){
+    if (!(*iv)->isDecay()) continue ;
+    if (msgLevel(MSG::VERBOSE)) verbose() << "Decay vertex of type " << (*iv)->type() << " with "
+                                          << (*iv)->products().size() << " products" << endmsg ;
+    const SmartRefVector< LHCb::MCParticle > pr = (*iv)->products() ;
+    for ( SmartRefVector< LHCb::MCParticle >::const_iterator ip = pr.begin() ; ip != pr.end() ; ++ip){
+      if (msgLevel(MSG::VERBOSE)) verbose() << "Pushing back " << (*ip)->particleID().pid() << endmsg ;
+      dauts.push_back(*ip);
+    }
+    break ;
+  }
+
+  if ( 2>=dauts.size() ){
+    return Warning("Will not fill Dalitz of two body decay "+prefix,StatusCode::SUCCESS,0);
+  }
+
+  return fill(dauts,"MC",tuple,(part->particleID().pid()<0)) ;
+
+}

first-analysis-steps/DecayTrees/MCTupleToolDalitz.h

@@ -0,0 +1,51 @@
+// $Id: MCTupleToolDalitz.h,v 1.3 2010-01-26 15:39:25 rlambert Exp $
+#ifndef MCTUPLETOOLDALITZ_H
+#define MCTUPLETOOLDALITZ_H 1
+
+// Include files
+// from Gaudi
+#include "Kernel/IMCParticleTupleTool.h"
+#include "DecayTreeTupleBase/TupleToolBase.h"
+#include "Kernel/Escape.h"
+#include "Event/MCParticle.h"
+#include "GaudiAlg/Tuple.h"
+#include "GaudiAlg/TupleObj.h"
+#include "Kernel/IParticlePropertySvc.h"
+#include "Kernel/ParticleProperty.h"
+
+// Boost
+#include <boost/lexical_cast.hpp>
+
+/** @class MCTupleToolDalitz MCTupleToolDalitz.h
+ *
+ *  Fills square masses of all combinations of direct daughters
+ *
+ *  @author Patrick Koppenburg
+ *  @date   2009-02-17
+ */
+class MCTupleToolDalitz : public TupleToolBase,
+                          virtual public IMCParticleTupleTool
+{
+
+public:
+
+  /// Standard constructor
+  MCTupleToolDalitz( const std::string& type,
+                     const std::string& name,
+                     const IInterface* parent);
+
+  virtual ~MCTupleToolDalitz( ); ///< Destructor
+
+  /// The filling method
+  StatusCode fill( const LHCb::MCParticle*
+                   , const LHCb::MCParticle*
+                   , const std::string&
+                   , Tuples::Tuple& ) override;
+
+private:
+
+  // Horrid way of sharing code. Should use inheritance instead ...
+#include "TupleToolDalitz.icpp"
+
+};
+#endif // MCTUPLETOOLDALITZ_H