| [658] | 1 | // This may look like C code, but it is really -*- C++ -*-
 | 
|---|
 | 2 | //                         C.Magneville          05/99
 | 
|---|
 | 3 | #ifndef TVector_SEEN
 | 
|---|
 | 4 | #define TVector_SEEN
 | 
|---|
 | 5 | 
 | 
|---|
 | 6 | #include "tmatrix.h"
 | 
|---|
 | 7 | 
 | 
|---|
 | 8 | namespace SOPHYA {
 | 
|---|
 | 9 | 
 | 
|---|
 | 10 | class GeneralFit;
 | 
|---|
 | 11 | 
 | 
|---|
 | 12 | template <class T>
 | 
|---|
 | 13 | class TVector : public TMatrix<T> {
 | 
|---|
 | 14 | public:
 | 
|---|
 | 15 | 
 | 
|---|
 | 16 |   // Creation / destruction 
 | 
|---|
 | 17 |   TVector(uint_4 n=1);
 | 
|---|
 | 18 |   TVector(uint_4 n, T* values,Bridge* br=NULL);
 | 
|---|
 | 19 |   TVector(const TVector<T>& v);
 | 
|---|
 | 20 |   TVector(const TVector<T>& v,bool share);
 | 
|---|
 | 21 |   TVector(const TMatrix<T>& a);
 | 
|---|
 | 22 | 
 | 
|---|
 | 23 |   // Gestion taille/Remplissage
 | 
|---|
 | 24 |   inline void ReSize(uint_4 n) {TMatrix<T>::ReSize(n,1);} // Reallocation de place
 | 
|---|
 | 25 |   inline void Realloc(uint_4 n,bool force=false) {TMatrix<T>::Realloc(n,1,force);}
 | 
|---|
 | 26 | 
 | 
|---|
 | 27 |   // Informations pointeur/data
 | 
|---|
 | 28 |   inline uint_4 NElts() const {return NRows();}
 | 
|---|
 | 29 |   
 | 
|---|
 | 30 |   // Acces aux elements
 | 
|---|
 | 31 |   inline T& operator()(uint_4 n) {return (*this)[n];}
 | 
|---|
 | 32 |   inline T const& operator()(uint_4 n) const {return (*this)[n];}
 | 
|---|
 | 33 |   inline T& Element(uint_4 n) {return (*this)[n];}
 | 
|---|
 | 34 |   inline T const& Element(uint_4 n) const {return (*this)[n];}
 | 
|---|
 | 35 | 
 | 
|---|
 | 36 |   // Operateur d'affectation
 | 
|---|
 | 37 |   inline TVector& operator = (const TVector& v)
 | 
|---|
 | 38 |                   {TMatrix<T>::operator =(v); return *this;}
 | 
|---|
 | 39 |   inline TVector& operator = (T x)
 | 
|---|
 | 40 |                   {for(uint_4 i=0;i<NRows();i++) (*this)(i)=x; return *this;}
 | 
|---|
 | 41 | 
 | 
|---|
 | 42 |   // Residus et fonction fittees.
 | 
|---|
 | 43 |   TVector<T> FitResidus(GeneralFit& gfit,double xorg=0.,double dx=1.);
 | 
|---|
 | 44 |   TVector<T> FitFunction(GeneralFit& gfit,double xorg=0.,double dx=1.);
 | 
|---|
 | 45 | 
 | 
|---|
 | 46 | };
 | 
|---|
 | 47 | 
 | 
|---|
 | 48 | // produit scalaire, matrice*vecteur
 | 
|---|
 | 49 | template <class T>
 | 
|---|
 | 50 | inline T operator* (const TVector<T>& v1, const TVector<T>& v2)
 | 
|---|
 | 51 |   {if(v1.NRows() != v2.NRows())
 | 
|---|
 | 52 |      throw(SzMismatchError("TVector::operator*(TVector& v1,TVector v2) size mismatch"));
 | 
|---|
 | 53 |    T *p = const_cast<T *>(v1.Data()), *pEnd = p+v1.NElts(),
 | 
|---|
 | 54 |      *q = const_cast<T *>(v2.Data()), r = 0;
 | 
|---|
 | 55 |    while (p<pEnd) r += *p++ * *q++;
 | 
|---|
 | 56 |    return r;}
 | 
|---|
 | 57 | 
 | 
|---|
 | 58 | template <class T>
 | 
|---|
 | 59 | inline TVector<T> operator* (const TMatrix<T>& a, const TVector<T>& b)
 | 
|---|
 | 60 |    {return TVector<T>(a * ((TMatrix<T> const&)(b)));}
 | 
|---|
 | 61 | 
 | 
|---|
 | 62 | // Resolution du systeme A*C = B
 | 
|---|
 | 63 | inline r_8 LinSolveInPlace(TMatrix<r_8>& a, TVector<r_8>& b)
 | 
|---|
 | 64 | {
 | 
|---|
 | 65 | if(a.NCols() != b.NRows() || a.NCols() != a.NRows())
 | 
|---|
 | 66 |   throw(SzMismatchError("LinSolveInPlace(TMatrix<r_8>,TVector<r_8>) size mismatch"));
 | 
|---|
 | 67 | return TMatrix<r_8>::GausPiv(a,b);
 | 
|---|
 | 68 | }
 | 
|---|
 | 69 | 
 | 
|---|
 | 70 | // Resolution du systeme A*C = B, avec C retourne dans B
 | 
|---|
 | 71 | inline r_8 LinSolve(const TMatrix<r_8>& a, const TVector<r_8>& b, TVector<r_8>& c)
 | 
|---|
 | 72 | { 
 | 
|---|
 | 73 | if(a.NCols() != b.NRows() || a.NCols() != a.NRows())
 | 
|---|
 | 74 |   throw(SzMismatchError("LinSolve(TMatrix<r_8>,TVector<r_8>) size mismatch"));
 | 
|---|
 | 75 | c = b;
 | 
|---|
 | 76 | TMatrix<r_8> a1(a);
 | 
|---|
 | 77 | return TMatrix<r_8>::GausPiv(a1,c);
 | 
|---|
 | 78 | }
 | 
|---|
 | 79 | 
 | 
|---|
 | 80 | // Typedef pour simplifier et compatibilite Peida
 | 
|---|
 | 81 | typedef TVector<r_8> Vector;
 | 
|---|
 | 82 | 
 | 
|---|
 | 83 | /////////////////////////////////////////////////////////////////////////
 | 
|---|
 | 84 | // Classe pour la gestion de persistance
 | 
|---|
 | 85 | template <class T>
 | 
|---|
 | 86 | class FIO_TVector : public  PPersist  {
 | 
|---|
 | 87 | public:
 | 
|---|
 | 88 |   FIO_TVector();
 | 
|---|
 | 89 |   FIO_TVector(string const & filename); 
 | 
|---|
 | 90 |   FIO_TVector(const TVector<T> & obj);
 | 
|---|
 | 91 |   FIO_TVector(TVector<T> * obj);
 | 
|---|
 | 92 |   virtual ~FIO_TVector();
 | 
|---|
 | 93 |   virtual AnyDataObj* DataObj();
 | 
|---|
 | 94 |   inline operator TVector<T>() { return(*dobj); }
 | 
|---|
 | 95 | protected :
 | 
|---|
 | 96 |   virtual void ReadSelf(PInPersist&);           
 | 
|---|
 | 97 |   virtual void WriteSelf(POutPersist&) const;  
 | 
|---|
 | 98 |   TVector<T> * dobj;
 | 
|---|
 | 99 |   bool ownobj;
 | 
|---|
 | 100 | };
 | 
|---|
 | 101 | 
 | 
|---|
 | 102 | template <class T>
 | 
|---|
 | 103 | inline POutPersist& operator << (POutPersist& os, TVector<T> & obj)
 | 
|---|
 | 104 | { FIO_TVector<T> fio(&obj);  fio.Write(os);  return(os); }
 | 
|---|
 | 105 | template <class T>
 | 
|---|
 | 106 | inline PInPersist& operator >> (PInPersist& is, TVector<T> & obj)
 | 
|---|
 | 107 | { FIO_TVector<T> fio(&obj);  fio.Read(is);  return(is); }
 | 
|---|
 | 108 | 
 | 
|---|
 | 109 | } // Fin du namespace
 | 
|---|
 | 110 | 
 | 
|---|
 | 111 | #endif
 | 
|---|