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

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

A bunch of changes aimed at streamlining the FITS file access at the start, particularly for the case of large images where we access a subsection (this can be slow to access). We now only open the file once, and keep the fitsfile pointer in the FitsHeader? class. Once the image access is finished the file is closed.

File size: 28.1 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 Is the object next to the MW range?
362    bool        objNextToMW(Detection obj);
363
364    /// @brief Set warning flags for the detections.
365    void        setObjectFlags();
366
367    // ----------------------------
368    // Dealing with the WCS
369    //
370    /// @brief Return the FitsHeader object.
371    FitsHeader  getHead(){ return head; };
372    /// @brief Define the FitsHeader object.
373    void        setHead(FitsHeader F){head = F;};
374    /// @brief Provide a reference to the FitsHeader object.
375    FitsHeader& header(){ FitsHeader &h = head; return h; };
376
377    //-------------------------------------------
378    // FITS-I/O related functions -- not in cubes.cc
379    //
380    /// @brief Function to read in FITS file.
381    OUTCOME     getMetadata(std::string fname);  // in Cubes/getImage.cc
382    OUTCOME     getCube(std::string fname);  // in Cubes/getImage.cc
383
384    /// @brief Convert the flux units to something user-specified.
385    void        convertFluxUnits(std::string oldUnit, std::string newUnit, ARRAYREF whichArray=ARRAY); // in Cubes/getImage.cc
386
387    /// @brief Function to retrieve FITS data array
388    OUTCOME         getFITSdata(std::string fname);   // in FitsIO/dataIO.cc
389    OUTCOME         getFITSdata();   // in FitsIO/dataIO.cc
390    OUTCOME         getFITSdata(fitsfile *fptr);   // in FitsIO/dataIO.cc
391
392    OUTCOME         writeBasicHeader(fitsfile *fptr, int bitpix, bool is2D=false);
393
394    /// @brief Save the moment map to a FITS file
395    OUTCOME        saveMomentMapImage();
396
397    /// @brief Save a mask array to disk.
398    OUTCOME        saveMaskCube();       // in Cubes/saveImage.cc
399
400    /// @brief Save Smoothed array to disk.
401    OUTCOME        saveSmoothedCube();       // in Cubes/saveImage.cc
402
403    /// @brief Save Reconstructed array to disk.
404    OUTCOME        saveReconstructedCube();  // in Cubes/saveImage.cc
405
406    /// @brief Read in reconstructed array from FITS file.
407    OUTCOME        readReconCube();  // in Cubes/readRecon.cc
408 
409    /// @brief Read in Hanning-smoothed array from FITS file.
410    OUTCOME        readSmoothCube();     // in Cubes/readSmooth.cc 
411
412    //-------------------------------------
413    // Functions that act on the cube
414    //
415
416    /// @brief Remove excess BLANK pixels from spatial edge of cube.
417    void        trimCube();         // in Cubes/trimImage.cc
418
419    /// @brief Replace BLANK pixels to spatial edge of cube.
420    void        unTrimCube();       // in Cubes/trimImage.cc
421
422    /// @brief Removes the baselines from the spectra, and stores in Cube::baseline
423    void        removeBaseline();   // in Cubes/baseline.cc
424
425    /// @brief Replace the baselines stored in Cube::baseline
426    void        replaceBaseline();  // in Cubes/baseline.cc
427
428    /// @brief Multiply all pixel values by -1.
429    void        invert();           // in Cubes/invertCube.cc
430
431    /// @brief Undo the inversion, and invert fluxes of all detected objects.
432    void        reInvert();         // in Cubes/invertCube.cc
433
434    //-------------------------------------
435    // Reconstruction, Searching and Merging functions
436    //
437    // in cubes.cc
438    /// @brief A front-end to all the searching functions
439    void        Search();
440
441    // in ATrous/ReconSearch.cc
442    /// @brief Front-end to reconstruction & searching functions.
443    void        ReconSearch();
444    /// @brief Switcher to reconstruction functions
445    void        ReconCube();
446    /// @brief Performs 1-dimensional a trous reconstruction on the Cube.
447    void        ReconCube1D();
448    /// @brief Performs 2-dimensional a trous reconstruction on the Cube.
449    void        ReconCube2D();
450    /// @brief Performs 3-dimensional a trous reconstruction on the Cube.
451    void        ReconCube3D();
452
453    // in Cubes/CubicSearch.cc
454    /// @brief Front-end to the cubic searching routine.
455    void        CubicSearch();
456
457    // in Cubes/smoothCube.cc
458    /// @brief Smooth the cube with the requested method.
459    void        SmoothCube();
460    /// @brief Front-end to the smoothing and searching functions.
461    void        SmoothSearch();
462    /// @brief A function to spatially smooth the cube and search the result.
463    void        SpatialSmoothNSearch();
464    /// @brief A function to Hanning-smooth the cube.
465    void        SpectralSmooth();
466    /// @brief A function to spatially-smooth the cube.
467    void        SpatialSmooth();
468
469    void        Simple3DSearch(){
470      /// @brief Basic front-end to the simple 3d searching function -- does
471      /// nothing more.
472      ///
473      /// @details This function just runs the search3DArraySimple function,
474      /// storing the results in the Cube::objectList vector. No stats
475      /// are calculated beforehand, and no logging or detection map
476      /// updating is done.
477      *objectList = search3DArray(axisDim,array,par,Stats);
478    };
479
480    void        Simple3DSearchRecon(){
481      /// @brief Basic front-end to the 3d searching function used in the
482      /// reconstruction case -- does nothing more.
483      ///
484      /// @details This function just runs the searchReconArraySimple
485      /// function, storing the results in the Cube::objectList
486      /// vector. No stats are calculated beforehand, and no logging
487      /// or detection map updating is done. The recon array is
488      /// assumed to have been defined already.
489
490      *objectList = searchReconArray(axisDim,array,recon,par,Stats);
491    };
492
493    void        Simple3DSearchSmooth(){
494      /// @brief Basic front-end to the simple 3d searching function
495      /// run on the smoothed array -- does nothing more.
496      ///
497      /// @details This function just runs the search3DArraySimple
498      /// function on the recon array (which contains the smoothed
499      /// array), storing the results in the Cube::objectList
500      /// vector. No stats are calculated beforehand, and no logging
501      /// or detection map updating is done. The recon array is
502      /// assumed to have been defined already.
503
504      *objectList = search3DArray(axisDim,recon,par,Stats);
505    };
506
507    // in Cubes/Merger.cc
508    /// @brief Merge all objects in the detection list so that only distinct ones are left.
509    void        ObjectMerger();
510 
511    // in Cubes/existingDetections.cc
512    /// @brief Read a previously-created log file to get the detections without searching
513    OUTCOME     getExistingDetections();
514
515    //-------------------------------------
516    // Text outputting of detected objects.
517    //   in Cubes/detectionIO.cc
518    //
519    /// @brief Set up the output file with parameters and stats.
520    void        prepareOutputFile();
521
522    /// @brief Write the statistical information to the output file.
523    void        outputStats();
524
525    /// @brief Output detections to the output file and standard output.
526    void        outputDetectionList();
527
528    /// @brief Prepare the log file for output.
529    void        prepareLogFile(int argc, char *argv[]);
530
531    /// @brief Output detections to the log file.
532    void        logDetectionList(bool calcFluxes=true);
533
534    /// @brief Output a single detection to the log file.
535    void        logDetection(Detection obj, int counter);
536
537    /// @brief Output detections to a Karma annotation file.
538    void        outputDetectionsKarma(std::ostream &stream);
539
540    /// @brief Output detections to a VOTable.
541    void        outputDetectionsVOTable(std::ostream &stream);
542
543
544    // ----------------------------------
545    // Calculation of overall moment maps
546
547    /// @brief Return a moment-0 map plus a map of where the object pixels are
548    std::vector<bool> getMomentMap(float *momentMap);
549    /// @brief Return a moment-0 map formatted for logarithmic greyscale plotting, with greyscale limits
550    void       getMomentMapForPlot(float *momentMap, float &z1, float &z2);
551
552    // ----------------------------------
553    // Graphical plotting of the cube and the detections.
554    //
555    //  in Cubes/plotting.cc
556    /// @brief Plot a spatial map of detections based on number of detected channels.
557    void        plotDetectionMap(std::string pgDestination);
558
559    /// @brief Plot a spatial map of detections based on 0th moment map of each object.
560    void        plotMomentMap(std::string pgDestination);
561
562    /// @brief Plot a spatial map of detections based on 0th moment map of each object to a number of PGPLOT devices.
563    void        plotMomentMap(std::vector<std::string> pgDestination);
564
565    /// @brief Draw WCS axes over a PGPLOT map.
566    void        plotWCSaxes(){duchamp::plotWCSaxes(head.getWCS(),axisDim);};
567
568    //  in Cubes/outputSpectra.cc
569    /// @brief Print spectra of each detected object.
570    void        outputSpectra();
571
572    /// @brief Write out text file of all spectra.
573    void        writeSpectralData();
574
575    /// @brief Print spectrum of a single object
576    void        plotSpectrum(int objNum, Plot::SpectralPlot &plot);
577    /// @brief Plot the image cutout for a single object
578    void        plotSource(Detection obj, Plot::CutoutPlot &plot);
579
580    /// @brief Get the spectral arrays
581    void        getSpectralArrays(int objNumber, float *specx, float *specy, float *specRecon, float *specBase);
582
583    //  in Cubes/drawMomentCutout.cc
584    /// @brief Draw the 0th moment map for a single object.
585    void        drawMomentCutout(Detection &object);
586
587    /// @brief Draw a scale bar indicating angular size of the cutout.
588    void        drawScale(float xstart,float ystart,float channel);
589
590    /// @brief Draw a yellow line around the edge of the spatial extent of pixels.
591    void        drawFieldEdge();
592
593  private:
594    float      *recon;            ///< reconstructed array - used when doing a trous reconstruction.
595    bool        reconExists;      ///< flag saying whether there is a reconstruction
596    short      *detectMap;        ///< "moment map" - x,y locations of detected pixels
597    float      *baseline;         ///< array of spectral baseline values.
598                             
599    bool        reconAllocated;   ///< have we allocated memory for the recon array?
600    bool        baselineAllocated;///< have we allocated memory for the baseline array?
601    FitsHeader  head;             ///< the WCS and other header information.
602    std::vector<Column::Col> fullCols;    ///< the list of all columns as printed in the results file
603    std::vector<Column::Col> logCols;     ///< the list of columns as printed in the log file
604
605  };
606
607
608  //============================================================================
609
610  ///  @brief A DataArray object limited to two dimensions, and with some additional
611  ///   special functions.
612  ///
613  ///  @details It is used primarily for searching a 1- or 2-D array with
614  ///   lutz_detect() and spectrumDetect().
615
616  class Image : public DataArray
617  {
618  public:
619    Image(){numPixels=0; numDim=2; minSize=2;};
620    Image(size_t nPix);
621    Image(size_t *dimensions);
622    virtual ~Image(){};
623    Image(const Image &i);
624    Image& operator=(const Image &i);
625
626    //--------------------
627    // Defining the array
628    //
629    /// @brief Save an external array to the Cube's pixel array
630    void      saveArray(float *input, size_t size);
631
632    /// @brief Extract a spectrum from an array and save to the local pixel array.
633    void      extractSpectrum(float *Array, size_t *dim, size_t pixel);
634
635    /// @brief Extract an image from an array and save to the local pixel array.
636    void      extractImage(float *Array, size_t *dim, size_t channel);
637
638    /// @brief Extract a spectrum from a cube and save to the local pixel array.
639    void      extractSpectrum(Cube &cube, size_t pixel);
640
641    /// @brief Extract an image from a cube and save to the local pixel array.
642    void      extractImage(Cube &cube, size_t channel);
643
644    //--------------------
645    // Accessing the data.
646    //
647    float     getPixValue(size_t pos){ return array[pos]; };
648    float     getPixValue(size_t x, size_t y){return array[y*axisDim[0] + x];};
649    // the next few should have checks against array overflow...
650    void      setPixValue(size_t pos, float f){array[pos] = f;};
651    void      setPixValue(size_t x, size_t y, float f){array[y*axisDim[0] + x] = f;};
652    bool      isBlank(int vox){return par.isBlank(array[vox]);};
653    bool      isBlank(size_t x,size_t y){return par.isBlank(array[y*axisDim[0]+x]);};
654
655    //-----------------------
656    // Detection-related
657    //
658    /// @brief Detect objects in a 2-D image
659    std::vector<PixelInfo::Object2D> findSources2D();
660
661    /// @brief Detect objects in a 1-D spectrum
662    std::vector<PixelInfo::Scan> findSources1D();
663
664    unsigned int getMinSize(){return minSize;};
665    void         setMinSize(int i){minSize=i;};
666
667    /// @brief  A detection test for a pixel location. 
668    bool      isDetection(size_t x, size_t y){
669      /// @details Test whether a pixel (x,y) is a statistically
670      /// significant detection, according to the set of statistics in
671      /// the local StatsContainer object.
672
673      size_t voxel = y*axisDim[0] + x;
674      if(isBlank(x,y)) return false;
675      else return Stats.isDetection(array[voxel]);
676    }; 
677
678    /// @brief Blank out a set of channels marked as being Milky Way channels
679    void      removeMW();
680
681  private:
682    unsigned int minSize;    ///< the minimum number of pixels for a detection to be accepted.
683  };
684
685
686}
687
688#endif
Note: See TracBrowser for help on using the repository browser.