Add routine printing equations where a searched variable appears in.
parent
2eb793a1f9
commit
b4cd1c0f57
|
@ -0,0 +1,93 @@
|
||||||
|
function print_equations(variable_name, withexpansion)
|
||||||
|
|
||||||
|
% Prints equations where the variable appears in.
|
||||||
|
%
|
||||||
|
% INPUTS
|
||||||
|
% - variable_name [string] Name of the variable to be traced.
|
||||||
|
% - withexpansion [logical] Prints expanded equation of the VAR_EXPECTATION or PAC_EXPECTATION term if true.
|
||||||
|
%
|
||||||
|
% OUTPUTS
|
||||||
|
% None
|
||||||
|
|
||||||
|
% Copyright (C) 2019 Dynare Team
|
||||||
|
%
|
||||||
|
% This file is part of Dynare.
|
||||||
|
%
|
||||||
|
% Dynare is free software: you can redistribute it and/or modify
|
||||||
|
% it under the terms of the GNU General Public License as published by
|
||||||
|
% the Free Software Foundation, either version 3 of the License, or
|
||||||
|
% (at your option) any later version.
|
||||||
|
%
|
||||||
|
% Dynare 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 General Public License for more details.
|
||||||
|
%
|
||||||
|
% You should have received a copy of the GNU General Public License
|
||||||
|
% along with Dynare. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
global M_
|
||||||
|
|
||||||
|
if nargin == 0
|
||||||
|
error('Provide variable name as input argument.');
|
||||||
|
elseif nargin == 1
|
||||||
|
% Assign default value for VAR_EXPECTATION/PAC_EXPECTATION expression expansion.
|
||||||
|
withexpansion = false;
|
||||||
|
elseif nargin == 2
|
||||||
|
if ~islogical(withexpansion)
|
||||||
|
error('Second input argument must be Boolean.');
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
% Check if corresponding JSON file exists.
|
||||||
|
fname = [M_.fname filesep 'model' filesep 'json' filesep 'modfile-original.json'];
|
||||||
|
if exist(fname, 'file') ~= 2
|
||||||
|
error('Could not find %s! Please use the json option (See the Dynare invocation section in the reference manual).', fname);
|
||||||
|
end
|
||||||
|
|
||||||
|
% Check that the first input is a character array.
|
||||||
|
if ~ischar(variable_name)
|
||||||
|
error('First input argument must be a string.');
|
||||||
|
end
|
||||||
|
|
||||||
|
% Check that the variable is actually a variable in the model.
|
||||||
|
if ~ismember(variable_name, [M_.exo_names; M_.endo_names])
|
||||||
|
error('There is no variable named %s!', variable_name);
|
||||||
|
end
|
||||||
|
|
||||||
|
% Load the JSON file.
|
||||||
|
jsonfile = loadjson(fname);
|
||||||
|
model = jsonfile.model;
|
||||||
|
|
||||||
|
% Print the equations the variable appears in.
|
||||||
|
for it = 1:length(M_.mapping.(variable_name).eqidx)
|
||||||
|
rhs = model{M_.mapping.(variable_name).eqidx(it)}.rhs;
|
||||||
|
if withexpansion
|
||||||
|
if isfield(M_, 'pac') && contains(rhs, 'pac_expectation')
|
||||||
|
% Get the index of the equation's PAC model.
|
||||||
|
models = fieldnames(M_.pac);
|
||||||
|
idx = find(~cellfun('isempty',cellfun(@(s)find(contains(rhs,s)),models,'uni',0)));
|
||||||
|
% Get the expanded PAC_EXPECTATION term.
|
||||||
|
[pac_expression, growthneutralitycorrection] = write_expectations(M_.pac.(models{idx}).tag_map(:,1), models{idx}, 'pac', true);
|
||||||
|
expression = [sprintf('\n\t + %s', growthneutralitycorrection) TransformExpandedExpr(pac_expression)];
|
||||||
|
rhs = strrep(rhs, ['+pac_expectation(model_name = ' models{idx} ')'], expression);
|
||||||
|
elseif isfield(M_, 'var_expectation') && contains(rhs, 'var_expectation')
|
||||||
|
% Get the index of the equation's VAR model.
|
||||||
|
models = fieldnames(M_.var_expectation);
|
||||||
|
idx = find(~cellfun('isempty',cellfun(@(s)find(contains(rhs,s)),models,'uni',0)));
|
||||||
|
% Get the expanded VAR_EXPECTATION term.
|
||||||
|
expression = write_expectations('fake', models{idx}, 'var', true);
|
||||||
|
expression = TransformExpandedExpr(expression);
|
||||||
|
rhs = strrep(rhs, ['+var_expectation(model_name = ' models{idx} ')'], expression);
|
||||||
|
elseif ~isfield(M_, 'pac') && ~isfield(M_, 'var_expectation')
|
||||||
|
warning('No VAR or PAC expectations found, continuing without expansion');
|
||||||
|
withexpansion = false;
|
||||||
|
end
|
||||||
|
end
|
||||||
|
fprintf('%s = %s;\n', model{M_.mapping.(variable_name).eqidx(it)}.lhs, rhs);
|
||||||
|
end
|
||||||
|
|
||||||
|
function [transformed_expression] = TransformExpandedExpr(expression)
|
||||||
|
transformed_expression = splitlines(expression);
|
||||||
|
transformed_expression{1} = sprintf(' + %s', transformed_expression{1});
|
||||||
|
transformed_expression = sprintf('\n\t%s', transformed_expression{1:end-1});
|
|
@ -1 +1 @@
|
||||||
Subproject commit 2a127b1f2396fa52e26773b9314ec4e7621dde4e
|
Subproject commit 0f0299bc1c796b489da7fd485266a66f58c92236
|
|
@ -108,6 +108,15 @@ pac.estimate.iterative_ols('zpac', eparams, edata, 2005Q1:2000Q1+200);
|
||||||
// Test printing of PAC expectations
|
// Test printing of PAC expectations
|
||||||
pac.print('pacman','zpac');
|
pac.print('pacman','zpac');
|
||||||
|
|
||||||
|
// Print equations where the variable appears in
|
||||||
|
fprintf('x1bar is in: \n')
|
||||||
|
print_equations('x1bar')
|
||||||
|
fprintf('\n')
|
||||||
|
|
||||||
|
fprintf('x2bar is in: \n')
|
||||||
|
print_equations('x2bar', true);
|
||||||
|
fprintf('\n')
|
||||||
|
|
||||||
e_c_m_iterative_ols = M_.params(strmatch('e_c_m', M_.param_names, 'exact'));
|
e_c_m_iterative_ols = M_.params(strmatch('e_c_m', M_.param_names, 'exact'));
|
||||||
c_z_1_iterative_ols = M_.params(strmatch('c_z_1', M_.param_names, 'exact'));
|
c_z_1_iterative_ols = M_.params(strmatch('c_z_1', M_.param_names, 'exact'));
|
||||||
c_z_2_iterative_ols = M_.params(strmatch('c_z_2', M_.param_names, 'exact'));
|
c_z_2_iterative_ols = M_.params(strmatch('c_z_2', M_.param_names, 'exact'));
|
||||||
|
|
|
@ -57,6 +57,15 @@ var_expectation.initialize('varexp')
|
||||||
// Update VAR_EXPECTATION reduced form parameters
|
// Update VAR_EXPECTATION reduced form parameters
|
||||||
var_expectation.update('varexp');
|
var_expectation.update('varexp');
|
||||||
|
|
||||||
|
// Print equations where the variable appears in
|
||||||
|
fprintf('x is in: \n')
|
||||||
|
print_equations('x')
|
||||||
|
fprintf('\n')
|
||||||
|
|
||||||
|
fprintf('y is in: \n')
|
||||||
|
print_equations('y', true)
|
||||||
|
fprintf('\n')
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** REMARK The VAR model is such that x depends on past values of x
|
** REMARK The VAR model is such that x depends on past values of x
|
||||||
** (xₜ₋₁ and xₜ₋₂) and on zₜ₋₂ ⇒ yₜ₋₁, yₜ₋₂ and zₜ₋₁ do not bring any
|
** (xₜ₋₁ and xₜ₋₂) and on zₜ₋₂ ⇒ yₜ₋₁, yₜ₋₂ and zₜ₋₁ do not bring any
|
||||||
|
|
Loading…
Reference in New Issue