panomosity 0.1.39 → 0.1.40

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d0babb2f2886451531b10cafb8fb2b8ccce33edd260247d1a100001120c3af57
-  data.tar.gz: 93c56a79260969f6a77c87d019bbd5efb11c6bcd908fd6ea05f5ad13334c2b3a
+  metadata.gz: 508c47f3caa59487c8e85f4787044dd174b9e43a63fce0ee0ecfc210c88ecef7
+  data.tar.gz: d7e9a8a263379d90f984746e0b3f2215943322418c59ce4f9f25012563fbce82
 SHA512:
-  metadata.gz: 2f373113dd8fbb7e70045e94cd20d97342f1d955f64b221208288246f8fbcc486e05785e518769d51b73096029b8570861929a9eb10fd1ed0081c448d9111895
-  data.tar.gz: e82ff01175bbf2010e102fceae0acc512517ac51037b2e70a1b893ecd6325c0afc251e4a3525fe003528a557b5849590c56229b3c51edab896eb4a79b71bc4d0
+  metadata.gz: 73a953315be6abba6b26a8371dbf5f0ad131fbae1733b59c12e243d6e31e531a2b9bf3e908b2d5b5d03423ef3006347dd15be55dafcd38c019735e57ea998ad4
+  data.tar.gz: 380b7ac01f1306116cbdc2c5982760f2495ca107321a6564bfca266b449e6c631ec8752b40228e12b8ffb8127f8f3305c3451a93a5ba2a084a91f3bd838ad30d

data/Gemfile.lock

@@ -1,31 +1,38 @@
 PATH
   remote: .
   specs:
-    panomosity (0.1.38)
+    panomosity (0.1.39)
+      write_xlsx
 
 GEM
   remote: https://rubygems.org/
   specs:
     coderay (1.1.2)
-    diff-lcs (1.3)
+    diff-lcs (1.4.4)
     method_source (0.9.0)
     pry (0.11.3)
       coderay (~> 1.1.0)
       method_source (~> 0.9.0)
     rake (10.5.0)
