MixedIntVars.h

/*  _________________________________________________________________________
 *
 *  UTILIB: A utility library for developing portable C++ codes.
 *  Copyright (c) 2001, Sandia National Laboratories.
 *  This software is distributed under the GNU Lesser General Public License.
 *  For more information, see the README file in the top UTILIB directory.
 *  _________________________________________________________________________
 */

//
// MixedIntVars.h
//
#ifndef __MixedIntVars_h
#define __MixedIntVars_h

#ifdef __GNUC__
#pragma interface
#endif

#ifdef NON_ANSI
#include <iostream.h>
#else
#include <iostream>
using namespace std;
#endif

#include "BitArray.h"
#include "IntVector.h"
#include "DoubleVector.h"

class UTILIB_API MixedIntVarsRep {
 
  friend class MixedIntVars;

protected:
 
  MixedIntVarsRep(int nBinVars=0, int nIntVars=0, int nDblVars=0) 
        : ref(1), 
        own(AssumeOwnership),
        NumBinaryVars(nBinVars), 
        NumIntegerVars(nIntVars),
        NumVars(nBinVars+nIntVars+nDblVars),
        bitVars(nBinVars), gintVars(nIntVars), doubleVars(nDblVars)
        {}

  int ref;
  EnumDataOwned own;

  int NumBinaryVars;
  int NumIntegerVars;
  int NumVars;

  BitArray bitVars;
  IntVector gintVars;
  DoubleVector doubleVars;

};



class UTILIB_API MixedIntVars 
{
public:

  MixedIntVars(const int numBits=0, const int numGenInts=0, 
                        const int numDoubles=0); 
  virtual ~MixedIntVars()
                {free();}

  void resize(const int numBits, const int numGenInts, const int numDoubles);
  int nrefs() const
        {return (a ? a->ref : 0);}

  int numIntegerVars() const
                { return (a ? a->NumIntegerVars : 0); }
  int numBinaryVars() const
                { return (a ? a->NumBinaryVars : 0); }
  int numGeneralIntVars() const
                { return (a ? a->NumIntegerVars - a->NumBinaryVars : 0); }
  int numDoubles() const
                { return (a ? a->NumVars - a->NumIntegerVars : 0); }
  int numAllVars() const
                { return (a ? a->NumVars : 0); }

  void setBit(const int ndx, int val);
  void setGeneralInt(const int ndx, int val);
  void setInt(const int ndx, int val);
  void setDouble(const int ndx, double val);
  void set(const int ndx, double val);

  int getBit(const int ndx);
  int getGeneralInt(const int ndx);
  int getInt(const int ndx);
  double getDouble(const int ndx);
  double get(const int ndx);

  MixedIntVars& operator=(const MixedIntVars& array);
  MixedIntVars& operator&=(const MixedIntVars& array);
  MixedIntVars& operator<<(const MixedIntVars& array);
  MixedIntVars& operator=(const char val);
  MixedIntVars& operator=(const int val);
  MixedIntVars& operator=(const double val);

  int operator==(const MixedIntVars& array) const;
  int compare(const MixedIntVars& array) const;

  int write(ostream& os) const;
  int read(istream& is);
  int write(PackBuffer& os) const;
  int read(UnPackBuffer& is);

protected:

  MixedIntVarsRep* a;

  void construct(const int nbits, const int nints, const int ndlbs);
  void free();

};


inline UTILIB_API PackBuffer& operator<< (PackBuffer& buff, 
                                                const MixedIntVars& vars)
        { vars.write(buff); return buff; }

inline UTILIB_API UnPackBuffer& operator>> (UnPackBuffer& buff, 
                                                MixedIntVars& vars)
        { vars.read(buff); return buff;}   

inline UTILIB_API ostream& operator<< (ostream& buff, const MixedIntVars& vars)
        { vars.write(buff); return buff; }

inline UTILIB_API istream& operator>> (istream& buff, MixedIntVars& vars)
        { vars.read(buff); return buff;}   
#endif