source: trunk/src/Cubes/cubes.hh @ 931

Last change on this file since 931 was 923, checked in by MatthewWhiting, 12 years ago

Removing un-used prototypes and inline functions - these are all in FitsHeader?

File size: 27.9 KB
Line 
1// -----------------------------------------------------------------------
2// cubes.hh: Definitions of the DataArray, Cube and Image classes.
3// -----------------------------------------------------------------------
4// Copyright (C) 2006, Matthew Whiting, ATNF
5//
6// This program is free software; you can redistribute it and/or modify it
7// under the terms of the GNU General Public License as published by the
8// Free Software Foundation; either version 2 of the License, or (at your
9// option) any later version.
10//
11// Duchamp is distributed in the hope that it will be useful, but WITHOUT
12// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13// FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14// for more details.
15//
16// You should have received a copy of the GNU General Public License
17// along with Duchamp; if not, write to the Free Software Foundation,
18// Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA
19//
20// Correspondence concerning Duchamp may be directed to:
21//    Internet email: Matthew.Whiting [at] atnf.csiro.au
22//    Postal address: Dr. Matthew Whiting
23//                    Australia Telescope National Facility, CSIRO
24//                    PO Box 76
25//                    Epping NSW 1710
26//                    AUSTRALIA
27// -----------------------------------------------------------------------
28#ifndef CUBES_H
29#define CUBES_H
30
31#include <iostream>
32#include <string>
33#include <vector>
34
35#include <duchamp/duchamp.hh>
36#include <duchamp/param.hh>
37#include <duchamp/fitsHeader.hh>
38#include <duchamp/Detection/detection.hh>
39#include <duchamp/Detection/columns.hh>
40#include <duchamp/Cubes/plots.hh>
41#include <duchamp/Utils/Statistics.hh>
42#include <duchamp/PixelMap/Scan.hh>
43#include <duchamp/PixelMap/Object2D.hh>
44
45
46namespace duchamp
47{
48
49
50  /// @brief Search a reconstructed array for significant detections.
51  std::vector <Detection> searchReconArray(size_t *dim, float *originalArray, float *reconArray,
52                                           Param &par, Statistics::StatsContainer<float> &stats);
53  std::vector <Detection> searchReconArraySpectral(size_t *dim, float *originalArray, float *reconArray,
54                                                   Param &par, Statistics::StatsContainer<float> &stats);
55  std::vector <Detection> searchReconArraySpatial(size_t *dim, float *originalArray, float *reconArray,
56                                                  Param &par, Statistics::StatsContainer<float> &stats);
57
58  /// @brief Search a 3-dimensional array for significant detections.
59  std::vector <Detection> search3DArray(size_t *dim, float *Array, Param &par,
60                                        Statistics::StatsContainer<float> &stats);
61  std::vector <Detection> search3DArraySpectral(size_t *dim, float *Array, Param &par,
62                                                Statistics::StatsContainer<float> &stats);
63  std::vector <Detection> search3DArraySpatial(size_t *dim, float *Array, Param &par,
64                                               Statistics::StatsContainer<float> &stats);
65
66  //=========================================================================
67
68  /// An enumeration allowing us to refer to a particular array within functions
69  enum ARRAYREF {ARRAY=0, RECON, BASELINE};
70
71
72  /// @brief Base class for the image container.
73  ///
74  /// @details Definition of an n-dimensional data array: array of
75  ///     pixel values, size & dimensions array of Detection objects
76
77  class DataArray
78  {
79  public:
80    DataArray();  ///< Basic DataArray constructor
81    DataArray(short int nDim); ///< Basic nDim-dimensional DataArray constructor
82    DataArray(short int nDim, size_t size);///< Basic nDim-dimensional DataArray constructor, specifying size.
83    DataArray(short int nDim, size_t *dimensions); ///< Basic nDim-dimensional DataArray constructor, specifying size of dimensions.
84    virtual ~DataArray(); ///< Basic DataArray constructor
85    DataArray(const DataArray &d);
86    DataArray& operator=(const DataArray &d);
87
88    //-----------------------------------------
89    // Obvious inline accessor functions.
90    //
91    size_t             getDimX(){if(numDim>0) return axisDim[0];else return 0;};
92    size_t             getDimY(){if(numDim>1) return axisDim[1];else return 1;};
93    size_t             getDimZ(){if(numDim>2) return axisDim[2];else return 1;};
94    size_t             getSize(){ return numPixels; };
95    short int          getNumDim(){ return numDim; };
96    virtual float      getPixValue(size_t pos){ return array[pos]; };
97    virtual void       setPixValue(size_t pos, float f){array[pos] = f;};
98    Detection          getObject(size_t number){ return objectList->at(number); };
99    Detection *        pObject(size_t number){ return &(objectList->at(number));};
100    std::vector <Detection> getObjectList(){ return *objectList; };
101    std::vector <Detection> *pObjectList(){ return objectList; };
102    std::vector <Detection> &ObjectList(){ std::vector<Detection> &rlist = *objectList; return rlist; };
103    size_t              getNumObj(){ return objectList->size(); };
104
105    /// @brief Delete all objects from the list of detections.
106    void               clearDetectionList(){
107      //objectList->clear();
108      delete objectList;
109      objectList = new std::vector<Detection>;
110    };
111
112    /// @brief Read a parameter set from file.
113    OUTCOME readParam(std::string paramfile){
114      /// @brief
115      ///  Uses Param::readParams() to read parameters from a file.
116      ///  \param paramfile The file to be read.
117       
118      return par.readParams(paramfile);
119    };
120
121    //-----------------------------------------
122    // Related to the various arrays
123    //
124    /// @brief  Return first three dimensional axes.
125    void               getDim(size_t &x, size_t &y, size_t &z);
126    /// @brief Return array of dimensional sizes.
127    void               getDimArray(size_t *output);
128    size_t *           getDimArray(){return axisDim;};
129
130    /// @brief Return pixel value array.
131    void               getArray(float *output);
132    float *            getArray(){return array;};
133
134    /// @brief Save pixel value array.
135    virtual void       saveArray(float *input, size_t size);
136
137    //-----------------------------------------
138    // Related to the object lists
139    //
140    /// @brief Adds a single detection to the object list.
141    void               addObject(Detection object);
142
143    /// @brief Adds all objects in a detection list to the object list.
144    void               addObjectList(std::vector <Detection> newlist);   
145
146    /// @brief Add pixel offsets to object coordinates.
147    void               addObjectOffsets();
148
149    //-----------------------------------------
150    // Parameter list related.
151    //
152    /// @brief Output the Param set.
153    void               showParam(std::ostream &stream){ stream << par; };
154    /// @brief Return the Param set.
155    Param              getParam(){ return par; };
156    /// @brief Save a Param set to the Cube.
157    void               saveParam(Param &newpar){par = newpar;};
158    /// @brief Provides a reference to the Param set.
159    Param&             pars(){ Param &rpar = par; return rpar; };
160    /// @brief Is the voxel number given by vox a BLANK value?
161    bool               isBlank(int vox){return par.isBlank(array[vox]);};
162
163    // --------------------------------------------
164    // Statistics
165    //
166    /// @brief  Returns the StatsContainer.
167    Statistics::StatsContainer<float> getStats(){ return Stats; };
168
169    /// @brief Provides a reference to the StatsContainer.
170    Statistics::StatsContainer<float>& stats(){
171      Statistics::StatsContainer<float> &rstats = Stats;  return rstats;
172    };
173 
174    /// @brief Save a StatsContainer to the Cube.
175    void saveStats(Statistics::StatsContainer<float> newStats){ Stats = newStats;};
176
177    /// @brief A detection test for value.
178    bool isDetection(float value);
179
180    /// @brief  A detection test for pixel. 
181    bool isDetection(size_t voxel);
182
183
184    /// @brief Output operator for DataArray.
185    friend std::ostream& operator<< (std::ostream& theStream, DataArray &array);
186 
187
188  protected:
189    short int                numDim;     ///< Number of dimensions.
190    size_t                  *axisDim;    ///< Array of axis dimensions of cube
191    bool                     axisDimAllocated; ///< has axisDim been allocated?
192    size_t                   numPixels;  ///< Total number of pixels in cube.
193    float                   *array;      ///< Array of data.
194    bool                     arrayAllocated; ///< has the array been allocated?
195    std::vector <Detection> *objectList; ///< The list of detected objects.
196    Param                    par;        ///< A parameter list.
197    Statistics::StatsContainer<float> Stats; ///< The statistics for the DataArray.
198  };
199
200
201
202  //=========================================================================
203
204  /// @brief Definition of an data-cube object (3D):
205  ///     a DataArray object limited to dim=3
206
207  class Cube : public DataArray
208  {
209  public:
210    Cube();                 ///< Basic Cube constructor.
211    Cube(size_t nPix);        ///< Alternative Cube constructor.
212    Cube(size_t *dimensions); ///< Alternative Cube constructor.
213    virtual ~Cube();        ///< Basic Cube destructor.
214    Cube(const Cube &c);    ///< Copy constructor
215    Cube& operator=(const Cube &c); ///< Copy operator
216
217    /// @brief Return a Cube holding only a subsection of the original
218    Cube* slice(Section subsection);
219   
220
221
222    bool        is2D();
223    void        checkDim(){head.set2D(is2D());};
224    void        reportMemorySize(std::ostream &theStream, bool allocateArrays);
225
226    // INLINE functions -- definitions included after class declaration.
227    /// @brief Is the voxel number given by vox a BLANK value?
228    bool        isBlank(size_t vox){ return par.isBlank(array[vox]); };
229
230    /// @brief Is the voxel at (x,y,z) a BLANK value?
231    bool        isBlank(size_t x, size_t y, size_t z){
232      return par.isBlank(array[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x]); };
233
234    /// @brief Return a bool array masking blank pixels: 1=good, 0=blank
235    bool *      makeBlankMask(){return par.makeBlankMask(array, numPixels);};
236
237    /// @brief Does the Cube::recon array exist?
238    bool        isRecon(){ return reconExists; };
239
240    float       getPixValue(size_t pos){ return array[pos]; };
241    float       getPixValue(size_t x, size_t y, size_t z){
242      return array[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x]; };
243    short       getDetectMapValue(size_t pos){ return detectMap[pos]; };
244    short       getDetectMapValue(size_t x, size_t y){ return detectMap[y*axisDim[0]+x]; };
245    float       getReconValue(size_t pos){ return recon[pos]; };
246    float       getReconValue(size_t x, size_t y, size_t z){
247      return recon[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x];};
248    float       getBaselineValue(size_t pos){ return baseline[pos]; };
249    float       getBaselineValue(size_t x, size_t y, size_t z){
250      return baseline[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x]; };
251    void        setPixValue(size_t pos, float f){array[pos] = f;};
252    void        setPixValue(size_t x, size_t y, size_t z, float f){
253      array[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x] = f;};
254    void        setDetectMapValue(size_t pos, short f){ detectMap[pos] = f;};
255    void        setDetectMapValue(size_t x, size_t y, short f){ detectMap[y*axisDim[0] + x] = f;};
256    void        incrementDetectMap(size_t x, size_t y){detectMap[y*axisDim[0]+x]++;};
257    void        incrementDetectMap(size_t pos){detectMap[pos]++;};
258    void        setReconValue(size_t pos, float f){recon[pos] = f;};
259    void        setReconValue(size_t x, size_t y, size_t z, float f){
260      recon[z*axisDim[0]*axisDim[1] + y*axisDim[0] + x] = f; };
261    void        setReconFlag(bool f){reconExists = f;};
262
263    std::vector<Column::Col> getLogCols(){return logCols;};
264    void        setLogCols(std::vector<Column::Col> C){logCols=C;};
265    std::vector<Column::Col> getFullCols(){return fullCols;};
266    void        setFullCols(std::vector<Column::Col> C){fullCols=C;};
267
268    // additional functions -- in Cubes/cubes.cc
269    /// @brief Allocate memory correctly, with WCS defining the correct axes.
270    OUTCOME     initialiseCube(size_t *dimensions, bool allocateArrays=true);
271    OUTCOME     initialiseCube(long *dimensions, bool allocateArrays=true);
272
273    /// @brief Read in a FITS file, with subsection correction.
274    OUTCOME     getCube();
275    /// @brief Read in a FITS file, with subsection correction.
276    OUTCOME     getMetadata();
277
278    /// @brief Read in command-line options.
279    //   int         getopts(int argc, char ** argv);
280    int         getopts(int argc, char ** argv, std::string progname="Duchamp"){return par.getopts(argc,argv,progname);};
281
282    /// @brief Read in reconstructed & smoothed arrays from FITS files on disk.
283    void        readSavedArrays();
284
285    /// @brief Save an external array to the Cube's pixel array
286    void        saveArray(float *input, size_t size);
287
288    /// @brief Save an external array to the Cube's pixel array
289    void        saveArray(std::vector<float> &input);
290
291    /// @brief Save an external array to the Cube's reconstructed array.
292    void        saveRecon(float *input, size_t size);
293
294    /// @brief Save reconstructed array to an external array.
295    void        getRecon(float *output);
296    float *     getRecon(){return recon; };
297
298    /// @brief Set Milky Way channels to zero.
299    void        removeMW();
300
301    //------------------------
302    // Statistics for cube
303    //
304
305    /// @brief Calculate the statistics for the Cube (older version).
306    void        setCubeStatsOld();
307
308    /// @brief Calculate the statistics for the Cube.
309    void        setCubeStats();
310
311    /// @brief Set up thresholds for the False Discovery Rate routine.
312    void        setupFDR();
313    /// @brief Set up thresholds for the False Discovery Rate routine using a particular array.
314    void        setupFDR(float *input);
315
316    /// @brief A detection test for a given voxel.
317    bool        isDetection(size_t x, size_t y, size_t z);
318
319    //-----------------------------
320    // Dealing with the detections
321    //
322   
323    /// @brief Get the set of voxels pertaining to the detected objects.
324    std::vector< std::vector<PixelInfo::Voxel> > getObjVoxList();
325
326    /// @brief Calculate the object fluxes
327    void        calcObjectFluxes();
328
329    /// @brief Calculate the WCS parameters for each Cube Detection.
330    void        calcObjectWCSparams();
331    /// @brief Calculate the WCS parameters for each Cube Detection, using flux information in Voxels.
332    void        calcObjectWCSparams(std::vector< std::vector<PixelInfo::Voxel> > bigVoxList);
333    void        calcObjectWCSparams(std::map<PixelInfo::Voxel,float> &voxelMap);
334 
335    /// @brief Sort the list of detections.
336    void        sortDetections();
337
338    /// @brief Update the map of detected pixels.
339    void        updateDetectMap();
340
341    /// @brief Update the map of detected pixels for a given Detection.
342    void        updateDetectMap(Detection obj);
343
344    /// @brief Clear the map of detected pixels.
345    void        clearDetectMap(){
346      for(size_t i=0;i<axisDim[0]*axisDim[1];i++) detectMap[i]=0;
347    };
348
349    /// @brief Find the flux enclosed by a Detection.
350    float       enclosedFlux(Detection obj);
351
352    /// @brief Set up the output column definitions for the Cube and its Detection list.
353    void        setupColumns();
354
355    /// @brief Is the object at the edge of the image?
356    bool        objAtSpatialEdge(Detection obj);
357
358    /// @brief Is the object at an end of the spectrum?
359    bool        objAtSpectralEdge(Detection obj);
360
361    /// @brief Set warning flags for the detections.
362    void        setObjectFlags();
363
364    // ----------------------------
365    // Dealing with the WCS
366    //
367    /// @brief Return the FitsHeader object.
368    FitsHeader  getHead(){ return head; };
369    /// @brief Define the FitsHeader object.
370    void        setHead(FitsHeader F){head = F;};
371    /// @brief Provide a reference to the FitsHeader object.
372    FitsHeader& header(){ FitsHeader &h = head; return h; };
373
374    //-------------------------------------------
375    // FITS-I/O related functions -- not in cubes.cc
376    //
377    /// @brief Function to read in FITS file.
378    OUTCOME     getMetadata(std::string fname);  // in Cubes/getImage.cc
379    OUTCOME     getCube(std::string fname);  // in Cubes/getImage.cc
380
381    /// @brief Convert the flux units to something user-specified.
382    void        convertFluxUnits(std::string oldUnit, std::string newUnit, ARRAYREF whichArray=ARRAY); // in Cubes/getImage.cc
383
384    /// @brief Function to retrieve FITS data array
385    OUTCOME         getFITSdata(std::string fname);   // in FitsIO/dataIO.cc
386
387    OUTCOME         writeBasicHeader(fitsfile *fptr, int bitpix, bool is2D=false);
388
389    /// @brief Save the moment map to a FITS file
390    OUTCOME        saveMomentMapImage();
391
392    /// @brief Save a mask array to disk.
393    OUTCOME        saveMaskCube();       // in Cubes/saveImage.cc
394
395    /// @brief Save Smoothed array to disk.
396    OUTCOME        saveSmoothedCube();       // in Cubes/saveImage.cc
397
398    /// @brief Save Reconstructed array to disk.
399    OUTCOME        saveReconstructedCube();  // in Cubes/saveImage.cc
400
401    /// @brief Read in reconstructed array from FITS file.
402    OUTCOME        readReconCube();  // in Cubes/readRecon.cc
403 
404    /// @brief Read in Hanning-smoothed array from FITS file.
405    OUTCOME        readSmoothCube();     // in Cubes/readSmooth.cc 
406
407    //-------------------------------------
408    // Functions that act on the cube
409    //
410
411    /// @brief Remove excess BLANK pixels from spatial edge of cube.
412    void        trimCube();         // in Cubes/trimImage.cc
413
414    /// @brief Replace BLANK pixels to spatial edge of cube.
415    void        unTrimCube();       // in Cubes/trimImage.cc
416
417    /// @brief Removes the baselines from the spectra, and stores in Cube::baseline
418    void        removeBaseline();   // in Cubes/baseline.cc
419
420    /// @brief Replace the baselines stored in Cube::baseline
421    void        replaceBaseline();  // in Cubes/baseline.cc
422
423    /// @brief Multiply all pixel values by -1.
424    void        invert();           // in Cubes/invertCube.cc
425
426    /// @brief Undo the inversion, and invert fluxes of all detected objects.
427    void        reInvert();         // in Cubes/invertCube.cc
428
429    //-------------------------------------
430    // Reconstruction, Searching and Merging functions
431    //
432    // in cubes.cc
433    /// @brief A front-end to all the searching functions
434    void        Search();
435
436    // in ATrous/ReconSearch.cc
437    /// @brief Front-end to reconstruction & searching functions.
438    void        ReconSearch();
439    /// @brief Switcher to reconstruction functions
440    void        ReconCube();
441    /// @brief Performs 1-dimensional a trous reconstruction on the Cube.
442    void        ReconCube1D();
443    /// @brief Performs 2-dimensional a trous reconstruction on the Cube.
444    void        ReconCube2D();
445    /// @brief Performs 3-dimensional a trous reconstruction on the Cube.
446    void        ReconCube3D();
447
448    // in Cubes/CubicSearch.cc
449    /// @brief Front-end to the cubic searching routine.
450    void        CubicSearch();
451
452    // in Cubes/smoothCube.cc
453    /// @brief Smooth the cube with the requested method.
454    void        SmoothCube();
455    /// @brief Front-end to the smoothing and searching functions.
456    void        SmoothSearch();
457    /// @brief A function to spatially smooth the cube and search the result.
458    void        SpatialSmoothNSearch();
459    /// @brief A function to Hanning-smooth the cube.
460    void        SpectralSmooth();
461    /// @brief A function to spatially-smooth the cube.
462    void        SpatialSmooth();
463
464    void        Simple3DSearch(){
465      /// @brief Basic front-end to the simple 3d searching function -- does
466      /// nothing more.
467      ///
468      /// @details This function just runs the search3DArraySimple function,
469      /// storing the results in the Cube::objectList vector. No stats
470      /// are calculated beforehand, and no logging or detection map
471      /// updating is done.
472      *objectList = search3DArray(axisDim,array,par,Stats);
473    };
474
475    void        Simple3DSearchRecon(){
476      /// @brief Basic front-end to the 3d searching function used in the
477      /// reconstruction case -- does nothing more.
478      ///
479      /// @details This function just runs the searchReconArraySimple
480      /// function, storing the results in the Cube::objectList
481      /// vector. No stats are calculated beforehand, and no logging
482      /// or detection map updating is done. The recon array is
483      /// assumed to have been defined already.
484
485      *objectList = searchReconArray(axisDim,array,recon,par,Stats);
486    };
487
488    void        Simple3DSearchSmooth(){
489      /// @brief Basic front-end to the simple 3d searching function
490      /// run on the smoothed array -- does nothing more.
491      ///
492      /// @details This function just runs the search3DArraySimple
493      /// function on the recon array (which contains the smoothed
494      /// array), storing the results in the Cube::objectList
495      /// vector. No stats are calculated beforehand, and no logging
496      /// or detection map updating is done. The recon array is
497      /// assumed to have been defined already.
498
499      *objectList = search3DArray(axisDim,recon,par,Stats);
500    };
501
502    // in Cubes/Merger.cc
503    /// @brief Merge all objects in the detection list so that only distinct ones are left.
504    void        ObjectMerger();
505 
506    // in Cubes/existingDetections.cc
507    /// @brief Read a previously-created log file to get the detections without searching
508    OUTCOME     getExistingDetections();
509
510    //-------------------------------------
511    // Text outputting of detected objects.
512    //   in Cubes/detectionIO.cc
513    //
514    /// @brief Set up the output file with parameters and stats.
515    void        prepareOutputFile();
516
517    /// @brief Write the statistical information to the output file.
518    void        outputStats();
519
520    /// @brief Output detections to the output file and standard output.
521    void        outputDetectionList();
522
523    /// @brief Prepare the log file for output.
524    void        prepareLogFile(int argc, char *argv[]);
525
526    /// @brief Output detections to the log file.
527    void        logDetectionList(bool calcFluxes=true);
528
529    /// @brief Output a single detection to the log file.
530    void        logDetection(Detection obj, int counter);
531
532    /// @brief Output detections to a Karma annotation file.
533    void        outputDetectionsKarma(std::ostream &stream);
534
535    /// @brief Output detections to a VOTable.
536    void        outputDetectionsVOTable(std::ostream &stream);
537
538
539    // ----------------------------------
540    // Calculation of overall moment maps
541
542    /// @brief Return a moment-0 map plus a map of where the object pixels are
543    void       getMomentMap(float *momentMap, std::vector<bool> &detectedPixels);
544    /// @brief Return a moment-0 map formatted for logarithmic greyscale plotting, with greyscale limits
545    void       getMomentMapForPlot(float *momentMap, float &z1, float &z2);
546
547    // ----------------------------------
548    // Graphical plotting of the cube and the detections.
549    //
550    //  in Cubes/plotting.cc
551    /// @brief Plot a spatial map of detections based on number of detected channels.
552    void        plotDetectionMap(std::string pgDestination);
553
554    /// @brief Plot a spatial map of detections based on 0th moment map of each object.
555    void        plotMomentMap(std::string pgDestination);
556
557    /// @brief Plot a spatial map of detections based on 0th moment map of each object to a number of PGPLOT devices.
558    void        plotMomentMap(std::vector<std::string> pgDestination);
559
560    /// @brief Draw WCS axes over a PGPLOT map.
561    void        plotWCSaxes(){duchamp::plotWCSaxes(head.getWCS(),axisDim);};
562
563    //  in Cubes/outputSpectra.cc
564    /// @brief Print spectra of each detected object.
565    void        outputSpectra();
566
567    /// @brief Write out text file of all spectra.
568    void        writeSpectralData();
569
570    /// @brief Print spectrum of a single object
571    void        plotSpectrum(int objNum, Plot::SpectralPlot &plot);
572    /// @brief Plot the image cutout for a single object
573    void        plotSource(Detection obj, Plot::CutoutPlot &plot);
574
575    /// @brief Get the spectral arrays
576    void        getSpectralArrays(int objNumber, float *specx, float *specy, float *specRecon, float *specBase);
577
578    //  in Cubes/drawMomentCutout.cc
579    /// @brief Draw the 0th moment map for a single object.
580    void        drawMomentCutout(Detection &object);
581
582    /// @brief Draw a scale bar indicating angular size of the cutout.
583    void        drawScale(float xstart,float ystart,float channel);
584
585    /// @brief Draw a yellow line around the edge of the spatial extent of pixels.
586    void        drawFieldEdge();
587
588  private:
589    float      *recon;            ///< reconstructed array - used when doing a trous reconstruction.
590    bool        reconExists;      ///< flag saying whether there is a reconstruction
591    short      *detectMap;        ///< "moment map" - x,y locations of detected pixels
592    float      *baseline;         ///< array of spectral baseline values.
593                             
594    bool        reconAllocated;   ///< have we allocated memory for the recon array?
595    bool        baselineAllocated;///< have we allocated memory for the baseline array?
596    FitsHeader  head;             ///< the WCS and other header information.
597    std::vector<Column::Col> fullCols;    ///< the list of all columns as printed in the results file
598    std::vector<Column::Col> logCols;     ///< the list of columns as printed in the log file
599
600  };
601
602
603  //============================================================================
604
605  ///  @brief A DataArray object limited to two dimensions, and with some additional
606  ///   special functions.
607  ///
608  ///  @details It is used primarily for searching a 1- or 2-D array with
609  ///   lutz_detect() and spectrumDetect().
610
611  class Image : public DataArray
612  {
613  public:
614    Image(){numPixels=0; numDim=2; minSize=2;};
615    Image(size_t nPix);
616    Image(size_t *dimensions);
617    virtual ~Image(){};
618    Image(const Image &i);
619    Image& operator=(const Image &i);
620
621    //--------------------
622    // Defining the array
623    //
624    /// @brief Save an external array to the Cube's pixel array
625    void      saveArray(float *input, size_t size);
626
627    /// @brief Extract a spectrum from an array and save to the local pixel array.
628    void      extractSpectrum(float *Array, size_t *dim, size_t pixel);
629
630    /// @brief Extract an image from an array and save to the local pixel array.
631    void      extractImage(float *Array, size_t *dim, size_t channel);
632
633    /// @brief Extract a spectrum from a cube and save to the local pixel array.
634    void      extractSpectrum(Cube &cube, size_t pixel);
635
636    /// @brief Extract an image from a cube and save to the local pixel array.
637    void      extractImage(Cube &cube, size_t channel);
638
639    //--------------------
640    // Accessing the data.
641    //
642    float     getPixValue(size_t pos){ return array[pos]; };
643    float     getPixValue(size_t x, size_t y){return array[y*axisDim[0] + x];};
644    // the next few should have checks against array overflow...
645    void      setPixValue(size_t pos, float f){array[pos] = f;};
646    void      setPixValue(size_t x, size_t y, float f){array[y*axisDim[0] + x] = f;};
647    bool      isBlank(int vox){return par.isBlank(array[vox]);};
648    bool      isBlank(size_t x,size_t y){return par.isBlank(array[y*axisDim[0]+x]);};
649
650    //-----------------------
651    // Detection-related
652    //
653    /// @brief Detect objects in a 2-D image
654    std::vector<PixelInfo::Object2D> findSources2D();
655
656    /// @brief Detect objects in a 1-D spectrum
657    std::vector<PixelInfo::Scan> findSources1D();
658
659    unsigned int getMinSize(){return minSize;};
660    void         setMinSize(int i){minSize=i;};
661
662    /// @brief  A detection test for a pixel location. 
663    bool      isDetection(size_t x, size_t y){
664      /// @details Test whether a pixel (x,y) is a statistically
665      /// significant detection, according to the set of statistics in
666      /// the local StatsContainer object.
667
668      size_t voxel = y*axisDim[0] + x;
669      if(isBlank(x,y)) return false;
670      else return Stats.isDetection(array[voxel]);
671    }; 
672
673    /// @brief Blank out a set of channels marked as being Milky Way channels
674    void      removeMW();
675
676  private:
677    unsigned int minSize;    ///< the minimum number of pixels for a detection to be accepted.
678  };
679
680
681}
682
683#endif
Note: See TracBrowser for help on using the repository browser.