source: Sophya/trunk/SophyaLib/BaseTools/ndatablock.cc@ 2615

Last change on this file since 2615 was 2615, checked in by cmv, 21 years ago

using namespace sophya enleve de machdefs.h, nouveau sopnamsp.h cmv 10/09/2004

File size: 26.2 KB
Line 
1// Gestion de block de donnees avec partage de references
2// malheureusement tres mal concu... C.Magneville 04/99
3// LAL (Orsay) / IN2P3-CNRS DAPNIA/SPP (Saclay) / CEA
4#include "sopnamsp.h"
5#include "machdefs.h"
6#include <stdio.h>
7#include <stdlib.h>
8#include <iostream>
9#include <complex>
10#include "pexceptions.h"
11#include "ndatablock.h"
12
13/*!
14 \class SOPHYA::NDataBlock
15 \ingroup BaseTools
16 Management of data blocks
17*/
18
19//////////////////////////////////
20// Fonctionnement en mode debug //
21//////////////////////////////////
22
23template <class T> int NDataBlock<T>::Debug_NDataBlock = 0;
24template <class T> size_t NDataBlock<T>::NallocData = 0;
25template <class T> size_t NDataBlock<T>::NallocSRef = 0;
26
27//! Set debug (and level print) for allocation and references debug.
28/*!
29 \param prtlevel : activate/des-activate debug mode
30 and select print level
31
32 \arg prtlevel <= 0 : no debug
33 \arg prtlevel == 1 : debug activated, no print
34 \arg prtlevel >=2 : debug activated,
35 print infos in all routines that have something to do with
36 allocations or des-allocation of datas or references.
37 */
38template <class T>
39void NDataBlock<T>::SetPrintDebug(int prtdbglevel)
40{
41 Debug_NDataBlock = prtdbglevel;
42}
43
44//! Reset debug counter values.
45/*!
46 \param nallocdata : reset number of allocated data structures to \b nallocdata
47 \param nallocsref : reset number of allocated references to \b nallocsref
48 \warning In principle this routine should not be use (only experts)
49 */
50template <class T>
51void NDataBlock<T>::ResetDebug(size_t nallocdata, size_t nallocsref)
52{
53NallocData = nallocdata;
54NallocSRef = nallocsref;
55}
56
57//! Print debug current status.
58/*!
59 Print debug current status for number of allocated
60 data structures and number of allocated references.
61 */
62template <class T>
63void NDataBlock<T>::PrintDebug()
64{
65cout<<"... ... ... NallocData = "<<NallocData
66 <<" , NallocSRef = "<<NallocSRef
67 <<" ... ... ..."<<endl;
68}
69
70///////////////////////////
71// Createur, Destructeur //
72///////////////////////////
73
74//! Constructor for \b n datas. if \b zero=true, filled with zeros
75template <class T>
76NDataBlock<T>::NDataBlock(size_t n, bool fzero)
77// Createur d'une structure de "n" donnees
78: mSz(0), mSRef(NULL), mIsTemp(false)
79{
80if(Debug_NDataBlock>1)
81 cout<<"?_NDataBlock::NDataBlock("<<this<<",n="<<n<<")"<<endl;
82
83Alloc(n, NULL, NULL, fzero); // allocation et mise a zero
84}
85
86//! Constructor for \b n datas shared with external
87/*!
88 Datas are previously allocated by an other external source.
89 \warning This require particular care (see Alloc)
90 \sa Alloc
91 */
92template <class T>
93NDataBlock<T>::NDataBlock(size_t n, T* data, Bridge* br)
94// Createur d'une structure de "n" donnees, avec donnees preallouees.
95// Attention createur TRES DANGEREUX (Voir explications dans Alloc()).
96: mSz(0), mSRef(NULL), mIsTemp(false)
97{
98if(Debug_NDataBlock>1)
99 cout<<"?_NDataBlock::NDataBlock("<<this
100 <<",data="<<data<<",br="<<br<<")"<<endl;
101
102Alloc(n,data,br);
103}
104
105//! Default constructor
106template <class T>
107NDataBlock<T>::NDataBlock()
108// Createur par default
109: mSz(0), mSRef(NULL), mIsTemp(false)
110{
111if(Debug_NDataBlock>1)
112 cout<<"?_NDataBlock::NDataBlock("<<this<<") default"<<endl;
113}
114
115//! Copy constructor
116/*!
117 \warning datas are \b SHARED with \b a.
118 */
119template <class T>
120NDataBlock<T>::NDataBlock(const NDataBlock<T>& a)
121// Createur par copie: partage les donnees dans tous les cas
122: mSz(0), mSRef(NULL), mIsTemp(false)
123{
124if(Debug_NDataBlock>1)
125 cout<<"?_NDataBlock::NDataBlock("<<this<<",&a="<<&a<<" a.mSz="<<a.mSz<<")"<<endl;
126
127if(a.mSRef && a.mSz>0) Share(a);
128}
129
130//! Copy constructor with \b share option
131/*!
132 \warning datas are shared if \b share is \b true, cloned if not.
133 */
134template <class T>
135NDataBlock<T>::NDataBlock(const NDataBlock<T>& a,bool share)
136// Createur avec choix de partager ou non selon "share"
137: mSz(0), mSRef(NULL), mIsTemp(false)
138{
139if(Debug_NDataBlock>1)
140 cout<<"?_NDataBlock::NDataBlock("<<this<<",&a="<<&a
141 <<",sh=<<"<<share<<")"<<endl;
142
143if(a.mSRef && a.mSz>0) {if(share) Share(a); else Clone(a);}
144}
145
146//! Destructor
147template <class T>
148NDataBlock<T>::~NDataBlock()
149// Destructeur
150{
151if(Debug_NDataBlock>1)
152 cout<<"?_NDataBlock::~NDataBlock("<<this<<")"<<endl;
153
154Delete();
155}
156
157////////////////////////
158// Gestion de donnees //
159////////////////////////
160
161//! Clone datas from \b a
162template <class T>
163void NDataBlock<T>::Clone(const NDataBlock<T>& a)
164// Clone: copie de donnees a partir de "a"
165{
166if(Debug_NDataBlock>1)
167 cout<<"?_NDataBlock::Clone("<<this<<","<<&a<<") a.(mSz="
168 <<a.mSz<<" mSRef="<<a.mSRef<<" IsTemp="<<a.IsTemp()
169 <<"), mSz="<<mSz<<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp<<endl;
170
171if(&a==NULL) throw(NullPtrError("NDataBlock::Clone &a==NULL\n"));
172if(!a.mSRef || a.mSz==0) throw(NullPtrError("NDataBlock::Clone a.mSz=0\n"));
173Alloc(a.mSz, NULL, NULL, false); // pas de mise a zero
174memcpy(Data(),a.Data(),mSz*sizeof(T));
175}
176
177//! Share datas with \b a
178template <class T>
179void NDataBlock<T>::Share(const NDataBlock<T>& a)
180// Share: Partage les donnees avec "a"
181{
182if(Debug_NDataBlock>1) {
183 cout<<"?_NDataBlock::Share("<<this<<","<<&a<<")";
184 if(&a!=NULL) cout<<" a.(mSz="<<a.mSz<<" mSRef="<<a.mSRef
185 <<" IsTemp="<<a.IsTemp()<<")";
186 cout<<", mSz="<<mSz<<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp<<endl;
187}
188
189if(&a==NULL) throw(NullPtrError("NDataBlock::Share &a==NULL\n"));
190if(!a.mSRef || a.mSz==0) throw(NullPtrError("NDataBlock::Share a.mSz=0\n"));
191if(mSRef) Delete();
192mSz = a.mSz; mSRef = a.mSRef; mSRef->nref++;
193
194if(Debug_NDataBlock>1)
195 cout<<"...?_NDataBlock::Share mSz="<<mSz<<" mSRef="<<mSRef
196 <<" mSRef->nref="<<mSRef->nref<<" mSRef->data="<< mSRef->data
197 <<" mSRef->bridge="<<mSRef->bridge
198 <<" IsTemp="<<mIsTemp<<endl;
199}
200
201//! \b Share with \b a if \b temporary, \b clone from \b a if not.
202/*! \warning For most purposes, users don't have to worry with
203 the "temporary" nature of a NDataBlock. That is used
204 internaly to avoid memory allocation in operation
205 like A = B + C + D for instance. The method is not
206 protected to allow users to write complicated functions
207 on NDataBlock.
208 \verbatim
209 ----------------------------------------------------------
210 Pourquoi une complication avec la notion de "temporaire" :
211 ----------------------------------------------------------
212 - Le constructeur par copie partageant les donnees,
213 dans une methode un { NDataBlock<T> result; ...; return result;}
214 ne va pas allouer de la memoire pour retourner "result".
215 - La gestion de temporaire sert quand on enchaine plusieurs
216 operations sur la meme ligne, par exemple : A = B+C+D;
217 Dans ce cas l'objet CD=C+D est d'abord alloue et rempli
218 avec C+D, puis CD est mis a "temporaire".
219 Quand on ajoute B a CD, la methode d'addition va se rendre compte
220 que CD est "temporaire" et additionner B "in-place" dans CD
221 sans allouer une fois de plus de la place (pas d'allocation
222 de place BCD pour mettre B+CD mais une operation CD += B).
223 Si la notion d'objet "temporaire" n'avait pas ete consideree
224 l'addition A = B+C+D aurait alloue de la place pour "CD=C+D"
225 puis pour BCD=B+CD : 2 allocations auraient ete necessaires
226 contre 1 seule dans notre cas de geston de "temporaire".
227 \endverbatim
228*/
229template <class T>
230void NDataBlock<T>::CloneOrShare(const NDataBlock<T>& a)
231// CloneOrShare: Share si "a" temporaire, Clone sinon.
232{
233if(Debug_NDataBlock>1)
234 cout<<"?_NDataBlock::CloneOrShare("<<this<<","<<&a<<")"<<endl;
235
236if(&a==NULL) throw(NullPtrError("NDataBlock::CloneOrShare &a==NULL\n"));
237if(a.IsTemp()) Share(a); else Clone(a);
238}
239
240////////////////////////////////////////////////////////////
241// Allocation , destruction , remplissage et reallocation //
242////////////////////////////////////////////////////////////
243
244//! Allocation management
245/*!
246 Allocation d'un NOUVEL espace de stoquage de "n" donnees
247 \verbatim
248 Si data==NULL : allocation de l'espace memoire
249 si zero == true , l'espace est remplis de zeros
250 data!=NULL : partage des donnees avec l'adresse data
251 Si br==NULL : les donnees nous appartiennent
252 br!=NULL : les donnees ne nous appartiennent pas (ex: Blitz)
253
254 Exemple: on veut connecter a un tableau de T*
255 > float *x = new float[5]; ... remplissage de x[] ...;
256 1- On veut que NDataBlock NE DESALLOUE PAS le tableau "x[]"
257 a- Premiere solution
258 > NDataBlock A(5,x,new Bridge);
259 ......
260 > delete [] x;
261 - Il faut deleter x[] explicitement.
262 - Le destructeur de "A" ne detruit pas x[].
263 ATTENTION: Une fois x[] detruit, "A" ne peut
264 plus acceder les donnees!
265 - Bridge est detruit par le destructeur de "A"
266 b- Autre solution:
267 > NDataBlock A(5); A.FillFrom(5,x);
268 > delete [] x;
269 ......
270 - Il faut deleter x[] explicitement.
271 - "A" possede une copie en local de x[].
272 - Le destructeur de "A" ne detruit pas x[] mais la copie locale.
273 2- On veut que NDataBlock desalloue le tableau
274 > NDataBlock A(5,x);
275 - Ne Pas Faire "delete [] x;"
276 - "A" partage les donnees avec x[].
277 - Le destructeur de "A" detruit x[].
278
279 --- REMARQUE SUR LE DANGER DE CERTAINES SITUATIONS (CMV):
280 1-/ x = new float[n1]; NDataBlock A(n2,x);
281 1er danger: si n2>n1 depassement de tableaux (core dump)
282 2sd danger: celui qui alloue x[] ne doit pas faire le "delete"
283 en desaccord avec toutes les regles de bonne conduite.
284 2-/ float x[5]={1,2,3,4,5}; {NDataBlock A(n2,&x[0]);} cout<<x[2];
285 Ici, a la sortie du bloc {}, le destructeur de "A" va detruire
286 l'adresse de &x[0]: je n'ose imaginer que ca se fasse sans probleme
287 et de toute facon, cout<<x[2]; va surement faire des etincelles.
288 3-/ x = new float[n1]; NDataBlock A(n2,x,new Bridge);
289 1er danger: si n2>n1 depassement de tableaux (core dump)
290 2sd danger: si la methode bridgee (blitz?) detruit x[]
291 "A" n'a plus de donnees connectees!
292 --- CONCLUSION
293 Cette classe est franchement merdique.
294 - On peut accepter la prise de risque liee a NDataBlock(n2,x,new Bridge);
295 car je ne vois pas comment on pourrait faire autrement pour connecter
296 un tableau de type blitz par exemple.
297 - Par contre le createur NDataBlock(n2,x); doit etre interdit
298 dans sa forme actelle car trop dangereux et il me semble inutile.
299 - Dans cette nouvelle optique:
300 NDataBlock(n2,x,new Bridge) et NDataBlock(n2,x) disparaissent
301 On remplace par NDataBlock(n2,x) {Alloc(n2,x,new Bridge);}
302 qui force le Bridge dans tout les cas puisque NDataBlock
303 ne possede pas les donnees.
304 Mais puis-je encore le faire vu que NDataBlock est a la base
305 de TVector,TMatrix et qu'il faut donc reprendre tout le code DPC
306 - Quoiqu'il arrive Alloc est une methode privee et peut donc rester
307 sous sa forme actuelle.
308
309 \endverbatim
310 */
311
312static uint_8 _ndrefid_ = 0; // Identificateur de NDREF cree
313
314template <class T>
315void NDataBlock<T>::Alloc(size_t n,T* data,Bridge* br,bool zero)
316{
317if(Debug_NDataBlock>1)
318 cout<<"?_NDataBlock::Alloc("<<this<<","
319 <<n<<","<<data<<","<<br<<") mSz="<<mSz
320 <<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp<<endl;
321
322if(br && !data)
323 throw(NullPtrError("NDataBlock::Alloc br!=NULL && data==NULL\n"));
324if(n==0) throw(SzMismatchError("NDataBlock::Alloc n==0\n"));
325if(mSRef) Delete();
326mSz = n;
327mSRef = new NDREF;
328mSRef->nref = 1;
329mSRef->dsid = ++_ndrefid_;
330if(data) mSRef->data = data;
331else {mSRef->data = new T[n]; if (zero) memset(mSRef->data,0,n*sizeof(T));}
332mSRef->bridge = br;
333
334if(Debug_NDataBlock>0) {
335 // Meme dans le cas data!=0 et br==0 (connexion d'un tableau
336 // avec destruction geree par ~NDataBlock (cas 2-) on compte
337 // comme si on avait fait une allocation du tableau (ce qui a ete
338 // fait au niveau du dessus!).
339 if(!br) NallocData++; NallocSRef++;
340 if(Debug_NDataBlock>1)
341 cout<<"...?_NDataBlock::Alloc mSz="<<mSz<<" mSRef="<<mSRef
342 <<" mSRef->nref="<<mSRef->nref<<" mSRef->data="<<mSRef->data
343 <<" mSRef->bridge="<<mSRef->bridge
344 <<" IsTemp="<<mIsTemp
345 <<" Total("<<NallocData<<","<<NallocSRef<<")"<<endl;
346}
347}
348
349//! Management of de-allocation
350template <class T>
351void NDataBlock<T>::Delete(void)
352// Pour detruire les pointeurs en tenant compte des references
353{
354if(Debug_NDataBlock>1) {
355 cout<<"?_NDataBlock::Delete("<<this<<") mSz="<<mSz
356 <<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp;
357 if(mSRef)
358 cout<<" mSRef->nref="<<mSRef->nref<<" mSRef->data="
359 <<mSRef->data<<" mSRef->bridge="<<mSRef->bridge;
360 cout<<endl;
361}
362
363if(mSRef==NULL) return;
364
365mSRef->nref--;
366if(mSRef->nref != 0) {
367
368if(Debug_NDataBlock>1)
369 cout<<"...?_NDataBlock::Delete() pas de desallocation il reste nref="
370 <<mSRef->nref<<" Total("<<NallocData<<","<<NallocSRef<<")"<<endl;
371
372 mSz = 0; mSRef=NULL;
373 return;
374}
375
376if(Debug_NDataBlock>0) {
377 if(!mSRef->bridge) NallocData--; NallocSRef--;
378 if(Debug_NDataBlock>1)
379 cout<<"...?_NDataBlock::Delete() desallocation complete il reste nref="
380 <<mSRef->nref<<" Total("<<NallocData<<","<<NallocSRef<<")"<<endl;
381}
382
383// Si il y a un Bridge les donnees ne n'appartiennent pas, on detruit le Bridge
384// sinon, les donnees ont ete allouees par nos soins, on libere l'espace
385if(mSRef->bridge) delete mSRef->bridge; else delete [] mSRef->data;
386mSRef->bridge=NULL; mSRef->data=NULL;
387delete mSRef; mSRef=NULL; mSz = 0;
388}
389
390//! Fill dats of this NDataBlock with the \b n datas pointed by \b data
391/*!
392 \warning If class empty : allocate space in memory
393 \warning If class already connected : overwrite with minimum size
394 (\b n or \b mSz)
395 */
396template <class T>
397void NDataBlock<T>::FillFrom(size_t n,T* data)
398// Remplissage par un tableau de donnees
399// - Si classe vide : creation de l'espace memoire
400// - Si classe connectee : on ecrit selon la longueur minimale
401// (cad this->mSz ou "n")
402{
403if(data==NULL) throw(NullPtrError("NDataBlock::FillFrom data==NULL\n"));
404if(n==0) throw(ParmError("NDataBlock::FillFrom n<=0\n"));
405if(mSRef==NULL) Alloc(n, NULL, NULL, false); // Pas de mise a zero
406if(mSz<n) n = mSz;
407memcpy(Data(),data,n*sizeof(T));
408}
409
410//! Re-allocate space for \b nnew datas
411/*!
412 \param nnnew : new size
413 \param force : to manage the way re-allocation will be done (see after).
414 \verbatim
415 Re-allocation de "nnew" place memoire pour les donnees
416 avec conservation des "nold" donnees precedentes si possible.
417 "force" gere la re-allocation de la place memoire pour les donnees.
418 Divers cas se presentent:
419 a-/ *** nnew>nold force=quelconque ***
420 place re-allouee, donnees [0,nold[ copiees, surplus [nold,new[ mis a zero
421 b-/ *** nnew<=nold force=true ***
422 place re-allouee, donnees [0,nnew[ copiees, pas de surplus
423 c-/ *** nnew<=nold force=false ***
424 place non re-allouee, seule la valeur de la taille est diminuee
425 - On tient compte du partage des donnees dans tous les cas.
426 - Si il n'y a pas de donnees connectees a la classe, on re-alloue
427 dans tous les cas
428 \endverbatim
429 */
430template <class T>
431void NDataBlock<T>::Realloc(size_t nnew,bool force)
432{
433if(nnew==0) throw(ParmError("NDataBlock::Realloc n<=0\n"));
434
435// Cas sans re-allocation memoire
436if(mSRef && nnew<=mSz && ! force) { mSz=nnew; return;}
437
438// Cas avec re-allocation memoire
439size_t ncop;
440if(!mSRef || mSz==0) ncop=0; else if(mSz<nnew) ncop=mSz; else ncop=nnew;
441T* dataloc = new T[nnew];
442if(ncop>0) memcpy(dataloc,mSRef->data,ncop*sizeof(T));
443if(nnew>ncop) memset(dataloc+ncop,0,(nnew-ncop)*sizeof(T));
444Alloc(nnew,dataloc,NULL); //Alloc gere partage de reference et bridge
445}
446
447////////////////
448// Impression //
449////////////////
450
451//! Give infos and print \b n datas beginning at \b i1 on stream \b os.
452template <class T>
453void NDataBlock<T>::Print(ostream& os,size_t i1,size_t n) const
454// Impression de n elements a partir de i1
455{
456size_t nr = 0;
457T* p = NULL; Bridge* br = NULL;
458if(mSRef) {nr = mSRef->nref; p = mSRef->data; br = mSRef->bridge;}
459os<<"NDataBlock::Print("<<this<<",Sz="<<mSz<<",IsTemp="<<mIsTemp<<")\n"
460 <<" mSRef="<<mSRef<<"(nref="<<nr<<",data="<<p
461 <<",bridge="<<br<<")"<<endl;
462if(i1>=mSz || n<=0 || !p) return;
463size_t i2 = i1+n; if(i2>mSz) i2=mSz;
464size_t im = 1; bool enl=false;
465while(i1<i2) {
466 enl = false;
467 os<<" "<<(*this)(i1); i1++;
468 if(im==8) {os<<"\n"; im=1; enl=true;} else im++;
469}
470if(!enl) os<<endl;
471}
472
473//////////////////////////////////////////////
474// Calcul de la somme / produit des donnees //
475//////////////////////////////////////////////
476
477//! Return sum of \b n datas beginning at data \b i1.
478template <class T>
479T NDataBlock<T>::Sum(size_t i1,size_t n) const
480// Somme des elements de i1 a i1+n-1
481{
482if(i1>=mSz) return 0;
483if(n>mSz) n = mSz; if(n==0) n = mSz-i1;
484T const *p=Begin()+i1, *pe=p+n;
485T val = 0;
486while (p<pe) val += *p++;
487return val;
488}
489
490//! Return product of \b n datas beginning at data \b i1.
491template <class T>
492T NDataBlock<T>::Product(size_t i1,size_t n) const
493// Produit des elements de i1 a i1+n-1
494{
495if(i1>=mSz) return 0;
496if(n>mSz) n = mSz; if(n==0) n = mSz-i1;
497T const *p=Begin()+i1, *pe=p+n;
498T val = 0;
499while (p<pe) val *= *p++;
500return val;
501}
502
503///////////////////////////////////////////////////////////////
504// Surcharge de = : NDataBlock=NDataBlock; NDataBlock=<T> b; //
505///////////////////////////////////////////////////////////////
506
507//! Operator = : ND = NDa
508/*! \warning Datas are copied (cloned) from \b a. */
509template <class T>
510NDataBlock<T>& NDataBlock<T>::operator = (const NDataBlock<T>& a)
511// Affectation: partage des donnees si "a" temporaire, clone sinon.
512{
513if(Debug_NDataBlock>1)
514 cout<<"?_NDataBlock::operator=("<<this<<","<<&a<<") a.(mSz="
515 <<a.mSz<<" mSRef="<<a.mSRef<<" IsTemp="<<a.IsTemp()
516 <<"), mSz="<<mSz<<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp<<endl;
517
518if(this == &a) return *this;
519if(a.mSz==0)
520 throw(SzMismatchError("NDataBlock::operator=A null size \n"));
521if (mSz==0) { CloneOrShare(a); return *this; }
522if (a.mSz != mSz)
523 throw(SzMismatchError("NDataBlock::operator=A Unequal sizes \n"));
524memcpy(Data(),a.Data(),mSz*sizeof(T));
525return *this;
526}
527
528//! Operator = : ND = \b v (at dats set to \b v).
529template <class T>
530NDataBlock<T>& NDataBlock<T>::operator = (T v)
531// Affectation de tous les elements a une constante "v"
532{
533if(Debug_NDataBlock>1)
534 cout<<"?_NDataBlock::operator=("<<this<<","<<v<<")"
535 <<" mSz="<<mSz<<" mSRef="<<mSRef<<" IsTemp="<<mIsTemp<<endl;
536
537if(mSz==0) throw(SzMismatchError("NDataBlock::operator=v null size\n"));
538T *p=Begin(), *pe=End(); while (p<pe) *p++ = v;
539return *this;
540}
541
542//////////////////////////////////////////////////////////////
543// Surcharge de +=,-=,*=,/= (INPLACE): NDataBlock += <T> b; //
544//////////////////////////////////////////////////////////////
545
546//! Add a constant : ND += b
547template <class T>
548NDataBlock<T>& NDataBlock<T>::operator += (T b)
549{
550if(mSz==0) throw(SzMismatchError("NDataBlock::operator+=v null size\n"));
551T *p=Begin(), *pe=End(); while (p<pe) *p++ += b;
552return *this;
553}
554
555//! Substract a constant : ND -= b
556template <class T>
557NDataBlock<T>& NDataBlock<T>::operator -= (T b)
558{
559if(mSz==0) throw(SzMismatchError("NDataBlock::operator-=v null size\n"));
560T *p=Begin(), *pe=End(); while (p<pe) *p++ -= b;
561return *this;
562}
563
564//! Multiply by a constant : ND *= b
565template <class T>
566NDataBlock<T>& NDataBlock<T>::operator *= (T b)
567{
568if(mSz==0) throw(SzMismatchError("NDataBlock::operator*=v null size\n"));
569T *p=Begin(), *pe=End(); while (p<pe) *p++ *= b;
570return *this;
571}
572
573//! Divide by a constant : ND /= b
574template <class T>
575NDataBlock<T>& NDataBlock<T>::operator /= (T b)
576{
577if(b==(T) 0) throw(ParmError("NDataBlock::operator/=v divide by zero\n"));
578if(mSz==0) throw(SzMismatchError("NDataBlock::operator/=v null size\n"));
579T *p=Begin(), *pe=End(); while (p<pe) *p++ /= b;
580return *this;
581}
582
583////////////////////////////////////////////////////////////////////
584// Surcharge de +=,-=,*=,/= (INPLACE): NDataBlock += NDataBlock1; //
585////////////////////////////////////////////////////////////////////
586
587//! Add a NDataBlock : ND += NDa
588template <class T>
589NDataBlock<T>& NDataBlock<T>::operator += (const NDataBlock<T>& a)
590{
591if(mSz==0 || mSz!=a.mSz)
592 throw(SzMismatchError("NDataBlock::operator+=A size mismatch/null"));
593T *p=Begin(), *pe=End();
594T const * pa=a.Begin();
595while (p<pe) *p++ += *pa++;
596return *this;
597}
598
599//! Substract a NDataBlock : ND -= NDa
600template <class T>
601NDataBlock<T>& NDataBlock<T>::operator -= (const NDataBlock<T>& a)
602{
603if(mSz==0 || mSz!=a.mSz)
604 throw(SzMismatchError("NDataBlock::operator-=A size mismatch/null"));
605T *p=Begin(), *pe=End();
606T const *pa=a.Begin();
607while (p<pe) *p++ -= *pa++;
608return *this;
609}
610
611//! Multiply by a NDataBlock : ND *= NDa
612template <class T>
613NDataBlock<T>& NDataBlock<T>::operator *= (const NDataBlock<T>& a)
614{
615if(mSz==0 || mSz!=a.mSz)
616 throw(SzMismatchError("NDataBlock::operator*=A size mismatch/null"));
617T *p=Begin(), *pe=End();
618T const *pa=a.Begin();
619while (p<pe) *p++ *= *pa++;
620return *this;
621}
622
623//! Divide by a NDataBlock : ND /= NDa
624template <class T>
625NDataBlock<T>& NDataBlock<T>::operator /= (const NDataBlock<T>& a)
626// Attention, aucune protection si un element de "a" est nul.
627{
628if(mSz==0 || mSz!=a.mSz)
629 throw(SzMismatchError("NDataBlock::operator/=A size mismatch/null"));
630T *p=Begin(), *pe=End();
631T const *pa=a.Begin();
632while (p<pe) *p++ /= *pa++; // Division par zero non protegee
633return *this;
634}
635
636//////////////////////////////////////////////////////////////////
637// Pour surcharge de +,-,*,/ : NDataBlock = NDataBlock1+<T>b; //
638// NDataBlock = <T>b+NDataBlock1; //
639// Pour la notion de "temporaire" voir blabla dans CloneOrShare //
640//////////////////////////////////////////////////////////////////
641
642//! Add a constant and return NDataBlock : NDret = ND + b
643template <class T>
644NDataBlock<T> NDataBlock<T>::Add(T b) const
645// Pour A+b
646{
647NDataBlock<T> result;
648result.CloneOrShare(*this); result.SetTemp(true);
649result += b;
650return result;
651}
652
653//! Substract a constant and return NDataBlock : NDret = ND - b or NDret = b - ND
654/*! Substract a constant or from a constant
655 \param fginv==false : performs NDret = ND - b (default)
656 \param fginv==true : performs NDret = b - ND
657*/
658template <class T>
659NDataBlock<T> NDataBlock<T>::Sub(T b,bool fginv) const
660// Pour A-b sauf si fginv==true b-A (- n'est pas commutatif!)
661{
662NDataBlock<T> result;
663result.CloneOrShare(*this); result.SetTemp(true);
664if(fginv) {
665 T *p=result.Begin(), *pe=result.End();
666 T const *pa=this->Begin();
667 while(p<pe) {*p++ = b - *pa++;}
668} else result -= b;
669return result;
670}
671
672//! Multiply by a constant and return NDataBlock : NDret = ND * b
673template <class T>
674NDataBlock<T> NDataBlock<T>::Mul(T b) const
675// Pour A*b
676{
677NDataBlock<T> result;
678result.CloneOrShare(*this); result.SetTemp(true);
679result *= b;
680return result;
681}
682
683//! Divide by a constant and return NDataBlock : NDret = ND / b or NDret = b / ND
684/*! Divide by a constant or from a constant
685 \param fginv==false : performs NDret = ND / b (default)
686 \param fginv==true : performs NDret = b / ND
687*/
688template <class T>
689NDataBlock<T> NDataBlock<T>::Div(T b,bool fginv) const
690// Pour A/b sauf si fginv==true b/A (/ n'est pas commutatif!)
691{
692NDataBlock<T> result;
693result.CloneOrShare(*this); result.SetTemp(true);
694 if(fginv) {
695 T *p=result.Begin(), *pe=result.End();
696 T const *pa = this->Begin();
697 while(p<pe) {*p++ = b / *pa++;} // Division par zero non protegee
698} else {
699 if( b == (T) 0 ) throw MathExc("NDataBlock<T>::Div(T) - Divide by zero ! ");
700 result /= b;
701}
702return result;
703}
704
705///////////////////////////////////////////////////////////////////////
706// Pour surcharge de +,-,*,/ : NDataBlock = NDataBlock1+NDataBlock2; //
707///////////////////////////////////////////////////////////////////////
708
709//! Add a NDataBlock and return a NDataBlock: ND = NDthis + NDb
710template <class T>
711NDataBlock<T> NDataBlock<T>::Add(const NDataBlock<T>& b) const
712// Pour A+B
713{
714if(mSz!=b.mSz)
715 throw(SzMismatchError("NDataBlock operator C=A+B size mismatch/null\n"));
716NDataBlock<T> result; result.SetTemp(true);
717if(b.IsTemp()) {result.Share(b); result += *this;}
718 else {result.CloneOrShare(*this); result += b;}
719return result;
720}
721
722//! Multiply by a NDataBlock and return a NDataBlock: ND = NDthis * NDb
723template <class T>
724NDataBlock<T> NDataBlock<T>::Mul(const NDataBlock<T>& b) const
725// Pour A*B
726{
727if(mSz!=b.mSz)
728 throw(SzMismatchError("NDataBlock operator C=A*B size mismatch/null\n"));
729NDataBlock<T> result; result.SetTemp(true);
730if(b.IsTemp()) {result.Share(b); result *= *this;}
731 else {result.CloneOrShare(*this); result *= b;}
732return result;
733}
734
735//! Substract a NDataBlock and return a NDataBlock: ND = NDthis - NDb
736template <class T>
737NDataBlock<T> NDataBlock<T>::Sub(const NDataBlock<T>& b) const
738// Pour A-B
739{
740if(mSz!=b.mSz)
741 throw(SzMismatchError("NDataBlock operator C=A-B size mismatch/null\n"));
742NDataBlock<T> result; result.SetTemp(true);
743if(b.IsTemp()) {
744 result.Share(b);
745 T *p=result.Begin(), *pe=result.End(); T const *pa=Begin();
746 while(p<pe) {*p = *pa++ - *p; p++;}
747} else {result.CloneOrShare(*this); result -= b;}
748return result;
749}
750
751//! Divide by a NDataBlock and return a NDataBlock: ND = NDthis / NDb
752template <class T>
753NDataBlock<T> NDataBlock<T>::Div(const NDataBlock<T>& b) const
754// Pour A/B
755{
756if(mSz!=b.mSz)
757 throw(SzMismatchError("NDataBlock operator C=A/B size mismatch/null\n"));
758NDataBlock<T> result; result.SetTemp(true);
759if(b.IsTemp()) {
760 result.Share(b);
761 T *p=result.Begin(), *pe=result.End(); T const *pa=Begin();
762 while(p<pe) {*p = *pa++ / *p; p++;} // Division par zero non protegee
763} else {result.CloneOrShare(*this); result /= b;}
764return result;
765}
766
767
768///////////////////////////////////////////////////////////////
769#ifdef __CXX_PRAGMA_TEMPLATES__
770#pragma define_template NDataBlock<uint_1>
771#pragma define_template NDataBlock<uint_2>
772#pragma define_template NDataBlock<int_2>
773#pragma define_template NDataBlock<int_4>
774#pragma define_template NDataBlock<int_8>
775#pragma define_template NDataBlock<uint_4>
776#pragma define_template NDataBlock<uint_8>
777#pragma define_template NDataBlock<r_4>
778#pragma define_template NDataBlock<r_8>
779#pragma define_template NDataBlock< complex<r_4> >
780#pragma define_template NDataBlock< complex<r_8> >
781#endif
782
783#if defined(ANSI_TEMPLATES) || defined(GNU_TEMPLATES)
784template class NDataBlock<uint_1>;
785template class NDataBlock<uint_2>;
786template class NDataBlock<int_2>;
787template class NDataBlock<int_4>;
788template class NDataBlock<int_8>;
789template class NDataBlock<uint_4>;
790template class NDataBlock<uint_8>;
791template class NDataBlock<r_4>;
792template class NDataBlock<r_8>;
793template class NDataBlock< complex<r_4> >;
794template class NDataBlock< complex<r_8> >;
795#endif
Note: See TracBrowser for help on using the repository browser.