Pspm.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
//
// $Id$
//
// Copyright 2008, 2009, 2010, 2011, 2012  Antonio Franchi and Paolo Stegagno
//
// This file is part of MIP.
//
// MIP is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// MIP is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with MIP. If not, see <http://www.gnu.org/licenses/>.
//
// Contact info: antonio.franchi@tuebingen.mpg.de stegagno@diag.uniroma1.it
//
// ----------------------------------------------------------------------------




/* @{ */

#ifndef __PSPM_ALGO_H__
#define __PSPM_ALGO_H__

#ifdef MIP_HOST_APPLE
#include <applePatch.h>
#endif

#include <baselib.h>

#include <boost/numeric/ublas/vector.hpp>
#include <boost/numeric/ublas/matrix.hpp>
#include <boost/numeric/ublas/io.hpp>

#include <ublasTools.h>


typedef boost::numeric::ublas::vector<Decimal> uvector;
typedef boost::numeric::ublas::matrix<Decimal> umatrix;

using namespace MipBaselib;

class PspmOptions : public Options {
 public:
  DecimalOption *diagStiff;
  DecimalOption *diagDamp;
  DecimalOption *initEnergy;
  DecimalOption *maxEnergy;

  PspmOptions();

  string getObjectName() const {
   return "PspmOptions";
  }
};


class PspmPar : public MIPObject {
 private:
  Decimal _maxEnergy, _initEnergy; //max and initial energy level
  umatrix _dampMat, _stiffMat;

  stringstream _ssfatal;

 public:
  PspmPar(Decimal maxEnrg=0, Decimal initEnergy=0, umatrix B=boost::numeric::ublas::identity_matrix<Decimal>(3), umatrix K=boost::numeric::ublas::identity_matrix<Decimal>(3));

  PspmPar(const PspmPar& p);

  PspmPar& operator=(const PspmPar& rhs);

  Decimal getMaxEnergy();

  Decimal getInitEnergy();

  umatrix getDampMat();

  umatrix getStiffMat();

  void setMaxEnergy(Decimal maxEnrg);

  void setInitEnergy(Decimal maxEnrg);

  void setDampMat(umatrix &dampMat);

  void setStiffMat(umatrix &stiffMat);

  string getObjectName() const {
   return "PspmPar";
  }
};


class Pspm : public MIPObject{
 private:
  PspmOptions _op;

  PspmPar *_par;

  stringstream _ssfatal;

  Decimal _currEnrg; 
  Decimal _currDampEnrg; 
  Decimal _pastDampEnrg; 

  Decimal _oldTime;

  uvector _oldY, _oldHatY, _xMax, _xMin;

  int _dim;

  bool _firstStep;

  void _numIntDamp(uvector xDot, Decimal dT, bool reset=false);

  void _commonConstr();
  
  FILE* _logFile;

 public:

   Pspm();

  Pspm(PspmPar &par);

  Pspm(int argc, const char* argv[]);

  ~Pspm();

  Pspm(const Pspm& t);

  Pspm& operator=(const Pspm& t);

  void step(vector<Decimal> y_, vector<Decimal> x_, vector<Decimal> xDot, Decimal inEnergy, Decimal& outEnergy, Decimal time, Decimal dT,
       vector<Decimal>& out, bool newData=false, bool approxDamping=true, bool reset=false);

   Decimal getEnergy();

//   Decimal getDampingEnergy();

//   void setEnergy(Decimal energy);

//  /// \brief
//   Decimal getOldTime();

   void setLogFile(FILE* logFile);

  string getObjectName() const {
   return "Pspm";
  }
};


#endif


/* @} */