| 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 | 
|---|