Homec4science
Diffusion RationalROMPy (master) Browse

RROMPy/rrompy/sampling/engines/sampling_engine_pod.py
e2ed7448d9f1master

Last Change
Raw File

sampling_engine_pod.py
View Options

# Copyright (C) 2018-2020 by the RROMPy authors
#
# This file is part of RROMPy.
#
# RROMPy is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# RROMPy 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
#
import numpy as np
from scipy.sparse import block_diag
from .sampling_engine_normalize import SamplingEngineNormalize
from rrompy.utilities.base.types import Any, sampList
from rrompy.utilities.base import verbosityManager as vbMng
__all__ = ['SamplingEnginePOD']
class SamplingEnginePOD(SamplingEngineNormalize):
def _mergeFeature(self, name:str, value:Any):
if name == "Rscale":
self.Rscale = block_diag((self.Rscale, value), "csc")
else:
super()._mergeFeature(name, value)
def resetHistory(self):
super().resetHistory()
self.Rscale = np.zeros((0, 0), dtype = np.complex)
def popSample(self):
if hasattr(self, "nsamples") and self.nsamples > 1:
self.Rscale = self.Rscale[:, : -1]
super().popSample()
def postprocessu(self, u:sampList, overwrite : bool = False):
"""Postprocess by orthogonalizing snapshot."""
self.setsample(u, overwrite)
vbMng(self, "INIT", "Starting orthogonalization.", 20)
u, r, _ = self.PODEngine.GS(u, self.samples_normal, is_state = True)
self.Rscale = np.pad(self.Rscale, ((0, 1), (0, 1)), 'constant')
self.Rscale[:, -1] = r
vbMng(self, "DEL", "Done orthogonalizing.", 20)
self.setsample_normal(u, overwrite)
def postprocessuBulk(self):
"""Postprocess by orthogonalizing snapshots in bulk."""
vbMng(self, "INIT", "Starting orthogonalization.", 10)
samples_normal, self.Rscale = self.PODEngine.generalizedQR(
self.samples, is_state = True)
vbMng(self, "DEL", "Done orthogonalizing.", 10)
nsamples, self.nsamples = self.nsamples, 0
self.setsample_normal(samples_normal)
self.nsamples = nsamples
c4science · Help