jpayne@68: package kmer;
jpayne@68: 
jpayne@68: import java.util.ArrayList;
jpayne@68: import java.util.concurrent.atomic.AtomicInteger;
jpayne@68: 
jpayne@68: import shared.KillSwitch;
jpayne@68: import shared.Shared;
jpayne@68: import shared.Tools;
jpayne@68: 
jpayne@68: public class OwnershipThread extends Thread {
jpayne@68: 	
jpayne@68: 	public static void clear(AbstractKmerTable[] tables){
jpayne@68: 		process(tables, CLEAR);
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public static void initialize(AbstractKmerTable[] tables){
jpayne@68: 		process(tables, INITIALIZE);
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	private static void process(AbstractKmerTable[] tables, int mode){
jpayne@68: 		if(tables.length<2){
jpayne@68: 			if(mode==INITIALIZE){
jpayne@68: 				for(AbstractKmerTable akt : tables){akt.initializeOwnership();}
jpayne@68: 			}else if(mode==CLEAR){
jpayne@68: 				for(AbstractKmerTable akt : tables){akt.clearOwnership();}
jpayne@68: 			}else{
jpayne@68: 				KillSwitch.kill("Bad mode: "+mode);
jpayne@68: 			}
jpayne@68: 			return;
jpayne@68: 		}
jpayne@68: 		final int threads=Tools.min(Shared.threads(), tables.length);
jpayne@68: 		final AtomicInteger next=new AtomicInteger(0);
jpayne@68: 		ArrayList<OwnershipThread> alpt=new ArrayList<OwnershipThread>(threads);
jpayne@68: 		for(int i=0; i<threads; i++){alpt.add(new OwnershipThread(tables, mode, next));}
jpayne@68: 		for(OwnershipThread pt : alpt){pt.start();}
jpayne@68: 		
jpayne@68: 		for(OwnershipThread pt : alpt){
jpayne@68: 			while(pt.getState()!=Thread.State.TERMINATED){
jpayne@68: 				try {
jpayne@68: 					pt.join();
jpayne@68: 				} catch (InterruptedException e) {
jpayne@68: 					// TODO Auto-generated catch block
jpayne@68: 					e.printStackTrace();
jpayne@68: 				}
jpayne@68: 			}
jpayne@68: 		}
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	public OwnershipThread(AbstractKmerTable[] tables_, int mode_, AtomicInteger next_){
jpayne@68: 		tables=tables_;
jpayne@68: 		mode=mode_;
jpayne@68: 		next=next_;
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	@Override
jpayne@68: 	public void run(){
jpayne@68: 		for(int i=next.getAndIncrement(); i<tables.length; i=next.getAndIncrement()){
jpayne@68: 			if(mode==INITIALIZE){
jpayne@68: 				tables[i].initializeOwnership();
jpayne@68: 			}else if(mode==CLEAR){
jpayne@68: 				tables[i].clearOwnership();
jpayne@68: 			}else{
jpayne@68: 				KillSwitch.kill("Bad mode: "+mode);
jpayne@68: 			}
jpayne@68: 		}
jpayne@68: 	}
jpayne@68: 	
jpayne@68: 	private final AbstractKmerTable[] tables;
jpayne@68: 	private final AtomicInteger next;
jpayne@68: 	private final int mode;
jpayne@68: 
jpayne@68: 	public static final int INITIALIZE=0;
jpayne@68: 	public static final int CLEAR=1;
jpayne@68: 	
jpayne@68: }