Module: Mustachio

Defined in:
lib/mustachio.rb,
lib/mustachio/app.rb

Defined Under Namespace

Classes: App

Constant Summary collapse

FACE_POS_ATTRS = 
['center', 'eye_left', 'eye_right', 'mouth_left', 'mouth_center', 'mouth_right', 'nose']
FACE_SPAN_SCALE = 
2.0

Class Method Summary collapse

Class Method Details

.face_client ⇒ Object 



11
12
13
 
# File 'lib/mustachio.rb', line 11

def face_client
  @@face_client
end

.face_data(file_or_job) ⇒ Object 



40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
 
# File 'lib/mustachio.rb', line 40

def face_data(file_or_job)
  file = case file_or_job
  when Dragonfly::Job
    file_or_job.temp_object
  when Dragonfly::TempObject
    file_or_job.file
  when File
    file_or_job
  else
    raise ArgumentError, "A #{file_or_job.class} is not a valid argument for #face_data.  Please provide a File or a Dragonfly::Job."
  end

  face_data = Mustachio.face_client.faces_detect(:file => file, :attributes => 'none')
  face_data['photos'].first
end

.face_data_as_px(file_or_job, width = nil, height = nil) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/mustachio.rb', line 56

def face_data_as_px(file_or_job, width=nil, height=nil)
  data = self.face_data(file_or_job)
  width ||= data['width']
  height ||= data['height']

  new_tags = []
  data['tags'].map do |face|
    has_all_attrs = FACE_POS_ATTRS.all? do |pos_attr|
      if face[pos_attr]
        face[pos_attr]['x'] *= (width / 100.0)
        face[pos_attr]['y'] *= (height / 100.0)
        true
      else # face attribute missing
        false
      end
    end

    new_tags << face if has_all_attrs
  end

  data['tags'] = new_tags
  data
end

.face_span(file_or_job) ⇒ Object 



80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
# File 'lib/mustachio.rb', line 80

def face_span(file_or_job)
  face_data = self.face_data_as_px(file_or_job)
  faces = face_data['tags']
  
  left_face, right_face = faces.minmax_by{|face| face['center']['x'] }
  top_face, bottom_face = faces.minmax_by{|face| face['center']['y'] }
  
  top = top_face['eye_left']['y']
  bottom = bottom_face['mouth_center']['y']
  right = right_face['eye_right']['x']
  left = left_face['eye_left']['x']
  width = right - left
  height = bottom - top
  
  # compute adjusted values for padding around face span
  adj_width = width * FACE_SPAN_SCALE
  adj_height = height * FACE_SPAN_SCALE
  adj_top = top - ((adj_height - height) / 2.0)
  adj_bottom = bottom + ((adj_height - height) / 2.0)
  adj_right = right + ((adj_width - width) / 2.0)
  adj_left = left - ((adj_width - width) / 2.0)
  
  {
    :top => adj_top,
    :bottom => adj_bottom,
    :right => adj_right,
    :left => adj_left,
    :width => adj_width,
    :height => adj_height,
    :center_x => (adj_left + adj_right) / 2,
    :center_y => (adj_top + adj_bottom) / 2
  }
end

.mustaches ⇒ Object 



15
16
17
 
# File 'lib/mustachio.rb', line 15

def mustaches
  @@mustaches
end

.setup ⇒ Object 



19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/mustachio.rb', line 19

def setup
  @@face_client = Face.get_client(
    :api_key => (ENV['MUSTACHIO_FACE_API_KEY'] || raise("Please set MUSTACHIO_FACE_API_KEY.")),
    :api_secret => (ENV['MUSTACHIO_FACE_API_SECRET'] || raise("Please set MUSTACHIO_FACE_API_SECRET."))
  )
  
  staches = YAML.load(File.read(File.join(File.dirname(__FILE__), '..', 'config', 'staches.yml')))
  staches.map! do |stache|
    stache['vert_offset'] ||= 0
    stache['mouth_overlap'] ||= 0
    
    stache['file_path'] = File.expand_path(File.join(File.dirname(__FILE__), 'mustachio', 'public', 'images', 'staches', stache['filename']))
    unless stache['width'] && stache['height']
      stache['width'], stache['height'] = ImageSize.new(File.new(stache['file_path'])).get_size
    end
    stache
  end
  @@mustaches = staches
end