[1365] | 1 | #include "fitstoiwtr.h"
|
---|
| 2 | #include "toimanager.h"
|
---|
| 3 |
|
---|
| 4 |
|
---|
| 5 |
|
---|
| 6 | extern void fits_lock();
|
---|
| 7 | extern void fits_unlock();
|
---|
| 8 |
|
---|
| 9 | FITSTOIWriter::FITSTOIWriter(string fn) {
|
---|
| 10 | fname = fn;
|
---|
| 11 | fstatus = 0;
|
---|
[1369] | 12 | outFlags = false;
|
---|
[1365] | 13 |
|
---|
| 14 | // Open file
|
---|
| 15 | remove(fname.c_str());
|
---|
| 16 | fits_lock();
|
---|
| 17 | fits_create_file(&fptr,fname.c_str(),&fstatus);
|
---|
| 18 | fits_unlock();
|
---|
| 19 | name = "wtr";
|
---|
| 20 | }
|
---|
| 21 |
|
---|
| 22 | FITSTOIWriter::~FITSTOIWriter() {
|
---|
| 23 | }
|
---|
| 24 |
|
---|
[1369] | 25 | void FITSTOIWriter::setOutFlags(bool yn) {
|
---|
| 26 | outFlags = yn;
|
---|
| 27 | }
|
---|
| 28 |
|
---|
[1365] | 29 | void FITSTOIWriter::addInput(string name, TOI* toi) {
|
---|
| 30 | declareInput(name);
|
---|
| 31 | fwinputs.push_back(toi);
|
---|
| 32 | }
|
---|
| 33 |
|
---|
[1439] | 34 | void FITSTOIWriter::afterinit()
|
---|
| 35 | {
|
---|
| 36 | if (inTOIs) delete[] inTOIs;
|
---|
[1365] | 37 | inTOIs = new (TOI*[fwinputs.size()]);
|
---|
| 38 |
|
---|
| 39 | for (int i=0; i<fwinputs.size(); i++) {
|
---|
| 40 | inTOIs[i] = fwinputs[i];
|
---|
| 41 | }
|
---|
[1439] | 42 |
|
---|
| 43 | }
|
---|
[1365] | 44 |
|
---|
[1439] | 45 | void FITSTOIWriter::run() {
|
---|
| 46 | cout << "fitstoiwriter running" << endl;
|
---|
| 47 | // init done here
|
---|
| 48 | afterinit();
|
---|
| 49 | inited=true;
|
---|
| 50 |
|
---|
[1369] | 51 | int ndata = inIx.size();
|
---|
| 52 | int ncols = inIx.size() * (outFlags ? 2 : 1) + 1;
|
---|
[1365] | 53 |
|
---|
| 54 | char** colnames = new (char*[ncols]);
|
---|
| 55 | char** coltypes = new (char*[ncols]);
|
---|
| 56 | char** colunits = new (char*[ncols]);
|
---|
| 57 |
|
---|
[1369] | 58 | colnames[0] = "samplenum";
|
---|
| 59 | coltypes[0] = "1D";
|
---|
| 60 | colunits[0] = "integer";
|
---|
| 61 |
|
---|
[1439] | 62 | cout << " FITSTOIWriter::run() - Creating output FITS file: "
|
---|
| 63 | << name << endl;
|
---|
| 64 |
|
---|
| 65 | vector<string> coln;
|
---|
| 66 | int ck=0;
|
---|
[1410] | 67 | for (map<string, int>::iterator ii = inIx.begin(); ii != inIx.end(); ii++) {
|
---|
| 68 | int j = (*ii).second;
|
---|
[1439] | 69 | coln.push_back((*ii).first);
|
---|
[1369] | 70 | if (outFlags) j += j;
|
---|
[1439] | 71 | colnames[j+1] = const_cast<char*>(coln[ck].c_str());
|
---|
| 72 | cout << " Column[" << j+1 << "] Name=" << coln[ck] << endl;
|
---|
| 73 | ck++;
|
---|
[1369] | 74 | coltypes[j+1] = "1D";
|
---|
[1439] | 75 | colunits[j+1] = "double";
|
---|
[1369] | 76 | if (outFlags) {
|
---|
[1439] | 77 | coln.push_back("fg_" + coln[ck-1]);
|
---|
| 78 | colnames[j+2] = const_cast<char*>(coln[ck].c_str());
|
---|
| 79 | cout << " Column[" << j+1 << "] -Flag- Name=" << coln[ck] << endl;
|
---|
| 80 | ck++;
|
---|
[1369] | 81 | coltypes[j+2] = "1D";
|
---|
[1439] | 82 | colunits[j+2] = "IntFlag";
|
---|
[1369] | 83 | }
|
---|
[1365] | 84 | }
|
---|
| 85 |
|
---|
| 86 | fits_lock();
|
---|
| 87 | fits_create_tbl(fptr, BINARY_TBL, 0, ncols, colnames, coltypes, colunits, NULL, &fstatus);
|
---|
| 88 | fits_write_date(fptr, &fstatus);
|
---|
| 89 | fits_unlock();
|
---|
| 90 |
|
---|
| 91 | delete[] colunits;
|
---|
| 92 | delete[] coltypes;
|
---|
| 93 | delete[] colnames;
|
---|
| 94 |
|
---|
| 95 | // Add headers ?
|
---|
| 96 |
|
---|
| 97 | // loop
|
---|
| 98 |
|
---|
| 99 | int fitsLine = 1;
|
---|
| 100 | int snb = getMinIn();
|
---|
| 101 | int sne = getMaxIn();
|
---|
| 102 |
|
---|
[1442] | 103 | double* tabdata = new double[ndata];
|
---|
| 104 | double* tabflag = new double[ndata]; // ? pas integer ?? $CHECK$ - Reza
|
---|
| 105 | bool* tabck = new bool[ndata];
|
---|
| 106 | int i;
|
---|
| 107 | for(i=0; i<ndata; i++) {
|
---|
| 108 | tabdata[i] = -9.e19; // $CHECK$ - Reza valeur par defaut !
|
---|
| 109 | tabflag[i] = -9.e19; // $CHECK$ - Reza valeur par defaut !
|
---|
| 110 | tabck[i] = checkInputTOIIndex(i);
|
---|
| 111 | }
|
---|
| 112 |
|
---|
[1365] | 113 | for (int sn = snb; sn <= sne; sn++) {
|
---|
[1442] | 114 | // if ((sn%2000 == 0) || (sn<snb+5))
|
---|
| 115 | // cout << " DBG-A-FitsWriter::run()" << sn << endl;
|
---|
[1365] | 116 | try {
|
---|
[1462] | 117 | int_4 out_flg;
|
---|
| 118 | double out_val;
|
---|
[1442] | 119 | for (i=0; i<ndata; i++) {
|
---|
| 120 | if (tabck[i]) {
|
---|
[1462] | 121 | getData(i,sn, out_val, out_flg);
|
---|
| 122 | tabdata[i] = out_val;
|
---|
| 123 | if (outFlags) tabflag[i] = out_flg;
|
---|
[1442] | 124 | }
|
---|
| 125 | }
|
---|
[1369] | 126 | fits_lock();
|
---|
[1442] | 127 | // if ((sn%2000 == 0) || (sn<snb+5))
|
---|
| 128 | //cout << " DBG-B-FitsWriter::run()" << sn << endl;
|
---|
[1369] | 129 | double xx = sn;
|
---|
| 130 | fits_write_col_dbl(fptr, 1, fitsLine, 1, 1, &xx, &fstatus);
|
---|
[1442] | 131 |
|
---|
| 132 | for (i=0; i<ndata; i++) {
|
---|
[1369] | 133 | if (outFlags) {
|
---|
[1442] | 134 | fits_write_col_dbl(fptr, 2*i+2, fitsLine, 1, 1, tabdata+i, &fstatus);
|
---|
| 135 | fits_write_col_dbl(fptr, 2*i+3, fitsLine, 1, 1, tabflag+i, &fstatus);
|
---|
[1369] | 136 | } else {
|
---|
[1442] | 137 | fits_write_col_dbl(fptr, i+2, fitsLine, 1, 1, tabdata+i, &fstatus);
|
---|
[1369] | 138 | }
|
---|
[1365] | 139 | if (fstatus != 0) {
|
---|
| 140 | cerr << "fitstoiwtr error sn = " << sn << " i = "<< i << endl;
|
---|
| 141 | fits_report_error(stderr, fstatus);
|
---|
| 142 | abort();
|
---|
| 143 | }
|
---|
| 144 | }
|
---|
[1442] | 145 | fits_unlock();
|
---|
| 146 | // if ((sn%2000 == 0) || (sn<snb+5))
|
---|
| 147 | // cout << " DBG-C-FitsWriter::run()" << sn << " line=" << fitsLine << endl;
|
---|
| 148 | fitsLine++;
|
---|
[1365] | 149 | } catch (PException e) {
|
---|
[1442] | 150 | cout << "fitstoiwtr exception " << e.Msg() << endl;
|
---|
| 151 | continue;
|
---|
[1365] | 152 | }
|
---|
| 153 | }
|
---|
| 154 |
|
---|
[1442] | 155 | delete[] tabdata;
|
---|
| 156 | delete[] tabflag;
|
---|
| 157 | delete[] tabck;
|
---|
| 158 |
|
---|
[1365] | 159 | fits_lock();
|
---|
| 160 | fits_close_file(fptr, &fstatus);
|
---|
| 161 | fits_report_error(stderr, fstatus);
|
---|
| 162 | fits_unlock();
|
---|
[1442] | 163 | cout << "fitstoiwriter done fitsLine= " << fitsLine << endl;
|
---|
[1365] | 164 | }
|
---|
| 165 |
|
---|
| 166 |
|
---|
| 167 |
|
---|
| 168 |
|
---|
| 169 |
|
---|
| 170 |
|
---|
| 171 |
|
---|
| 172 |
|
---|