예제 #1
0
def main(fpath, viscous_on, force_decay, nsampling):
    cwd = os.getcwd()
    os.chdir('../build')
    #call(['cmake', '..'])
    #call(['make'])

    ns_str = '%s' % nsampling
    cmd = ['./main', fpath, viscous_on, force_decay, ns_str]

    proc = check_output(cmd)

    fstr = None
    try:
        # used in python 2.*
        from StringIO import StringIO as fstring
        fstr = fstring(str(proc))
    except ImportError:
        # used in python 3.*
        #from io import BytesIO as fstring
        from io import StringIO as fstring
        fstr = fstring(str(proc, 'utf-8'))

    particle_lists = pd.read_csv(fstr,
                                 sep=' ',
                                 header=None,
                                 dtype=np.float32,
                                 comment='#').values

    #print('particle list read in')
    #np.savetxt('mc_particle_list.txt', particle_lists)
    #print('particle list saved')

    os.chdir(cwd)

    plot(fpath, particle_lists, nsampling=nsampling)
예제 #2
0
def main(fpath, viscous_on, force_decay, nsampling):
    from subprocess import call, check_output
    cwd = os.getcwd()
    os.chdir('../build')
    call(['cmake', '..'])
    call(['make'])

    ns_str = '%s' % nsampling
    cmd = ['./main', fpath, viscous_on, force_decay, ns_str]

    proc = check_output(cmd)

    try:
        # used in python 2.*
        from StringIO import StringIO as fstring
    except ImportError:
        # used in python 3.*
        from io import StringIO as fstring

    #particle_lists = np.genfromtxt(fstring(proc))

    particle_lists = pd.read_csv(fstring(proc),
                                 sep=' ',
                                 header=None,
                                 dtype=np.float32,
                                 comment='#').values

    create_figure_matrix(fpath, particle_lists)

    #from mcspec import mcspec
    #mcspec(fstring(proc))

    os.chdir(cwd)
예제 #3
0
def main(fpath, viscous_on, force_decay, nsampling):
    from subprocess import call, check_output
    cwd = os.getcwd()
    os.chdir('../build')
    call(['cmake', '..'])
    call(['make'])

    ns_str = '%s'%nsampling
    cmd = ['./main', fpath, viscous_on, force_decay, ns_str]

    proc = check_output(cmd)

    try:
        # used in python 2.*
        from StringIO import StringIO as fstring
    except ImportError:
        # used in python 3.*
        from io import StringIO as fstring

    #particle_lists = np.genfromtxt(fstring(proc))

    particle_lists = pd.read_csv(fstring(proc), sep=' ', header=None, dtype=np.float32, comment='#').values

    print('particle list read in')

    #np.savetxt('mc_particle_list.txt', particle_lists)

    print('particle list saved')

    os.chdir(cwd)

    plot(fpath, particle_lists, nsampling = nsampling)
예제 #4
0
    def __init__(self, events_str, rapidity_kind='eta', fpath='./'):
        self.rapidity_col = 4
        if rapidity_kind == 'eta':
            self.rapidity_col = 6

        self.events = [
            np.genfromtxt(fstring(event))
            for event in events_str.split('#finished')[:-1]
        ]

        self.num_of_events = len(self.events)
        print('in mcspec, num of events=', self.num_of_events)
예제 #5
0
 def __parse(self):
     for cent, headers in self.info.items():
         idx = headers['table_id']
         base_path = os.path.join(os.getcwd(),
                                  'data/ALICE_pbpb2760_vn_vs_eta/',
                                  'Table%s.csv' % idx)
         cent_data = {}
         with open(base_path, 'r') as myfile:
             data = myfile.read().split('\n\n')[:-1]
             for idx, vn_type in enumerate(headers['files']):
                 data_str = fstring(data[idx])
                 cent_data[vn_type] = pd.read_csv(data_str,
                                                  sep=',',
                                                  comment='#')
         self.data[cent] = cent_data
예제 #6
0
    def __init__(self, events_str, rapidity_kind='eta', fpath='./'):
        self.rapidity_col = 4
        if rapidity_kind == 'eta':
            self.rapidity_col = 6

        self.events = [
            np.genfromtxt(fstring(event))
            for event in events_str.split('#finished')[:-1]
        ]

        #self.events = [pd.read_csv(fstring(event), sep=' ',
        #               header=None, dtype=np.float64).values
        #               for event in events_str.split('#finished')[:-1]]

        self.num_of_events = len(self.events)
        print('in mcspec, num of events=', self.num_of_events)
        print(self.events[9999])

        self.fpath = fpath
예제 #7
0
    args = parser.parse_args()

    for eid in range(args.start_id, args.end_id):
        fpath = os.path.join(args.input_dir, "event%d" % eid)

        if os.path.exists(os.path.join(fpath, 'vn24_vs_eta.txt')):
            print("vn24_vs_eta exists")
            continue

        cwd = os.getcwd()
        os.chdir('../build')
        #call(['cmake', '..'])
        #call(['make'])

        try:
            cmd = ['./main', fpath, 'true', 'true', '500']

            proc = check_output(cmd)

            stio = fstring()
            stio.write(proc)

            mc = mcspec(stio.getvalue(), fpath=fpath)

            mc.vn_vs_eta()

            #mc.plot_vn_pt()
            os.chdir(cwd)
        except:
            print("event does not have full evolution")