add graphs on DB

Models/VICAN/test_ages.py

 
+import sqlite3
+
 import numpy.random as rd
 import numpy as np
 import pandas as pd
-from scipy.stats import truncnorm, bernoulli
-import matplotlib
+
+from scipy.stats import truncnorm, gaussian_kde
 import matplotlib.pyplot as plt
-from scipy.stats import gaussian_kde
+
 from SNDSGenerator.simuExternal import simuExternal
 from SNDSGenerator.simulationDB import simDB
 
@@ -25,8 +27,6 @@ from SNDSGenerator.pathway_factory import pathwayInjector
 from SNDSGenerator.database_model import (
     Patient,
     ShortHospStay,
-    DrugDelivery,
-    MedicalAct,
 )
 from SNDSGenerator.simulation import simulation
 
@@ -157,18 +157,24 @@ for patient in simulator.patients:
     first_event_ages.append( (patient.hospitalStays[0].start_date-patient.BD).days/365.25 )
     first_event_ages_internal.append( patient.age( patient.hospitalStays[0].start_date)  )
 
-    print(dates[-1], patient.hospitalStays[0].start_date, first_event_ages[-1])
+    #print(dates[-1], patient.hospitalStays[0].start_date, first_event_ages[-1])
+
+
+dbgen = simDB()
+dbgen.output_db_name = "Models/VICAN/snds_testage.db"
+dbgen.generate(simulator, rootschemas="schema-snds/schemas")
 
-#dbgen = simDB()
-#dbgen.output_db_name = "Models/VICAN/snds_testage.db"
-#dbgen.generate(simulator, rootschemas="schema-snds/schemas")
+conn = sqlite3.connect(dbgen.output_db_name)
+final_ages = pd.read_sql_query("SELECT AGE_ANN FROM T_MCOaaB;", conn)['AGE_ANN'].to_numpy()
 
 density = gaussian_kde(samples1)
 density_first_event = gaussian_kde(first_event_ages)
 density_first_event_internal = gaussian_kde(first_event_ages_internal)
+density_final = gaussian_kde(final_ages)
 xs = np.linspace(0, 100, 200)
 fig=plt.figure(figsize=(14, 8))
 plt.plot(xs,density(xs),c='r')
 plt.plot(xs,density_first_event(xs),c='b')
 plt.plot(xs,density_first_event_internal(xs),c='g')
-fig.savefig('testage.pdf')
\ No newline at end of file
+plt.scatter(xs,density_final(xs),c='k',marker="x")
+fig.savefig('testage.pdf')