qttk 0.1.0

data/lib/quadtone/tool.rb

@@ -0,0 +1,61 @@
+module Quadtone
+
+  class ToolUsageError < Exception; end
+
+  class Tool
+
+    attr_accessor :profile
+    attr_accessor :verbose
+
+    def self.process_args(args)
+      begin
+        name = args.shift or raise ToolUsageError, "No subcommand specified"
+        klass_name = 'Quadtone::Tools::' + name.split('-').map { |p| p.capitalize }.join
+        begin
+          klass = Kernel.const_get(klass_name)
+          raise NameError unless klass.respond_to?(:process_args)
+        rescue NameError => e
+          raise ToolUsageError, "Unknown subcommand specified: #{name.inspect} (#{klass_name})"
+        end
+        tool = klass.new
+        tool.process_environment
+        tool.load_profile
+        while args.first && args.first[0] == '-'
+          option = args.shift
+          tool.parse_global_option(option, args) or tool.parse_option(option, args) or raise ToolUsageError, "Unknown option for #{name.inspect} tool: #{option}"
+        end
+        tool.run(*args)
+      rescue ToolUsageError => e
+        warn e
+        exit 1
+      end
+    end
+
+    def process_environment
+      if (profile_name = ENV['PROFILE'])
+        @profile = Profile.load(profile_name)
+      end
+    end
+
+    def parse_global_option(option, args)
+      case option
+      when '--verbose'
+        @verbose = true
+      when '--profile'
+        @profile = Profile.load(args.shift)
+      end
+    end
+
+    def parse_option(option, args)
+      # overridden by subclass
+    end
+
+    # subclass can override this to avoid requirement of profile being set
+
+    def load_profile
+      raise ToolUsageError, "No profile set" unless @profile
+    end
+
+  end
+
+end

data/lib/quadtone/tools/add_printer.rb

