source: branches/hpc34/src/FillerWrapper.h@ 2605

Last change on this file since 2605 was 2580, checked in by ShinnosukeKawakami, 12 years ago

hpc33 merged asap-trunk

File size: 2.7 KB
Line 
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
29namespace asap
30{
31class FillerWrapper
32{
33public:
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
84private:
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 // if we want to compare to 6 characters we should only read in 6
95 FILE *f = fopen( filename.c_str(), "r") ;
96 char buf[7] ;
97 size_t tmp = fread( buf, 6, 1, f ) ;
98 (void *)tmp;
99 fclose( f ) ;
100 buf[6]='\0' ;
101 // NRO data has two types:
102 // 1) specific binary data for OTF observation
103 // 2) (pseudo-)FITS data that doesn't have primary HDU
104 // So, one can distinguish NRO and non-NRO data by examining
105 // first keyword name.
106 if ( casa::String( buf ) == "SIMPLE" ) {
107 ret = pks ;
108 }
109 else {
110 ret = nro ;
111 }
112 }
113 return ret ;
114 }
115
116 FillerWrapper();
117 FillerWrapper(const FillerWrapper&);
118 FillerWrapper& operator=(const FillerWrapper&);
119
120 casa::CountedPtr<FillerBase> filler_;
121 bool attached_;
122 casa::CountedPtr<Scantable> stable_;
123};
124
125};
126#endif
Note: See TracBrowser for help on using the repository browser.