vollmersj / coexist.jl Goto Github PK

ERROR: TensorOperations.IndexError{String}("invalid permutation of length 1: (1, 2, 3, 4)")
Stacktrace:
 [1] contract!(::Bool, ::Array{Operation,1}, ::Symbol, ::Array{Float64,7}, ::Symbol, ::Bool, ::SubArray{Float64,3,Array{Float64,4},Tuple{Int64,Base.Slice{Base.OneTo{Int64}},Base.Slice{Base.OneTo{Int64}},Base.Slice{Base.OneTo{Int64}}},true}, ::Tuple{}, ::NTuple{4,Int64}, ::Tuple{Int64,Int64,Int64}, ::NTuple{4,Int64}, ::Tuple{Int64,Int64,Int64}, ::Tuple{Symbol,Symbol,Symbol}) at /Users/vollmer/.julia/packages/TensorOperations/PuhjE/src/implementation/stridedarray.jl:262
 [2] contract!(::Bool, ::Array{Operation,1}, ::Symbol, ::Array{Float64,7}, ::Symbol, ::Bool, ::SubArray{Float64,3,Array{Float64,4},Tuple{Int64,Base.Slice{Base.OneTo{Int64}},Base.Slice{Base.OneTo{Int64}},Base.Slice{Base.OneTo{Int64}}},true}, ::Tuple{}, ::NTuple{4,Int64}, ::Tuple{Int64,Int64,Int64}, ::NTuple{4,Int64}, ::Tuple{Int64,Int64,Int64}, ::Tuple{}, ::Tuple{Symbol,Symbol,Symbol}) at /Users/vollmer/.julia/packages/TensorOperations/PuhjE/src/implementation/stridedarray.jl:85
 [3] f!(::Array{Any,1}, ::Array{Operation,1}, ::Array{Any,1}, ::Operation) at /Users/vollmer/covidwarwick/coexist_julia/julia/highlevel.jl:89
 [4] (::ODEFunction{true,typeof(f!),LinearAlgebra.UniformScaling{Bool},Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing})(::Array{Any,1}, ::Vararg{Any,N} where N) at /Users/vollmer/.julia/packages/DiffEqBase/BZxEO/src/diffeqfunction.jl:248
 [5] modelingtoolkitize(::ODEProblem{Array{Float64,4},Tuple{Float64,Float64},true,DiffEqBase.NullParameters,ODEFunction{true,typeof(f!),LinearAlgebra.UniformScaling{Bool},Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing,Nothing},Base.Iterators.Pairs{Symbol,Nothing,Tuple{Symbol},NamedTuple{(:p,),Tuple{Nothing}}},DiffEqBase.StandardODEProblem}) at /Users/vollmer/.julia/packages/ModelingToolkit/aKa4S/src/systems/diffeqs/modelingtoolkitize.jl:19
 [6] top-level scope at none:0

using ModelingToolkit
using DifferentialEquations



nIso=4
nTest=4
nHS=8
nAge=9

x=1
# yeah

ssize=(nAge, nHS, nIso, nTest)


function f_vec!(dstateTensor_vec,stateTensor_vec,p,t)
  dstateTensor=reshape(dstateTensor_vec,(nAge, nHS, nIso, nTest))
  stateTensor=reshape(stateTensor_vec,(nAge, nHS, nIso, nTest))
  trTensor_complete = zeros((nAge, nHS, nIso, nTest, nHS, nIso, nTest))
  for iso=1:nIso
    for ag=1:nAge
      for hs=1:nHS
        for tst=1:nTest
          for nhs=1:nHS #new health state
              trTensor_complete[ag,hs,iso,tst,nhs,iso,tst]+=1.0
          end
        end
      end
    end
  end
  dstateTensor=zeros((nAge, nHS, nIso, nTest))

  for ag=1:nAge
    for hs=1:nHS
      for iso=1:nIso
        for tst=1:nTest
          for nhs=1:nHS
            for niso=1:nIso
              for ntst=1:nTest
                dstateTensor[ag,hs,iso,tst]+=stateTensor[1,nhs,niso,ntst,]* trTensor_complete[ag,nhs,niso,ntst,hs,iso,tst]
              end
            end
          end
        end
      end
    end
  end
  dstate[:]=reshape(dstateTensor,(nIso*nAge*nHS*nTest))
end

n=prod([ssize...])
prob = ODEProblem(f_vec!,100*ones(n),(0.0,80.0),p=nothing)
sys = modelingtoolkitize(prob)

(;  social_distiance= (;lockdown_lift=Date("2020-06-01")) )