ResidualsControllerFast.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "Align/ResidualsControllerFast.h"
#include "Align/AlignmentTrack.h"
#include "Align/AlignmentPoint.h"
#include "Align/AlignableSensor.h"
#include "Framework/Logger.h"
#include <TString.h>
#include <TMath.h>
#include <cstdio>
 
using namespace TMath;
 
ClassImp(o2::align::ResidualsControllerFast);
 
namespace o2
{
namespace align
{
 
//____________________________________
ResidualsControllerFast::ResidualsControllerFast()
  : mNPoints(0), mNMatSol(0), mNBook(0), mChi2(0), mChi2Ini(0), mD0(nullptr), mD1(nullptr), mSig0(nullptr), mSig1(nullptr), mVolID(nullptr), mLabel(nullptr), mSolMat(nullptr), mMatErr(nullptr)
{
  // def c-tor
  for (int i = 0; i < 5; ++i) {
    mTrCorr[i] = 0;
  }
}
 
//________________________________________________
ResidualsControllerFast::~ResidualsControllerFast()
{
  // d-tor
  delete[] mD0;
  delete[] mD1;
  delete[] mSig0;
  delete[] mSig1;
  delete[] mVolID;
  delete[] mLabel;
  delete[] mSolMat;
  delete[] mMatErr;
}
 
//________________________________________________
void ResidualsControllerFast::resize(int np)
{
  // resize container
  if (np > mNBook) {
    delete[] mD0;
    delete[] mD1;
    delete[] mSig0;
    delete[] mSig1;
    delete[] mVolID;
    delete[] mLabel;
    delete[] mSolMat;
    delete[] mMatErr;
    //
    mNBook = 30 + np;
    mD0 = new float[mNBook];
    mD1 = new float[mNBook];
    mSig0 = new float[mNBook];
    mSig1 = new float[mNBook];
    mVolID = new int[mNBook];
    mLabel = new int[mNBook];
    mSolMat = new float[mNBook * 4]; // at most 4 material params per point
    mMatErr = new float[mNBook * 4]; // at most 4 material params per point
    //
    memset(mD0, 0, mNBook * sizeof(float));
    memset(mD1, 0, mNBook * sizeof(float));
    memset(mSig0, 0, mNBook * sizeof(float));
    memset(mSig1, 0, mNBook * sizeof(float));
    memset(mVolID, 0, mNBook * sizeof(int));
    memset(mLabel, 0, mNBook * sizeof(int));
    memset(mSolMat, 0, 4 * mNBook * sizeof(int));
    memset(mMatErr, 0, 4 * mNBook * sizeof(int));
  }
  //
}
 
//____________________________________________
void ResidualsControllerFast::Clear(const Option_t*)
{
  // reset record
  mNPoints = 0;
  mNMatSol = 0;
  mTrCorr[4] = 0; // rest will be 100% overwritten
  //
}
 
//____________________________________________
void ResidualsControllerFast::Print(const Option_t* /*opt*/) const
{
  // print info
  printf("%3s:%1s (%9s/%5s) %6s | [ %7s:%7s ]\n", "Pnt", "M", "Label",
         "VolID", "Sigma", "resid", "pull/LG");
  for (int irs = 0; irs < mNPoints; irs++) {
    printf("%3d:%1d (%9d/%5d) %6.4f | [%+.2e:%+7.2f]\n",
           irs, 0, mLabel[irs], mVolID[irs], mSig0[irs], mD0[irs],
           mSig0[irs] > 0 ? mD0[irs] / mSig0[irs] : -99);
    printf("%3d:%1d (%9d/%5d) %6.4f | [%+.2e:%+7.2f]\n",
           irs, 1, mLabel[irs], mVolID[irs], mSig1[irs], mD1[irs],
           mSig1[irs] > 0 ? mD1[irs] / mSig1[irs] : -99);
  }
  //
  printf("CorrETP: ");
  for (int i = 0; i < 5; i++) {
    printf("%+.3f ", mTrCorr[i]);
  }
  printf("\n");
  printf("MatCorr (corr/sig:pull)\n");
  int nmp = mNMatSol / 4;
  int cnt = 0;
  for (int imp = 0; imp < nmp; imp++) {
    for (int ic = 0; ic < 4; ic++) {
      printf("%+.2e/%.2e:%+8.3f|", mSolMat[cnt], mMatErr[cnt],
             mMatErr[cnt] > 0 ? mSolMat[cnt] / mMatErr[cnt] : -99);
      cnt++;
    }
    printf("\n");
  }
  //
}
 
//____________________________________________
void ResidualsControllerFast::setResSigMeas(int ip, int ord, float res, float sig)
{
  // assign residual and error for measurement
  if (ord == 0) {
    mD0[ip] = res;
    mSig0[ip] = sig;
  } else {
    mD1[ip] = res;
    mSig1[ip] = sig;
  }
}
 
//____________________________________________
void ResidualsControllerFast::setMatCorr(int id, float res, float sig)
{
  // assign residual and error for material correction
  mSolMat[id] = res;
  mMatErr[id] = sig;
}
 
//____________________________________________
void ResidualsControllerFast::setLabel(int ip, int lab, int vol)
{
  // set label/volid of measured volume
  mVolID[ip] = vol;
  mLabel[ip] = lab;
}
 
} // namespace align
} // namespace o2