dates 1.0.1

data/README.mdown

@@ -0,0 +1,46 @@
+Dates keeps track of things you two have been meaning to do, and learns what's been fun and what hasn't along several different dimensions (adventure, romantic, intellectual, athletic).
+
+## Usage
+
+    $ bundle exec bin/dates.rb --help
+    dates 1.0.0-beta, max & daphne.
+    Usage: dates.rb [command] [options] <ideas-file>
+
+    To see help for a specific command, use the --help switch on that command.
+
+    Commands:
+      choose  Choose a good date idea for right now! [default command]
+      list    Parse and list the contents of the date ideas file.
+
+    Global options:
+      --version, -v:   Print version and exit
+         --help, -h:   Show this message
+
+To run it, the simplest way is just to provide it with an ideas file and nothing else:
+
+    $ bundle exec bin/dates.rb ideas.rb
+    watch _serenity_: (intellectual: 5, athletic: 0, adventure: 2, romantic: 4)
+
+### Commands
+
+The `choose` command picks a date, and it takes two options:
+
+    $ bundle exec bin/dates.rb choose --help
+    dates 1.0.0-beta, max & daphne.
+    Options for `choose` command:
+      --weather, -w <f>:   Current weather score [0-9] (default: 5.0)
+         --time, -t <f>:   Time commitment score [0-9] (default: 5.0)
+
+The `list` command prints out the contents of an ideas file:
+
+    $ bundle exec bin/dates.rb list --help
+    dates 1.0.0-beta, max & daphne.
+    Options for `list` command:
+      --incomplete, -i:   Only the incomplete dates
+        --complete, -c:   Only the completed dates (with feedback)
+
+## Tests
+
+Dates uses RSpec 2 (core and expectations) for tests.  To run them, just call:
+
+    bundle exec rspec test/spec.rb -c -f d

data/bin/dates

@@ -0,0 +1,61 @@
+#!/usr/bin/env ruby
+
+require File.dirname(__FILE__) + '/../lib/dates'
+
+SUB_COMMANDS = %w(list choose)
+VERSION_STR = "dates 1.0.0-beta, max & daphne."
+global_opts = Trollop::options do
+  version VERSION_STR
+  banner <<-EOS
+#{VERSION_STR}
+Usage: dates.rb [command] [options] <ideas-file>
+
+To see help for a specific command, use the --help switch on that command.
+
+Commands:
+  choose  Choose a good date idea for right now! [default command]
+  list    Parse and list the contents of the date ideas file.
+
+Global options:
+EOS
+
+  stop_on SUB_COMMANDS
+end
+
+cmd = ARGV.shift if ARGV.length > 1
+
+cmd_opts = case cmd
+  when "list"
+    Trollop::options do
+      version VERSION_STR
+      banner "#{VERSION_STR}\nOptions for `list` command:"
+      opt :incomplete, "Only the incomplete dates", :short => :i
+      opt :complete, "Only the completed dates (with feedback)", :short => :c
+    end
+  else # `choose` is default command: can be called without explicit specification
+    Trollop::options do
+      version VERSION_STR
+      banner "#{VERSION_STR}\nOptions for `choose` command:"
+      opt :weather, "Current weather score [0-9]", :short => :w, :default => 5.0
+      opt :time, "Time commitment score [0-9]", :short => :t, :default => 5.0
+    end
+  end
+
+if ARGV.length < 1
+  Trollop::die "Ideas file required"
+  exit 1
+end
+
+ideas = eval(File.read(ARGV.first))
+
+if "list" == cmd
+  if cmd_opts[:complete] and cmd_opts[:incomplete]
+    Trollop::die "-i and -c are mutually exclusive options for `list`"
+  else
+    puts ideas.list(cmd_opts)
+  end
+elsif "choose" == cmd
+  puts ideas.choose(cmd_opts[:weather], cmd_opts[:time])
+else
+  puts ideas.choose(cmd_opts[:weather], cmd_opts[:time])
+end

data/lib/date_idea.rb

@@ -0,0 +1,70 @@
+class DateIdea
+
+  def initialize(name = "")
+    @name = name.to_s
+    @attributes = {}
+    @requirements = {}
+    @outcome = nil
+  end
+
+  def kind(args)
+    args.each do |field, value|
+      @attributes[field] = value
+    end
+  end
+
+  def requires(args)
+    args.each do |field, value|
+      @requirements[field] = value
+    end
+  end
+
+  def outcome(score = nil)
+    if score.nil?
+      @outcome
+    else
+      @outcome = score
+    end
+  end
+
+  def outcome_safe
+    return @outcome unless @outcome.nil?
+    -1
+  end
+
+  def attributes
+    @attributes
+  end
+
+  def attribute(name)
+    @attributes[name]
+  end
+
+  def requirements
+    @requirements
+  end
+
+  def name
+    @name
+  end
+
+  def score
+    @outcome
+  end
+
+  def stub?
+    @attributes.empty?
+  end
+
+  def complete?
+    not @outcome.nil?
+  end
+
+  def to_s
+    attr_str = @attributes.map { |k,v| "#{k}: #{v}" }.join(", ")
+    date_str = "#{@name}: (#{attr_str})"
+    date_str = date_str + ", outcome: #{@outcome}" unless @outcome.nil?
+    "#{date_str}\n"
+  end
+
+end

data/lib/date_ideas.rb

