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

Last change on this file since 3208 was 3208, checked in by ansari, 18 years ago

Ajout classe ThSafeOp (thsafeop.h) pour rendre NDataBlock<T> (et SegDataBlock) thread-safe, ajout methode DataTypeInfo::getTypeId() , implementation ThSafeOp ds ndatablock.cc (.h), Reza 10/04/2007

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