[220] | 1 | #include <stdio.h>
|
---|
| 2 | #include <string.h>
|
---|
| 3 |
|
---|
| 4 | #include "perrors.h"
|
---|
| 5 | #include "ntuple.h"
|
---|
| 6 |
|
---|
| 7 |
|
---|
| 8 | #define BADVAL -1.e19
|
---|
| 9 | #define LENNAME 8
|
---|
| 10 | #define LENNAME1 (LENNAME+1)
|
---|
| 11 |
|
---|
| 12 | //++
|
---|
| 13 | // Class NTuple
|
---|
| 14 | // Lib Outils++
|
---|
| 15 | // include ntuple.h
|
---|
| 16 | //
|
---|
| 17 | // Classe de ntuples
|
---|
| 18 | //--
|
---|
| 19 |
|
---|
| 20 | /* --Methode-- */
|
---|
| 21 | //++
|
---|
| 22 | NTuple::NTuple()
|
---|
| 23 | //
|
---|
| 24 | // Createur par defaut
|
---|
| 25 | //--
|
---|
| 26 | {
|
---|
| 27 | mNVar = mNEnt = mBlk = mNBlk = 0;
|
---|
| 28 | mVar = NULL;
|
---|
| 29 | mNames = NULL;
|
---|
| 30 | mInfo = NULL;
|
---|
| 31 | }
|
---|
| 32 |
|
---|
| 33 |
|
---|
| 34 | //++
|
---|
| 35 | NTuple::NTuple(int nvar, char** noms, int blk)
|
---|
| 36 | //
|
---|
| 37 | // Createur d'un ntuple de `nvar' variables dont les
|
---|
| 38 | // noms sont dans le tableau de cahines de caracteres `noms'
|
---|
| 39 | // avec `blk' d'evenements par blocks.
|
---|
| 40 | //--
|
---|
| 41 | {
|
---|
| 42 | mNVar = mNEnt = mBlk = mNBlk = 0;
|
---|
| 43 | mVar = NULL;
|
---|
| 44 | mNames = NULL;
|
---|
| 45 | mInfo = NULL;
|
---|
| 46 | if (nvar <= 0) THROW(sizeMismatchErr);
|
---|
| 47 | mNVar = nvar;
|
---|
| 48 | mVar = new r_4[nvar];
|
---|
| 49 | if (blk < 10) blk = 10;
|
---|
| 50 | mBlk = blk;
|
---|
| 51 | // On prend des noms de LENNAME char pour le moment
|
---|
| 52 | mNames = new char[nvar*LENNAME1];
|
---|
| 53 | r_4* pt = new r_4[nvar*blk];
|
---|
| 54 | mNBlk = 1;
|
---|
| 55 | mPtr.push_back(pt);
|
---|
| 56 | int i;
|
---|
| 57 | for(i=0; i<nvar; i++)
|
---|
| 58 | { strncpy(mNames+i*LENNAME1, noms[i], LENNAME);
|
---|
| 59 | mNames[i*LENNAME1+LENNAME] = '\0'; }
|
---|
| 60 | return;
|
---|
| 61 | }
|
---|
| 62 |
|
---|
| 63 | /* --Methode-- */
|
---|
| 64 | //++
|
---|
| 65 | NTuple::NTuple(char *flnm)
|
---|
| 66 | //
|
---|
| 67 | // Createur lecture fichier ppersist.
|
---|
| 68 | //--
|
---|
| 69 | {
|
---|
| 70 | mNVar = mNEnt = mBlk = mNBlk = 0;
|
---|
| 71 | mVar = NULL;
|
---|
| 72 | mNames = NULL;
|
---|
| 73 | mInfo = NULL;
|
---|
| 74 | PInPersist s(flnm);
|
---|
| 75 | Read(s);
|
---|
| 76 | }
|
---|
| 77 |
|
---|
| 78 | /* --Methode-- */
|
---|
| 79 | NTuple::~NTuple()
|
---|
| 80 | {
|
---|
| 81 | Clean();
|
---|
| 82 | }
|
---|
| 83 |
|
---|
| 84 | /* --Methode-- */
|
---|
| 85 | void NTuple::Clean()
|
---|
| 86 | {
|
---|
| 87 | if (mVar) delete[] mVar;
|
---|
| 88 | if (mNames) delete[] mNames;
|
---|
| 89 | if (mInfo) delete mInfo;
|
---|
| 90 | int i;
|
---|
| 91 | for(i=0; i<mNBlk; i++) delete[] mPtr[i];
|
---|
| 92 | mPtr.erase(mPtr.begin(), mPtr.end());
|
---|
| 93 | mNVar = mNEnt = mBlk = mNBlk = 0;
|
---|
| 94 | mVar = NULL;
|
---|
| 95 | mNames = NULL;
|
---|
| 96 | return;
|
---|
| 97 | }
|
---|
| 98 |
|
---|
| 99 | /* --Methode-- */
|
---|
| 100 | //++
|
---|
| 101 | void NTuple::Fill(r_4* x)
|
---|
| 102 | //
|
---|
| 103 | // Remplit le ntuple avec le tableau cd reels `x'.
|
---|
| 104 | //--
|
---|
| 105 | {
|
---|
| 106 | int numb = mNEnt/mBlk;
|
---|
| 107 | if (numb >= mNBlk) {
|
---|
| 108 | r_4* pt = new r_4[mNVar*mBlk];
|
---|
| 109 | mNBlk++;
|
---|
| 110 | mPtr.push_back(pt);
|
---|
| 111 | }
|
---|
| 112 | int offb = mNEnt-numb*mBlk;
|
---|
| 113 | memcpy((mPtr[numb]+offb*mNVar), x, mNVar*sizeof(r_4));
|
---|
| 114 | mNEnt++;
|
---|
| 115 | return;
|
---|
| 116 | }
|
---|
| 117 |
|
---|
| 118 |
|
---|
| 119 | /* --Methode-- */
|
---|
| 120 | //++
|
---|
| 121 | float NTuple::GetVal(int n, int k) const
|
---|
| 122 | //
|
---|
| 123 | // Retourne la valeur de la variable `k' de l'evenement `n'.
|
---|
| 124 | //--
|
---|
| 125 | {
|
---|
| 126 | if (n >= mNEnt) return(BADVAL);
|
---|
| 127 | if ( (k < 0) || (k >= mNVar) ) return(BADVAL);
|
---|
| 128 | int numb = n/mBlk;
|
---|
| 129 | int offb = n-numb*mBlk;
|
---|
| 130 | return(*(mPtr[numb]+offb*mNVar+k));
|
---|
| 131 | }
|
---|
| 132 |
|
---|
| 133 |
|
---|
| 134 | /* --Methode-- */
|
---|
| 135 | //++
|
---|
| 136 | int NTuple::IndexNom(const char* nom) const
|
---|
| 137 | //
|
---|
| 138 | // Retourne le numero de la variable de nom `nom'.
|
---|
| 139 | //--
|
---|
| 140 | {
|
---|
| 141 | int i;
|
---|
| 142 | for(i=0; i<mNVar; i++)
|
---|
| 143 | if ( strcmp(nom, mNames+i*LENNAME1) == 0) return(i);
|
---|
| 144 | return(-1);
|
---|
| 145 | }
|
---|
| 146 |
|
---|
| 147 |
|
---|
| 148 | static char nomretour[2*LENNAME1];
|
---|
| 149 | /* --Methode-- */
|
---|
| 150 | //++
|
---|
| 151 | char* NTuple::NomIndex(int k) const
|
---|
| 152 | //
|
---|
| 153 | // Retourne le nom de la variable numero 'k'
|
---|
| 154 | //--
|
---|
| 155 | {
|
---|
| 156 | nomretour[0] = '\0';
|
---|
| 157 | if ((k >= 0) && (k < mNVar)) strcpy(nomretour, mNames+k*LENNAME1);
|
---|
| 158 | return(nomretour);
|
---|
| 159 | }
|
---|
| 160 |
|
---|
| 161 | /* --Methode-- */
|
---|
| 162 | //++
|
---|
| 163 | string NTuple::VarList_C(const char* nomx) const
|
---|
| 164 | //
|
---|
| 165 | // Retourne une chaine de caracteres avec la declaration des noms de
|
---|
| 166 | // variables. si "nomx!=NULL" , des instructions d'affectation
|
---|
| 167 | // a partir d'un tableau "nomx[i]" sont ajoutees.
|
---|
| 168 | //--
|
---|
| 169 | {
|
---|
| 170 | string rets;
|
---|
| 171 | int i;
|
---|
| 172 | for(i=0; i<mNVar; i++) {
|
---|
| 173 | if ( (i%5 == 0) && (i > 0) ) rets += ";";
|
---|
| 174 | if (i%5 == 0) rets += "\ndouble ";
|
---|
| 175 | else rets += ",";
|
---|
| 176 | rets += mNames+i*LENNAME1;
|
---|
| 177 | }
|
---|
| 178 | rets += "; \n";
|
---|
| 179 | if (nomx) {
|
---|
| 180 | char buff[256];
|
---|
| 181 | for(i=0; i<mNVar; i++) {
|
---|
| 182 | sprintf(buff,"%s=%s[%d]; ", mNames+i*LENNAME1, nomx, i);
|
---|
| 183 | rets += buff;
|
---|
| 184 | if ( (i%3 == 0) && (i > 0) ) rets += "\n";
|
---|
| 185 | }
|
---|
| 186 | }
|
---|
| 187 |
|
---|
| 188 | return(rets);
|
---|
| 189 | }
|
---|
| 190 |
|
---|
| 191 |
|
---|
| 192 | /* --Methode-- */
|
---|
| 193 | //++
|
---|
| 194 | r_4* NTuple::GetVec(int n, r_4* ret) const
|
---|
| 195 | //
|
---|
| 196 | // Retourne l'evenement `n' dans le vecteur `ret'.
|
---|
| 197 | //--
|
---|
| 198 | {
|
---|
| 199 | int i;
|
---|
| 200 | if (ret == NULL) ret = mVar;
|
---|
| 201 | if (n >= mNEnt) {
|
---|
| 202 | for(i=0; i<mNVar; i++) ret[i] = BADVAL;
|
---|
| 203 | return(ret);
|
---|
| 204 | }
|
---|
| 205 |
|
---|
| 206 | int numb = n/mBlk;
|
---|
| 207 | int offb = n-numb*mBlk;
|
---|
| 208 | memcpy(ret, (mPtr[numb]+offb*mNVar), mNVar*sizeof(r_4));
|
---|
| 209 | return(ret);
|
---|
| 210 | }
|
---|
| 211 |
|
---|
| 212 | /* --Methode-- */
|
---|
| 213 | //++
|
---|
| 214 | void NTuple::GetMinMax(int k, float& min, float& max) const
|
---|
| 215 | //
|
---|
| 216 | // Retourne le minimum et le maximum de la variable `k'.
|
---|
| 217 | //--
|
---|
| 218 | {
|
---|
| 219 | min = 9.e19; max = -9.e19;
|
---|
| 220 | if ( (k < 0) || (k >= mNVar) ) return;
|
---|
| 221 | int jb,ib,i;
|
---|
| 222 | float x;
|
---|
| 223 | i=0;
|
---|
| 224 | for(jb=0; jb< mNBlk; jb++)
|
---|
| 225 | for(ib=0; ib< mBlk; ib++) {
|
---|
| 226 | if (i >= mNEnt) break;
|
---|
| 227 | i++;
|
---|
| 228 | x = *(mPtr[jb]+ib*mNVar+k);
|
---|
| 229 | if(i==1) {min = x; max = x;}
|
---|
| 230 | if (x < min) min = x;
|
---|
| 231 | if (x > max) max = x;
|
---|
| 232 | }
|
---|
| 233 | return;
|
---|
| 234 | }
|
---|
| 235 |
|
---|
| 236 | /* --Methode-- */
|
---|
| 237 | //++
|
---|
| 238 | DVList& NTuple::Info()
|
---|
| 239 | //
|
---|
| 240 | // Renvoie une référence sur l'objet DVList Associé
|
---|
| 241 | //--
|
---|
| 242 | {
|
---|
| 243 | if (mInfo == NULL) mInfo = new DVList;
|
---|
| 244 | return(*mInfo);
|
---|
| 245 | }
|
---|
| 246 |
|
---|
| 247 | /* --Methode-- */
|
---|
| 248 | //++
|
---|
| 249 | void NTuple::Print(int num, int nmax) const
|
---|
| 250 | //
|
---|
| 251 | // Imprime `nmax' evenements a partir du numero `num'.
|
---|
| 252 | //--
|
---|
| 253 | {
|
---|
| 254 | int i,j;
|
---|
| 255 |
|
---|
| 256 | printf("Num ");
|
---|
| 257 | for(i=0; i<mNVar; i++) printf("%8s ", mNames+i*LENNAME1);
|
---|
| 258 | putchar('\n');
|
---|
| 259 |
|
---|
| 260 | if (nmax <= 0) nmax = 1;
|
---|
| 261 | if (num < 0) num = 0;
|
---|
| 262 | nmax += num;
|
---|
| 263 | if (nmax > mNEnt) nmax = mNEnt;
|
---|
| 264 | for(i=num; i<nmax; i++) {
|
---|
| 265 | GetVec(i, NULL);
|
---|
| 266 | printf("%6d ", i);
|
---|
| 267 | for(j=0; j<mNVar; j++) printf("%8g ", (float)mVar[j]);
|
---|
| 268 | putchar('\n');
|
---|
| 269 | }
|
---|
| 270 | return;
|
---|
| 271 | }
|
---|
| 272 |
|
---|
| 273 | /* --Methode-- */
|
---|
| 274 | //++
|
---|
| 275 | void NTuple::Show(ostream& os) const
|
---|
| 276 | //
|
---|
| 277 | // Imprime l'information generale sur le ntuple.
|
---|
| 278 | //--
|
---|
| 279 | {
|
---|
| 280 | os << "NTuple: NVar= " << mNVar << " NEnt=" << mNEnt
|
---|
| 281 | << " (Blk Sz,Nb= " << mBlk << " ," << mNBlk << ")\n";
|
---|
| 282 | os << " Variables Min Max \n";
|
---|
| 283 | int i;
|
---|
| 284 | float min, max;
|
---|
| 285 | char buff[128];
|
---|
| 286 | for(i=0; i<mNVar; i++) {
|
---|
| 287 | GetMinMax(i, min, max);
|
---|
| 288 | sprintf(buff, "%3d %16s %10g %10g \n", i, mNames+i*LENNAME1, min, max);
|
---|
| 289 | os << (string)buff ;
|
---|
| 290 | }
|
---|
| 291 | os << endl;
|
---|
| 292 | }
|
---|
| 293 |
|
---|
| 294 |
|
---|
| 295 | /* --Methode-- */
|
---|
| 296 | //++
|
---|
| 297 | void NTuple::WriteSelf(POutPersist& s) const
|
---|
| 298 | //
|
---|
| 299 | // Ecriture ppersist du ntuple.
|
---|
| 300 | //--
|
---|
| 301 | {
|
---|
| 302 | char strg[256];
|
---|
| 303 | if (mInfo) sprintf(strg, "NVar=%6d NEnt=%9d BlkSz=%6d NBlk=%6d HasInfo",
|
---|
| 304 | (int)mNVar, (int)mNEnt, (int)mBlk, (int)mNBlk);
|
---|
| 305 | else sprintf(strg, "NVar=%6d NEnt=%9d BlkSz=%6d NBlk=%6d ",
|
---|
| 306 | (int)mNVar, (int)mNEnt, (int)mBlk, (int)mNBlk);
|
---|
| 307 | s.PutLine(strg);
|
---|
| 308 | s.PutI4(mNVar);
|
---|
| 309 | s.PutBytes(mNames, mNVar*LENNAME1);
|
---|
| 310 | s.PutI4(mNEnt);
|
---|
| 311 | s.PutI4(mBlk);
|
---|
| 312 | s.PutI4(mNBlk);
|
---|
[278] | 313 | if (mInfo) s << (*mInfo);
|
---|
[220] | 314 | int jb;
|
---|
| 315 | for(jb=0; jb<mNBlk; jb++)
|
---|
| 316 | s.PutR4s(mPtr[jb], mNVar*mBlk);
|
---|
| 317 | return;
|
---|
| 318 | }
|
---|
| 319 |
|
---|
| 320 | /* --Methode-- */
|
---|
| 321 | //++
|
---|
| 322 | void NTuple::ReadSelf(PInPersist& s)
|
---|
| 323 | //
|
---|
| 324 | // Lecture ppersist du ntuple.
|
---|
| 325 | //--
|
---|
| 326 | {
|
---|
| 327 |
|
---|
| 328 | Clean();
|
---|
| 329 |
|
---|
| 330 | char strg[256];
|
---|
| 331 | s.GetLine(strg, 255);
|
---|
| 332 | // Pour savoir s'il y avait un DVList Info associe
|
---|
| 333 | bool hadinfo = false;
|
---|
| 334 | if (strncmp(strg+strlen(strg)-7, "HasInfo", 7) == 0) hadinfo = true;
|
---|
| 335 |
|
---|
| 336 | s.GetI4(mNVar);
|
---|
| 337 | mNames = new char[mNVar*LENNAME1];
|
---|
| 338 | mVar = new r_4[mNVar];
|
---|
| 339 | s.GetBytes(mNames, mNVar*LENNAME1);
|
---|
| 340 | s.GetI4(mNEnt);
|
---|
| 341 | s.GetI4(mBlk);
|
---|
| 342 | s.GetI4(mNBlk);
|
---|
| 343 |
|
---|
| 344 | if (hadinfo) { // Lecture eventuelle du DVList Info
|
---|
| 345 | if (mInfo == NULL) mInfo = new DVList;
|
---|
[278] | 346 | s >> (*mInfo);
|
---|
[220] | 347 | }
|
---|
| 348 |
|
---|
| 349 | int jb;
|
---|
| 350 | for(jb=0; jb<mNBlk; jb++) {
|
---|
| 351 | r_4* pt = new r_4[mNVar*mBlk];
|
---|
| 352 | mPtr.push_back(pt);
|
---|
| 353 | s.GetR4s(mPtr[jb], mNVar*mBlk);
|
---|
| 354 | }
|
---|
| 355 |
|
---|
| 356 | }
|
---|