Exemplo n.º 1
0
def plot_hor_mag(gplt):
    text = gen_codisp_hor_mag_file()
    gmt = pGMT.GMT()
    gmt.nearneighbor(text.name, G='~ver.grd', I='1k', N='8', R='', S='60k')

    gmt = pGMT.GMT()
    gmt.makecpt(C='seminf-haxby.cpt', T='-3/.6/0.1', Q='', M='')
    gmt.save_stdout('~hor_mag.cpt')

    gplt.grdimage('~ver.grd', J='', R='', C='~hor_mag.cpt', O='', K='')
    text.close()

    # fill water with white color
    gplt.pscoast(R='', J='', S='white', O='', K='')
Exemplo n.º 2
0
    def plot_slip_contour(
        self,
        contours=[5, 10, 20, 40, 60],
        W='thickest',
    ):
        _txt = ''
        for ii in contours:
            _txt += '%f A\n' % ii

        with tempfile.NamedTemporaryFile('w+t') as fid:
            fid.write(_txt)
            fid.seek(0, 0)

            gmt = pGMT.GMT()

            self.gplt.grdcontour(
                self.slip_file_grd_cutted,
                C=fid.name,
                A='1+f9+um',
                G='n1/.5c',
                J='',
                R='',
                O='',
                K='',
                W=W,
            )
Exemplo n.º 3
0
    def plot(self):
        self._get_info_from_fault_file_and_result_file()

        self.gmt = gmt = pGMT.GMT()
        self.gplt = gplt = gmt.gplt        

        gmt.gmtset('ANNOT_FONT_SIZE_PRIMARY','6',
                   'LABEL_FONT_SIZE','6',
                   'FONT_ANNOT_PRIMARY','4',
                   'MAP_FRAME_TYPE','plain',
                   'MAP_FRAME_PEN','thinner,black',
                   'PS_MEDIA','letter',
                   'MAP_TICK_LENGTH_PRIMARY','.05',
                   'MAP_ANNOT_OFFSET_PRIMARY','.05')

        self._plot_coseismic_slip()

        # plot afterslip:
        self._num_of_subplots_in_a_row = 1
        for nth, si in enumerate(self.slip[1:,:,:],1):
            offset_X, offset_Y = self._next_offset()

            _plt = self._plot_afterslip_at_nth_epoch(
                nth = nth,
                offset_X = offset_X,
                offset_Y = offset_Y
                )
            if nth ==1:
                _plt.add_psscale(
                    gridline_interval = self.color_label_interval_aslip)                
Exemplo n.º 4
0
def plot_vertical_coseismic_disp(gplt):
    text = gen_coseismic_vertical_file()
    gmt = pGMT.GMT()
    gmt.nearneighbor(text.name, G='~ver.grd', I='1k', N='8', R='', S='60k')
    gplt.grdimage('~ver.grd', J='', R='', C='vertical_disp.cpt', O='', K='')

    text.close()
    # fill water with white color
    gplt.pscoast(R='', J='', S='white', O='', K='')
Exemplo n.º 5
0
    def _trench_cut_slip_grd(self, grd_in, grd_out):
        gmt = pGMT.GMT()
        file_mask = join(self._workdir, 'trench_cut_mask.grd')
        gmt.grdmask(file_kur_top,
                    G=file_mask,
                    A='',
                    N='1/1/NaN',
                    I=self.I,
                    R='')

        gmt.grdmath(grd_in, file_mask, 'OR', '= ', grd_out)
Exemplo n.º 6
0
def plot_etopo1(gplt, A='-70/20', file_topo_cpt=topo_cpts['afrikakarte']):
    gmt = pGMT.GMT()

    file_topo_cut = tempfile.NamedTemporaryFile()
    gmt.grdcut(file_etopo1, G=file_topo_cut.name, R='')

    gmt = pGMT.GMT()
    file_topo_grad = tempfile.NamedTemporaryFile()
    gmt.grdgradient(file_topo_cut.name, G=file_topo_grad.name, A=A, R='')

    gplt.grdimage(file_topo_cut.name,
                  J='',
                  C=file_topo_cpt,
                  I=file_topo_grad.name,
                  O='',
                  K='')

    # close temporary files:
    file_topo_cut.close()
    file_topo_grad.close()
