Fit_Result_Vec.cc

Go to the documentation of this file.

//
//
// File: src/Fit_Result_Vec.cc
// Purpose: Hold a set of Fit_Result structures.
// Created: Jul, 2000, sss, based on run 1 mass analysis code.
//
// CMSSW File      : src/Fit_Result_Vec.cc
// Original Author : Scott Stuart Snyder <snyder@bnl.gov> for D0
// Imported to CMSSW by Haryo Sumowidagdo <Suharyo.Sumowidagdo@cern.ch>
//

#include "TopQuarkAnalysis/TopHitFit/interface/Fit_Result_Vec.h"
#include "TopQuarkAnalysis/TopHitFit/interface/Fit_Result.h"
#include <cassert>
#include <ostream>
#include <algorithm>


using std::ostream;
using std::vector;
using std::lower_bound;


namespace hitfit {


Fit_Result_Vec::Fit_Result_Vec (std::vector<Fit_Result*>::size_type max_len)
//
// Purpose Constructor.
//
// Inputs:
//   max_len -     The maximum length of the vector.
//
  : _max_len (max_len)
{
  assert (max_len > 0);
  _v.reserve (max_len + 1);
}


Fit_Result_Vec::Fit_Result_Vec (const Fit_Result_Vec& vec)
//
// Purpose: Copy constructor.
//
// Inputs:
//   vec -         The vector to copy.
//
  : _v (vec._v),
    _max_len (vec._max_len)
{
  // Gotta increase the reference count on the contents.
  for (std::vector<Fit_Result*>::size_type i=0; i < _v.size(); i++)
    _v[i]->incref ();
}


Fit_Result_Vec::~Fit_Result_Vec ()
//
// Purpose: Destructor.
//
{
  for (std::vector<Fit_Result*>::size_type i=0; i < _v.size(); i++)
    _v[i]->decref ();
}


Fit_Result_Vec& Fit_Result_Vec::operator= (const Fit_Result_Vec& vec)
//
// Purpose: Assignment.
//
// Inputs:
//   vec -         The vector to copy.
//
// Returns:
//   This object.
//
{
  for (std::vector<Fit_Result*>::size_type i=0; i < _v.size(); i++)
    _v[i]->decref ();
  _v = vec._v;
  _max_len = vec._max_len;
  for (std::vector<Fit_Result*>::size_type i=0; i < _v.size(); i++)
    _v[i]->incref ();
  return *this;
}


std::vector<Fit_Result*>::size_type Fit_Result_Vec::size () const
//
// Purpose: Get back the number of results in the vector.
//
// Returns:
//   The number of results in the vector.
//
{
  return _v.size ();
}


const Fit_Result& Fit_Result_Vec::operator[] (std::vector<Fit_Result*>::size_type i) const
//
// Purpose: Get back the Ith result in the vector.
//
// Inputs:
//   i -           The index of the desired result.
//
// Returns:
//   The Ith result.
//
{
  assert (i < _v.size());
  return *_v[i];
}


namespace {


struct Compare_Fitresptr
//
// Purpose: Helper for push().
//
{
  bool operator() (const Fit_Result* a, const Fit_Result* b) const
  {
    return *a < *b;
  }
};


} // unnamed namespace


void Fit_Result_Vec::push (Fit_Result* res)
//
// Purpose: Add a new result to the vector.
//
// Inputs:
//   res -         The result to add.
//
{
  // Find where to add it.
  vector<Fit_Result*>::iterator it = lower_bound (_v.begin(),
                                                  _v.end(),
                                                  res,
                                                  Compare_Fitresptr());

  // Insert it.
  _v.insert (it, res);
  res->incref ();

  // Knock off the guy at the end if we've exceeded our maximum size.
  if (_v.size() > _max_len) {
    _v.back()->decref ();
    _v.erase (_v.end()-1);
  }
}


std::ostream& operator<< (std::ostream& s, const Fit_Result_Vec& resvec)
//
// Purpose: Print the object to S.
//
// Inputs:
//   s -           The stream to which to write.
//   resvec -      The object to write.
//
// Returns:
//   The stream S.
//
{
  for (std::vector<Fit_Result*>::size_type i=0; i < resvec._v.size(); i++)
    s << "Entry " << i << "\n" << *resvec._v[i];
  return s;
}


} // namespace hitfit