forked from LLNL/libROM
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BasisWriter.C
107 lines (98 loc) · 3.84 KB
/
BasisWriter.C
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
/******************************************************************************
*
* Copyright (c) 2013-2016, Lawrence Livermore National Security, LLC.
* Produced at the Lawrence Livermore National Laboratory
* Written by William Arrighi [email protected]
* CODE-686965
* All rights reserved.
*
* This file is part of libROM.
* For details, see https://computation.llnl.gov/librom
* Please also read README_BSD_NOTICE.
*
* Redistribution and use in source and binary forms, with or without
* modifications, are permitted provided that the following conditions are met:
*
* o Redistributions of source code must retain the above copyright notice,
* this list of conditions and the disclaimer below.
* o Redistribution in binary form must reproduce the above copyright
* notice, this list of conditions and the disclaimer (as noted below) in
* the documentation and/or other materials provided with the
* distribution.
* o Neither the name of the LLNS/LLNL nor the names of its contributors may
* be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL LAWRENCE LIVERMORE NATIONAL SECURITY,
* LLC, THE U.S. DEPARTMENT OF ENERGY OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OR SUCH DAMAGE.
*
*****************************************************************************/
// Description: A class that writes basis vectors to a file.
#include "BasisWriter.h"
#include "HDFDatabase.h"
#include "Matrix.h"
#include "SVDBasisGenerator.h"
#include "Utilities.h"
#include "mpi.h"
namespace CAROM {
BasisWriter::BasisWriter(
SVDBasisGenerator* basis_generator,
const std::string& base_file_name,
Database::formats db_format) :
d_basis_generator(basis_generator),
d_num_intervals_written(0)
{
CAROM_ASSERT(basis_generator != 0);
CAROM_ASSERT(!base_file_name.empty());
int mpi_init;
MPI_Initialized(&mpi_init);
int rank;
if (mpi_init) {
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
}
else {
rank = 0;
}
char tmp[100];
sprintf(tmp, ".%06d", rank);
std::string full_file_name = base_file_name + tmp;
if (db_format == Database::HDF5) {
d_database = new HDFDatabase();
}
d_database->create(full_file_name);
}
BasisWriter::~BasisWriter()
{
d_database->putInteger("num_time_intervals", d_num_intervals_written);
d_database->close();
delete d_database;
}
void
BasisWriter::writeBasis()
{
char tmp[100];
double time_interval_start_time =
d_basis_generator->getBasisIntervalStartTime(d_num_intervals_written);
sprintf(tmp, "time_%06d", d_num_intervals_written);
d_database->putDouble(tmp, time_interval_start_time);
const Matrix* basis = d_basis_generator->getBasis();
int num_rows = basis->numRows();
sprintf(tmp, "num_rows_%06d", d_num_intervals_written);
d_database->putInteger(tmp, num_rows);
int num_cols = basis->numColumns();
sprintf(tmp, "num_cols_%06d", d_num_intervals_written);
d_database->putInteger(tmp, num_cols);
sprintf(tmp, "basis_%06d", d_num_intervals_written);
d_database->putDoubleArray(tmp, &basis->item(0, 0), num_rows*num_cols);
++d_num_intervals_written;
}
}