Last change
on this file since 1647 was
1647,
checked in by Max Voronkov, 15 years ago
|
slight modification of the output of the simple line finder test
|
-
Property svn:executable set to
*
|
File size:
1.3 KB
|
Rev | Line | |
---|
[1646] | 1 | #!/usr/bin/env python |
---|
| 2 | from asap import * |
---|
| 3 | import time,sys |
---|
| 4 | |
---|
| 5 | wait_str = "" |
---|
| 6 | if len(sys.argv)>=2: |
---|
| 7 | if sys.argv[1].lower() == "wait": |
---|
| 8 | wait_str = " Press Enter." |
---|
| 9 | |
---|
| 10 | # read the spectrum from ascii file |
---|
| 11 | f = open('data/atca_spectrum.dat') |
---|
| 12 | spc=[] |
---|
| 13 | try: |
---|
| 14 | for line in f: |
---|
| 15 | parts=line.split() |
---|
| 16 | if len(parts)!=3: |
---|
| 17 | raise RuntimeError, "Expect 3 parts, you have %s"% (parts,) |
---|
| 18 | spc.append(float(parts[2])) |
---|
| 19 | finally: |
---|
| 20 | f.close() |
---|
| 21 | |
---|
| 22 | |
---|
| 23 | #run the line finder |
---|
| 24 | |
---|
| 25 | lf = simplelinefinder() |
---|
| 26 | rng=lf.find_lines(spc,threshold=3,splitFeatures=False) |
---|
| 27 | |
---|
| 28 | print "Found %i lines.%s" % (len(rng),wait_str) |
---|
| 29 | b=len(spc)/4 |
---|
| 30 | e=len(spc)*3/4 |
---|
| 31 | plotter._plotter.plot(range(b,e),spc[b:e]) |
---|
[1647] | 32 | plotter._plotter.set_line(number=0,label='Input spectrum') |
---|
[1646] | 33 | plotter._plotter.show() |
---|
| 34 | if wait_str == "": |
---|
| 35 | time.sleep(1.5) |
---|
| 36 | else: |
---|
| 37 | raw_input() |
---|
| 38 | cnt = 1 |
---|
| 39 | for line in rng: |
---|
| 40 | if len(line)!=2: |
---|
| 41 | raise RuntimeError, "Line is supposed to have two parameters" |
---|
| 42 | b = line[0]-1 |
---|
| 43 | if b<0: |
---|
| 44 | b=0 |
---|
| 45 | e = line[1]+1 |
---|
| 46 | if e>len(spc): |
---|
| 47 | e=len(spc) |
---|
[1647] | 48 | print "Showing line %i located at %s.%s" % (cnt,line,wait_str) |
---|
[1646] | 49 | plotter._plotter.set_line(label='Line %i' % cnt) |
---|
| 50 | cnt = cnt + 1 |
---|
| 51 | plotter._plotter.plot(range(b,e),spc[b:e]) |
---|
| 52 | plotter._plotter.show() |
---|
| 53 | if wait_str == "": |
---|
| 54 | time.sleep(1.5) |
---|
| 55 | else: |
---|
| 56 | raw_input() |
---|
| 57 | |
---|
[1647] | 58 | print "Test of simple line finder is successful" |
---|
Note: See
TracBrowser
for help on using the repository browser.