IpDiagMatrix.hpp

Go to the documentation of this file.

// Copyright (C) 2004, 2006 International Business Machines and others.
// All Rights Reserved.
// This code is published under the Common Public License.
//
// $Id: IpDiagMatrix.hpp 759 2006-07-07 03:07:08Z andreasw $
//
// Authors:  Carl Laird, Andreas Waechter     IBM    2004-08-13

#ifndef __IPDIAGMATRIX_HPP__
#define __IPDIAGMATRIX_HPP__

#include "IpUtils.hpp"
#include "IpSymMatrix.hpp"

namespace Ipopt
{

  class DiagMatrix : public SymMatrix
  {
  public:


    DiagMatrix(const SymMatrixSpace* owner_space);

    ~DiagMatrix();

    void SetDiag(const Vector& diag)
    {
      diag_ = &diag;
    }

    SmartPtr<const Vector> GetDiag() const
    {
      return diag_;
    }

  protected:
    virtual void MultVectorImpl(Number alpha, const Vector& x,
                                Number beta, Vector& y) const;

    virtual bool HasValidNumbersImpl() const;

    virtual void PrintImpl(const Journalist& jnlst,
                           EJournalLevel level,
                           EJournalCategory category,
                           const std::string& name,
                           Index indent,
                           const std::string& prefix) const;

  private:
    DiagMatrix();

    DiagMatrix(const DiagMatrix&);

    void operator=(const DiagMatrix&);

    SmartPtr<const Vector> diag_;
  };

  class DiagMatrixSpace : public SymMatrixSpace
  {
  public:
    DiagMatrixSpace(Index dim)
        :
        SymMatrixSpace(dim)
    {}

    virtual ~DiagMatrixSpace()
    {}

    virtual SymMatrix* MakeNewSymMatrix() const
    {
      return MakeNewDiagMatrix();
    }

    DiagMatrix* MakeNewDiagMatrix() const
    {
      return new DiagMatrix(this);
    }

  private:
    DiagMatrixSpace();

    DiagMatrixSpace(const DiagMatrixSpace&);

    void operator=(const DiagMatrixSpace&);

  };

} // namespace Ipopt
#endif