require 'spec_helper'

module Mork
  describe SheetOMR do
    
    context 'highlighting' do
      # since these specs change the @crop, they must be run in isolation
      # with the SheetOMR rebuilt each time, even though it is time consuming!
      let(:shinfo) { sample_img 'sample-gray' }
      let(:sheet)  { SheetOMR.new shinfo.filename, shinfo.grid_file }
      
      it 'highlights the registration areas and frame' do
        sheet.highlight_registration
        sheet.write_raw 'spec/out/reg_areas.jpg'
      end
      
      it 'highlights all choice cells' do
        sheet.highlight_all_choices
        sheet.write 'spec/out/all_highlights.jpg'
      end
      
      it 'highlights marked cells' do
        sheet.highlight_marked
        sheet.write 'spec/out/marked_highlights.jpg'
      end

      it 'outlines some responses' do
        sheet.outline [[1],[1],[2],[2],[3,4],[],[0,1,2,3,4], [],[1],[2],[2],[3,4],[],[0,1,2,3,4]]
        sheet.write 'spec/out/outlines.jpg'
      end

      it 'highlights marked cells and outline correct responses', focus: true do
        sheet.highlight_marked
        sheet.outline [[1],[1],[2],[2],[3,4],[],[0,1,2,3,4], [],[1],[2],[2],[3,4],[],[0,1,2,3,4]]
        sheet.write 'spec/out/marks_and_outs.jpg'
      end
      
      it 'highlights marked cells of a problematic one' do
        si = sample_img 'silvia'
        s = SheetOMR.new si.filename, si.grid_file
        s.highlight_marked
        s.write 'spec/out/problem.jpg'
      end
      
      it 'highlights the barcode' do
        si = sample_img 'sample-gray'
        s = SheetOMR.new si.filename, si.grid_file
        s.highlight_barcode
        s.write 'spec/out/code_bits.jpg'
      end
    end

    context 'marking a nicely printed and scanned sheet' do
      before(:all) do
        @shinfo = sample_img 'sample-gray'
        @sheet = SheetOMR.new @shinfo.filename, @shinfo.grid_file
      end
      
      describe '#valid?' do
        it 'returns true' do
          @sheet.valid?.should be_truthy
        end
      end
      
      describe '#marked?' do
        it 'returns true for some darkened choices' do
          expect(@sheet.marked?(0,0)).to be_truthy
          expect(@sheet.marked?(1,1)).to be_truthy
          expect(@sheet.marked?(2,2)).to be_truthy
        end
        
        it 'return false for some blank choices' do
          expect(@sheet.marked?(0,1)).to be_falsy
          expect(@sheet.marked?(1,0)).to be_falsy
          expect(@sheet.marked?(2,3)).to be_falsy
        end

        it 'writes out markedness' do
          puts "Choice threshold: #{@sheet.send :choice_threshold}"
          mf = File.open('spec/out/marked.txt',   'w')
          120.times do |q|
            x = 5.times.collect do |c|
              @sheet.marked?(q,c) ? '1' : '0'
            end
            mf.puts x.join(' ')
          end
          mf.close
        end
      end

      describe '#mark_array' do
        it 'should return an array of marked choices for each of the first 5 questions' do
          @sheet.mark_array(5).should == [[0], [1], [2], [3], [4]]
        end

        it 'should return an array of marked choices for the specified question set' do
          @sheet.mark_array([0, 1, 13, 16, 31, 104, 21]).should == [[0], [1], [3,4], [2], [1], [], [0,1,2,3,4]]
        end

        it 'should return an array of @grid.max_questions length if called without arguments' do
          @sheet.mark_array.length.should == 120
        end
      end

      describe '#mark_logical_array' do
        it 'should return an array of booleans for each of the first questions' do
          @sheet.mark_logical_array(5).should == [
            [true, false, false, false, false],
            [false, true, false, false, false],
            [false, false, true, false, false],
            [false, false, false, true, false],
            [false, false, false, false, true],
          ]
        end
      end
      
      describe 'barcodes' do
        it 'should read the bit string as all ones' do
          @sheet.barcode_string.should == '1111111111111111111111111111111111111111'
          @sheet.barcode.should == 1099511627775
        end
        
        it 'should read another bit string' do
          barcode_string = '0000000000000000000000000010000110100000'
          s2 = SheetOMR.new('spec/samples/sample02.jpg')
          s2.barcode_string.should == barcode_string
          s2.barcode.should == 8608
        end

        it 'should read the 666 bit string' do
          sh = sample_img 'code666'
          s2 = SheetOMR.new sh.filename, sh.grid_file
          s2.barcode.should == sh.barcode_int
        end
      end
      
    end

    # context "multi-page pdf" do
    #   before(:all) do
    #     @mlist = MimageList.new('spec/samples/two_pages.pdf')
    #   end
    #
    #   describe "reading the codes" do
    #     it "should read the right code for the first page" do
    #       s = SheetOMR.new(@mlist[0], Grid.new)
    #       s.code.should == 18446744073709551615
    #     end
    #     it "should read the right code for the second page" do
    #       s = SheetOMR.new(@mlist[1], Grid.new)
    #       s.code.should == 283764283738
    #     end
    #   end
    #
    #   describe "getting the answers" do
    #     it "should read the correct choices for the first page" do
    #       s = SheetOMR.new(@mlist[0], Grid.new)
    #       s.marked?( 0, 0).should be_truthy
    #       s.marked?( 0, 1).should be_falsy
    #       s.marked?(15, 3).should be_falsy
    #       s.marked?(15, 4).should be_truthy
    #     end
    #
    #     it "should read the correct choices for the second page" do
    #       s = SheetOMR.new(@mlist[1], Grid.new)
    #       s.mark_array(15).should == [[0], [1], [2], [3], [4], [0], [1], [2, 3], [4], [1, 2, 3], [0], [1], [2], [3], [4]]
    #     end
    #   end
    # end
  end
end