UnBinData.cxx

Go to the documentation of this file.

// @(#)root/mathcore:$Id$
// Author: L. Moneta Wed Aug 30 11:10:03 2006

/**********************************************************************
 *                                                                    *
 * Copyright (c) 2006  LCG ROOT Math Team, CERN/PH-SFT                *
 *                                                                    *
 *                                                                    *
 **********************************************************************/

// Implementation file for class UnBinData

#include "Fit/UnBinData.h"
#include "Fit/DataVector.h"
#include "Math/Error.h"

#include <cassert>
#include <cmath>

namespace ROOT {

   namespace Fit {

UnBinData::UnBinData(unsigned int maxpoints, unsigned int dim, bool isWeighted  ) :
   FitData(),
   fDim(dim),
   fPointSize( (isWeighted) ? dim +1 : dim),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor with default option and range
   unsigned int n = fPointSize*maxpoints;
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0)
      fDataVector = new DataVector(n);
}

UnBinData::UnBinData (const DataRange & range,  unsigned int maxpoints , unsigned int dim, bool isWeighted  ) :
   FitData(range),
   fDim(dim),
   fPointSize( (isWeighted) ? dim +1 : dim),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor from option and default range
   unsigned int n = fPointSize*maxpoints;
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0)
      fDataVector = new DataVector(n);
}

UnBinData::UnBinData (const DataOptions & opt, const DataRange & range,  unsigned int maxpoints, unsigned int dim, bool isWeighted) :
   FitData( opt, range),
   fDim(dim),
   fPointSize( (isWeighted) ? dim +1 : dim),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor from options and range
   unsigned int n = fPointSize*maxpoints;
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0)
      fDataVector = new DataVector(n);
}

UnBinData::UnBinData(unsigned int n, const double * dataX ) :
   FitData( ),
   fDim(1),
   fPointSize(1),
   fNPoints(n),
   fDataVector(0)
{
   // constructor for 1D external data
   fDataWrapper = new DataWrapper(dataX);
}

UnBinData::UnBinData(unsigned int n, const double * dataX, const double * dataY, bool isWeighted ) :
   FitData( ),
   fDim( (isWeighted) ? 1 : 2),
   fPointSize(2),
   fNPoints(n),
   fDataVector(0),
   fDataWrapper(0)
{
   //    constructor for 2D external data
   fDataWrapper = new DataWrapper(dataX, dataY, 0, 0, 0, 0);
}

UnBinData::UnBinData(unsigned int n, const double * dataX, const double * dataY, const double * dataZ, bool isWeighted ) :
   FitData( ),
   fDim( (isWeighted) ? 2 : 3),
   fPointSize(3),
   fNPoints(n),
   fDataVector(0)
{
   //   constructor for 3D external data
   fDataWrapper = new DataWrapper(dataX, dataY, dataZ, 0, 0, 0, 0, 0);
}

UnBinData::UnBinData(unsigned int n, const double * dataX, const DataRange & range ) :
   FitData(range),
   fDim(1),
   fPointSize(1),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor for 1D array data using a range to select the data
   // copy the data inside
   // when n = 0 construct just an empty data set
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0) {
      fDataVector = new DataVector(n);

      for (unsigned int i = 0; i < n; ++i)
         if ( range.IsInside(dataX[i]) ) Add(dataX[i] );

      if (fNPoints < n) (fDataVector->Data()).resize(fNPoints);
   }
}

UnBinData::UnBinData(unsigned int n, const double * dataX, const double * dataY, const DataRange & range, bool isWeighted ) :
   FitData(range),
   fDim( (isWeighted) ? 1 : 2 ),
   fPointSize(2),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor for 2D array data using a range to select the data
   // copy the data inside
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0) {
      fDataVector = new DataVector(2*n);

      for (unsigned int i = 0; i < n; ++i)
         if ( range.IsInside(dataX[i],0) &&
              range.IsInside(dataY[i],1) )
            Add(dataX[i], dataY[i] );

      if (fNPoints < n) (fDataVector->Data()).resize(2*fNPoints);
   }
}

UnBinData::UnBinData(unsigned int n, const double * dataX, const double * dataY, const double * dataZ,
                     const DataRange & range, bool isWeighted ) :
   FitData(range ),
   fDim( (isWeighted) ? 2 : 3 ),
   fPointSize(3),
   fNPoints(0),
   fDataVector(0),
   fDataWrapper(0)
{
   // constructor for 3D array data using a range to select the data
   if ( n > MaxSize() )
      MATH_ERROR_MSGVAL("UnBinData","Invalid data size n - no allocation done", n )
   else if (n > 0) {
      fDataVector = new DataVector(3*n);

      for (unsigned int i = 0; i < n; ++i)
         if ( range.IsInside(dataX[i],0) &&
              range.IsInside(dataY[i],1) &&
              range.IsInside(dataZ[i],2) )
            Add(dataX[i], dataY[i], dataZ[i] );

      if (fNPoints < n) (fDataVector->Data()).resize(3*fNPoints);
   }
}

void UnBinData::Initialize(unsigned int maxpoints, unsigned int dim, bool isWeighted ) {
   //   preallocate a data set given size and dimension
   unsigned int pointSize = (isWeighted) ? dim+1 : dim;
   if ( (dim != fDim || pointSize != fPointSize) && fDataVector) {
//       MATH_INFO_MSGVAL("BinData::Initialize"," Reset amd re-initialize with a new fit point size of ",
//                        dim);
      delete fDataVector;
      fDataVector = 0;
   }
   fDim = dim;
   fPointSize = pointSize;
   unsigned int n = fPointSize*maxpoints;
   if ( n > MaxSize() ) {
      MATH_ERROR_MSGVAL("UnBinData::Initialize","Invalid data size", n );
      return;
   }
   if (fDataVector)
      (fDataVector->Data()).resize( fDataVector->Size() + n );
   else
      fDataVector = new DataVector( n);
}

void UnBinData::Resize(unsigned int npoints) {
   // resize vector to new points
   if (fDim == 0) return;
   if ( npoints > MaxSize() ) {
      MATH_ERROR_MSGVAL("BinData::Resize"," Invalid data size  ", npoints );
      return;
   }
   if (fDataVector != 0)  {
      int nextraPoints = npoints -  fDataVector->Size()/fPointSize;
      if  (nextraPoints < 0) {
         // delete extra points
         (fDataVector->Data()).resize( npoints * fPointSize);
      }
      else if (nextraPoints > 0) {
         // add extra points
         Initialize(nextraPoints, fDim, IsWeighted()  );
      }
      else // nextraPoints == 0
         return;
   }
   else // no DataVector create it
      fDataVector = new DataVector( npoints*fPointSize);
}



   } // end namespace Fit

} // end namespace ROOT