framingham 0.1.20130712221655 → 0.1.20130719170918

data/Rakefile

@@ -9,7 +9,17 @@ task :deploy do
   sh "gem push framingham.gem"
 end
 
+task :upgrade do
+  Rake::Task[:build].execute
+  Rake::Task[:test].execute
+  begin
+    sh "gem uninstall framingham && gem install ./framingham.gem"
+  rescue => exception
+    abort "\033[31merror: did you not sudo? \033[0m"
+  end
+end
+
 Rake::TestTask.new { |_|
-  _.test_files = FileList['framingham/*_tests.rb']
+  _.test_files = FileList['framingham/tests/*.rb']
   _.verbose = true
 }

data/framingham.gemspec

@@ -22,7 +22,7 @@ Gem::Specification.new{ |_|
   _.homepage      = "https://github.com/3poundhealth/framingham#framingham"
   _.summary       = "Gem to implement Framingham Heart Study calculators"
   _.license       = "MIT"
-  _.files         = `git ls-files`.split($/)
+  _.files         = `git ls-files`.split($/).push("framingham/version.rb")
   _.executables   = _.files.grep(%r{^/}) { |f| File.basename(f) }
   _.test_files    = _.files.grep(%r{^(test|spec|features)/})
   _.require_path  = "."

data/framingham.rb

@@ -1,6 +1,5 @@
 #!/usr/bin/ruby
-load 'framingham/version.rb'
-load 'framingham/heartdisease.rb'
+require './framingham/version'
 
 module Framingham extend self
 
@@ -10,11 +9,21 @@ end
 
 protected
 
+$, = ", " #default join
+
+GENDERS = [
+  :male,
+  :female
+] unless defined? GENDERS
+
 def to_i value
   (value.eql? true or value.to_s == "true" or (
     value.to_f != 0 if value.respond_to? 'to_f'
   )) ? 1 : 0
 end
 
+load './framingham/heartdisease.rb'
+load './framingham/diabetes.rb'
+
 end
 

data/framingham/diabetes.rb

@@ -0,0 +1,76 @@
+module Framingham::Diabetes extend Framingham, self
+include Framingham
+
+def eval options = {}
+  begin
+    #initialize
+    options = NORMAL.merge options
+    @age = options[:age].to_f
+    @blood_pressure = options[:blood_pressure].to_f
+    @body_mass_index = options[:body_mass_index].to_f
+    @diabetes_relative = options[:diabetes_relative].to_s.to_sym
+    @fasting_glucose = options[:fasting_glucose].to_f
+    @gender = options[:gender].to_sym
+    @hdl = options[:hdl].to_f
+    @triglyceride = options[:triglyceride].to_f
+    #validate
+    raise ERRORS[:gender] unless GENDERS.include? @gender
+  rescue => exception
+    abort "\033[31merror: " + exception.message + "\033[0m"
+  else
+    #calculate
+=begin debug stuff
+    puts options.to_s
+    puts "50-65:               " + (50...65).include?(@age).to_s
+    puts "65+:                 " + (@age >= 65).to_s
+    puts "blood_pressure >130: " + (@blood_pressure > 130).to_s
+    puts "bmi 25-30:           " + (25...30).include?(@body_mass_index).to_s
+    puts "bmi >=30:            " + (@body_mass_index >= 30).to_s
+    puts "diabetes_relative:   " + @diabetes_relative.eql?(:true).to_s
+    puts "fasting_glucose:     " + (100..126).include?(@fasting_glucose).to_s
+    puts "gender:              " + @gender.to_s
+    puts "hdl:                 " + (@gender.eql?(:female) and @hdl < 50 or @hdl < 40).to_s
+    puts "triglyceride >= 150: " + (@triglyceride >= 150).to_s
+=end
+    risk_factor = BETA_ZERO[@gender]
+    risk_factor += 0.018 if (50...65).include? @age
+    risk_factor += 0.081 if @age >= 65
+    risk_factor -= 0.565 if @diabetes_relative.eql? :true
+    risk_factor -= 0.301 if (25...30).include? @body_mass_index
+    risk_factor -= 0.920 if @body_mass_index >= 30
+    risk_factor -= 0.498 if @blood_pressure > 130
+    risk_factor -= 0.944 if @gender.eql? :female and @hdl < 50 or @hdl < 40
+    risk_factor -= 0.575 if @triglyceride >= 150
+    risk_factor -= 1.980 if (100..126).include? @fasting_glucose
+    1 / (1 + Math.exp(risk_factor))
+  end
+end
+
+def help
+  "Evaluate 8 year diabetes risk"
+end
+
+protected
+
+NORMAL = { #default options
+  age:                      30,
+  blood_pressure:           125,
+  body_mass_index:          22.5,
+  diabetes_relative:        false,
+  fasting_glucose:          85,
+  gender:                  :female,
+  hdl:                      60,
+  triglyceride:             130
+} unless defined? NORMAL
+
+ERRORS = {
+  gender: "invalid gender, options: " + GENDERS.join
+} unless defined? ERRORS
+
+BETA_ZERO = {
+  male:   5.517 + 0.010,
+  female: 5.517
+}
+
+end
+

data/framingham/heartdisease.rb

@@ -9,9 +9,11 @@ def eval options = {}
     @blood_pressure = options[:blood_pressure].to_f
     @blood_pressure_treatment = options[:blood_pressure_treatment] ? :treated : :untreated
     @body_mass_index = options[:body_mass_index].to_f
-    @gender = options[:gender]
+    @gender = options[:gender].to_sym
     @diabetes = to_i options[:diabetes]
     @smoker = to_i options[:smoker]
+    #validate
+    raise ERRORS[:gender] unless GENDERS.include? @gender
   rescue => exception
     abort "\033[31merror: " + exception.message + "\033[0m"
   else
@@ -79,8 +81,6 @@ def internal_debug _ = {}
   '%5.1f' % ((1000 * _[:optimal]).round / 10.0)
 end
 
-$, = ", " #default join
-
 NORMAL = { #default options
   age:                      30,
   blood_pressure:           125,
@@ -102,16 +102,11 @@ BLOOD_PRESSURE_RANGE = (90..200) unless defined? BLOOD_PRESSURE_RANGE
 
 BMI_RANGE = (15..50) unless defined? BMI_RANGE
 
-GENDERS = [
-  :male,
-  :female
-] unless defined? GENDERS
-
 ERRORS = {
   age:              "age must be in range: " + AGE_RANGE.to_s,
   blood_pressure:   "blood pressure must be in range: " + BLOOD_PRESSURE_RANGE.to_s,
   bmi:              "bmi must be in range: " + BMI_RANGE.to_s,
-  gender:           "invalid gender, options: " + GENDERS.join,
+  gender:           "invalid gender, options: " + GENDERS.join
 } unless defined? ERRORS
 
 BETA_ZERO = {

data/framingham/tests/diabetes_tests.rb

@@ -0,0 +1,16 @@
+require 'test/unit'
+require './framingham'
+
+class DiabetesTests < Test::Unit::TestCase
+include Framingham::Diabetes
+
+def yellow _
+  "\e[1;7;41;33mFailed:\e[1;3;40m " + _ + " \e[0m"
+end
+
+def test_normal
+  assert_equal 0.004001804744896386, eval, (yellow "Normal test")
+end
+
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: framingham
 version: !ruby/object:Gem::Version
-  version: 0.1.20130712221655
+  version: 0.1.20130719170918
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-12 00:00:00.000000000 Z
+date: 2013-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -57,8 +57,11 @@ files:
 - bin/rake
 - framingham.gemspec
 - framingham.rb
+- framingham/diabetes.rb
 - framingham/heartdisease.rb
-- framingham/heartdisease_tests.rb
+- framingham/tests/diabetes_tests.rb
+- framingham/tests/heartdisease_tests.rb
+- framingham/version.rb
 homepage: https://github.com/3poundhealth/framingham#framingham
 licenses:
 - MIT