Yet another directory change.

Author: PhilipGriffith

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2018 Philip Griffith
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1 @@
+Do this at some point.

pyutm/__init__.py

@@ -0,0 +1,59 @@
+def from_list(data, col):
+
+    coords = ()
+    xs = None
+    ys = None
+    error = None
+    try:
+        # Grab the column containing coordinates if applicable
+        if col is not None:
+            coords = tuple(zip(*data))[col]
+        else:
+            if not isinstance(data[0], (tuple, list)) and not isinstance(data[1], (tuple, list)):
+                coords = [(data[0], data[1])]
+            else:
+                coords = data
+        xs = tuple(coord[0] for coord in coords)
+        ys = tuple(coord[1] for coord in coords)
+    except (TypeError, IndexError):
+        error = 'Invalid coordinates: {}'.format(coords)
+    finally:
+        return xs, ys, error
+
+
+def from_csv(data, col1, col2):
+
+    import pandas
+    xs = None
+    ys = None
+    error = None
+    try:
+        df = pandas.read_csv(data)
+        xs = df[col1].values
+        ys = df[col2].values
+    except (FileNotFoundError, KeyError) as e:
+        if type(e).__name__ == 'KeyError':
+            error = 'Invalid column names: col1={}, col2={}'.format(col1, col2)
+        else:
+            error = e
+    finally:
+        return xs, ys, error
+
+
+def from_shapefile(data):
+
+    import shapefile
+    xs = None
+    ys = None
+    error = None
+    try:
+        sf = shapefile.Reader(data)
+        shapes = sf.shapes()
+        # Only accept Point, PointZ and PointM geometries
+        if shapes[0].shapeType in (1, 11, 21):
+            xs = tuple(shape.points[0][0] for shape in shapes)
+            ys = tuple(shape.points[0][1] for shape in shapes)
+    except shapefile.ShapefileException as e:
+        error = e
+    finally:
+        return xs, ys, error

pyutm/getcoords.py