-    rspec (3.8.0)
-      rspec-core (~> 3.8.0)
-      rspec-expectations (~> 3.8.0)
-      rspec-mocks (~> 3.8.0)
-    rspec-core (3.8.0)
-      rspec-support (~> 3.8.0)
-    rspec-expectations (3.8.1)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.2)
+      rspec-support (~> 3.9.3)
+    rspec-expectations (3.9.2)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-mocks (3.8.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
       diff-lcs (>= 1.2.0, < 2.0)
-      rspec-support (~> 3.8.0)
-    rspec-support (3.8.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.3)
+    rubyzip (2.3.0)
+    write_xlsx (0.85.9)
+      rubyzip (>= 1.0.0)
+      zip-zip
+    zip-zip (0.3)
+      rubyzip (>= 1.0.0)
 
 PLATFORMS
   ruby

data/lib/panomosity/runner.rb

@@ -1,14 +1,16 @@
 require 'logger'
 require 'json'
 require 'csv'
+require 'panomosity/xlsx_writer'
 
 module Panomosity
   class Runner
     include Panomosity::Utils
+    include Panomosity::XLSXWriter
 
     attr_reader :logger
 
-    AVAILABLE_COMMANDS = %w(
+    AVAILABLE_COMMANDS = %w[
       add_calibration_flag
       apply_calibration_values
       check_position_changes
@@ -20,6 +22,7 @@ module Panomosity
       crop_centers
       diagnose
       export_control_point_csv
+      export_panorama_attributes_to_csv
       fix_conversion_errors
       fix_unconnected_image_pairs
       generate_border_line_control_points
@@ -34,7 +37,7 @@ module Panomosity
       prepare_for_pr0ntools
       remove_anchor_variables
       standardize_roll
-    )
+    ]
 
     def initialize(options)
       @options = options
@@ -299,7 +302,7 @@ module Panomosity
       panorama.calculate_neighborhoods
 
       filename = 'control_points.csv'
-      headers = %w(image_1_name image_2_name image_1_id image_2_id type width height d1 e1 d2 e2 x1 y1 x2 y2 rx ry r)
+      headers = %w[image_1_name image_2_name image_1_id image_2_id type width height d1 e1 d2 e2 x1 y1 x2 y2 rx ry r]
       CSV.open(filename, 'w+') do |csv|
         csv << headers
         panorama.control_points.each do |cp|
@@ -490,6 +493,70 @@ module Panomosity
       puts panorama.attributes.to_json
     end
 
+    def export_panorama_attributes_to_csv
+      panorama = Panorama.new(@input_file, @options)
+      panorama.calculate_neighborhoods
+
+      logger.info "Making #{Dir.pwd}/#{@input}_attributes Directory"
+
+      dir = "#{Dir.pwd}/#{@input}_attributes"
+      Dir.mkdir(dir) unless File.exists?(dir)
+
+      logger.info "Done. Moving into #{dir} Directory"
+
+      Dir.chdir "#{@input}_attributes"
+
+      logger.info "Making control_points.csv"
+
+      filename = 'control_points.csv'
+      headers = %w[image_1_name image_2_name image_1_id image_2_id type width height d1 e1 d2 e2 x1 y1 x2 y2 rx ry r]
+      CSV.open(filename, 'w+') do |csv|
+        csv << headers
+        panorama.control_points.each do |cp|
+          pair = Pair.all.find { |p| p.control_points.include?(cp) }
+          csv << [cp.i1.name, cp.i2.name, cp.i1.id, cp.i2.id, pair.type, cp.i1.w, cp.i1.h, cp.i1.d, cp.i1.e,
+                  cp.i2.d, cp.i2.e, cp.x1, cp.y1, cp.x2, cp.y2, cp.px, cp.py, cp.pdist]
+        end
+      end
+
+      logger.info "Done. Check for #{filename}"
+
+      attributes = panorama.attributes
+
+      logger.info "Making cp_pairs.csv"
+
+      filename = 'cp_pairs.csv'
+      headers = attributes[:pairs].first.keys
+      CSV.open(filename,"w") do |csv|
+        csv << headers
+        attributes[:pairs].each do |pair|
+          csv << pair.values
+        end
+      end
+
+      logger.info "Done. Check for #{filename}"
+
+      logger.info "Making neighborhood_pairs.csv"
+
+      filename = 'neighborhood_pairs.csv'
+      headers = ['data_type', attributes[:similar_neighborhoods].first.keys].flatten
+      CSV.open(filename,'w') do |csv|
+        csv << headers
+        attributes[:similar_neighborhoods].each do |neighborhood|
+          csv << ['similar_neighborhoods', neighborhood.values].flatten(1)
+        end
+        attributes[:neighborhoods_by_similar_neighborhood].each do |neighborhood|
+          csv << ['neighborhoods_by_similar_neighborhood', neighborhood.values].flatten(1)
+        end
+      end
+
+      logger.info "Done. Check for #{filename}"
+
+      csv_to_xlsx(['control_points.csv', 'cp_pairs.csv', 'neighborhood_pairs.csv'], 'attributes')
+
+      Dir.chdir '..'
+    end
+
     def import_control_point_csv
       logger.info 'importing control point csv'
       panorama = Panorama.new(@input_file, @options)

data/lib/panomosity/version.rb

@@ -1,3 +1,3 @@
 module Panomosity
-  VERSION = '0.1.39'
+  VERSION = '0.1.40'
 end

data/lib/panomosity/xlsx_writer.rb

@@ -0,0 +1,40 @@
+require 'write_xlsx'
+
+module Panomosity
+  module XLSXWriter
+    SEPERATE_CSV_COLUMNS = /,(?![^\[]*\])/.freeze
+
+
+    def csv_to_xlsx(input_files, output_file)
+      input_files = Array(input_files)
+
+      logger.info "Creating #{output_file}.xlsx"
+
+      workbook = WriteXLSX.new("#{output_file}.xlsx")
+
+      input_files.each do |csv|
+        logger.info "Creating #{csv} Worksheet"
+
+        worksheet = workbook.add_worksheet(csv)
+
+        seperate_csv_columns = /,(?![^\[]*\])/
+
+        File.open(csv, "r") do |f|
+          f.each_line do |rows|
+            cells = rows.split(SEPERATE_CSV_COLUMNS)
+
+            cells.each_with_index do |cell, column|
+              row = f.lineno - 1
+              data = cell.tr_s('"', '').strip
+              worksheet.write(row, column, data)
+            end
+          end
+        end
+      end
+
+      workbook.close
+
+      logger.info "Done. Check for #{output_file}.xlsx"
+    end
+  end
+end

data/panomosity.gemspec

@@ -24,4 +24,6 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'bundler', '~> 1.16'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_runtime_dependency 'write_xlsx'
+
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: panomosity
 version: !ruby/object:Gem::Version
-  version: 0.1.39
+  version: 0.1.40
 platform: ruby
 authors:
 - Oliver Garcia
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-09-30 00:00:00.000000000 Z
+date: 2020-10-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: write_xlsx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Custom scripts to help with PTO parsing and different strategies.
 email:
 - ogarci5@gmail.com
@@ -91,6 +105,7 @@ files:
 - lib/panomosity/runner.rb
 - lib/panomosity/utils.rb
 - lib/panomosity/version.rb
+- lib/panomosity/xlsx_writer.rb
 - panomosity.gemspec
 homepage: https://github.com/elevatesystems/panomosity
 licenses: