source: Sophya/trunk/AddOn/TAcq/brproc.h@ 3897

Last change on this file since 3897 was 3888, checked in by ansari, 15 years ago

Introduction correction de gain/reponse filtre et amelioration n-tuple ds BRMeanSpecCalculator, Reza+JEC 24/09/2010

File size: 6.8 KB
Line 
1#ifndef BRPROC_H_SEEN
2#define BRPROC_H_SEEN
3
4//----------------------------------------------------------------
5// Projet BAORadio - (C) LAL/IRFU 2008-2010
6// Classes de threads de traitememt donnees BAORadio
7// Principales classes utilisables
8// - BRMeanSpecCalculator (donnees FFT)
9// - BRFFTCalculator (donnees RAW)
10//----------------------------------------------------------------
11
12
13#include <string>
14#include <vector>
15#include <iostream>
16
17#include "racqumem.h"
18#include "brbaseproc.h"
19#include "parlex.h"
20#include "sopnamsp.h"
21#include "tvector.h"
22#include "ntuple.h"
23#include "datatable.h"
24#include "swfitsdtable.h"
25#include "FFTW/fftw3.h"
26
27using namespace std;
28
29
30
31//---------------------------------------------------------------------
32// Classe de traitement simple - calcul de spectres moyennes / voie
33// a partir de donnees de type FFT (firmware FFT)
34//---------------------------------------------------------------------
35class BRMeanSpecCalculator : public BRBaseProcessor
36{
37public:
38 BRMeanSpecCalculator(RAcqMemZoneMgr& memgr, string outpath, uint_4 nmean=1000,
39 bool fgdatafft=true, bool fgsinglechan=false);
40 virtual ~BRMeanSpecCalculator(); // Fait le nettoyage final , sauvegarde des objets ...
41
42 inline void SetVarianceLimits(double vmin=0., double vmax=9.e99)
43 { varmin_=vmin; varmax_=vmax; }
44
45 void ReadGainFitsFile(string filename);
46
47protected:
48 virtual int Process();
49 virtual void FlagBadPackets();
50 virtual void SaveSpectra();
51
52
53 uint_4 nmean_; // Nombre de spectres pour le calcul des moyennes
54 string outpath_; // directory pour fichiers de sortie
55 bool fgdatafft_;
56 bool fgsinglechannel_;
57 uint_4 numfile_;
58 uint_8 nbpaq4mean_;
59
60 uint_8 totnbpaq_;
61 TMatrix< r_4 > mspecmtx_; // Matrice des spectres moyennees
62 TMatrix< r_4 > sigspecmtx_; // Matrice des sigmas des spectres
63
64 TMatrix< r_4 > sgain_; // gain en fonction de la frequence
65
66 vector<int> clnflg_; // resultat de la routine "cleaner"
67 double varmin_, varmax_; // limites pour la coupure sur la variance
68 vector<uint_8> nbadpaq_;
69
70 FitsInOutFile* ofsdtp_; // Output fits stream for datatable
71 SwFitsDataTable* dtp_; // DataTable
72 double* xnt_;
73};
74
75//-------------------------------------------------------------------------
76// Classe WBRFFT : Calcul de TF sur donnees brutes (firmware RAW)
77// en utilisant FFTW et en effectuant une copie ds tableau local
78//-------------------------------------------------------------------------
79typedef unsigned char IDT ; // definition type de donnees en entree de WBRFFT
80typedef int_2 IIDT ; // definition type de donnees intermediaire
81typedef float ODT; // type de donnees en sortie (complex < ODT > )
82//----Classe WBRFFT----
83class WBRFFT {
84public:
85 WBRFFT(uint_4 sz=0);
86 virtual ~WBRFFT();
87
88 void SetInDataSize(uint_4 sz);
89 void DoFFT(IDT *indata, complex<ODT> * ofc);
90 static void PrintData(IDT *indata, complex<ODT> * ofc, uint_4 sz);
91
92 uint_4 sz_; // Taille tableau d'entree
93 TVector< ODT > inp;
94 TVector< complex<ODT> > outfc;
95 fftwf_plan myplan_;
96 static ZMutex* mtx_fftwp_; // Mutex de protection pour la creation du plan fftw
97};
98//--------------------
99
100//---------------------------------------------------------------------
101// Classe de thread de calcul de FFT sur donnees RAW
102//---------------------------------------------------------------------
103class BRFFTCalculator : public BRBaseProcessor
104{
105public:
106 BRFFTCalculator(RAcqMemZoneMgr& memgr, bool fgsinglechannel=false);
107 virtual ~BRFFTCalculator(); // Fait le nettoyage final , sauvegarde des objets ...
108
109protected:
110 virtual int Process();
111
112 bool fgsinglechannel_;
113 uint_8 totnbfftpaq_;
114 WBRFFT ffts_;
115};
116
117
118//-------------------------------------------------------------------------
119// **** OBSOLETE : BRProcA2C et BRProcB4C
120// Anciennes classes de traitement (Avant Nov2009) utilises par mcrd.cc
121//-------------------------------------------------------------------------
122//---------------------------------------------------------------------
123// Classe thread de traitement avec 2 voies par frame (donnees raw/FFT
124//---------------------------------------------------------------------
125
126/*-- Arguments du constructeur :
127 o mem : gestionnaire zones memoires avec contenant les "frames" avec 2 voies/frame
128 o path : Chemin et nom des fichiers PPF produits (on ajoute _numero.ppf )
129 o nmean : Nombre de frame (paquets) moyennes
130 o nmax : Nombre maxi de frames traites
131 o fghist = true -> fait aussi l'histo des valeurs des echantillons temps
132 o nfsmap>0 : fabrication de cartes (image 2D) des spectres,
133 avec nfsmap=nombre de frequences sommes
134 o fgnotrl = true -> fichiers fits / paquets sans trailer
135 o card : numero de carte , juste pour les impressions
136*/
137class BRProcA2C : public ZThread { // !!!! OBSOLETE !!!!
138public:
139 BRProcA2C(RAcqMemZoneMgr& mem, string& path, bool fgraw,
140 uint_4 nmean=10, uint_4 nmax=100, bool fgtshist=false,
141 uint_4 nfsmap=0, bool fgnotrl=false, int card=1);
142 virtual void run();
143 void Stop();
144 inline void STOP() { stop_ = true; }
145protected:
146 RAcqMemZoneMgr& memgr;
147 bool stop_;
148 bool fgraw_; // true -> raw data
149 uint_4 nmax_; // Nombre maxi de blocs MemZone traites
150 uint_4 nmean_; // Nombre de spectres pour le calcul des moyennes
151 uint_4 nfsmap_; // Binning en frequence pour les cartes 2D temps-freq
152 string path_;
153 bool fghist_; // if true, fill and save histogram with time sample values
154 bool fgnotrl_; // if true, don't check packet trailer - when using pre-june fits files
155 int card_; // Numeros de voies 2*card_-1, 2*card_
156};
157
158//---------------------------------------------------------------------
159// Classe thread de traitement 2 x 2 voies/paquet (Apres BRProcA2C)
160// !!!! OBSOLETE !!!!
161//---------------------------------------------------------------------
162/*-- Arguments du constructeur :
163 o mem1,mem2 : gestionnairea zones memoires avec contenant les donnees des 4 voies (2 voies/frame)
164 o path : Chemin et nom des fichiers PPF produits (on ajoute _numero.ppf )
165 o nmean : Nombre de frame (paquets) moyennes
166 o nmax : Nombre maxi de frames traites
167 o fgnotrl = true -> fichiers fits / paquets sans trailer
168 o card : numero de carte , juste pour les impressions
169*/
170
171class BRProcB4C : public ZThread { // !!!! OBSOLETE !!!!
172
173public:
174 BRProcB4C(RAcqMemZoneMgr& mem1, RAcqMemZoneMgr& mem2, string& path,
175 bool fgraw, uint_4 nmean=10, uint_4 nmax=100, bool fgnotrl=false);
176 virtual void run();
177 void Stop();
178 inline void STOP() { stop_ = true; }
179protected:
180 RAcqMemZoneMgr& memgr1;
181 RAcqMemZoneMgr& memgr2;
182 bool stop_;
183 bool fgraw_; // true -> raw data
184 uint_4 nmax_; // Nombre maxi de blocs MemZone traites
185 uint_4 nmean_; // Nombre de blocs pour le calcul des moyennes
186 string path_;
187 bool fgnotrl_; // if true, don't check packet trailer - when using pre-june fits files
188};
189
190
191
192#endif
Note: See TracBrowser for help on using the repository browser.