CouenneSparseBndVec.hpp

Go to the documentation of this file.

/* $Id: CouenneSparseBndVec.hpp 490 2011-01-14 16:07:12Z pbelotti $
 *
 * Name:    CouenneSparseBndVec.hpp
 * Author:  Pietro Belotti
 * Purpose: Keep track of tightened bounds with a sparse vector
 *
 * (C) Pietro Belotti, 2010.
 * This file is licensed under the Eclipse Public License (EPL)
 */

#ifndef COUENNESPARSEBNDVEC_HPP
#define COUENNESPARSEBNDVEC_HPP

namespace Couenne {

  template <class T> class CouenneSparseBndVec {


  private:

    unsigned int  size_; 
    unsigned int  n_;    
    unsigned int *dInd_; 
    unsigned int *sInd_; 
    T            *data_; 

  public:

    CouenneSparseBndVec (unsigned int size):

      size_ (size),
      n_    (0) {

      dInd_ = new unsigned int [size_];
      sInd_ = new unsigned int [size_];
      data_ = new T   [size_];
    }

    CouenneSparseBndVec (CouenneSparseBndVec &src):

      size_ (src.size_),
      n_    (src.n_) {

      for (register unsigned int i=0; i<n_; i++) {

        register unsigned int ind = (dInd_ [i] = src.dInd_ [i]);
        data_ [ind] = src.data_ [ind];  
        sInd_ [ind] = i; 
      }
    }

    ~CouenneSparseBndVec () {
      delete [] sInd_;
      delete [] dInd_;
      delete [] data_;
    }

    void reset () 
    {n_ = 0;}

    T &operator[] (register unsigned int index) {

      register unsigned int &sind = sInd_ [index];

      if ((sind >= n_) || 
          (dInd_ [sind] != index))
        dInd_ [sind = n_++] = index;    // this entry is new and has to be initialized  

      return data_ [sind];
    }

    T *data ()
    {return data_;}

    unsigned int *indices ()
    {return dInd_;}

    unsigned int nElements ()
    {return n_;}

    void resize (unsigned int newsize) 
    {size_ = newsize;}
  };
}

#endif

// #include <stdio.h>
// #include <stdlib.h>

// int main () {

//   Couenne::CouenneSparseBndVec <int> v (100);

//   v[84] = 10;
//   v[0]  = 60;
//   v[99] = 63;
//   v[72] = 16;
//   v[84] = 70;
//   v[25] = 33;
//   v[21] = 15;
//   v[21] = 12;
//   v[21] = 12;
//   v[8]  = 22;
//   v[4]  = 66;

//   srand48(1243235);
//   for (int i=0; i<1e9; i++)
//     v [(int)(99.999 * drand48())] = (int)(10000 * drand48());

//   for (int i=0; i< v.nElements(); i++)
//     printf ("v [%d] = %d\n", v.indices () [i], v.data () [i]);
// }