ClipBorder.py

# -*- coding: utf-8 -*-
#
#     Author:     Liu xianyao
#     Email:      flashlxy@qq.com
#     Update:     2017-04-06
#     Copyright:  ©江西省气象台 2017
#     Version:    1.1.20170409
from __future__ import print_function

from datetime import datetime

from matplotlib.path import Path

import maskout
from Projection import Projection


class ClipBorder:
    """
    边界裁切类
    """

    def __init__(self, leaf):
        start_pos = 0
        filehead = Projection.leaf_to_string(leaf, "File")
        filetype = Projection.leaf_to_string(leaf, "Type", 'shp')
        code = Projection.leaf_to_list(leaf, "Code", [360000])
        drawswitch = str.upper(Projection.leaf_to_string(leaf, "Draw", 'off'))
        self.path = self.getPath(filehead=filehead, code=code, filetype=filetype, start_pos=start_pos)
        self.draw = str.upper(Projection.leaf_to_string(leaf, "Draw", 'off'))
        self.using = Projection.leaf_to_bool(leaf, "Using", True)
        self.linewidth = Projection.leaf_to_float(leaf, "LineWidth", 1)
        self.linecolor = Projection.leaf_to_string(leaf, "LineColor", 'k') if drawswitch == 'ON' else 'none'

    @staticmethod
    def getPath(filehead, code, filetype, start_pos):
        """
        根据文件类型获取path对象
        :param start_pos: 正文开始的位置，对txt文件有效
        :param filehead: 文件名
        :param code: 闭合区域行政区号-对shp文件有效
        :param filetype: 文件类型
        :return: path对象的一个实例
        """
        if filetype == 'shp':
            path = maskout.getPathFromShp(filehead, code)
        else:
            path = ClipBorder.readPath(filehead, start_pos)
        return path

    @staticmethod
    def readPath(filename, start_pos=13):
        """
        从类似第9类micaps数据中获取path
        :param start_pos: 
        :param filename: 数据文件全名
        :return: path对象的一个实例
        """
        try:
            file_object = open(filename)
            all_the_text = file_object.read()
            file_object.close()
            poses = all_the_text.strip().split()
            lon = [float(p) for p in poses[start_pos::2]]
            lat = [float(p) for p in poses[start_pos+1::2]]
            path = Path(zip(lon, lat))
            return path
        except Exception as err:
            print(u'【{0}】{1}-{2}'.format(filename, err, datetime.now()))
            return None