Exemplo n.º 7
0
    def init(self):
        self.gmt = pGMT.GMT()
        self.gplt = self.gmt.gplt

        self.gmt.gmtset('ANNOT_FONT_SIZE_PRIMARY', '10', 'LABEL_FONT_SIZE',
                        '6', 'FONT_ANNOT_PRIMARY', '10', 'MAP_FRAME_TYPE',
                        'plain', 'MAP_FRAME_PEN', 'thinner,black', 'PS_MEDIA',
                        'letter', 'MAP_TICK_LENGTH_PRIMARY', '.05',
                        'MAP_ANNOT_OFFSET_PRIMARY', '.05')

        self._make_cpt_file()
Exemplo n.º 8
0
    def _interpolate_slip_file(self, out_grd):
        make_basename_dir(out_grd)
        gmt = pGMT.GMT()
        gmt.nearneighbor(self.slip_file_txt,
                         G='"%s"' % out_grd,
                         I=self.I,
                         N='8',
                         R='',
                         S='40k')

        gmt.grdfilter(out_grd, G=out_grd, D='4', F='c25')
Exemplo n.º 9
0
    def test(self):
        gmt = pGMT.GMT()

        gmt.gplt.psbasemap(
            R='5/15/52/58',  # region
            J='B10/55/55/60/10c',  # projection
            B='4g4',
            K='')

        gmt.gplt.pscoast(R='', J='', D='f', W='thinnest', O='')

        gmt.save(join(self.outs_dir, 'out.pdf'))
        gmt.save_shell_script(join(self.outs_dir, 'shell.sh'),
                              output_file='>out.ps')
Exemplo n.º 10
0
    def _prepare_intensity_file(self):
        self.intensity_file = join(self._workdir, 'intensity.grd')
        self._cutted_topo = join(self._workdir, 'cutted_topo.grd')
        _intensity_file_before_resample = join(
            self._workdir, '~intensity_before_resample.grd')
        gmt = pGMT.GMT()
        gmt.grdcut(self.topo_file, G=self._cutted_topo, R='')
        gmt.grdgradient(self._cutted_topo,
                        G=_intensity_file_before_resample,
                        A=self.A,
                        R='')

        gmt.grdsample(_intensity_file_before_resample,
                      G=self.intensity_file,
                      I=self.I)
Exemplo n.º 11
0
def plot(
    infile,
    outfile,
    if_log_color_scale,
    cpt_scale,
    contours,
    scale_interval,
):
    gmt = pGMT.GMT()
    gmt.gmtset('ANNOT_FONT_SIZE_PRIMARY', '9', 'LABEL_FONT_SIZE', '9',
               'COLOR_NAN', 'white', 'COLOR_BACKGROUND', 'white',
               'COLOR_FOREGROUND', 'white')

    gmt.gplt.psbasemap(
        R='128/148/30/46',  # region
        J='B138/38/30/46/16c',  # projection
        B='4',
        U='20/0/25/Yang',
        P='',
        K='')

    plt_ver = vj.gmt.GMTXYZ(
        gmt,
        infile,
        cpt_scale=cpt_scale,
        if_log_color_scale=if_log_color_scale,
    )
    plt_ver.plot_xyz()
    plt_ver.plot_scale(scale_interval='1')
    plt_ver.plot_contour(contours=contours, W='thick')

    vj.gmt.plot_plate_boundary(gmt.gplt, color='150')

    gmt.gplt.pscoast(R='',
                     J='',
                     D='h',
                     N='a/faint,50,--',
                     W='faint,100',
                     L='f145/31/38/200+lkm+jt',
                     O='',
                     K='')

    vj.gmt.plot_Tohoku_focal_mechanism(gmt.gplt, K=None)

    gmt.save(outfile)
    gmt.gplt.save_shell_script('shell.sh', output_file=' > out.ps')
