1 | #include <fstream> |
---|
2 | #include <iostream> |
---|
3 | #include <iomanip> |
---|
4 | //#include <Cubes/cubes.hh> |
---|
5 | using namespace std; |
---|
6 | |
---|
7 | //void searchVelocity(Cube &cube, Detection *objectList, int &numObjects) |
---|
8 | DetectionList *searchVelocity(Cube &cube, Param &par) |
---|
9 | { |
---|
10 | DetectionList *objectList = new DetectionList; |
---|
11 | int xdim=cube.getDimX(),ydim=cube.getDimY(),zdim = cube.getDimZ(); |
---|
12 | long *dim = new long[2]; |
---|
13 | dim[0] = zdim; |
---|
14 | dim[1] = 1; |
---|
15 | |
---|
16 | ofstream fout(par.getLogFile().c_str(),ios::app); |
---|
17 | outputDetectionTextHeader(fout); |
---|
18 | // outputDetectionTextHeader(cerr); |
---|
19 | for(int npixel=0;npixel<xdim*ydim;npixel++){ |
---|
20 | // at each spatial pixel in the cube, search in the spectrum |
---|
21 | |
---|
22 | if((100*npixel/(xdim*ydim))%5==0) |
---|
23 | cout<<setw(3)<<100*npixel/(xdim*ydim)<<"% done"<<"\b\b\b\b\b\b\b\b\b"<<flush; |
---|
24 | |
---|
25 | Image *spectrum = new Image(dim); |
---|
26 | float *array = new float[zdim]; |
---|
27 | for(long zpos=0;zpos<zdim;zpos++){ |
---|
28 | int cubepos = npixel + zpos*xdim*ydim; |
---|
29 | array[zpos] = cube.getPixValue(cubepos); |
---|
30 | } |
---|
31 | spectrum->saveArray(array,zdim); |
---|
32 | |
---|
33 | /* This was the old way of doing it */ |
---|
34 | //findSources(*spectrum,par); |
---|
35 | |
---|
36 | /* This is the new way using the atrous reconstruction */ |
---|
37 | for(int i=0;i<zdim;i++) spectrum->setMaskValue(i,0); |
---|
38 | findSourcesAtrous1D(*spectrum,par); |
---|
39 | for(long zpos=0;zpos<zdim;zpos++) { |
---|
40 | int mask = spectrum->getMaskValue(zpos); |
---|
41 | if(mask>1) cerr << "**** "<<npixel<<" "<<zpos<<" "<<mask<<endl; |
---|
42 | } |
---|
43 | |
---|
44 | Image *newSpectrum = new Image(dim); |
---|
45 | newSpectrum->saveArray(array,zdim); |
---|
46 | // set the proper positional parameters for each object |
---|
47 | for(int objCtr=0;objCtr<spectrum->getNumObj();objCtr++){ |
---|
48 | Detection *obj = new Detection; |
---|
49 | *obj = spectrum->getObject(objCtr); |
---|
50 | |
---|
51 | for(int pixCtr=0;pixCtr<obj->getSize();pixCtr++){ |
---|
52 | int zpos = obj->getX(pixCtr); |
---|
53 | int xpos = npixel%xdim; |
---|
54 | int ypos = npixel/xdim; |
---|
55 | obj->setZ(pixCtr,zpos); |
---|
56 | obj->setY(pixCtr,ypos); |
---|
57 | obj->setX(pixCtr,xpos); |
---|
58 | } |
---|
59 | obj->calcParams(); |
---|
60 | newSpectrum->addObject(*obj); |
---|
61 | delete obj; |
---|
62 | } |
---|
63 | for(long zpos=0;zpos<zdim;zpos++) { |
---|
64 | int mask = spectrum->getMaskValue(zpos); |
---|
65 | if(mask>1) cerr << "**** "<<zpos<<" "<<mask<<endl; |
---|
66 | newSpectrum->setMaskValue(zpos, mask ); |
---|
67 | } |
---|
68 | delete spectrum; |
---|
69 | spectrum = newSpectrum; |
---|
70 | delete [] array; |
---|
71 | |
---|
72 | for(int objCtr=0;objCtr<spectrum->getNumObj();objCtr++){ |
---|
73 | Detection *obj = new Detection; |
---|
74 | *obj = spectrum->getObject(objCtr); |
---|
75 | outputDetectionText(fout,*obj,objectList->getSize()+1+objCtr); |
---|
76 | // outputDetectionText(cerr,*obj,objectList->getSize()+1+objCtr); |
---|
77 | delete obj; |
---|
78 | } |
---|
79 | |
---|
80 | short detect = cube.getDetectMapValue(npixel); |
---|
81 | for(int i=0;i<zdim;i++){ |
---|
82 | short mask = spectrum->getMaskValue(i); |
---|
83 | detect += mask; |
---|
84 | } |
---|
85 | cube.setDetectMapValue(npixel, detect); |
---|
86 | |
---|
87 | // add the newly found objects to the list |
---|
88 | for(int i=0;i<spectrum->getNumObj();i++) |
---|
89 | objectList->addObject(spectrum->getObject(i)); |
---|
90 | |
---|
91 | delete spectrum; |
---|
92 | |
---|
93 | } |
---|
94 | delete [] dim; |
---|
95 | fout<<endl<<endl; |
---|
96 | fout.close(); |
---|
97 | |
---|
98 | return objectList; |
---|
99 | |
---|
100 | } |
---|