1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
//# tomf.cc: This program copies files to a multifile
//# Copyright (C) 2014
//# Associated Universities, Inc. Washington DC, USA.
//#
//# This program is free software; you can redistribute it and/or modify it
//# under the terms of the GNU General Public License as published by the Free
//# Software Foundation; either version 2 of the License, or (at your option)
//# any later version.
//#
//# This program is distributed in the hope that it will be useful, but WITHOUT
//# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
//# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
//# more details.
//#
//# You should have received a copy of the GNU General Public License along
//# with this program; if not, write to the Free Software Foundation, Inc.,
//# 675 Massachusetts Ave, Cambridge, MA 02139, USA.
//#
//# Correspondence concerning AIPS++ should be addressed as follows:
//# Internet email: casa-feedback@nrao.edu.
//# Postal address: AIPS++ Project Office
//# National Radio Astronomy Observatory
//# 520 Edgemont Road
//# Charlottesville, VA 22903-2475 USA
#include <casacore/casa/IO/MFFileIO.h>
#include <casacore/casa/IO/MultiFile.h>
#include <casacore/casa/IO/MultiHDF5.h>
#include <casacore/casa/IO/RegularFileIO.h>
#include <casacore/casa/IO/FiledesIO.h>
#include <casacore/casa/Containers/Block.h>
#include <vector>
#include <stdexcept>
#include <iostream>
using namespace casacore;
using namespace std;
int main (int argc, char* argv[])
{
try {
vector<String> fname;
String outName;
Int64 blockSize = 1048576;
Bool useHDF5 = False;
for (int argnr=1; argnr<argc; ++argnr) {
if (String(argv[argnr]) == "-b") {
argnr++;
if (argnr < argc) {
blockSize = atoi (argv[argnr]);
}
} else if (String(argv[argnr]) == "-h") {
useHDF5 = True;
} else if (argnr == argc-1) {
outName = argv[argnr];
} else {
fname.push_back (argv[argnr]);
}
}
if (fname.empty() || outName.empty()) {
cerr << "Run as: tomf [-h] [-b blocksize] filename1 ... outname" << endl;
cerr << " -h create MultiFile as HDF5 instead of regular file" << endl;
cerr << " -b blocksize in bytes; default 1048576" << endl;
return 0;
}
// Open each file and copy to the MultiFile.
std::shared_ptr<MultiFileBase> mfile;
if (useHDF5) {
mfile.reset (new MultiHDF5 (outName, ByteIO::New, blockSize));
} else {
mfile.reset (new MultiFile (outName, ByteIO::New, blockSize));
}
Block<char> buffer (blockSize);
for (vector<String>::const_iterator iter=fname.begin();
iter!=fname.end(); ++iter) {
if (iter->empty()) {
cerr << "*** Empty file name given" << endl;
} else {
int fd = RegularFileIO::openCreate (*iter, ByteIO::Old);
FiledesIO file(fd, *iter);
Int64 todo = file.length();
cout << " copying " << todo << " bytes of " << *iter
<< " ..." << endl;
MFFileIO outfile (mfile, *iter, ByteIO::New);
while (todo > 0) {
Int64 sz = file.read (std::min(todo, blockSize), buffer.storage());
outfile.write (sz, buffer.storage());
todo -= sz;
}
}
}
cout << endl;
} catch (const std::exception& x) {
cerr << x.what() << endl;
return 1;
}
return 0;
}
|
