LibtorchObservationHistoryHelper.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#ifdef MOOSE_LIBTORCH_ENABLED
 
#pragma once
 
#include "MooseTypes.h"
 
#include <vector>
 
class LibtorchObservationHistoryHelper
{
public:
  LibtorchObservationHistoryHelper(unsigned int input_timesteps);
 
  unsigned int inputTimesteps() const { return _input_timesteps; }
 
  void initializeHistory(const std::vector<Real> & observation,
                         std::vector<std::vector<Real>> & old_observations) const;
 
  void advanceHistory(const std::vector<Real> & observation,
                      std::vector<std::vector<Real>> & old_observations) const;
 
  std::vector<Real> expandObservationFactors(const std::vector<Real> & observation_factors) const;
 
  std::vector<Real>
  stackCurrentObservation(const std::vector<Real> & observation,
                          const std::vector<std::vector<Real>> & old_observations) const;
 
  std::vector<Real>
  stackTrajectoryObservation(const std::vector<std::vector<Real>> & component_trajectories,
                             unsigned int time_index) const;
 
private:
  void validateTrajectoryShape(const std::vector<std::vector<Real>> & component_trajectories) const;
 
  const unsigned int _input_timesteps;
};
 
#endif