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

Last change on this file since 944 was 944, checked in by ansari, 25 years ago

doc + nouvelle gestion debug cmv 16/04/00

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