1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
package bin;
import java.io.File;
import java.util.ArrayList;
import json.JsonObject;
import shared.PreParser;
import sketch.DisplayParams;
import sketch.SendSketch;
import sketch.Sketch;
import sketch.SketchIdComparator;
import sketch.SketchObject;
import sketch.SketchTool;
/** Renames files with top sketch hit taxid */
public class FileRenamer {
public static void main(String[] args) {
PreParser pp=new PreParser(args, null, false);
args=pp.args;
DisplayParams params=new DisplayParams();
params.format=DisplayParams.FORMAT_JSON;
SketchObject.postParse();
SketchTool tool=new SketchTool(10000, params);
ArrayList<Sketch> inSketches=tool.loadSketches_MT(params, args);
assert(inSketches.size()==args.length) : inSketches.size()+" != "+args.length;
final int numLoaded=(inSketches.size());
if(numLoaded>1 && params.mode==Sketch.PER_FILE){
inSketches.sort(SketchIdComparator.comparator);//Otherwise they come out disordered
}
ArrayList<JsonObject> results=SendSketch.sendSketches(inSketches, "refseq", params);
assert(results.size()==inSketches.size()) : inSketches.size()+" != "+results.size();
for(int i=0; i<args.length; i++) {
String fname=args[i];
JsonObject result=results.get(i);
JsonObject top=null;
if(result!=null && result.jmapSize()>0) {
for(String key : result.jmap.keySet()){
top=result.jmap.get(key);
break;
}
}
SketchRecord topHit=(top==null ? null : new SketchRecord(top));
int taxid=(topHit==null ? -1 : topHit.taxid);
File f=new File(fname);
String fname2="tid_"+taxid+"_"+fname;
File f2=new File(fname2);
assert(f.exists());
assert(!f2.exists());
f.renameTo(f2);
}
}
}
|
