SSIT/Examples/example_2_SolveSSITModels_ODE.m at main · MunskyGroup/SSIT · GitHub

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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
%% SSIT/Examples/example_2_SolveSSITModels_ODE

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% Section 2.2: Finding and visualizing master equation solutions
%   * Compute Ordinary Differential Equations (ODEs)
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%% Preliminaries
% Use the models from example_1_CreateSSITModels
% clear
% close all

% example_1_CreateSSITModels

% Load the models created in example_1_CreateSSITModels
% load('example_1_CreateSSITModels.mat')

% View model summaries:
Model.summarizeModel
STL1.summarizeModel
STL1_4state.summarizeModel

% Set the times at which distributions will be computed:
Model.tSpan = linspace(0,50,101);
STL1.tSpan = linspace(0,50,101);
STL1_4state.tSpan = linspace(0,50,101);

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% Ex(1): Use deterministic, ordinary differential equations (ODEs)
% to average the time evolution of state space probabilities for
% the bursting gene example model from example_1_CreateSSITModels
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%% Model:
    % Create a copy of the bursting gene model for ODEs:
    Model_ODE = Model;

    % Set solution scheme to 'ODE':
    Model_ODE.solutionScheme = 'ODE';

    % This function compiles and stores the given reaction propensities
    % into symbolic expression functions that use sparse matrices to
    % operate on the system based on the current state. The functions are
    % stored with the given prefix, in this case, 'Model_ODE':
    Model_ODE = Model_ODE.formPropensitiesGeneral('Model_ODE');

    % Solve ODEs:
    Model_ODE.Solutions = Model_ODE.solve;

    % Plot ODE solutions:
    Model_ODE.plotODE(speciesNames=Model_ODE.species,...
        timeVec=Model_ODE.tSpan, lineProps={'linewidth',4},...
        Title='Busting Gene', TitleFontSize=32, AxisLabelSize=24,...
        TickLabelSize=24, LegendFontSize=15, LegendLocation='southeast',...
        XLabel='Time', YLabel='Molecule Count')

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% Ex(2): Use deterministic, ordinary differential equations (ODEs)
% to average the time evolution of state space probabilities for
% the time-varying STL1 yeast model from example_1_CreateSSITModels
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%% STL1 Model:
    % Create a copy of the time-varying STL1 yeast model for ODEs:
    STL1_ODE = STL1;

    % Set solution scheme to 'ODE':
    STL1_ODE.solutionScheme = 'ODE';

    % This function compiles and stores the given reaction propensities
    % into symbolic expression functions that use sparse matrices to
    % operate on the system based on the current state. The functions are
    % stored with the given prefix, in this case, 'STL1_ODE':
    STL1_ODE = STL1_ODE.formPropensitiesGeneral('STL1_ODE');

    % Solve ODEs:
    STL1_ODE.Solutions = STL1_ODE.solve;

    % Plot ODE solutions:
    STL1_ODE.plotODE(speciesNames=STL1_ODE.species,...
        timeVec=STL1_ODE.tSpan, lineProps={'linewidth',4},...
        Title='STL1', TitleFontSize=32,...
        AxisLabelSize=24, TickLabelSize=24,...
        LegendFontSize=15, LegendLocation='northwest',...
        XLabel='Time', YLabel='Molecule Count')

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%% Ex(3): Use deterministic, ordinary differential equations (ODEs)
% to average the time evolution of state space probabilities for the
% 4-state time-varying STL1 yeast model from example_1_CreateSSITModels
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

%% 4-state STL1 Model:
    % Create a copy of the time-varying STL1 yeast model for ODEs:
    STL1_4state_ODE = STL1_4state;

    % Set solution scheme to 'ODE':E
    STL1_4state_ODE.solutionScheme = 'ODE';

    % Set the ODE integrator (default 'ode23s'):
    STL1_4state_ODE.odeIntegrator = 'ode23s';

    % This function compiles and stores the given reaction propensities
    % into symbolic expression functions that use sparse matrices to
    % operate on the system based on the current state. The functions are
    % stored with the given prefix, in this case, 'STL1_4state_ODE':
    STL1_4state_ODE = ...
        STL1_4state_ODE.formPropensitiesGeneral('STL1_4state_ODE');

    % Solve ODEs:
    STL1_4state_ODE.Solutions = STL1_4state_ODE.solve;

    % Plot ODE solutions for mRNA:
    STL1_4state_ODE.plotODE(speciesNames=STL1_4state_ODE.species(5),...
        timeVec=STL1_4state_ODE.tSpan, lineProps={'linewidth',4},...
        TitleFontSize=26, Title='4-state STL1 (mRNA)',...
        AxisLabelSize=20, TickLabelSize=20, LegendFontSize=20,...
        LegendLocation='east', Colors=[0.23,0.67,0.20],...
        XLabel='Time', YLabel='Molecule Count')

    % Plot ODE solutions for the four gene states:
    STL1_4state_ODE.plotODE(speciesNames=STL1_4state_ODE.species(1:4),...
        timeVec=STL1_4state_ODE.tSpan, lineProps={'linewidth',4},...
        TitleFontSize=26, Title='4-state STL1 (gene states)',...
        AxisLabelSize=20, TickLabelSize=20, LegendFontSize=20,...
        LegendLocation='east', XLabel='Time', YLabel='Molecule Count',...
        makeMovie=true, videoFileName='STL1_4state_ODE.mp4')

    %% Make a movie of the ODE solution being plotted:
    % makeODEmovie(STL1_4state_ODE.Solutions, STL1_4state_ODE.species, ...
    %              STL1_4state_ODE.tSpan, 'STL1_4state_ODE.mp4');


%% Save ODE models & solutions
saveNames = unique({'Model_ODE'
    'STL1_ODE'
    'STL1_4state_ODE'
    });

save('example_2_SolveSSITModels_ODE',saveNames{:})