@@ -0,0 +1,148 @@
+import pyproj
+
+import getcoords
+import locate
+
+
+class Grid:
+
+    def __init__(self, data, col1=None, col2=None, epsg=4326):
+
+        self.data = data
+        self.col1 = col1
+        self.col2 = col2
+        self.epsg = epsg
+        self.xs = None
+        self.ys = None
+        self.error = None
+        self.grid_references = []
+
+        self.set_coords()
+        if self.epsg != 4326:
+            self.transform_coords()
+
+    def set_coords(self):
+
+        if isinstance(self.data, (tuple, list)) and (len(self.data) > 1):
+            self.xs, self.ys, self.error = getcoords.from_list(self.data, self.col1)
+        else:
+            try:
+                if self.data.endswith('.csv'):
+                    self.xs, self.ys, self.error = getcoords.from_csv(self.data, self.col1, self.col2)
+                elif self.data.endswith('.shp'):
+                    self.xs, self.ys, self.error = getcoords.from_shapefile(self.data)
+                else:
+                    raise AttributeError
+            except AttributeError:
+                self.set_error('Invalid parameters: Grid(data={}, col1={}, col2={}, epsg={})'.format(
+                    repr(self.data), self.col1, self.col2, self.epsg))
+
+    def transform_coords(self):
+
+        try:
+            p = pyproj.Proj(init='epsg:{}'.format(self.epsg))
+            self.xs, self.ys = p(self.xs, self.ys, inverse=True)
+        except RuntimeError:
+            self.error('EPSG:{} not found'.format(self.epsg))
+
+    def get_grid_refs(self):
+
+        for x, y in zip(self.xs, self.ys):
+            self.grid_references.append(locate.Point(x, y).get_grid_reference())
+
+    def write_references(self, fname=None, col='GRID_REFS'):
+
+        if fname is None:
+            fname = self.data
+            self.get_grid_refs()
+
+    def set_error(self, message):
+
+        preamble = 'Error creating Grid object:'
+        self.error = ('{} {}'.format(preamble, message))
+        print(self.error)
+
+
+if __name__ == "__main__":
+
+    lonlats = [(-34.907587535813704, 50.58441270574641),
+               (108.93083026662671, 32.38153601114477),
+               (-36.69218329018642, -45.06991972863084),
+               (43.97154480746007, -46.140677181254475)]
+
+    lonlats2 = [[(-34.907587535813704, 50.58441270574641), True, [1, 2], 1, 'a'],
+                [(108.93083026662671, 32.38153601114477), False, [3, 4], 2, 'b'],
+                [(-36.69218329018642, -45.06991972863084), False, [5, 6], 3, 'c'],
+                [(43.97154480746007, -46.140677181254475), True, [7, 8], 4, 'd']]
+
+    lonlats3 = [((-34.907587535813704, 50.58441270574641), 1), ((108.93083026662671, 32.38153601114477), 2),
+                ((-36.69218329018642, -45.06991972863084), 3), ((43.97154480746007, -46.140677181254475), 4)]
+
+    g = Grid((-34.907587535813704, 50.58441270574641))
+    g.write_references()
+    # g = Grid(lonlats)
+    # g = Grid(lonlats2, 0, epsg=3086)
+    # g = Grid(lonlats3, 0)
+    g = Grid('.tests/data/points.csv', 'POINT_X', 'POINT_Y')
+    print(g.get_grid_refs())
+    # g = Grid('points.shp')
+    # g = Grid('points.shp', epsg=3086)
+
+    # g = Grid('good_crimes.csv', 'Longitude', 'Latitude', epsg=3086)
+
+    # g = Grid('chicago_crimes_2016.csv', 'Longitude', 'Latitude')
+
+    # print(g.grid_refs)
+
+    # These are the desired method calls
+    # g = Grid('myassets.csv', 'Lon', 'Lat', epsg=3086)
+    # g.write_refs('myassets.shp', 'MY_OFFICIAL_GRID_REFERENCES')
+    # g.write_assets('myassets.shp', 'MY_UNIQUE_ASSET_NAMES',
+    # prefix='H', col=None, gzd=False, k100=False, delimiter='-')
+
+
+# These possibilities are done
+# print('Grid(True)')
+# g = Grid(True)
+# print('Grid(0)')
+# g = Grid(0)
+# print('Grid(1)')
+# g = Grid(1)
+# print('Grid(2)')
+# g = Grid(2)
+# print('Grid("i")')
+# g = Grid('i')
+# print('Grid("ijeg")')
+# g = Grid('ijeg')
+# print('Grid(0.123456)')
+# g = Grid(0.123456)
+# print('Grid([])')
+# g = Grid([])
+# print('Grid(())')
+# g = Grid(())
+# print('Grid([1])')
+# g = Grid([1])
+# print('Grid((1,))')
+# g = Grid((1,))
+# print('Grid((1))')
+# g = Grid((1))
+# g = Grid('points1.shp')
+# g = Grid('points.shp', epsg=45673086)
+# g = Grid('points.csv', 'POINT_X', 'POINT')
+# g = Grid('points.csv', 'POINT_X')
+# g = Grid('points1.csv', 'POINT_X', 'B')
+# print('Grid([1, 2])')
+# g = Grid([1, 2])
+# print('Grid((1, 2))')
+# g = Grid((1, 2))
+# print('Grid(lonlats2, 1)')
+# g = Grid(lonlats2, 1)
+# print('Grid(lonlats2, 2)')
+# g = Grid(lonlats2, 2)
+# print('Grid(lonlats2, 3)')
+# g = Grid(lonlats2, 3)
+# print('Grid(lonlats2, 4)')
+# g = Grid(lonlats2, 4)
+# print('Grid(lonlats2, 5)')
+# g = Grid(lonlats2, 5)
+# g = Grid(lonlats3)

pyutm/main.py

@@ -0,0 +1 @@
+UTF-8

pyutm/setrefs.py

@@ -0,0 +1,27 @@
+OID,Id,POINT_X,POINT_Y
+-1,0,-118.418681394000000,44.820703608400002
+-1,0,-103.188850642000010,35.480078531799997
+-1,0,-152.562683848000010,65.520768720500001
+-1,0,-156.057234486999990,19.742539480200001
+-1,0,-81.670585520600000,28.127543362400001
+-1,0,-72.914167112000001,43.498865047499997
+-1,0,-66.533303400500003,17.894144738000001
+-1,0,1.498165853390000,52.012089309200000
+-1,0,-36.692183290199999,-45.069919728599999
+-1,0,43.971544807500003,-46.140677181299999
+-1,0,59.675987445899999,4.898761393810000
+-1,0,108.930830267000000,32.381536011100003
+-1,0,74.309672631799998,63.076582986399998
+-1,0,153.902643277000010,40.947595632099997
+-1,0,120.709162245000000,-32.220830297100001
+-1,0,170.320924217000000,-42.571485672500003
+-1,0,-70.956421774099994,-28.294719637499998
+-1,0,-126.278890160000000,-40.786889918100002
+-1,0,-99.509953844099996,0.258812432436000
+-1,0,-21.344659802599999,6.683357148180000
+-1,0,-34.907587535799998,50.584412705699997
+-1,0,62.888259803799997,38.449161576000002
+-1,0,12.205740379600000,-24.011689827000001
+-1,0,86.444923761499993,-19.014821714800000
+-1,0,123.564515451999990,64.861178740699998
+-1,0,251.402070826000000,4.257190516060000