File: TransitionStepMultiStageRegressionDP.h

package info (click to toggle)
stopt 5.12%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 8,860 kB
  • sloc: cpp: 70,456; python: 5,950; makefile: 72; sh: 57
file content (124 lines) | stat: -rw-r--r-- 6,225 bytes parent folder | download | duplicates (2)
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
// Copyright (C) 2023 EDF
// All Rights Reserved
// This code is published under the GNU Lesser General Public License (GNU LGPL)
#ifndef TRANSITIONSTEPMULTISTAGEREGRESSIONDP_H
#define TRANSITIONSTEPMULTISTAGEREGRESSIONDP_H
#ifdef OMP
#include <omp.h>
#endif
#include <memory>
#ifdef USE_MPI
#include <boost/mpi.hpp>
#endif
#include <Eigen/Dense>
#include "geners/BinaryFileArchive.hh"
#include "StOpt/core/grids/FullGrid.h"
#include "StOpt/regression/BaseRegression.h"
#include "StOpt/regression/ContinuationValue.h"
#include "StOpt/dp/OptimizerMultiStageDPBase.h"

/** \file TransitionStepMultiStageRegressionDP.h
 * \brief Solve the dynamic programming  problem on one time step by regression with multi thread and mpi without distribution of the data
 *        In this version, on each time time a multistage deterministic problem is solved using DP
 * \author Benoit Clair , Xavier Warin
  */


namespace StOpt
{

/// \class TransitionStepMultiStageRegressionDP TransitionStepMultiStageRegressionDP.h
///        One step of dynamic programming without using mpi
class TransitionStepMultiStageRegressionDP
{
private :

    std::shared_ptr<FullGrid>  m_pGridCurrent ; ///< global grid at current time step
    std::shared_ptr<FullGrid>  m_pGridPrevious ; ///< global grid at previous time step
    std::shared_ptr<OptimizerMultiStageDPBase  >  m_pOptimize ; ///< optimizer solving the problem for one point and one step
    std::shared_ptr<gs::BinaryFileArchive> m_arGen ; ///< geners archive
    std::string  m_nameDump ; ///< name to dump deterministic values

#ifdef USE_MPI
    boost::mpi::communicator  m_world; ///< Mpi communicator
#endif


    /// \brief calculate one step period
    /// \param p_phiIn             cash values at the end of period (nb sim * nb stock points)
    /// \param p_phiOut            cash to be filled at the beginning of period  (nb sim * nb stock points)
    /// \param p_contVal           continuation object
    /// \param p_pGridCurTrans     current grid in transition
    /// \param p_pGridPrevTrans    previous grid in transition
    /// \param p_phiOutLoc        utility for MPI calculations
    /// \param p_ilocToGLobal     utility for MPI calculations
    /// \param p_ilocToGLobalGlob utility for MPI calculations
    /// \param p_storeGlob        utility for MPI calculations
    void oneStageInStep(const std::vector< std::shared_ptr< Eigen::ArrayXXd > > &p_phiIn,
                        std::vector< std::shared_ptr< Eigen::ArrayXXd > > &p_phiOut,
                        std::vector<std::shared_ptr<ContinuationValue> > &p_contVal,
                        const std::shared_ptr<FullGrid> &p_pGridCurTrans,
                        const std::shared_ptr<FullGrid> &p_pGridPrevTrans
#ifdef USE_MPI
                        , std::vector< Eigen::ArrayXXd >   &p_phiOutLoc,
                        Eigen::ArrayXi &p_ilocToGLobal,
                        Eigen::ArrayXi &p_ilocToGLobalGlob,
                        Eigen::ArrayXXd   &p_storeGlob
#endif
                       ) const;


public :

    /// \brief default
    TransitionStepMultiStageRegressionDP() {}
    virtual ~TransitionStepMultiStageRegressionDP() {}

    /// \brief Constructor without arxive dump
    TransitionStepMultiStageRegressionDP(const  std::shared_ptr<FullGrid> &p_pGridCurrent,
                                         const  std::shared_ptr<FullGrid> &p_pGridPrevious,
                                         const  std::shared_ptr<OptimizerMultiStageDPBase > &p_pOptimize
#ifdef USE_MPI
                                         , const boost::mpi::communicator &p_world
#endif
                                        );

    /// \brief Constructor without arxive dump
    TransitionStepMultiStageRegressionDP(const  std::shared_ptr<FullGrid> &p_pGridCurrent,
                                         const  std::shared_ptr<FullGrid> &p_pGridPrevious,
                                         const  std::shared_ptr<OptimizerMultiStageDPBase > &p_pOptimize,
                                         const  std::shared_ptr<gs::BinaryFileArchive>   &p_arGen,
                                         const  std::string &p_nameDump
#ifdef USE_MPI
                                         , const boost::mpi::communicator &p_world
#endif
                                        );

    /// \brief One step for dynamic programming in optimization
    /// \param p_phiIn             for each regime the function value ( nb simulation, nb stocks )
    /// \param p_condExp           Conditional expectation object
    std::vector< std::shared_ptr< Eigen::ArrayXXd > >  oneStep(const std::vector< std::shared_ptr< Eigen::ArrayXXd > > &p_phiIn,
            const std::shared_ptr< BaseRegression>     &p_condExp) const ;


    /// \brief Permits to dump stochastic continuation values on archive during time step optimization
    /// \param p_ar                   archive to dump in
    /// \param p_name                 name used for object
    /// \param p_iStep                 Step number or identifier for time step
    /// \param p_phiIn                for each regime the function value ( nb simulation ,nb stocks)
    /// \param p_condExp               conditional expectation operator
    void dumpContinuationValues(std::shared_ptr<gs::BinaryFileArchive> p_ar, const std::string &p_name, const int &p_iStep, const std::vector< std::shared_ptr< Eigen::ArrayXXd > > &p_phiIn, const  std::shared_ptr<BaseRegression>    &p_condExp) const;


    /// \brief Permits to dump Bellman values on archive
    /// \param p_ar                   archive to dump in
    /// \param p_name                 name used for object
    /// \param p_iStep                 Step number or identifier for time step
    /// \param p_phiIn                for each regime the function value ( nb simulation ,nb stocks)
    /// \param p_condExp               conditional expectation operator
    void dumpBellmanValues(std::shared_ptr<gs::BinaryFileArchive> p_ar, const std::string &p_name, const int &p_iStep, const std::vector< std::shared_ptr< Eigen::ArrayXXd > > &p_phiIn,
                           const  std::shared_ptr<BaseRegression>    &p_condExp) const;
};
}
#endif /* TRANSITIONSTEPMULTISTAGEREGRESSIONDP_H */