File: AssemblyStats3.java

package info (click to toggle)
bbmap 39.20%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 26,024 kB
  • sloc: java: 312,743; sh: 18,099; python: 5,247; ansic: 2,074; perl: 96; makefile: 39; xml: 38
file content (143 lines) | stat: -rwxr-xr-x 3,866 bytes parent folder | download | duplicates (2)
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
package jgi;

import java.io.File;
import java.util.ArrayList;

import fileIO.ByteStreamWriter;
import fileIO.FileFormat;
import shared.Parser;
import shared.PreParser;
import shared.Shared;
import shared.Timer;
import shared.Tools;
import structures.ByteBuilder;

/**
 * Generates some stats from multiple files.
 * Uses the new Assembly class.
 * @author Brian Bushnell
 * @date January 21, 2025
 *
 */
public class AssemblyStats3 {

	public static void main(String[] args){
		//Start a timer immediately upon code entrance.
		Timer t=new Timer();
		
		//Create an instance of this class
		AssemblyStats3 x=new AssemblyStats3(args);
		
		//Run the object
		x.process(t);
		
		//Close the print stream if it was redirected
		Shared.closeStream(x.outstream);
	}
	
	public AssemblyStats3(String[] args){
		
		{//Preparse block for help, config files, and outstream
			PreParser pp=new PreParser(args, getClass(), false);
			args=pp.args;
			outstream=pp.outstream;
		}
		
		Parser parser=new Parser();
		parser.out1=out1;
		for(int i=0; i<args.length; i++){
			String arg=args[i];
			String[] split=arg.split("=");
			String a=split[0].toLowerCase();
			String b=split.length>1 ? split[1] : null;
			if(b!=null && b.equalsIgnoreCase("null")){b=null;}

			if(a.equals("parse_flag_goes_here")){
				//Set a variable here
			}else if(a.equals("in")){
				in.clear();
				String[] b2=(b==null) ? null : (new File(b).exists() ? new String[] {b} : b.split(","));
				for(String b3 : b2){in.add(b3);}
			}else if(b==null && new File(arg).exists()){
				in.add(arg);
			}else if(parser.parse(arg, a, b)){
				//do nothing
			}else{
				assert(false) : "Unknown parameter "+args[i];
				outstream.println("Unknown parameter "+args[i]);
			}
		}
		
		{//Process parser fields
			Parser.processQuality();
			
			maxReads=parser.maxReads;
			out1=parser.out1;
		}
		
		ffout1=FileFormat.testOutput(out1, FileFormat.TXT, null, true, true, false, false);
	}
	
	void process(Timer t){

		final ByteStreamWriter bsw=ByteStreamWriter.makeBSW(ffout1);
		if(bsw!=null) {bsw.println(makeHeader());}
		
		for(String s : in) {
			processInner(s, bsw);
		}
		
		if(bsw!=null) {bsw.poisonAndWait();}
		if(verbose){outstream.println("Finished.");}
		
		t.stop();
		Tools.timeLinesBytesProcessed(t, linesProcessed, bytesProcessed, 12);
	}
	
	public static String makeHeader() {
		ByteBuilder bb=new ByteBuilder();
		bb.append("fname");
		bb.tab().append("size");
		bb.tab().append("contigs");
		bb.tab().append("gc");
		bb.tab().append("maxContig");
		bb.tab().append("5kplus");
		bb.tab().append("10kplus");
		bb.tab().append("25kplus");
		bb.tab().append("50kplus");
		return bb.toString();
	}
	
	void processInner(String fname, ByteStreamWriter bsw) {
		Assembly a=new Assembly(fname);
		if(bsw==null) {return;}
		int max=a.contigs.size>0 ? a.contigs.get(0) : 0;
		bsw.print(fname).tab().print(a.length).tab().print(a.contigs.size);
		bsw.tab().print(a.gc(), 3).tab().print(max);
		bsw.tab().print(a.lengthAtLeast(5000));
		bsw.tab().print(a.lengthAtLeast(10000));
		bsw.tab().print(a.lengthAtLeast(25000));
		bsw.tab().print(a.lengthAtLeast(50000));
		bsw.nl();
	}
	
	/*--------------------------------------------------------------*/
	
	/*--------------------------------------------------------------*/
	
	private ArrayList<String> in=new ArrayList<String>();
	private String out1="stdout.txt";
	
	private final FileFormat ffout1;
	
	/*--------------------------------------------------------------*/

	private long maxReads=-1;
	private long linesProcessed=0, bytesProcessed=0;
	
	/*--------------------------------------------------------------*/
	
	private java.io.PrintStream outstream=System.err;
	public static boolean verbose=false;
	
}