@@ -0,0 +1,73 @@
+module Quadtone
+
+  module Tools
+
+    class AddPrinter < Tool
+
+      def load_profile
+        false
+      end
+
+      def run(printer)
+
+        #FIXME: move this into Printer class
+
+        unless %x{lpstat -v #{printer} 2>/dev/null}.empty?
+          raise ToolUsageError, "Printer #{printer.inspect} already exists"
+        end
+
+        curves_dir          = Pathname.new('/Library/Printers/QTR/quadtone') + printer
+        cups_data_dir       = Pathname.new(%x{cups-config --datadir}.chomp)
+        cups_serverbin_dir  = Pathname.new(%x{cups-config --serverbin}.chomp)
+
+        model = printer.split(/[-_=]/).first
+        model_ppd = "C/#{model}.ppd.gz"
+        ppd_file = cups_data_dir + 'model' + model_ppd
+
+        raise "QuadToneRIP does not support printer model #{model.inspect}" unless ppd_file.exist?
+
+        uri = loc = nil
+
+        File.popen(cups_serverbin_dir + 'backend' + 'usb').readlines.each do |line|
+          #FIXME: Too fragile -- use 'lpinfo' to find all printers
+          if line =~ /(usb:.*EPSON.*#{Regexp.escape(model.sub(/^Quad/, ''))})/
+            uri = $1
+            loc = "USB Printer"
+            break
+          end
+        end
+
+        if uri.nil?
+          loop do
+            print "Enter IP address of IPP printer, or <return> to cancel: "
+            ipp = STDIN.gets.chomp
+            if ipp.empty?
+              warn "Install cancelled."
+              return
+            end
+            if ipp =~ /^\d+\.\d+\.\d+\.\d+$/
+              uri = "lpd://#{ipp}"
+              loc = "Network Printer IPP = #{ipp}"
+              break
+            end
+            warn "Invalid IPP number: #{ipp.inspect}"
+          end
+        end
+
+        warn "Creating printer #{printer.inspect}"
+        system('lpadmin',
+          '-p', printer,
+          '-E',
+          '-m', model_ppd,
+          '-L', loc,
+          '-v', uri)
+        raise "Failed to create printer" unless $? == 0
+
+        curves_dir.mkpath
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/characterize.rb

@@ -0,0 +1,43 @@
+module Quadtone
+
+  module Tools
+
+    class Characterize < Tool
+
+      attr_accessor :force
+      attr_accessor :remeasure
+      attr_accessor :channel
+
+      def parse_option(option, args)
+        case option
+        when '--force'
+          @force = true
+        when '--remeasure'
+          @remeasure = args.shift.to_i
+        when '--channel'
+          @channels = args.shift.split(',').map(&:to_sym)
+        end
+      end
+
+      def run(*args)
+        case (action = args.shift)
+        when 'build'
+          @profile.characterization_curveset.build_target
+        when 'print'
+          @profile.characterization_curveset.print_target
+        when 'measure'
+          @profile.characterization_curveset.measure_target(force: @force, remeasure: @remeasure, channels: @channels)
+        when 'process'
+          @profile.characterization_curveset.process_target
+        when 'chart'
+          @profile.characterization_curveset.chart_target
+        else
+          raise ToolUsageError, "Unknown action: #{action.inspect}"
+        end
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/chart.rb

@@ -0,0 +1,31 @@
+module Quadtone
+
+  module Tools
+
+    class Chart < Tool
+
+      attr_accessor :open
+      attr_accessor :quick_look
+
+      def parse_option(option, args)
+        case option
+        when '--open'
+          @open = true
+        when '--quicklook'
+          @quicklook = true
+        end
+      end
+
+      def run
+        html_path = @profile.html_path
+        html_path.open('w') { |io| io.write(profile.to_html) }
+        ;;warn "Saved HTML to #{html_path}"
+        system 'open', html_path if @open
+        system 'qlmanage', '-p', html_path if @quick_look
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/check.rb

@@ -0,0 +1,16 @@
+module Quadtone
+
+  module Tools
+
+    class Check < Tool
+
+      def run
+        @profile.check
+        ;;warn "Profile checks out okay."
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/dir.rb

@@ -0,0 +1,15 @@
+module Quadtone
+
+  module Tools
+
+    class Dir < Tool
+
+      def run
+        puts @profile.dir_path
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/edit.rb

@@ -0,0 +1,23 @@
+module Quadtone
+
+  module Tools
+
+    class Edit < Tool
+
+      def run
+        # editor = ENV['EDITOR'] or raise "Can't determine editor"
+        editor = 'subl'
+        Quadtone.run(editor,
+          '--wait',
+          @profile.qtr_profile_path)
+        @profile = Profile.load(@profile.qtr_profile_path)
+        @profile.check
+        @profile.save
+        @profile.show
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/init.rb

@@ -0,0 +1,82 @@
+module Quadtone
+
+  module Tools
+
+    class Init < Tool
+
+      def load_profile
+        false
+      end
+
+      def run(*args)
+        get_printer
+        get_printer_options
+        get_inks
+        get_medium
+        profile = Profile.new(printer: @printer, printer_options: @printer_options, inks: @inks, medium: @medium)
+        profile.save
+        ;;warn "Created profile #{profile.name.inspect}"
+      end
+
+      def get_printer
+        printers = CupsPrinter.get_all_printer_names.map { |n| Printer.new(n) }.select(&:quadtone?)
+        @printer = prompt('Printer', printers, printers[0]) { |p| p.name }
+      end
+
+      def get_printer_options
+        @printer_options = {}
+        @printer.options.each do |name, option|
+          if Printer::ImportantOptions.include?(name)
+            choices = option.choices.map { |c| c.choice }
+            default = option.choices.find { |c| c.choice == option.default_choice }
+            @printer_options[name] = prompt(name, choices, default)
+          end
+        end
+      end
+
+      def get_medium
+        media = [
+          'Epson Velvet Fine Art',
+          'Epson Ultra Premium Presentation',
+          'Epson Premium Presentation',
+          'Epson Premium Luster',
+          'Hahnemuhle Bamboo',
+          'Southworth Antique Laid',
+          'Crane Museo',
+        ]
+        @medium = prompt('Media', media, media[0])
+      end
+
+      def get_inks
+        @inks = prompt('Inks', @printer.inks, @printer.inks.join(','))
+        @inks = [@inks] unless @inks.kind_of?(Array)
+      end
+
+      def prompt(label, values, default=nil, &block)
+        choices = {}
+        values.each_with_index { |value, i| choices[i + 1] = value }
+        STDERR.puts
+        STDERR.puts "#{label}:"
+        choices.each { |i, value| STDERR.puts '%2d. %s' % [i, block_given? ? yield(value) : value] }
+        STDERR.print "Choice" + (default ? " [#{block_given? ? yield(default) : default}]" : '') + "? "
+        selections = STDIN.gets.chomp
+        if selections.empty?
+          default
+        else
+          selections = selections.split(',').map(&:to_i)
+          case selections.length
+          when 0
+            default
+          when 1
+            choices[selections.first]
+          else
+            selections.map { |s| choices[s] }
+          end
+        end
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/install.rb

@@ -0,0 +1,15 @@
+module Quadtone
+
+  module Tools
+
+    class Install < Tool
+
+      def run
+        @profile.install
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/linearize.rb

@@ -0,0 +1,28 @@
+module Quadtone
+
+  module Tools
+
+    class Linearize < Tool
+
+      def run(*args)
+        case (action = args.shift)
+        when 'build'
+          @profile.linearization_curveset.build_target
+        when 'print'
+          @profile.linearization_curveset.print_target
+        when 'measure'
+          @profile.linearization_curveset.measure_target
+        when 'process'
+          @profile.linearization_curveset.process_target
+        when 'chart'
+          @profile.linearization_curveset.chart_target
+        else
+          raise ToolUsageError, "Unknown action: #{action.inspect}"
+        end
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/list.rb

@@ -0,0 +1,19 @@
+module Quadtone
+
+  module Tools
+
+    class List < Tool
+
+      def load_profile
+        false
+      end
+
+      def run
+        Profile.profile_names.each { |name| puts name }
+      end
+
+    end
+
+  end
+
+end

data/lib/quadtone/tools/print.rb

@@ -0,0 +1,38 @@
+module Quadtone
+
+  module Tools
+
+    class Print < Tool
+
+      attr_accessor :calibrate
+      attr_accessor :printer_options
+
+      def initialize
+        super
+        @printer_options = {}
+      end
+
+      def parse_option(option, args)
+        case option
+        when '--calibrate'
+          @calibrate = true
+        when '--option', '--options'
+          @printer_options.merge!(
+            Hash[
+              args.shift.split(',').map { |o| o.split('=') }
+            ]
+          )
+        end
+      end
+
+      def run(*args)
+        args.map { |p| Pathname.new(p) }.each do |image_path|
+          @profile.print_file(image_path, calibrate: @calibrate, printer_options: @printer_options)
+        end
+      end
+
+    end
+
+  end
+
+end