Exemplo n.º 12
0
    def _prepare_cpt_file(self,
                          original_cpt_file='no_green',
                          if_cpt_reverse=False):
        self.cpt_file = join(self._workdir, 'slip.cpt')
        gmt = pGMT.GMT()

        if if_cpt_reverse:
            I = ''
        else:
            I = None
        gmt.grd2cpt(
            self.slip_file_grd,
            #C='temperature.cpt',
            C=original_cpt_file,
            Z='',
            I=I,
        )
        gmt.save_stdout(self.cpt_file)
    def plot_stations_map(self):
        gmt = pGMT.GMT()

        gmt.gmtset('ANNOT_FONT_SIZE_PRIMARY', '9', 'LABEL_FONT_SIZE', '9',
                   'FONT_ANNOT_PRIMARY', '6', 'MAP_FRAME_TYPE', 'plain')

        gplt = gmt.gplt

        gplt.psbasemap(
            R='125/146/30/46',  # region
            JB='135.5/38.5/35/41.5/16c',  # projection
            B='5',
            U='18/25/0',
            P='',
            K='',
        )

        gplt.pscoast(R='',
                     J='',
                     D='h',
                     N='a/faint,150,-.',
                     W='faint,dimgray',
                     A='500',
                     L='144/32/38/200+lkm+jt',
                     O='',
                     K='')

        plot_stations(
            gplt,
            self.sites,
            S='s.2',
            color='red',
            fill_color='red',
            fontcolor='blue',
            fontsize='8',
            text_offset_X=0,
            text_offset_Y=0,
        )

        plot_focal_mechanism_JMA(gplt, scale=.2, fontsize=0)

        gplt.finish()
        gmt.save(self.file_sites_map)
Exemplo n.º 14
0
from os.path import join
import numpy as np
import tempfile

import pGMT

import viscojapan as vj

gmt = pGMT.GMT()
gmt.gmtset(
    'ANNOT_FONT_SIZE_PRIMARY',
    '9',
    'LABEL_FONT_SIZE',
    '9',
    'BASEMAP_TYPE',
    'PLAIN',
)

gplt = gmt.gplt

lon1 = 131.5
lon2 = 135.7
lat1 = 33
lat2 = 35.8

