1 | //
|
---|
2 | // C++ Interface: Filler
|
---|
3 | //
|
---|
4 | // Description:
|
---|
5 | //
|
---|
6 | //
|
---|
7 | // Author: Malte Marquarding <asap@atnf.csiro.au>, (C) 2010
|
---|
8 | //
|
---|
9 | // Copyright: See COPYING file that comes with this distribution
|
---|
10 | //
|
---|
11 | //
|
---|
12 | #ifndef ASAPFILLER_H
|
---|
13 | #define ASAPFILLER_H
|
---|
14 |
|
---|
15 | #include <casa/aips.h>
|
---|
16 | #include <casa/Exceptions.h>
|
---|
17 | #include <casa/Utilities/CountedPtr.h>
|
---|
18 | #include <casa/Containers/Record.h>
|
---|
19 | #include <casa/OS/File.h>
|
---|
20 |
|
---|
21 | #include <string>
|
---|
22 |
|
---|
23 | #include "ScantableWrapper.h"
|
---|
24 | #include "FillerBase.h"
|
---|
25 | #include "PKSFiller.h"
|
---|
26 | #include "NROFiller.h"
|
---|
27 |
|
---|
28 |
|
---|
29 | namespace asap
|
---|
30 | {
|
---|
31 | class FillerWrapper
|
---|
32 | {
|
---|
33 | public:
|
---|
34 |
|
---|
35 | explicit FillerWrapper(ScantableWrapper tbl) : filler_(0), attached_(false)
|
---|
36 | { stable_ = tbl.getCP(); }
|
---|
37 |
|
---|
38 | virtual ~FillerWrapper() { close(); }
|
---|
39 |
|
---|
40 |
|
---|
41 | void open(const std::string& filename, const casa::Record& rec) {
|
---|
42 | // void open(const std::string& filename) {
|
---|
43 | casa::File file(filename);
|
---|
44 | if ( !file.exists() ) {
|
---|
45 | throw(AipsError("File does not exist"));
|
---|
46 | }
|
---|
47 | int fileType = dataType( filename ) ;
|
---|
48 | if ( fileType == 0 ) {
|
---|
49 | filler_ = new PKSFiller(stable_);
|
---|
50 | if (filler_->open(filename, rec)) {
|
---|
51 | attached_ = true;
|
---|
52 | return;
|
---|
53 | }
|
---|
54 | }
|
---|
55 | else if ( fileType == 1 ) {
|
---|
56 | filler_ = new NROFiller(stable_);
|
---|
57 | if (filler_->open(filename, rec)) {
|
---|
58 | attached_ = true;
|
---|
59 | return;
|
---|
60 | }
|
---|
61 | }
|
---|
62 | filler_ = 0;
|
---|
63 | attached_ = false;
|
---|
64 | throw casa::AipsError("Unknown Data Format");
|
---|
65 | }
|
---|
66 | void close() {
|
---|
67 | if (attached_) {
|
---|
68 | filler_->close();
|
---|
69 | }
|
---|
70 | }
|
---|
71 |
|
---|
72 | void fill() {
|
---|
73 | if (attached_) {
|
---|
74 | filler_->fill();
|
---|
75 | }
|
---|
76 | }
|
---|
77 |
|
---|
78 | void setReferenceRegex(const std::string& rx) {
|
---|
79 | if (attached_) {
|
---|
80 | filler_->setReferenceRegex(rx);
|
---|
81 | }
|
---|
82 | }
|
---|
83 |
|
---|
84 | private:
|
---|
85 |
|
---|
86 | int dataType( const std::string &filename ) {
|
---|
87 | int ret = -1 ;
|
---|
88 | int pks = 0 ;
|
---|
89 | int nro = 1 ;
|
---|
90 | casa::File file( filename ) ;
|
---|
91 | if ( file.isDirectory() )
|
---|
92 | ret = pks ;
|
---|
93 | else if ( file.isReadable() ) {
|
---|
94 | FILE *f = fopen( filename.c_str(), "r") ;
|
---|
95 | char buf[8] ;
|
---|
96 | fread( buf, 6, 1, f ) ;
|
---|
97 | fclose( f ) ;
|
---|
98 | buf[7]='\0' ;
|
---|
99 | // NRO data has two types:
|
---|
100 | // 1) specific binary data for OTF observation
|
---|
101 | // 2) (pseudo-)FITS data that doesn't have primary HDU
|
---|
102 | // So, one can distinguish NRO and non-NRO data by examining
|
---|
103 | // first keyword name.
|
---|
104 | if ( casa::String( buf ) == "SIMPLE" ) {
|
---|
105 | ret = pks ;
|
---|
106 | }
|
---|
107 | else {
|
---|
108 | ret = nro ;
|
---|
109 | }
|
---|
110 | }
|
---|
111 | return ret ;
|
---|
112 | }
|
---|
113 |
|
---|
114 | FillerWrapper();
|
---|
115 | FillerWrapper(const FillerWrapper&);
|
---|
116 | FillerWrapper& operator=(const FillerWrapper&);
|
---|
117 |
|
---|
118 | casa::CountedPtr<FillerBase> filler_;
|
---|
119 | bool attached_;
|
---|
120 | casa::CountedPtr<Scantable> stable_;
|
---|
121 | };
|
---|
122 |
|
---|
123 | };
|
---|
124 | #endif
|
---|