summaryrefslogtreecommitdiff
path: root/lib/feature.rb
blob: f811aef6385b3a56335ad7d64a155f1ae7ca28ee (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
module OpenTox

  # Basic feature class
  class Feature
    field :measured, type: Boolean
    field :calculated, type: Boolean
    field :category, type: String
    field :unit, type: String
    field :conditions, type: Hash

    # Is it a nominal feature
    # @return [TrueClass,FalseClass]
    def nominal?
      self.class == NominalFeature
    end

    # Is it a numeric feature
    # @return [TrueClass,FalseClass]
    def numeric?
      self.class == NumericFeature
    end
  end

  # Feature for categorical variables
  class NominalFeature < Feature
    field :accept_values, type: Array
  end

  # Feature for quantitative variables
  class NumericFeature < Feature
  end

  # Feature for SMARTS fragments
  class Smarts < NominalFeature
    field :smarts, type: String 
    index "smarts" => 1
    # Create feature from SMARTS string
    # @param [String]
    # @return [OpenTox::Feature]
    def self.from_smarts smarts
      self.find_or_create_by :smarts => smarts
    end
  end

end