@@ -0,0 +1,96 @@
+class DateIdeas
+
+  ATTRIBUTES = %w(adventure romantic athletic intellectual)
+
+  def initialize(&block)
+    @date_ideas = []
+    instance_eval &block
+  end
+
+  def ideas
+    @date_ideas
+  end
+
+  def date(name, &block)
+    date = DateIdea.new(name)
+    if block
+      date.instance_eval(&block)
+    end
+    @date_ideas << date
+  end
+
+  def list(opts)
+    if opts[:complete]
+      render complete
+    elsif opts[:incomplete]
+      render incomplete
+    else
+      render
+    end
+  end
+
+  def render(ideas = nil)
+    if nil == ideas
+      ideas = @date_ideas
+    end
+    str = "Date ideas:\n"
+    ideas.each do |f|
+      if not f.complete?
+        str << "  #{f.to_s}".blue
+      elsif f.complete? and f.score >= 6.0
+        str << "  #{f.to_s}".green
+      elsif f.complete? and f.score < 6.0
+        str << "  #{f.to_s}".red
+      else
+        raise Exception("We should never reach this.")
+      end
+    end
+    str
+  end
+
+  def to_s
+    render
+  end
+
+  def complete
+    @date_ideas.reject { |idea| idea.stub? or not idea.complete? }
+  end
+
+  def incomplete
+    @date_ideas.reject { |idea| idea.stub? or idea.complete? }
+  end
+
+  def extract_features(idea_list)
+    keys = idea_list.reduce([]) { |acc, idea|
+      acc << idea.attributes.keys
+    }.flatten.uniq
+    features = idea_list.map { |idea|
+      attributes = idea.attributes.map { |key, val|
+        [keys.index(key), val]
+      }.sort_by {|k| k[0]}.map { |e| e[1] }
+      [attributes, idea.outcome_safe]
+    }
+    [keys, features]
+  end
+
+  def fitted_model
+    dimensions, examples = complete_features
+    terms = examples.map { |examples|
+      examples[0] + examples[0].permutation(2).to_a.map { |term| term.reduce(1) {|acc,k| acc*k } }
+    }
+  end
+
+  def complete_features
+    extract_features(complete)
+  end
+
+  def incomplete_features
+    extract_features(incomplete)
+  end
+
+  def choose(weather, time)
+    fitted_model
+    incomplete.first
+  end
+
+end

data/lib/dates.rb

@@ -0,0 +1,9 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'colorize'
+require 'statsample'
+require 'trollop'
+require 'date_ideas'
+require 'date_idea'
+
+$LOAD_PATH.shift

data/lib/version.rb

@@ -0,0 +1,3 @@
+module Dates
+  VERSION = "1.0.1"
+end

data/test/spec.rb

@@ -0,0 +1,76 @@
+require File.dirname(__FILE__)+ '/../lib/dates'
+
+ideas = DateIdeas.new do
+
+  date "watch primer in north" do
+    kind :adventure => 2, :romantic => 4, :intellectual => 6, :athletic => 0
+    requires :weather => 0, :hours => 2
+    outcome 20
+  end
+
+  date "very close to primer" do
+    kind :adventure => 2, :romantic => 4, :intellectual => 6, :athletic => 1
+    requires :weather => 0, :hours => 2
+  end
+
+  date "very far from primer" do
+    kind :adventure => 10, :romantic => 10, :intellectual => 0, :athletic => 10
+    requires :weather => 0, :hours => 2
+  end
+
+  date "escher in the DiVE" do
+    kind :adventure => 4, :romantic => 3, :intellectual => 4, :athletic => 0
+    requires :weather => 0, :hours => 2
+  end
+
+  date "mushroom hunting" do
+    kind :adventure => 6, :romantic => 3, :intellectual => 5, :athletic => 3
+    requires :weather => 5, :hours => 2
+  end
+
+  date "duke observatory" do
+    kind :adventure => 100, :romantic => 7, :intellectual => 5, :athletic => 1
+    requires :weather => 8, :hours => 4
+  end
+
+end
+
+valid_attributes = [:adventure, :romantic, :intellectual, :athletic]
+
+describe Date do
+
+  describe "interprets an ideas file" do
+    it "correctly identifies incomplete and completed dates" do
+      ideas.incomplete.length.should == 5
+      ideas.complete.length.should == 1
+      ideas.complete.first.name == "watch primer in north"
+    end
+
+    it "correctly loads the attributes and requirements for each date" do
+      ideas.ideas.map { |idea|
+        if idea.stub?
+          idea.attributes.empty?.should == true
+        else
+          idea.attributes.length.should == valid_attributes.length
+          idea.attributes.map { |name, val|
+            valid_attributes.member?(name).should == true
+          }
+        end
+      }
+    end
+  end
+
+  describe "list command" do
+    it "lists the contents of a dates file correctly" do
+      ideas.ideas.length.should == 6
+    end
+  end
+
+  describe "choose command" do
+    it "chooses a date from the incomplete dates list" do
+      choosed = ideas.choose(2, 2)
+      ideas.incomplete.member?(choosed).should == true
+    end
+  end
+
+end

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification
+name: dates
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Max Hodak, Daphne Ezer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-15 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: statsample
+  requirement: &70326830794380 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70326830794380
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: &70326830793940 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70326830793940
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: &70326830793520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70326830793520
+description: Dates keeps track of things you two have been meaning to do, and learns
+  to make good suggestions as you teach it.
+email: maxhodak@gmail.com
+executables:
+- dates
+extensions: []
+extra_rdoc_files: []
+files:
+- README.mdown
+- bin/dates
+- lib/date_idea.rb
+- lib/date_ideas.rb
+- lib/dates.rb
+- lib/version.rb
+- test/spec.rb
+homepage: http://www.github.com/maxhodak/dates
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Dates keeps track of things you two have been meaning to do.
+test_files: []
+has_rdoc: