예제 #1
0
    def __init__(self):
        #sys.path.append("/home/amigos/RX_system/base_param")

        self.sg2if1 = SG.secondsg01()
        self.sg2if2 = SG.secondsg02()

        #use device_table
        """
        for record in csv.reader(open(PATH_DEVICE_TABLE,"r")):
            r_dict[record[0]] =record[1]
            continue
        """

        pass
예제 #2
0
def interp2D(bfieldval, smoothval=501, colortype=yellowredblue, n=4096):
    parseData()
    data = getData()
    cmlabel = 'Normalized $\Delta$R$_D$ (k$\Omega$)'
    data = [
        x for (y, x) in sorted(zip(bfieldval, data), key=lambda pair: pair[0])
    ]
    bval = sorted(bfieldval)
    xmin = None
    xmax = 0
    for i in data:
        if i[0][0] > xmin:
            xmin = i[0][0]
        if i[0][-1] < xmax:
            xmax = i[0][-1]
    xmin += 0.5
    xmax -= 0.5
    xv = np.linspace(xmin, xmax, n)
    intv = list()
    if smoothval != 0:
        for i in data:
            dataclip = clip(i, xmin, xmax)
            #dataclip[1] = SG.savitzky_golay(dataclip[1],1751,3)
            tck = interpolate.splrep(dataclip[0], dataclip[1])
            y = interpolate.splev(xv, tck)
            y = SG.savitzky_golay(y, smoothval, 3)
            intv.append(normalize(y))
    else:
        for i in data:
            dataclip = clip(i, xmin, xmax)
            tck = interpolate.splrep(dataclip[0], dataclip[1])
            y = interpolate.splev(xv, tck)
            intv.append(normalize(y))
    xb = np.linspace(bval[0], bval[-1], n)
    b = list()
    for i in range(len(intv[0])):
        hold = list()
        for j in range(len(intv)):
            hold.append(intv[j][i])
        b.append(hold)
    intb = list()
    for i in b:
        tck = interpolate.splrep(bval, i)
        y = interpolate.splev(xb, tck)
        intb.append(y)
    Z = np.array(intb)
    fig = pylab.figure('ColorMap')
    X, Y = pylab.meshgrid(xb, xv)
    pylab.pcolormesh(X, Y, Z, vmin=0, vmax=1, cmap=colortype)
    pylab.ylim(xmin, xmax)
    pylab.xlim(bval[0], bval[-1])
    pylab.ylabel('Gate Voltage (mV)')
    pylab.xlabel('B Field (T)')
    cbar = pylab.colorbar()
    cbar.set_label(cmlabel)
    plt.ticker.ScalarFormatter(useOffset=None)
    pylab.show()
    gc.collect()
    return X, Y, Z
예제 #3
0
def basicBody(center):
    green = greenComp()
    green.rect.center = center

    red = redComp(green)
    blue = blueComp(green)

    return SG.LayeredSG(sprites=[red, blue, green])
예제 #4
0
 def setUpClass(cls):
     # Set some class properties so you can pass these off to all your test cases
     cls.baseUrl = os.environ.get("BASE_URL", "")
     cls.browser = os.environ.get("BROWSER", "chrome")
     cls.username = os.environ.get("USERNAME", "")
     cls.password = os.environ.get("PASSWORD", "")
     cls.project_id = os.environ.get("TEST_PROJECT_ID", 66)
     # Create a Shotgun API connection object
     cls.sg = Shotgun(cls.baseUrl,
                      login=cls.username,
                      password=cls.password)
     # Create a web driver
     cls.site = SG.ShotgunSite(baseUrl=cls.baseUrl, browser=cls.browser)
예제 #5
0
 def test_go_to_assets_page_and_run_quick_filter(self):
     # Create an asset, using the Shotgun API
     data = {
         "code": "Robot Test Asset",
         "project": {
             "type": "Project",
             "id": 66
         }
     }
     asset_id = self.sg.create("Asset", data)
     # Create a new page instance
     page = SG.Page(self.site, project_id=self.project_id)
     page.navigate_to_project_page(entity_type='Asset')
     time.sleep(2)
     page.toggle_page_mode('thumb')
     page.run_quick_filter(search_string="Robot Test Asset")
     # Save a screenshot
     self.site.driver.save_screenshot(
         'assets_page_thumb_after_quick_filter.png')
예제 #6
0
    depths = np.cumsum(th[::-1], axis=0)[::-1]
    DDL = np.zeros(len(c.yc))
    psi = c.get_psi_iso()
    for jj in range(len(c.yc)):
        if ma.all(PI[:, jj] == 1) or np.all(
                psi[:, jj] == -0) or PI[:, jj].mask.all():
            continue
        indx = ma.nonzero(PI[:, jj] < 0.9999999999)[0]
        a = indx[np.nonzero(indx > 3)[0]][0]
        if a < 41 and depths[a - 1, jj] - depths[a, jj] > 150:
            DDL[jj] = (depths[a - 1, jj] + depths[a, jj]) / 2
        else:
            DDL[jj] = depths[a, jj]

    r = ax.plot(c.yc / 1000,
                SG.savitzky_golay(-DDL / 1000, 21, 1),
                color='0.75',
                linewidth=4)
    # Lables
    ax.set_title(str(Figletter[Runs[i]]) + str(tau[Runs[i]]) + 'day',
                 fontsize=30)
    if str(tau[Runs[i]]) == 'Closed':
        ax.set_title(str(Figletter[Runs[i]]) + str(tau[Runs[i]]), fontsize=30)
    ax.set_xlabel('Distance (km)', fontsize=30)
    ax.set_ylabel('Depth (km)', fontsize=30)
    ax.tick_params(axis='both', which='major', labelsize=30)

plt.tight_layout()
cax = fig.add_axes([1, 0.1, 0.03, 0.8])
cbar = fig.colorbar(p, cax=cax, ticks=eke_ticks)
cbar.ax.set_ylabel(
예제 #7
0
def makeBodies():
    x = basicBody((500, 500))
    y = basicBody((700, 300))
    return SG.handoffRender(x, y)
예제 #8
0
    for jj in range(len(c.yc)):
        if ma.all(PI[:, jj] == 1) or np.all(
                psi[:, jj] == -0) or PI[:, jj].mask.all():
            continue
        indx = ma.nonzero(PI[:, jj] < 1)[0]
        b = indx[np.nonzero(indx > 3)[0]]
        if len(b) >= 2 and (b[1] - b[0]) > 1:
            a = b[1]
        else:
            a = b[0]
        if a < 41 and depths[a - 1, jj] - depths[a, jj] > 150:
            a = a - 1
        DDL[jj] = depths[a, jj]

    ax = fig.add_subplot(2, 2, i + 1)
    p = plt.plot(c.yc / 1000, SG.savitzky_golay(-DDL, 31, 1), 'r', linewidth=3)
    plt.ylim(-2895, 0)
    ax.set_title(str(Figletter[Runs[i]]) + str(tau[Runs[i]]) + 'day',
                 fontsize=30)
    if str(tau[Runs[i]]) == 'Closed':
        ax.set_title(str(Figletter[Runs[i]]) + str(tau[Runs[i]]), fontsize=30)
    ax.set_xlabel(r'$km$', fontsize=30)
    ax.set_ylabel(r'Depth', fontsize=30)
    ax.tick_params(axis='both', which='major', labelsize=30)

plt.tight_layout()
plt.savefig(x + '/Figures/Diabatic_layer_depth.png')

fig = plt.figure(figsize=(18.5, 16.5))
for i in range(len(Runs)):