source: rotdif/bin/preprocess_2d_3d.py @ 1539

Last change on this file since 1539 was 1539, checked in by yuexi, 3 years ago

plot 2d 3d

  • Property svn:executable set to *
File size: 2.9 KB
Line 
1#!/usr/bin/python
2import os
3import glob
4import re
5import shutil
6import sys
7import subprocess
8import json
9import cStringIO
10import shlex 
11import socket
12import time
13import pty
14import select
15import atexit
16import signal
17import functools
18import itertools
19import math
20import pandas as pd
21import numpy as np
22
23from itertools import groupby
24from matplotlib import cm, colors
25
26from StringIO import StringIO
27#from math import exp, expm1, log10, log, log1p
28from subprocess import Popen, PIPE, STDOUT
29
30def FileNameArr():
31    ExpArr       = []
32    FitArr_upper = []
33    FitArr_lower = []
34    LabelArr     = []
35    Surface3d    = []
36    Exp3d        = []
37    rotout       = []
38    for root, dirs, files in os.walk('./'):
39        for name in files:
40            filename = os.path.join(root, name)
41           
42            if '2d_fit_upper' in filename:
43                FitArr_upper.append(filename)
44
45            if '2d_fit_lower' in filename:
46                FitArr_lower.append(filename)
47               
48            if '2d_exp_' in filename:
49                ExpArr.append(filename)
50               
51            if '2d_label_' in filename:
52                LabelArr.append(filename) 
53               
54            if '3d_surface_' in filename:
55                Surface3d.append(filename)
56           
57            if '3d_exp_points_' in filename:
58                Exp3d.append(filename)
59            if 'rotdif' in filename:
60                rotout.append(filename)
61    ExpArr.sort()
62    FitArr_upper.sort()
63    FitArr_lower.sort()
64    LabelArr.sort()
65    Surface3d.sort()
66    Exp3d.sort()
67
68    return ExpArr, FitArr_upper, FitArr_lower, LabelArr, Surface3d, Exp3d
69
70def GetData(filename):
71    file = open(filename, 'rU')
72    data = file.readlines()
73    file.close()
74       
75    xData = []
76    yData = []
77    eData = []
78    tData = []
79   
80    for line in data:
81        xData.append(re.split(r'[\s]*', line)[0])
82        yData.append(re.split(r'[\s]*', line)[1])
83        if '2d_exp_' in filename:
84            eData.append(re.split(r'[\s]*', line)[2])
85           
86            tooltipArr = re.split(r'[\s]*', line)[-3:]
87            tooltipStr = ""
88            for ch in range(len(tooltipArr)):
89                tooltipStr += tooltipArr[ch] + " " 
90            tooltipStr.rstrip() 
91            tData.append(tooltipStr)
92
93    return xData, yData, eData, tData
94
95def GetData_3d(filename):
96    file = open(filename, 'rU')
97    data = file.readlines()
98    file.close()
99       
100    xData = []
101    yData = []
102    zData = []
103       
104    for line in data:
105        xData.append(re.split(r'[\s]*', line)[0])
106        yData.append(re.split(r'[\s]*', line)[1])
107        zData.append(re.split(r'[\s]*', line)[2])
108
109    return xData, yData, zData
110
111################################################################################
112def GetLabels(filename):
113    file = open(filename, 'rU')
114    data = file.readlines()
115    file.close()
116       
117    Labels = []
118   
119    for line in data:
120        Labels.append(line)
121       
122    return Labels
123
Note: See TracBrowser for help on using the repository browser.