| 1 | #include <stdlib.h> | 
|---|
| 2 | #include <string.h> | 
|---|
| 3 |  | 
|---|
| 4 | #include "branap.h" | 
|---|
| 5 | #include "minifits.h" | 
|---|
| 6 | #include "strutilxx.h" | 
|---|
| 7 | #include "sopnamsp.h" | 
|---|
| 8 |  | 
|---|
| 9 | //-------------------------------------------------------------- | 
|---|
| 10 | // Projet BAORadio - (C) LAL/IRFU  2008-2010 | 
|---|
| 11 | // Classe de gestion des parametres programmes d'analyse | 
|---|
| 12 | //-------------------------------------------------------------- | 
|---|
| 13 |  | 
|---|
| 14 | /* --Methode-- */ | 
|---|
| 15 | BRAnaParam::BRAnaParam(uint_4 nmean, uint_4 nzon, uint_4 npaqz) | 
|---|
| 16 | { | 
|---|
| 17 | outpath_="./"; | 
|---|
| 18 | nmean_=nmean; | 
|---|
| 19 | nbloc_=1; | 
|---|
| 20 | imin_=imax_=0; | 
|---|
| 21 | istep_=1; | 
|---|
| 22 | rdsamefc_=true;   // read paquets with same frame counter | 
|---|
| 23 | freqmin_=freqmax_=0; | 
|---|
| 24 | nbinfreq_=1; | 
|---|
| 25 | paqsize_=16424; | 
|---|
| 26 | nzones_=nzon; | 
|---|
| 27 | npaqinzone_=npaqz; | 
|---|
| 28 | fgdatafft_=false;  fgsinglechannel_=false; fgforcesinglechan_=false; | 
|---|
| 29 | prtlevel_=0; | 
|---|
| 30 | prtmodulo_=50000; | 
|---|
| 31 | nbcalgrp_=1; | 
|---|
| 32 | nthreads_=1; | 
|---|
| 33 |  | 
|---|
| 34 | spec_win_sz_=0; | 
|---|
| 35 | spw_ext_sz_=0; | 
|---|
| 36 | nbmax_specwfiles_=0; | 
|---|
| 37 |  | 
|---|
| 38 | vmin_=0.;  vmax_=9e99; | 
|---|
| 39 | nbands_=0; bandfirst_ = bandlast_ = 0; | 
|---|
| 40 | fgdt_ = false; | 
|---|
| 41 |  | 
|---|
| 42 | fgfreqfilter_=false; //JEC 1/2/11 | 
|---|
| 43 |  | 
|---|
| 44 | gainfile_=""; | 
|---|
| 45 | } | 
|---|
| 46 |  | 
|---|
| 47 | /* --Methode-- */ | 
|---|
| 48 | int BRAnaParam::DecodeArgs(int narg, char* arg[]) | 
|---|
| 49 | { | 
|---|
| 50 | if ((narg>1)&&(strcmp(arg[1],"-h")==0))  return Usage(false); | 
|---|
| 51 | if (narg<5) return Usage(true); | 
|---|
| 52 |  | 
|---|
| 53 | bool okarg=false; | 
|---|
| 54 | int ka=1; | 
|---|
| 55 | while (ka<(narg-1)) { | 
|---|
| 56 | if (strcmp(arg[ka],"-act")==0) { | 
|---|
| 57 | action_=arg[ka+1]; | 
|---|
| 58 | ka+=2; | 
|---|
| 59 | } | 
|---|
| 60 | else if (strcmp(arg[ka],"-out")==0) { | 
|---|
| 61 | outpath_=arg[ka+1]; | 
|---|
| 62 | size_t lenp=outpath_.size(); | 
|---|
| 63 | if ((lenp>0)&&(outpath_[lenp-1]!='/'))  outpath_+='/'; | 
|---|
| 64 | ka+=2; | 
|---|
| 65 | } | 
|---|
| 66 | else if (strcmp(arg[ka],"-nmean")==0) { | 
|---|
| 67 | nmean_=atoi(arg[ka+1]); | 
|---|
| 68 | ka+=2; | 
|---|
| 69 | } | 
|---|
| 70 | else if (strcmp(arg[ka],"-nbloc")==0) { | 
|---|
| 71 | nbloc_=atoi(arg[ka+1]); | 
|---|
| 72 | ka+=2; | 
|---|
| 73 | } | 
|---|
| 74 | else if (strcmp(arg[ka],"-freq")==0) { | 
|---|
| 75 | sscanf(arg[ka+1],"%d,%d,%d",&freqmin_,&freqmax_,&nbinfreq_); | 
|---|
| 76 | ka+=2; | 
|---|
| 77 | } | 
|---|
| 78 | else if (strcmp(arg[ka],"-zones")==0) { | 
|---|
| 79 | int nzon=4; | 
|---|
| 80 | int npaqz=128; | 
|---|
| 81 | sscanf(arg[ka+1],"%d,%d",&nzon,&npaqz); | 
|---|
| 82 | nzones_=nzon;  npaqinzone_=npaqz; | 
|---|
| 83 | ka+=2; | 
|---|
| 84 | } | 
|---|
| 85 | else if (strcmp(arg[ka],"-prt")==0) { | 
|---|
| 86 | sscanf(arg[ka+1],"%d,%ld",&prtlevel_,&prtmodulo_); | 
|---|
| 87 | ka+=2; | 
|---|
| 88 | } | 
|---|
| 89 | else if (strcmp(arg[ka],"-nthr")==0) { | 
|---|
| 90 | nthreads_=atoi(arg[ka+1]); | 
|---|
| 91 | ka+=2; | 
|---|
| 92 | } | 
|---|
| 93 | else if (strcmp(arg[ka],"-nvcal")==0) { | 
|---|
| 94 | nbcalgrp_=atoi(arg[ka+1]); | 
|---|
| 95 | ka+=2; | 
|---|
| 96 | } | 
|---|
| 97 | else if (strcmp(arg[ka],"-nosfc")==0) { | 
|---|
| 98 | rdsamefc_=false; | 
|---|
| 99 | ka++; | 
|---|
| 100 | } | 
|---|
| 101 | else if (strcmp(arg[ka],"-singlechan")==0) { | 
|---|
| 102 | fgsinglechannel_=true; fgforcesinglechan_ = true; | 
|---|
| 103 | ka++; | 
|---|
| 104 | } | 
|---|
| 105 | else if (strcmp(arg[ka],"-varcut")==0) { | 
|---|
| 106 | sscanf(arg[ka+1],"%lg,%lg",&vmin_,&vmax_); | 
|---|
| 107 | ka+=2; | 
|---|
| 108 | } | 
|---|
| 109 | else if (strcmp(arg[ka],"-nband")==0) { | 
|---|
| 110 | sscanf(arg[ka+1],"%d,%d,%d",&nbands_,&bandfirst_,&bandlast_); | 
|---|
| 111 | ka+=2; | 
|---|
| 112 | } | 
|---|
| 113 | else if (strcmp(arg[ka],"-filldt")==0) { | 
|---|
| 114 | fgdt_=true; | 
|---|
| 115 | ka++; | 
|---|
| 116 | } | 
|---|
| 117 | else if (strcmp(arg[ka],"-freqfilter")==0) { | 
|---|
| 118 | fgfreqfilter_=true; | 
|---|
| 119 | ka++; | 
|---|
| 120 | } | 
|---|
| 121 | else if (strcmp(arg[ka],"-gain")==0) { | 
|---|
| 122 | gainfile_=arg[ka+1]; | 
|---|
| 123 | ka+=2; | 
|---|
| 124 | } | 
|---|
| 125 | else if (strcmp(arg[ka],"-tspwin")==0) { | 
|---|
| 126 | int ai1=0,ai2=0,ai3=0; | 
|---|
| 127 | sscanf(arg[ka+1],"%d,%d,%d",&ai1,&ai2,&ai3); | 
|---|
| 128 | spec_win_sz_=ai1;  spw_ext_sz_=ai2;   nbmax_specwfiles_=ai3; | 
|---|
| 129 | ka+=2; | 
|---|
| 130 | } | 
|---|
| 131 | else if (strcmp(arg[ka],"-in")==0) { | 
|---|
| 132 | if ((narg-ka)<4)  { | 
|---|
| 133 | cout << " BRAnaParam::DecodeArgs() / Argument error " << endl; | 
|---|
| 134 | return Usage(true); | 
|---|
| 135 | } | 
|---|
| 136 | sscanf(arg[ka+1],"%d,%d,%d",&imin_,&imax_,&istep_);  ka+=2; | 
|---|
| 137 | while(ka<(narg-1)) { | 
|---|
| 138 | string inpath = arg[ka]; | 
|---|
| 139 | size_t lenp=inpath.size(); | 
|---|
| 140 | if (lenp<1)  inpath="./"; | 
|---|
| 141 | if ((lenp>0)&&(inpath[lenp-1]!='/'))  inpath+='/'; | 
|---|
| 142 | vector<string> fiblist; | 
|---|
| 143 | string sa1 = arg[ka+1]; | 
|---|
| 144 | FillVStringFrString(sa1, fiblist, ','); | 
|---|
| 145 | char dbuff[32]; | 
|---|
| 146 | for(size_t i=0; i<fiblist.size(); i++) { | 
|---|
| 147 | sprintf(dbuff,"Fiber%d/",(int)atoi(fiblist[i].c_str())); | 
|---|
| 148 | dirlist_.push_back(inpath+dbuff); | 
|---|
| 149 | } | 
|---|
| 150 | ka += 2; | 
|---|
| 151 | } | 
|---|
| 152 | okarg=true; | 
|---|
| 153 | } | 
|---|
| 154 | else ka++; | 
|---|
| 155 | } | 
|---|
| 156 |  | 
|---|
| 157 | if (!okarg) { | 
|---|
| 158 | cout << " BRAnaParam::DecodeArgs() / Argument error " << endl; | 
|---|
| 159 | return Usage(true); | 
|---|
| 160 | } | 
|---|
| 161 | return 0; | 
|---|
| 162 | } | 
|---|
| 163 |  | 
|---|
| 164 | /* --Methode-- */ | 
|---|
| 165 | int BRAnaParam::Usage(bool fgshort) | 
|---|
| 166 | { | 
|---|
| 167 | cout << " --- BRAnaParam : Reading/Processing BAORadio FITS files parameters " << endl; | 
|---|
| 168 | cout << " Usage:  prgname [-act ACT] [-out OutPath] [-nmean NMean] [-zones NZones,nPaqinZone] \n" | 
|---|
| 169 | << "                 [-nbloc NBloc] [-freq NumFreqMin,NumFreqMax,NBinFreq] \n" | 
|---|
| 170 | << "                 [-prt lev,modulo] [-nvcal n] [-nthr n] [-nosfc] [-singlechan]\n" | 
|---|
| 171 | << "                 [-freqfilter]\n" | 
|---|
| 172 | << "                 [-gain filename] [-varcut min,max] [-nband nband,first,last] \n" | 
|---|
| 173 | << "                 [-filldt] [-tspwin wsz,extsz,nfiles] \n" | 
|---|
| 174 | << "         -in Imin,Imax,Istep InPath FiberList [InPath2 FiberList2 InPath3 FiberList3 ...] \n" << endl; | 
|---|
| 175 | if (fgshort) { | 
|---|
| 176 | cout << " prgname -h for detailed instructions" << endl; | 
|---|
| 177 | return 5; | 
|---|
| 178 | } | 
|---|
| 179 | cout << " -act Action: cube3d or vis or viscktt or mspec \n" | 
|---|
| 180 | << "    cube3d: create 3D fits cubes, vis: compute visibilites,  \n" | 
|---|
| 181 | << "    viscktt: compute visibilities and check TimeTag/FrameCounter\n " | 
|---|
| 182 | << "    mspec: compute and save mean spectra for each channel \n " | 
|---|
| 183 | << "    bproc: run BRBaseProcessor for debug/printing (use -prt)  \n " | 
|---|
| 184 | << " -out OutPath: Output directory name \n" | 
|---|
| 185 | << " -nmean NMean: Number of packet used for spectra/visibility computation \n" | 
|---|
| 186 | << " -nbloc NBloc: Number of MemMgr blocs in output file\n" | 
|---|
| 187 | << " -zones NZones,NbPaqinZone : Number of Zones and number of paquets in one zone \n" | 
|---|
| 188 | << " -freq NumFreqMin,NumFreqMax,NBinFreq : Frequency zone and number of bins \n" | 
|---|
| 189 | << " -prt lev,modulo : Print level (0,1,2...) and print counter modulo (10000, 50000 ...) \n" | 
|---|
| 190 | << " -nvcal n : number of BRVisibilityCalculator objects running in parallel in BRVisCalcGroup  (default=1) \n" | 
|---|
| 191 | << " -nthr n : number of threads for parallel execution in BRVisibilityCalculator (default=1) \n" | 
|---|
| 192 | << " -nosfc : Don't force reading with SAME FrameCounter \n" | 
|---|
| 193 | << " -singlechan : Force one channel per fiber \n" | 
|---|
| 194 | << " -varcut min,max : min-max cut on variance \n" | 
|---|
| 195 | << " -nband nband,first,last: numbers of freq. bands and first and last bands used for cuts \n" | 
|---|
| 196 | << " -filldt : force data table filling \n" | 
|---|
| 197 | << " -freqfilter : force median filtering on the frequencies \n" | 
|---|
| 198 | << " -gain filename : spectral response fits file name \n" | 
|---|
| 199 | << " -tspwin wsz,extsz,nfiles : spectra time (paquet no) window (ex: -tspwin 120,4,5) \n" | 
|---|
| 200 | << "    wsz,extsz= window,extension size;  nfiles= maximum number of windows saved \n" | 
|---|
| 201 | << " -in : input files/directory definition : \n" | 
|---|
| 202 | << "       Imin,Imax,Istep: fits files signalII.fits Imin<=II<=Imax Istep=increment \n" | 
|---|
| 203 | << "       InPath: Input directerory fits files in InPath/FiberJJ directory \n" | 
|---|
| 204 | << "       FiberList: List of fiber numbers (JJ - Ex: 2,1,4 )  \n" << endl; | 
|---|
| 205 | return 1; | 
|---|
| 206 | } | 
|---|
| 207 |  | 
|---|
| 208 | /* --Fonction-- */ | 
|---|
| 209 | int BRAnaParam::PaqSizeFromFits() | 
|---|
| 210 | { | 
|---|
| 211 | uint_4 paqsz, npaqf; | 
|---|
| 212 | char flnm[1024]; | 
|---|
| 213 | sprintf(flnm,"%s/signal%d.fits",dirlist_[0].c_str(),imin_); | 
|---|
| 214 | int rc = DecodeMiniFitsHeader(flnm,paqsize_, npaqinfile_,fgdatafft_, fgsinglechannel_); | 
|---|
| 215 | if(fgforcesinglechan_) fgsinglechannel_= true; | 
|---|
| 216 | return rc; | 
|---|
| 217 | } | 
|---|
| 218 |  | 
|---|
| 219 | /* --Fonction-- */ | 
|---|
| 220 | ostream& BRAnaParam::Print(ostream& os) | 
|---|
| 221 | { | 
|---|
| 222 | os << " BRAnaParam::Print() dirlist_.size()=" << dirlist_.size() << " Input directories: " << endl; | 
|---|
| 223 | for(size_t k=0; k< dirlist_.size(); k++) | 
|---|
| 224 | cout << k+1 << " :  " << dirlist_[k] << endl; | 
|---|
| 225 | cout << " IMin= " << imin_ << " IMax= " << imax_ << " IStep= " << istep_ | 
|---|
| 226 | << ((rdsamefc_)?" SameFrameCounter read mode":" AllOKPaquets read mode ") << endl; | 
|---|
| 227 | cout << " OutPath= " << outpath_ << endl; | 
|---|
| 228 | cout << " Action=" << action_ << "  NMean=" << nmean_ << " NBloc=" << nbloc_ << endl; | 
|---|
| 229 | cout << " FreqMin= " << freqmin_ << " FreqMax= " << freqmax_ << " NBinFreq= " << nbinfreq_ << endl; | 
|---|
| 230 | cout<<  ((fgdt_)?" Fill DadaTable ":" NO DataTable") << endl; | 
|---|
| 231 | cout << " GainFileName=" << gainfile_ << " Variance: Min= " << vmin_ << " Max= " << vmax_ | 
|---|
| 232 | << " Bands: N=" <<  nbands_ << " First=" << bandfirst_ << " Last=" << bandlast_ | 
|---|
| 233 | << endl; | 
|---|
| 234 | cout << " force frequence median filtering (action gain)" << endl; | 
|---|
| 235 | cout << " Spectra TimeWindow (Nb.Paquets) Size=" << spec_win_sz_ << " ExtensionSize=" << spw_ext_sz_ | 
|---|
| 236 | << " MaxNbFile=" << nbmax_specwfiles_ << endl; | 
|---|
| 237 | cout << " PaqSize=" << paqsize_ << "  -  NZones=" << nzones_ << " NPaqZone=" << npaqinzone_ | 
|---|
| 238 | << " PrtLevel=" << prtlevel_ << " PrtCntModulo=" <<  prtmodulo_ << endl; | 
|---|
| 239 | cout << " AcqMode: " << ((fgdatafft_)?" Data_FFT " : " Data_Raw " ) | 
|---|
| 240 | << ((fgsinglechannel_)?" SingleChannel " : " TwoChannels " ) << endl; | 
|---|
| 241 | cout << " NbVisibCalculator in Group: " <<  nbcalgrp_ << " with N//threads: " << nthreads_ << endl; | 
|---|
| 242 |  | 
|---|
| 243 |  | 
|---|
| 244 | return os; | 
|---|
| 245 | } | 
|---|
| 246 |  | 
|---|
| 247 | /* --Fonction-- */ | 
|---|
| 248 | int BRAnaParam::DecodeMiniFitsHeader(const char* filename, uint_4& paqsz, uint_4& npaq, | 
|---|
| 249 | bool& fgdatafft, bool& fgsinglechannel) | 
|---|
| 250 | { | 
|---|
| 251 | cout << " DecodeMiniFitsHeader - Opening file: " << filename << endl; | 
|---|
| 252 | MiniFITSFile mff(filename, MF_Read); | 
|---|
| 253 | string acqmode=mff.GetKeyValue("ACQMODE"); | 
|---|
| 254 | cout << "DecodeMiniFitsHeader()... Type=" << mff.DataTypeToString() << " NAxis1=" << mff.NAxis1() | 
|---|
| 255 | << " NAxis2=" << mff.NAxis2() << " AcqMode=" << acqmode << endl; | 
|---|
| 256 | paqsz = mff.NAxis1(); | 
|---|
| 257 | npaq = mff.NAxis2(); | 
|---|
| 258 | if (acqmode.substr(0,3)=="fft") fgdatafft=true; | 
|---|
| 259 | if (acqmode.find("1c") < acqmode.length()) fgsinglechannel=true; | 
|---|
| 260 | return 0; | 
|---|
| 261 | } | 
|---|
| 262 |  | 
|---|