1 | |
---|
2 | #include "loggers.h" |
---|
3 | #include "../osutils.h" |
---|
4 | |
---|
5 | #include <fcntl.h> |
---|
6 | #ifdef WIN32 |
---|
7 | #include <io.h> |
---|
8 | #define open _open |
---|
9 | #define write _write |
---|
10 | #define close _close |
---|
11 | #endif |
---|
12 | |
---|
13 | namespace bdm { |
---|
14 | |
---|
15 | void memlog::init() { |
---|
16 | int i; |
---|
17 | int n = entries.length(); |
---|
18 | vectors.set_size ( n ); |
---|
19 | for ( i = 0; i < n; i++ ) { |
---|
20 | vectors ( i ).set_size ( maxlen, entries ( i )._dsize() ); |
---|
21 | |
---|
22 | // avoid uninitialized memory |
---|
23 | for ( int j = 0; j < vectors ( i ).rows(); ++j ) { |
---|
24 | for ( int k = 0; k < vectors ( i ).cols(); ++k ) { |
---|
25 | vectors ( i ) ( j, k ) = 0; |
---|
26 | } |
---|
27 | } |
---|
28 | } |
---|
29 | } |
---|
30 | |
---|
31 | void memlog::itsave ( const char* fname ) { |
---|
32 | it_file itf ( fname ); |
---|
33 | int i; |
---|
34 | string vec_name = ""; |
---|
35 | int istart, iend; |
---|
36 | for ( i = 0; i < entries.length(); i++ ) { |
---|
37 | istart = 0; |
---|
38 | for ( int j = 0; j < entries ( i ).length(); j++ ) { // same for as in add!!! |
---|
39 | vec_name = names ( i ) + separator + entries ( i ).name ( j ); |
---|
40 | iend = istart + entries ( i ).size ( j ) - 1; |
---|
41 | itf << Name ( vec_name ) << vectors ( i ).get_cols ( istart, iend ); |
---|
42 | // move start after current end... |
---|
43 | istart = iend + 1; |
---|
44 | } |
---|
45 | |
---|
46 | } |
---|
47 | } |
---|
48 | |
---|
49 | void memlog::from_setting ( const Setting &set ) { |
---|
50 | root::from_setting( set ); |
---|
51 | UI::get ( maxlen, set, "maxlen", UI::compulsory ); |
---|
52 | UI::get ( filename, set, "filename" ); |
---|
53 | } |
---|
54 | |
---|
55 | void dirfilelog::init() { |
---|
56 | int i, j, k; |
---|
57 | int nsc = 0; |
---|
58 | for ( i = 0; i < entries.length(); i++ ) { |
---|
59 | nsc += entries ( i )._dsize(); |
---|
60 | } |
---|
61 | ; //all entries!! |
---|
62 | |
---|
63 | char filename[200]; |
---|
64 | char num[3]; |
---|
65 | |
---|
66 | // Initialize directory |
---|
67 | makedir ( dirname ); // |
---|
68 | |
---|
69 | // directory OK, or it can be overwritten |
---|
70 | |
---|
71 | //Create filenames |
---|
72 | scalarnames.set_length ( nsc ); |
---|
73 | // For all entries |
---|
74 | int ii = 0; |
---|
75 | for ( i = 0; i < entries.length(); i++ ) { //for entries |
---|
76 | for ( j = 0; j < entries ( i ).length(); j++ ) { //for RVs in entries |
---|
77 | int rvsize = entries ( i ).size ( j ); |
---|
78 | // for non-empty names |
---|
79 | if ( names ( i ).length() > 0 ) { |
---|
80 | scalarnames ( ii ) = names ( i ) + separator; |
---|
81 | } |
---|
82 | // add name |
---|
83 | scalarnames ( ii ) += entries ( i ).name ( j ); |
---|
84 | // add number when needed |
---|
85 | if ( rvsize > 1 ) { |
---|
86 | // copy the same name for th whole RV |
---|
87 | for ( k = 1; k < rvsize; k++ ) { |
---|
88 | scalarnames ( ii + k ) = scalarnames ( ii ); |
---|
89 | } |
---|
90 | // add numbers |
---|
91 | for ( k = 0; k < rvsize; k++ ) { //for all scalars in given RV |
---|
92 | sprintf ( num, "%d", k ); |
---|
93 | scalarnames ( ii ) += separator + num; |
---|
94 | ii++; |
---|
95 | } |
---|
96 | } else { |
---|
97 | ii++; |
---|
98 | } |
---|
99 | } |
---|
100 | } |
---|
101 | //Create format |
---|
102 | string frm = "format"; |
---|
103 | get_fname ( filename, dirname, frm ); |
---|
104 | |
---|
105 | FILE* fpf; |
---|
106 | fpf = fopen ( filename, "w" ); |
---|
107 | for ( i = 0; i < ii; i++ ) { |
---|
108 | fprintf ( fpf, "%s RAW d 1\n", scalarnames ( i ).c_str() ); |
---|
109 | } |
---|
110 | fclose ( fpf ); |
---|
111 | |
---|
112 | memlog::init(); |
---|
113 | |
---|
114 | //Delete old data |
---|
115 | write_buffers ( 0 ); |
---|
116 | } |
---|
117 | |
---|
118 | |
---|
119 | void dirfilelog::write_buffers ( int Len ) { |
---|
120 | int fp; |
---|
121 | int nen = entries.length(); //all entries!! |
---|
122 | double *Dt; |
---|
123 | int DtRows; |
---|
124 | int i, j, k; |
---|
125 | int ii; //index in scalarnames; |
---|
126 | int jj; //index in vectors() |
---|
127 | char filename[200]; |
---|
128 | |
---|
129 | bdm_assert_debug ( Len <= maxlen, "diffilelog::write_buffers: Len too large" ); |
---|
130 | |
---|
131 | ii = 0; |
---|
132 | for ( i = 0; i < nen; i++ ) { //for entries |
---|
133 | Dt = vectors ( i )._data(); |
---|
134 | DtRows = vectors ( i ).rows(); |
---|
135 | |
---|
136 | jj = 0; |
---|
137 | for ( j = 0; j < entries ( i ).length(); j++ ) { //for RVs in entries |
---|
138 | int rvsize = entries ( i ).size ( j ); |
---|
139 | for ( k = 0; k < rvsize; k++ ) { { //for all scalars in given RV |
---|
140 | get_fname ( filename, dirname, scalarnames ( ii ) ); |
---|
141 | if ( Len == 0 ) //initialization |
---|
142 | fp = open ( filename, O_CREAT | O_WRONLY | O_TRUNC, 00644 ); |
---|
143 | else |
---|
144 | fp = open ( filename, O_CREAT | O_WRONLY | O_APPEND, 00644 ); |
---|
145 | |
---|
146 | write ( fp, &Dt[jj* ( DtRows ) ], ( Len ) *sizeof ( double ) ); |
---|
147 | close ( fp ); |
---|
148 | |
---|
149 | //next row in Dt |
---|
150 | jj++; |
---|
151 | //next scalarname |
---|
152 | ii++; |
---|
153 | } |
---|
154 | } |
---|
155 | } |
---|
156 | } |
---|
157 | } |
---|
158 | |
---|
159 | void dirfilelog::step ( ) { |
---|
160 | |
---|
161 | if ( ind == ( maxlen - 1 ) ) { |
---|
162 | write_buffers ( ind + 1 ); |
---|
163 | ind = 0; |
---|
164 | } else |
---|
165 | ind++; |
---|
166 | |
---|
167 | } |
---|
168 | |
---|
169 | void dirfilelog::finalize ( ) { |
---|
170 | if ( ind > 0 && !buffers_written) { |
---|
171 | write_buffers ( ind ); //assuming here that i+1 was not filled |
---|
172 | buffers_written = true; |
---|
173 | } |
---|
174 | } |
---|
175 | |
---|
176 | |
---|
177 | void dirfilelog::from_setting ( const Setting &set ) { |
---|
178 | dirname = ( const char* ) set["dirname"]; |
---|
179 | maxlen = set["maxlen"]; |
---|
180 | scalarnames.set_length ( 0 ); |
---|
181 | } |
---|
182 | |
---|
183 | } |
---|