source: branches/parallel/src/FillerWrapper.h@ 2291

Last change on this file since 2291 was 2263, checked in by Takeshi Nakazato, 13 years ago

New Development: No

JIRA Issue: No

Ready for Test: No

Interface Changes: No

What Interface Changed: Please list interface changes

Test Programs: List test programs

Put in Release Notes: Yes/No

Module(s): Module Names change impacts.

Description: Describe your changes here...

Tuning NROFiller.
Filler is about 6 times faster than before (~660sec->~110sec).

File size: 2.6 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 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
Note: See TracBrowser for help on using the repository browser.