gplt.psbasemap(
    R='{lon1}/{lon2}/{lat1}/{lat2}'.format(lon1=lon1,
                                           lon2=lon2,
                                           lat1=lat1,
                                           lat2=lat2),  # region
    J='B{lon0}/{lat0}/{lat1}/{lat2}/18c'.format(lon0=(lon1 + lon2) / 2.,
Exemplo n.º 15
0
 def __init__(self):
     self.gmt = pGMT.GMT()
     self.gplt = self.gmt.gplt
Exemplo n.º 16
0
def plot(
    disp_file1,
    disp_file2,
    outf,
    leg1='',
    leg2='',
    color1='black',
    color2='red',
):
    gmt = pGMT.GMT()
    gmt.gmtset(
        'ANNOT_FONT_SIZE_PRIMARY',
        '9',
        'LABEL_FONT_SIZE',
        '9',
        'BASEMAP_TYPE',
        'PLAIN',
    )

    gplt = gmt.gplt

    lon1 = 133.5
    lon2 = 135.1
    lat1 = 34.5
    lat2 = 35.5

    gplt.psbasemap(
        R='{lon1}/{lon2}/{lat1}/{lat2}'.format(lon1=lon1,
                                               lon2=lon2,
                                               lat1=lat1,
                                               lat2=lat2),  # region
        J='B{lon0}/{lat0}/{lat1}/{lat2}/18c'.format(lon0=(lon1 + lon2) / 2.,
                                                    lat0=(lat1 + lat2) / 2.,
                                                    lat1=lat1,
                                                    lat2=lat2),  # projection
        B='.5',
        U='20/0/22/Yang',
        K='')

    gplt.pscoast(R='',
                 J='',
                 D='h',
                 N='a/faint,50,--',
                 A='500',
                 G='lightblue2',
                 O='',
                 K='')

    scale = 27

    # plot disp file1
    plt_vec = vj.gmt.VecFieldPlotter(gmt, disp_file1, scale, color1)
    plt_vec.plot_vectors(arrow_width='.2',
                         head_length='.1',
                         head_width='.1',
                         pen_width='1.2')
    plt_vec.plot_vec_legend(
        lon=135,
        lat=33.4,
        leg_len=0.05,
        leg_txt=leg1,
        text_offset_lon=0,
        text_offset_lat=-0.06,
    )

    # plot disp file2
    plt_vec = vj.gmt.VecFieldPlotter(gmt, disp_file2, scale, color2)
    plt_vec.plot_vectors(arrow_width='.2',
                         head_length='.1',
                         head_width='.1',
                         pen_width='1.2')
    plt_vec.plot_vec_legend(
        lon=135,
        lat=33.2,
        leg_len=0.05,
        leg_txt=leg2,
        text_offset_lon=0,
        text_offset_lat=-0.06,
    )

    gplt.pscoast(R='',
                 J='',
                 D='h',
                 N='a/faint,50,--',
                 A='500',
                 W='faint,100',
                 L='f144/35.4/38/100+lkm+jt',
                 O='',
                 K='')

    #plot_horizontal_coseismic_vector(gplt)

    vj.gmt.plot_focal_mechanism_JMA(gplt)

    gplt.finish()
    gmt.save(outf)
Exemplo n.º 17
0
 def _land_cut_slip_grd(self, grd_in, grd_out):
     gmt = pGMT.GMT()
     file_mask = join(self._workdir, 'land_mask.grd')
     gmt.grdlandmask(R='', Dh='', I=self.I, N='1/NaN', G=file_mask)
     gmt.grdmath(grd_in, file_mask, 'OR', '= ', grd_out)
Exemplo n.º 18
0
    def plot(
        self,
        lons,
        lats,
        mags,
        output_file,
        contours=[0.001, 0.0026, 0.005, 0.01, 0.1, 0.2],
        if_topo=True,
        unit_label='m',
        title='',
    ):
        gmt = pGMT.GMT()
        gmt.gmtset('ANNOT_FONT_SIZE_PRIMARY', '9', 'LABEL_FONT_SIZE', '9',
                   'FONT_ANNOT_PRIMARY', '6', 'MAP_FRAME_TYPE', 'plain')

        gplt = gmt.gplt

        north = 60
        south = 12
        east = 190
        west = 100
        gplt.psbasemap(
            R = '{west}/{east}/{south}/{north}'.format(
                west = west,
                east = east,
                south = south,
                north = north),       # region
            JD = '{lon0}/{lat0}/{lat1}/{lat2}/{wid}'\
            .format(lon0 = vj.TOHOKU_EPICENTER[0],
                    lat0 = vj.TOHOKU_EPICENTER[1],
                    lat1 = south,
                    lat2 = north,
                    wid = '9i',
                    ), # projection
            B = '20:."%s":'%title, U='18/25/0',
            K = '')

        # topo
        if if_topo:
            vj.gmt.plot_etopo1(gplt)

        with tempfile.NamedTemporaryFile('w+t') as fid:
            self._write_to_temp_file(fid, lons, lats, mags)

            pltxyz = vj.gmt.GMTXYZ(
                gmt,
                file_xyz=fid.name,
                if_log_color_scale=True,
                cpt_scale='-3.1/0.6/0.001',
                interp_inc='40k',
                interp_searching_radius='10',
            )
            #pltxyz.maskout_water(A='1000k',D='h')
            #pltxyz.plot_xyz()
            pltxyz.plot_contour(
                contours=contours,
                W='thick,red',
                label_line=
                'L142.37/38.30/80/60,142.37/38.30/90/20,142.37/38.30/-160/40,158/38.30/180/5',
                label_font_size=8,
                smooth_factor=100,
                unit_label=unit_label,
            )

        #pltxyz.plot_scale(x=15, y=8)

        # plot coast
        gplt.pscoast(R='',
                     J='',
                     D='h',
                     N='a/faint,100,-.',
                     W='faint,50',
                     A='5000',
                     Lf='180/15/35/500+lkm+jt',
                     K='',
                     O='')

        # plot plate boundary
        vj.gmt.plot_plate_boundary(gplt, color='100')
        vj.gmt.plot_focal_mechanism_JMA(gplt, scale=0.2, fontsize=0)

        gplt.finish()

        gmt.save(output_file)
Exemplo n.º 19
0
 def _low_cut_slip_grd(self, grd_in, grd_out):
     gmt = pGMT.GMT()
     gmt.grdclip(grd_in, G=grd_out, Sb='%f/NaN' % self.low_cut_value)