Ejemplo n.º 1
0
print("SIMPLE EPIDEMIC TRANSMISION MODEL")
print("Human to human disease simulation")
print("---------------------------------")
print("---https://github/rvalla/SETM----")
print("#################################")

#Running desired number of simulations...
for i in range(simulationsCount):
    print("", end="\n")
    print("Starting simulation number " + str(i + 1), end="\n")
    simulationName = simulationsName
    simulationName = simulationsName + "_" + str(i)
    if i == 0:
        dt.saveConfigStart(simulationsPopulation, simulationsPeriod, simulationName, areaBDensity, runGovActions, \
         govActions, govFailureList, autoIsolationThreshold, startingImmunity, behavior, behaviorTrigger, \
         behaviorOff, behaviorFactor)
    s = sim(simulationsPopulation, simulationsPeriod, i + 1, casesCeroCount, simulationName, areaBDensity, \
     runGovActions, govActions, govFailureList, autoIsolationThreshold, startingImmunity, behavior, \
     behaviorTrigger, behaviorOff, behaviorFactor)
    vz.getFileNames(simulationName)
    vz.populationVisualization(simulationName)
    govActionsCycles = sim.getGovActionsCycles()
    behaviorCycles = sim.getBehaviorCycles()
    print("Building data visualization...", end="\r")
    vz.simulationVisualization(simulationName, runGovActions, govActionsCycles,
                               behavior, behaviorCycles, simulationsPopulation)
    vz.infectionsVisualization(simulationName, simulationsPeriod,
                               runGovActions, govActionsCycles, behavior,
                               behaviorCycles)
    print("Data visualization complete!     ", end="\n")
    print("", end="\n")