3 4[\@s2dZddlZddlZddlmZGdddZdS)u This class calculates all the risk related with the cardiovascular system. This is based on the article: D’Agostino, R. B., Vasan, R. S., Pencina, M. J., Wolf, P. A., Cobain, M., & Massaro, J. M. General cardiovascular risk profile for use in primary care the Framingham heart study. Circulation [Internet]. 2008; 117: 743-53. Doi: https://doi.org/10.1161/CIRCULATIONAHA.107.699579 @author: Vertigo Biosystems N)expc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/S)0CardiovascularRiskcCsT|jj|_|jj|_|jj|_|jj|_|jj|_|jj |_ |jj |_ |jj |_ dS)N) analytics cholesterolcholesterol_hdl anthropometryagesexhereditary_pathologiesdiabetes lifestylesmoker sbp_treatedsbp)self health_inforW/var/www/vitahealth/hygea.vitahealth/riskmap/src/vbs_riskmap/risk/CardiovascularRisk.py__init__s       zCardiovascularRisk.__init__cCs0|jdk r(|jdkrdS|jdkr,dSndSdS)aN This function runs a basic data for de baseline 10-year survival. This test is based on the Cox-formula article that can be found in documentation Parameters ---------- self: sex Returns ------- baseline_risk: int baseline 10-years survival NFgbg?MgKu?)r )rrrr baseline_survival_cardiovascular%s    z3CardiovascularRisk.baseline_survival_cardiovascularc Cs|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdkob|jdkrtj ddd(ddd d g}tj d d d)ddddg}tj |jtj |jtj |jg}|jdkr|j d|j tj |jn|j tj |j|j d|j |j|j |jtj dddddddg}tj dddddd d!g}|jd"krjt ||}t ||}n.|jd#krt ||}t ||}nt d$t||}d%|j|} | d&} nd} tjd't| | S)*a This function runs a basic test for cardiovascular risks. TBased on the original test made in the Cox.article. More information in the documentation. Parameters ---------- self: age (between 30 and 75 years old), cholesterol, cholesterol_hdl, sex, sbp, sbp_treated, smoker, diabetes. Returns ------- risk: float final risk NKgwkF}@gB>٬?gc?gvݰ?g%1 ?go?gAH0[?gwb֋@gS:X?g_?g@g @g%@M-[?g|~!?Trg +@g|?5^@g&@gBfj@gȘ?gCiq?gp= ף?gq@gAf@gn@g(\@gT?gcZB>?gUN@?rrz"Sex must be M (male) or F (female)dz2CardiovascularRisk: cox_cardiovascular_risk_test: gcg_)rrr rrrr r nparraylogappendsum ValueErrorrrloggingdebugstr) rZmale_ErCZ female_ErCZ risk_factorZmean_risk_factor_values_maleZmean_risk_factor_values_femaleZfirst_coefficientZsecond_coefficientexponentriskrrrrisk_cox_cardiovascular_test>sL                 z/CardiovascularRisk.risk_cox_cardiovascular_testcCs|jtddkrdS|jtddkr(dS|jtddkrr,r.N)rr9)rrrr"punctuation_non_treated_sbp_females  z5CardiovascularRisk.punctuation_non_treated_sbp_femalecCsp|jdkrd S|jtddkr"dS|jtddkr6dS|jtddkrJdS|jtdd kr^d S|jd krld Sd S)ac This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for treated sbp. Parameters ---------- self: sbp Returns ------- punctuation: int final punctuation for treated sbp in women rCrrDr*rErArFr.r>r0Nr<)rr9)rrrrpunctuation_treated_sbp_females  z1CardiovascularRisk.punctuation_treated_sbp_femalecCs|jdkrdSdSdS)a] This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for smoker. Parameters ---------- self: smoker Returns ------- punctuation: int final punctuation for smoking in women TrArN)r )rrrrpunctuation_smoker_female&s z,CardiovascularRisk.punctuation_smoker_femalecCs|jdkrdSdSdS)ab This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for diabetic. Parameters ---------- self: diabetes Returns ------- punctuation: int final punctuation for diabetic in women Tr,rN)r )rrrrpunctuation_diabetic_female<s z.CardiovascularRisk.punctuation_diabetic_femalecCs2|d.kr dS|d/krdS|dkr$dS|dkr0dS|dkr80r)rrVrrrheart_age_females@z#CardiovascularRisk.heart_age_femalecCs|jtddkrdS|jtddkr(dS|jtddkrrr?rr@r*irAr,N)rr9)rrrrpunctuation_cholesterol_males  z/CardiovascularRisk.punctuation_cholesterol_malecCs\|jdkrd S|jtddkr"dS|jtddkr6dS|jtddkrJdS|jdkrXdSd S) ah This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for non treated sbp. Parameters ---------- self: sbp Returns ------- punctuation: int final punctuation for non treated sbpin men rCr*rDrrErr>rANr;)rr9)rrrr punctuation_non_treated_sbp_males  z3CardiovascularRisk.punctuation_non_treated_sbp_malecCs\|jdkrdS|jtddkr"dS|jtddkr6dS|jtddkrJdS|jdkrXd Sd S) aa This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for treated sbp. Parameters ---------- self: sbp Returns ------- punctuation: int final punctuation for treated sbp in men rCrrDr*rErAr>r,r.N)rr9)rrrrpunctuation_treated_sbp_males  z/CardiovascularRisk.punctuation_treated_sbp_malecCs|jdkrdSdSdS)a[ This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for smoker. Parameters ---------- self: smoker Returns ------- punctuation: int final punctuation for smoking in men Tr,rN)r )rrrrpunctuation_smoker_male/s z*CardiovascularRisk.punctuation_smoker_malecCs|jdkrdSdSdS)a` This function compute the punctuation for point-based framingham test in cardivascular risk. Based on the punctuation given in the paper for diabetic. Parameters ---------- self: diabetes Returns ------- punctuation: int final punctuation for diabetic in men TrArN)r )rrrrpunctuation_diabetic_maleEs z,CardiovascularRisk.punctuation_diabetic_malecCs|d)kr dS|d*krdS|d+kr$dS|dkr0dS|dkr80r)rrVrrrheart_age_malesL z!CardiovascularRisk.heart_age_malecCsF|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdkob|jdkr,|jdkrvd}nd}|jdkr|j|j |j |j ||j |j|j |j}|j|}nX|jdkr0|j|j|j|j||j|j|j|j}|j|}nd}tjd t||S) z Calculates the framingham test Inputs: cholesterol, cholesterol_hdl, age, sex, diabetes, smoker, sbp, sbp_treated Outputs: risk NrrTrrrrz*CardiovascularRisk: risk_framingham_test: )rrrr r r rrr:r=rBrHrJrKrLrWrerfrgrhrirjrkrlr"r#r$)rsbp_aux_non_treatmentrVr&rrrrisk_framingham_tests(   8   8   z'CardiovascularRisk.risk_framingham_testcCsF|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdk ob|jdkob|jdkr,|jdkrvd}nd}|jdkr|j|j |j |j ||j |j|j |j}|j|}nX|jdkr0|j|j|j|j||j|j|j|j}|j|}nd}tjd t||S) z Calculates "the heart age" Inputs: cholesterol, cholesterol_hdl, age (between 30 and 75 years old), sex, diabetes, smoker, sbp, sbp_treated Outputs: heart_age NrrTrrrrz%CardiovascularRisk: risk_heart_age : )rrrr r r rrr:r=rBrHrJrKrLrdrerfrgrhrirjrkrsr"r#r$)rrtrV heart_agerrrrisk_heart_ages(   8   8   z!CardiovascularRisk.risk_heart_ageN)__name__ __module__ __qualname__rrr'r:r=rBrHrJrKrLrWrdrerfrgrhrirjrkrlrsrurwrrrrrs.N&2$.(2r)__doc__r"numpyrmathrrrrrr s