require 'spec_helper'
require 'fileutils'

module Mork
  describe SheetOMR do
    context 'with a valid response sheet' do
      let(:img) { sample_img 'jdoe1' }
      let(:fn)  { File.basename(img.image_path) }
      let(:omr) { SheetOMR.new img.image_path, choices: [img.nchoices]*img.nitems, layout: img.grid_path }
      describe '#new' do
        it 'creates a SheetOMR object' do
          expect(omr).to be_a SheetOMR
        end

        it 'raises an error if the provided path is invalid' do
          expect { SheetOMR.new 'non_existing_file.jpg'}.to raise_error IOError
        end
      end

      describe '#valid?' do
        it 'registers correctly' do
          expect(omr.valid?).to be_truthy
        end
      end

      describe '#status' do
        it 'returns the registration status for each of the four corners' do
          expect(omr.status).to eq({ tl: :ok, tr: :ok, br: :ok, bl: :ok })
        end
      end

      describe '#barcode' do
        it 'returns the integer form of the barcode' do
          expect(omr.barcode).to eq img.barcode_int
        end
      end

      describe '#barcode_string' do
        it 'returns the binary string version of the barcode' do
          expect(omr.barcode_string).to eq img.barcode_str
        end
      end

      describe '#marked?' do
        it 'returns true if the given cell was marked, false otherwise' do
          expect(omr.marked?   0, 0).to be_truthy
          expect(omr.marked?   0, 1).to be_falsy
          expect(omr.marked? 119, 4).to be_truthy
          expect(omr.marked? 119, 3).to be_falsy
        end
      end

      describe '#marked_choices' do
        it 'returns an array of marked choices as position indices' do
          expect(omr.marked_choices).to eq standard_mark_array(24)
        end
      end

      describe '#marked_choices' do
        it 'returns an array of marked choices as position indexes' do
          expect(omr.marked_choices ).to eq standard_mark_array(24)
        end

        let(:om2) { SheetOMR.new img.image_path, choices: [5, 4, 3, 2, 1], layout: img.grid_path }
        it 'returns marked choices only for existing choice cells' do
          expect(om2.marked_choices).to eq [[0], [1], [2], [], []]
        end
      end

      describe '#marked_logicals' do
        it 'returns an array of logicals for the marked choices' do
          expect(omr.marked_logicals).to eq standard_mark_logical_array(24)
        end
      end

      describe '#marked_letters' do
        it 'returns an array of characters for the marked choices' do
          expect(omr.marked_letters).to eq standard_mark_char_array(24)
        end
      end

      it 'writes out markedness' do
        mf = File.open('spec/out/text/marked.txt', 'w')
        img.nitems.times do |q|
          x = 5.times.collect do |c|
            omr.marked?(q,c) ? '1' : '0'
          end
          mf.puts x.join(' ')
        end
        mf.close
      end

      context 'creating overlays' do
        it 'registration highlighted' do
          omr.save_registration "spec/out/registration/#{fn}"
        end

        it 'highlights all choice cells' do
          omr.overlay :highlight, :all
          omr.save "spec/out/highlight/#{fn}"
        end

        it 'highlights marked cells' do
          omr.overlay :highlight, :marked
          omr.save "spec/out/highlight/marked-#{fn}"
        end

        it 'checks marked cells' do
          omr.overlay :check, :marked
          omr.save "spec/out/mark/#{fn}"
        end

        it 'outlines and crosses marked cells' do
          omr.overlay :outline, standard_mark_array(24)
          omr.overlay :check
          omr.save "spec/out/outline/#{fn}"
        end

        it 'highlights the barcode' do
          omr.overlay :outline, :barcode
          omr.save "spec/out/barcode/#{fn}"
        end
      end
    end

    context 'systematic tests', exclude: true do
      let(:bila)  { 'CCEBEBCEEACCDCABDBEBCADEADDCCCACCACDBBDAECDDABDEEBCEEDCBAAADEEEEDCADEABCBDECCCCDDDCABBECAADADBBEEABA'.split '' }
      let(:bila0) { SheetOMR.new 'spec/samples/syst/bila0.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bila1) { SheetOMR.new 'spec/samples/syst/bila1.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bila2) { SheetOMR.new 'spec/samples/syst/bila2.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bila3) { SheetOMR.new 'spec/samples/syst/bila3.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bila4) { SheetOMR.new 'spec/samples/syst/bila4.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:dald)  { 'DDDBECAAADBAEEEAEEEBAACAEDBDECDBDCDCDDEDCCDCDBDCADEEDBCCBEBBAADDCDBBECBBBDEABADABADADBABAEABACBDADDA'.split '' }
      let(:dald0) { SheetOMR.new 'spec/samples/syst/dald0.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:dald1) { SheetOMR.new 'spec/samples/syst/dald1.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:dald2) { SheetOMR.new 'spec/samples/syst/dald2.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:dald3) { SheetOMR.new 'spec/samples/syst/dald3.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:dald4) { SheetOMR.new 'spec/samples/syst/dald4.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:cost)  { 'ABBDDBAEAEBAADEAAECBCDBBDABABADEECCACBCAEDDAEBEABBCDABECAACEEEBADECBBEAADBBBEABDAEBDEEABBABEBEDDAEEC'.split '' }
      let(:cost0) { SheetOMR.new 'spec/samples/syst/cost0.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:cost1) { SheetOMR.new 'spec/samples/syst/cost1.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:cost2) { SheetOMR.new 'spec/samples/syst/cost2.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:cost3) { SheetOMR.new 'spec/samples/syst/cost3.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:cost4) { SheetOMR.new 'spec/samples/syst/cost4.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bone)  { 'CECBBABAECADEDCACBBDEECBADBECDCEDECABCAADCBDEDACAEEDCCADBEDCEBCCBBDCCACDEDDAAECEBDBADCBAAEBAEDABCBDC'.split '' }
      let(:bone0) { SheetOMR.new 'spec/samples/syst/bone0.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bone1) { SheetOMR.new 'spec/samples/syst/bone1.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:bone2) { SheetOMR.new 'spec/samples/syst/bone2.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:barr)  { 'ACECAAADDBCECCCDBEBECDEDAECEDDEEDCDEADDCCBCCCBBEACBCAEDEEDDDABBBBABEBDCEADEEDEBCBADBCEDCDBACEBCBDCDA'.split '' }
      let(:barr0) { SheetOMR.new 'spec/samples/syst/barr0.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:barr1) { SheetOMR.new 'spec/samples/syst/barr1.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}
      let(:barr2) { SheetOMR.new 'spec/samples/syst/barr2.jpg', choices: [5]*100, layout: 'spec/samples/syst/layout.yml'}

      it 'checks bila' do
        expect(bila0.mark_char_array.flatten).to eq(bila)
        expect(bila1.mark_char_array.flatten).to eq(bila)
        expect(bila2.mark_char_array.flatten).to eq(bila)
        expect(bila3.mark_char_array.flatten).to eq(bila)
        expect(bila4.mark_char_array.flatten).to eq(bila)
      end

      it 'checks dald' do
        expect(dald0.mark_char_array.flatten).to eq(dald)
        expect(dald1.mark_char_array.flatten).to eq(dald)
        expect(dald2.mark_char_array.flatten).to eq(dald)
        expect(dald3.mark_char_array.flatten).to eq(dald)
        expect(dald4.mark_char_array.flatten).to eq(dald)
      end

      it 'checks cost' do
        expect(cost0.mark_char_array.flatten).to eq(cost)
        expect(cost1.mark_char_array.flatten).to eq(cost)
        expect(cost2.mark_char_array.flatten).to eq(cost)
        expect(cost3.mark_char_array.flatten).to eq(cost)
        expect(cost4.mark_char_array.flatten).to eq(cost)
      end

      it 'checks bone' do
        expect(bone0.mark_char_array.flatten).to eq(bone)
        expect(bone1.mark_char_array.flatten).to eq(bone)
        expect(bone2.mark_char_array.flatten).to eq(bone)
      end

      it 'checks barr' do
        expect(barr0.mark_char_array.flatten).to eq(barr)
        expect(barr1.mark_char_array.flatten).to eq(barr)
        expect(barr2.mark_char_array.flatten).to eq(barr)
      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