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