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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/////////////////////////////////////////////////////////////
// //
// Copyright (c) 2003-2017 by The University of Queensland //
// Centre for Geoscience Computing //
// http://earth.uq.edu.au/centre-geoscience-computing //
// //
// Primary Business: Brisbane, Queensland, Australia //
// Licensed under the Open Software License version 3.0 //
// http://www.apache.org/licenses/LICENSE-2.0 //
// //
/////////////////////////////////////////////////////////////
// --- System includes ---
#include <string>
#include <sstream>
#include <iostream>
#include <cstdlib>
using std::cerr;
using std::endl;
using std::ostringstream;
// --- project includes ---
#include "frame_silo.h"
int main(int argc,char** argv)
{
string infilename;
string ofilename;
string basefilename;
string listfilename;
bool with_rot=false;
bool options_valid=true;
bool with_list=false;
bool with_slice_z=false;
int t0_frame=0;
int tstart=0;
int nt=0;
int dt=0;
int ret=0;
double slice_zmin=0.0,slice_zmax=0.0;
// process args
int args_read=1;
while(args_read<argc){
string option=string(argv[args_read]);
if(option=="-i"){
if(argc>args_read){
basefilename=string(argv[args_read+1]);
args_read+=2;
} else {
options_valid=false;
}
} else if(option=="-list"){
if(argc>args_read){
listfilename=string(argv[args_read+1]);
with_list=true;
args_read+=2;
} else {
options_valid=false;
}
} else if(option=="-o"){
if(argc>args_read){
ofilename=argv[args_read+1];
args_read+=2;
} else {
options_valid=false;
}
} else if(option=="-t0"){
if(argc>args_read){
t0_frame=atoi(argv[args_read+1]);
args_read+=2;
} else {
options_valid=false;
}
} else if(option=="-t"){
if(argc>args_read+2){
tstart=atoi(argv[args_read+1]);
nt=atoi(argv[args_read+2]);
dt=atoi(argv[args_read+3]);
args_read+=4;
} else {
options_valid=false;
}
} else if(option=="-sz"){
if(argc>args_read+1){
with_slice_z=true;
slice_zmin=atof(argv[args_read+1]);
slice_zmax=atof(argv[args_read+2]);
args_read+=3;
} else {
options_valid=false;
}
} else if(option=="-rot") {
with_rot=true;
options_valid=true;
args_read++;
} else {
cerr << "Unknown option " << option << endl;
options_valid = false;
break;
}
}
if(options_valid){
for (int i=0; i<nt; i++) {
// make filename
ostringstream filename;
filename << basefilename << "_t=" << tstart+i*dt << "_0.txt"; // write frame
if (with_slice_z) {
if (with_rot) {
saveSiloSnapSliced(filename.str(), ofilename, i+t0_frame, with_list, listfilename, slice_zmin, slice_zmax);
} else {
cerr << "slice not implemented for non-rotational particles" << endl;
}
} else {
saveSiloSnap(filename.str(), ofilename, i+t0_frame, with_list, listfilename, with_rot);
}
}
ret=0;
} else {
cerr << "Options invalid" << endl;
ret=1;
}
return ret;
}
|
