NeutrinoFitObject.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * Forked from https://github.com/iLCSoft/MarlinKinfit                    *
 *                                                                        *
 * Further information about the fit engine and the user interface        *
 * provided in MarlinKinfit can be found at                               *
 * https://www.desy.de/~blist/kinfit/doc/html/                            *
 * and in the LCNotes LC-TOOL-2009-001 and LC-TOOL-2009-004 available     *
 * from http://www-flc.desy.de/lcnotes/                                   *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include "analysis/OrcaKinFit/NeutrinoFitObject.h"
#include <framework/logging/Logger.h>
#include <cmath>
 
#undef NDEBUG
#include <cassert>
 
#include <algorithm>
 
using std::sqrt;
using std::sin;
using std::cos;
 
namespace Belle2 {
  namespace OrcaKinFit {
 
// constructor
    NeutrinoFitObject::NeutrinoFitObject(double E, double theta, double phi,
                                         double DE, double Dtheta, double Dphi)
      : ctheta(0), stheta(0), cphi(0), sphi(0), pt(0), px(0), py(0), pz(0), dptdE(0),
        dpxdE(0), dpydE(0), dpxdtheta(0), dpydtheta(0), chi2(0)
    {
 
      assert(int(NPAR) <= int(BaseDefs::MAXPAR));
 
      setMass(0);
      setParam(0, E, false);
      setParam(1, theta, false);
      setParam(2, phi, false);
      setError(0, DE);
      setError(1, Dtheta);
      setError(2, Dphi);
      invalidateCache();
    }
 
// destructor
    NeutrinoFitObject::~NeutrinoFitObject() = default;
 
    NeutrinoFitObject::NeutrinoFitObject(const NeutrinoFitObject& rhs)
      : ParticleFitObject(rhs), ctheta(0), stheta(0), cphi(0), sphi(0), pt(0), px(0), py(0), pz(0), dptdE(0),
        dpxdE(0), dpydE(0), dpxdtheta(0), dpydtheta(0), chi2(0)
    {
      NeutrinoFitObject::assign(rhs);
    }
 
    NeutrinoFitObject& NeutrinoFitObject::operator= (const NeutrinoFitObject& rhs)
    {
      if (this != &rhs) {
        assign(rhs);
      }
      return *this;
    }
 
    NeutrinoFitObject* NeutrinoFitObject::copy() const
    {
      return new NeutrinoFitObject(*this);
    }
 
    NeutrinoFitObject& NeutrinoFitObject::assign(const BaseFitObject& source)
    {
      if (const auto* psource = dynamic_cast<const NeutrinoFitObject*>(&source)) {
        if (psource != this) {
          ParticleFitObject::assign(source);
          // only mutable data members, need not to be copied, if cache is invalid
        }
      } else {
        assert(0);
      }
      return *this;
    }
 
    const char* NeutrinoFitObject::getParamName(int ilocal) const
    {
      switch (ilocal) {
        case 0: return "E";
        case 1: return "theta";
        case 2: return "phi";
      }
      return "undefined";
    }
 
 
    bool NeutrinoFitObject::updateParams(double pp[], int idim)
    {
 
      invalidateCache();
 
      int iE  = getGlobalParNum(0);
      int ith = getGlobalParNum(1);
      int iph = getGlobalParNum(2);
      assert(iE  >= 0 && iE  < idim);
      assert(ith >= 0 && ith < idim);
      assert(iph >= 0 && iph < idim);
 
      double e  = pp[iE];
      double th = pp[ith];
      double ph = pp[iph];
      if (e < 0) {
        e  = -e;
        th = M_PI - th;
        ph = M_PI + ph;
      }
      if (th < 0 || th > M_PI) {
        th = M_PI - th;
        ph = M_PI + ph;
      }
 
      bool result = (e - par[0]) * (e - par[0]) > eps2 * cov[0][0] ||
                    (th - par[1]) * (th - par[1]) > eps2 * cov[1][1] ||
                    (ph - par[2]) * (ph - par[2]) > eps2 * cov[2][2];
      par[0] = e;
      par[1] = (th >= 0 && th < M_PI) ?
               th : std::acos(std::cos(th));
      if (std::abs(ph) > M_PI) ph = atan2(sin(ph), cos(ph));
      par[2] = ph;
      pp[iE]  = par[0];
      pp[ith] = par[1];
      pp[iph] = par[2];
      return result;
    }
 
// these depend on actual parametrisation!
    double NeutrinoFitObject::getDPx(int ilocal) const
    {
      assert(ilocal >= 0 && ilocal < NPAR);
      if (!cachevalid) updateCache();
      switch (ilocal) {
        case 0: return dpxdE;
        case 1: return dpxdtheta;
        case 2: return -py;
      }
      return 0;
    }
 
    double NeutrinoFitObject::getDPy(int ilocal) const
    {
      assert(ilocal >= 0 && ilocal < NPAR);
      if (!cachevalid) updateCache();
      switch (ilocal) {
        case 0: return dpydE;
        case 1: return dpydtheta;
        case 2: return px;
      }
      return 0;
    }
 
    double NeutrinoFitObject::getDPz(int ilocal) const
    {
      assert(ilocal >= 0 && ilocal < NPAR);
      if (!cachevalid) updateCache();
      switch (ilocal) {
        case 0: return ctheta;
        case 1: return -pt;
        case 2: return 0;
      }
      return 0;
    }
 
    double NeutrinoFitObject::getDE(int ilocal) const
    {
      assert(ilocal >= 0 && ilocal < NPAR);
      switch (ilocal) {
        case 0: return 1;
        case 1: return 0;
        case 2: return 0;
      }
      return 0;
    }
 
    double NeutrinoFitObject::getFirstDerivative_Meta_Local(int iMeta, int ilocal, int metaSet) const
    {
      // iMeta = intermediate variable (i.e. E,px,py,pz)
      // ilocal = local variable (E, theta, phi)
      // metaSet = which set of intermediate varlables
 
      assert(metaSet == 0); // only defined for E,px,py,pz
 
      switch (iMeta) {
        case 0: // E
          return getDE(ilocal);
          break;
        case 1: // Px
          return getDPx(ilocal);
          break;
        case 2: // Py
          return getDPy(ilocal);
          break;
        case 3: // Pz
          return getDPz(ilocal);
          break;
        default:
          assert(0);
      }
      return -999;
    }
 
 
    double NeutrinoFitObject::getSecondDerivative_Meta_Local(int iMeta, int ilocal, int jlocal, int metaSet) const
    {
      assert(metaSet == 0);
      if (!cachevalid) updateCache();
 
      if (jlocal < ilocal) {
        int temp = jlocal;
        jlocal = ilocal;
        ilocal = temp;
      }
 
      // daniel hasn't checked these, copied from orig code
      switch (iMeta) {
 
        case 0:
          return 0;
          break;
        case 1:
          if (ilocal == 0 && jlocal == 1) return ctheta * cphi;
          else if (ilocal == 0 && jlocal == 2) return -dpydE;
          else if (ilocal == 1 && jlocal == 1) return -px;
          else if (ilocal == 1 && jlocal == 2) return -dpydtheta;
          else if (ilocal == 2 && jlocal == 2) return -px;
          else return 0;
          break;
        case 2:
          if (ilocal == 0 && jlocal == 1) return ctheta * sphi;
          else if (ilocal == 0 && jlocal == 2) return dpxdE;
          else if (ilocal == 1 && jlocal == 1) return -py;
          else if (ilocal == 1 && jlocal == 2) return dpxdtheta;
          else if (ilocal == 2 && jlocal == 2) return -py;
          else return 0;
          break;
        case 3:
          if (ilocal == 0 && jlocal == 1) return -stheta;
          else if (ilocal == 1 && jlocal == 1) return -pz;
          else return 0;
          break;
        default:
          assert(0);
      }
      return -999;
    }
 
 
    void NeutrinoFitObject::updateCache() const
    {
      double e     = par[0];
      double theta = par[1];
      double phi   = par[2];
 
      ctheta = cos(theta);
      stheta = sin(theta);
      cphi   = cos(phi);
      sphi   = sin(phi);
 
      pt = e * stheta;
 
      px = pt * cphi;
      py = pt * sphi;
      pz = e * ctheta;
 
      fourMomentum.setValues(e, px, py, pz);
 
      dpxdE = stheta * cphi;
      dpydE = stheta * sphi;
      dpxdtheta = pz * cphi;
      dpydtheta = pz * sphi;
 
      cachevalid = true;
    }
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace