~ubuntu-branches/ubuntu/lucid/tuxguitar/lucid

/*
 * Created on 04-ene-2006
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package org.herac.tuxguitar.gui.editors.tab.layout;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Font;
import org.eclipse.swt.graphics.FontData;
import org.eclipse.swt.graphics.Rectangle;
import org.herac.tuxguitar.gui.TuxGuitar;
import org.herac.tuxguitar.gui.editors.TGPainter;
import org.herac.tuxguitar.gui.editors.tab.TGChordImpl;
import org.herac.tuxguitar.gui.editors.tab.TGLyricImpl;
import org.herac.tuxguitar.gui.editors.tab.TGMeasureImpl;
import org.herac.tuxguitar.gui.editors.tab.TGTrackImpl;
import org.herac.tuxguitar.gui.editors.tab.Tablature;
import org.herac.tuxguitar.gui.printer.PrintDocument;
import org.herac.tuxguitar.gui.printer.PrintStyles;
import org.herac.tuxguitar.song.models.TGMeasure;
import org.herac.tuxguitar.song.models.TGTrack;

/**
 * @author julian
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
public class PrinterViewLayout extends ViewLayout{
	
	private static final int DEFAULT_SCORE_LINE_SPACING = 7;
	private static final int DEFAULT_STRING_SPACING = 8;
	private static final int DEFAULT_HORIZONTAL_SPACING = 15;
	private static final int MIN_SCORE_TAB_SPACING =  15;
	private static final int DEFAULT_TRACK_SPACING = 5;
	private static final int DEFAULT_FIRST_TRACK_SPACING = DEFAULT_TRACK_SPACING;
	private static final int DEFAULT_MIN_BUFFER_SEPARATOR = 15;
	private static final int DEFAULT_MIN_TOP_SPACING = 20;
	private static final int CHORD_FRET_INDEX_SPACING = 8;
	private static final int CHORD_STRING_SPACING = 4;
	private static final int CHORD_FRET_SPACING = 5;
	
	private PrintStyles styles;
	private PrintDocument document;
	private int page;
	
	private Font songNameFont;
	private Font trackNameFont;
	private Font songAuthorFont;
	
	public PrinterViewLayout(Tablature tablature,PrintStyles styles, float scale){
		super(tablature,( styles.getStyle() | DISPLAY_COMPACT ), scale );
		this.styles = styles;
	}
	
	protected void init( float scale ){
		this.setBufferEnabled(false);
		this.setFirstMeasureSpacing( getScaledValue(scale , DEFAULT_HORIZONTAL_SPACING ) );
		this.setMinBufferSeparator( getScaledValue(scale ,  DEFAULT_MIN_BUFFER_SEPARATOR ) );
		this.setMinTopSpacing( getScaledValue(scale ,  DEFAULT_MIN_TOP_SPACING ) );
		this.setMinScoreTabSpacing( getScaledValue(scale ,  MIN_SCORE_TAB_SPACING ) );
		this.setScoreLineSpacing( getScaledValue(scale ,  DEFAULT_SCORE_LINE_SPACING , 4) );
		this.setScoreSpacing( (( getScoreLineSpacing() * 4) + getMinScoreTabSpacing() ) );
		this.setFirstTrackSpacing( getScaledValue(scale ,  DEFAULT_FIRST_TRACK_SPACING ) );
		this.setTrackSpacing( getScaledValue(scale ,  DEFAULT_TRACK_SPACING ) );
		this.setStringSpacing( getScaledValue(scale ,  DEFAULT_STRING_SPACING  , 4) );
		this.setChordFretIndexSpacing( getScaledValue(scale ,  CHORD_FRET_INDEX_SPACING, 2 ) );
		this.setChordStringSpacing( getScaledValue(scale ,  CHORD_STRING_SPACING, 2 ) );
		this.setChordFretSpacing( getScaledValue(scale ,  CHORD_FRET_SPACING , 2 ) );
		this.setChordNoteSize(getScaledValue(scale ,  3 , 2 ));
		this.setRepeatEndingSpacing( getScaledValue(scale ,  20 ) );
		this.setTextSpacing( getScaledValue( scale, 15 ) );
		this.setMarkerSpacing( getScaledValue( scale, 15 ) );
		this.setTupletoSpacing( getScaledValue( scale,  10 ) );
		this.setEffectSpacing( getScaledValue(scale ,  8 ) );
		this.setScale( checkScale() );
		this.setFontScale( scale );
		
		// Initialize Tablature Defaults
		this.getTablature().setViewLayout( this ) ;
		this.getTablature().initDefaults();
		
		FontData[] fd = this.getResources().getPrinterDefaultFont().getFontData();
		String fontName = (fd != null && fd.length > 0 ? fd[0].getName() : new String());
		this.songNameFont = new Font(getTablature().getDisplay(),fontName,Math.round(16.0f * getFontScale()), SWT.BOLD | SWT.CENTER);
		this.trackNameFont = new Font(getTablature().getDisplay(),fontName,Math.round(8.0f * getFontScale()), SWT.BOLD | SWT.CENTER);
		this.songAuthorFont = new Font(getTablature().getDisplay(),fontName,Math.round(8.0f * getFontScale()), SWT.BOLD | SWT.CENTER);
	}
	
	public int getMode(){
		return 0;
	}
	
	public void makeDocument(PrintDocument document){
		this.page = 0;
		this.document = document;
		this.makeDocument();
	}
	
	private void makeDocument(){
		this.document.start();
		
		this.openPage();
		paintHeader(this.document.getPainter());
		paintSong(this.document.getPainter(),null,this.document.getBounds().x, ( this.document.getBounds().y + Math.round(80.0f * getFontScale() ) ) );
		paintFooter(this.document.getPainter());
		this.closePage();
		
		this.document.finish();
	}
	
	public void paintSong(TGPainter painter,Rectangle clientArea,int fromX,int fromY) {
		this.setWidth(0);
		this.setHeight(0);
		
		int style = getStyle();
		int posY = fromY + getFirstTrackSpacing();
		int height = getFirstTrackSpacing();
		int lineHeight = 0;
		
		TGTrackImpl track = (TGTrackImpl)getSongManager().getTrack(this.styles.getTrackNumber());
		((TGLyricImpl)track.getLyrics()).start(getSkippedBeats(track));
		
		TrackSpacing ts = new TrackSpacing(this) ;
		TempLine line = getTempLines(track,( this.styles.getFromMeasure() - 1 ),ts);
		while(!line.measures.isEmpty()){
			
			ts.setSize(TrackSpacing.POSITION_SCORE_MIDDLE_LINES, ((style & DISPLAY_SCORE) != 0?( (getScoreLineSpacing() * 5) ):0));
			if((style & DISPLAY_SCORE) != 0){
				ts.setSize(TrackSpacing.POSITION_SCORE_UP_LINES, Math.abs(line.minY));
				if(line.maxY + getMinScoreTabSpacing() > getScoreSpacing()){
					ts.setSize(TrackSpacing.POSITION_SCORE_DOWN_LINES, (line.maxY - (getScoreLineSpacing() * 4)) );
				}
			}
			if((style & DISPLAY_TABLATURE) != 0){
				ts.setSize(TrackSpacing.POSITION_TABLATURE_TOP_SEPARATOR, ((style & DISPLAY_SCORE) != 0 ? getMinScoreTabSpacing() : Math.max(Math.abs(line.minY), getStringSpacing()) ));
				ts.setSize(TrackSpacing.POSITION_TABLATURE, ((style & DISPLAY_SCORE) != 0 ?  track.getTabHeight() + getStringSpacing() + 1 : Math.max( line.maxY, track.getTabHeight() + getStringSpacing() + 1) ));
				//ts.setSize(TrackSpacing.POSITION_TABLATURE_TOP_SEPARATOR,((style & DISPLAY_SCORE) != 0 ?getMinScoreTabSpacing(): ((getStringSpacing() / 2) * 5) +1 ));
				//ts.setSize(TrackSpacing.POSITION_TABLATURE, ((style & DISPLAY_SCORE) != 0 ?track.getTabHeight() + getStringSpacing() +1:track.getTabHeight() + ((getStringSpacing() / 2) * 5) +1 ) );
			}
			ts.setSize(TrackSpacing.POSITION_LYRIC,10);
			checkDefaultSpacing(ts);
			
			lineHeight = ts.getSize();
			//Verifico si entra en la pagina actual
			if((posY + lineHeight + getTrackSpacing()) > (this.document.getBounds().y + getMaxHeight())){
				this.paintFooter(painter);
				this.closePage();
				this.openPage();
				posY = this.document.getBounds().y + getFirstTrackSpacing();
			}
			
			//pinto la linea
			paintLine(track,line,painter,fromX,posY,ts);
			
			posY += lineHeight + getTrackSpacing();
			height += lineHeight + getTrackSpacing();
			
			ts = new TrackSpacing(this) ;
			line = getTempLines(track,( line.lastIndex + 1 ),ts);
		}
		this.setHeight(height);
	}
	
	public void paintHeader(TGPainter painter){
		if(this.document.isPaintable(this.page) ){
			int x = this.document.getBounds().x;
			int y = this.document.getBounds().y;
			String songName = getSongManager().getSong().getName();
			String songAuthor = getSongManager().getSong().getAuthor();
			String trackName = "(" + getSongManager().getTrack(this.styles.getTrackNumber()).getName() + ")";
			
			if(songName == null || songName.length() == 0){
				songName = TuxGuitar.getProperty("print-header.default-song-name");
			}
			if(songAuthor == null || songAuthor.length() == 0){
				songAuthor = TuxGuitar.getProperty("print-header.default-song-author");
			}
			painter.setFont(this.songNameFont);
			painter.drawString(songName,(x + getCenter(painter,songName)),y);
			painter.setFont(this.trackNameFont);
			painter.drawString(trackName,(x + getCenter(painter,trackName)),(y + Math.round(30.0f * getFontScale())));
			painter.setFont(this.songAuthorFont);
			painter.drawString(songAuthor,(x + getRight(painter,songAuthor)),(y + Math.round(50.0f * getFontScale())));
		}
	}
	
	private void paintFooter(TGPainter painter){
		if(this.document.isPaintable(this.page) ){
			int x = this.document.getBounds().x;
			int y = this.document.getBounds().y;
			String pageNumber = Integer.toString(this.page);
			
			painter.setBackground(getResources().getColorWhite());
			painter.setForeground(getResources().getColorBlack());
			painter.drawString(pageNumber,(x + getRight(painter,pageNumber)),(y + getBottom(painter,pageNumber)));
		}
	}
	
	public void paintLine(TGTrackImpl track,TempLine line,TGPainter painter,int fromX, int fromY,TrackSpacing ts) {
		if(this.document.isPaintable(this.page) ){
			int posX = fromX;
			int posY = fromY;
			int width = 0;
			
			int measureSpacing = 0;
			if(line.fullLine){
				int diff = ( getMaxWidth() - line.tempWith);
				if(diff != 0 && line.measures.size() > 0){
					measureSpacing = diff / line.measures.size();
				}
			}
			
			for(int i = 0;i < line.measures.size();i ++){
				int index = ((Integer)line.measures.get(i)).intValue();
				TGMeasureImpl currMeasure = (TGMeasureImpl)track.getMeasure(index);
				
				//asigno la posicion dentro del compas
				currMeasure.setPosX(posX);
				currMeasure.setPosY(posY);
				currMeasure.setTs(ts);
				
				((TGLyricImpl)track.getLyrics()).setCurrentMeasure(currMeasure);
				
				currMeasure.setFirstOfLine(i == 0);
				
				paintMeasure(currMeasure,painter,measureSpacing);
				((TGLyricImpl)track.getLyrics()).paintCurrentNoteBeats(painter,this,currMeasure,posX, posY);
				
				int measureWidth = ( currMeasure.getWidth(this) + currMeasure.getSpacing() );
				posX += measureWidth;
				width += measureWidth;
			}
			this.setWidth(Math.max(getWidth(),width));
		}
	}
	
	private void openPage(){
		this.page ++;
		if(this.document.isPaintable(this.page)){
			this.document.pageStart();
		}
	}
	
	private void closePage(){
		if(this.document.isPaintable(this.page)){
			this.document.pageFinish();
		}
	}
	
	private int getCenter(TGPainter painter,String text){
		int textWidth = painter.getStringExtent(text).x;
		return ((getMaxWidth() - textWidth) / 2);
	}
	
	private int getRight(TGPainter painter,String text){
		int textWidth = painter.getStringExtent(text).x;
		return ((getMaxWidth() - textWidth));
	}
	
	private int getBottom(TGPainter painter,String text){
		int textHeight = painter.getStringExtent(text).y;
		return ((getMaxHeight() - textHeight));
	}
	
	private TempLine getTempLines(TGTrack track,int fromIndex,TrackSpacing ts) {
		TempLine line = new TempLine();
		int measureCount = track.countMeasures();
		for (int measureIdx = fromIndex; measureIdx < measureCount; measureIdx++) {
			TGMeasureImpl measure= (TGMeasureImpl) track.getMeasure(measureIdx);
			if( measure.getNumber() >= this.styles.getFromMeasure() && measure.getNumber() <= this.styles.getToMeasure()){
				
				//verifico si tengo que bajar de linea
				if((line.tempWith + measure.getWidth(this)) >=  getMaxWidth() && !line.measures.isEmpty()){
					line.fullLine = true;
					return line;
				}
				line.tempWith +=  measure.getWidth(this);
				line.maxY = (measure.getMaxY() > line.maxY)?measure.getMaxY():line.maxY;
				line.minY = (measure.getMinY() < line.minY)?measure.getMinY():line.minY;
				
				line.addMeasure(measureIdx);
				measure.registerSpacing(this,ts);
			}
		}
		
		return line;
	}
	
	private int getSkippedBeats(TGTrack track) {
		int beats = 0;
		
		for (int i = 0; i < (this.styles.getFromMeasure() - 1); i++) {
			TGMeasureImpl measure = (TGMeasureImpl) track.getMeasure(i);
			beats += measure.getNotEmptyBeats();
		}
		return beats;
	}
	
	public boolean isCaretVisible(){
		return false;
	}
	
	public boolean isPlayModeEnabled(){
		return false;
	}
	
	public void setLineStyle(TGPainter painter){
		painter.setLineWidth(1);
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setMeasureNumberStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground(getResources().getColorWhite());
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setDivisionsStyle(TGPainter painter, boolean fill){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground( (fill ? getResources().getColorBlack() :getResources().getColorWhite() ));
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setTempoStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground(getResources().getColorWhite());
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setTripletFeelStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground(getResources().getColorWhite());
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setTabNoteStyle(TGPainter painter,boolean playMode){
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
		painter.setFont(getResources().getPrinterNoteFont());
	}
	
	public void setTabNoteFooterStyle(TGPainter painter){
		painter.setForeground( getResources().getColorBlack());
		painter.setBackground( getResources().getColorBlack());
	}
	
	public void setTabEffectStyle(TGPainter painter){
		painter.setForeground( getResources().getColorBlack());
		painter.setBackground( getResources().getColorBlack());
	}
	
	public void setScoreNoteStyle(TGPainter painter,boolean playing){
		painter.setBackground( getResources().getColorBlack() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setScoreNoteFooterStyle(TGPainter painter){
		painter.setForeground( getResources().getColorBlack());
		painter.setBackground( getResources().getColorBlack());
	}
	
	public void setScoreEffectStyle(TGPainter painter){
		painter.setForeground( getResources().getColorBlack());
		painter.setBackground( getResources().getColorBlack());
	}
	
	public void setTimeSignatureStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterTimeSignatureFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setGraceStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterGraceFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setLyricStyle(TGPainter painter,boolean playMode){
		painter.setFont(getResources().getPrinterLyricFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setTextStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterTextFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setOfflineEffectStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setTupletoStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setRepeatEndingStyle(TGPainter painter){
		painter.setFont(getResources().getPrinterDefaultFont());
		painter.setBackground( getResources().getColorWhite() );
		painter.setForeground(getResources().getColorBlack());
	}
	
	public void setChordStyle(TGChordImpl chord){
		chord.setStyle(getStyle());
		chord.setFont(getResources().getPrinterChordFont());
		chord.setForegroundColor(getResources().getColorBlack());
		chord.setBackgroundColor(getResources().getColorWhite());
		chord.setColor(getResources().getColorBlack());
		chord.setNoteColor(getResources().getColorBlack());
		chord.setTonicColor(getResources().getColorBlack());
		chord.setFretSpacing(getChordFretSpacing());
		chord.setStringSpacing(getChordStringSpacing());
		chord.setNoteSize(getChordNoteSize());
		chord.setFirstFretSpacing(getChordFretIndexSpacing());
		chord.setFirstFretFont(getResources().getChordFretFont());
	}
	
	public int getMaxWidth(){
		return (this.document.getBounds().width - this.document.getBounds().x - 10);
	}
	
	public int getMaxHeight(){
		return (this.document.getBounds().height - this.document.getBounds().y - 10);
	}
	
	public boolean isFirstMeasure(TGMeasure measure){
		return (measure.getNumber() == this.styles.getFromMeasure());
	}
	
	public boolean isLastMeasure(TGMeasure measure){
		return (measure.getNumber() == this.styles.getToMeasure());
	}
	
	private int getScaledValue(float scale, int value){
		return getScaledValue(scale, value,1);
	}
	
	private int getScaledValue(float scale, int value, int minimum){
		float scaledValue = ( value * scale );
		return Math.max( Math.round( scaledValue ) , minimum );
	}
	
	public void disposeLayout(){
		super.disposeLayout();
		this.songNameFont.dispose();
		this.trackNameFont.dispose();
		this.songAuthorFont.dispose();
	}
	
	private class TempLine{
		protected int tempWith;
		protected int lastIndex;
		protected boolean fullLine;
		protected int maxY = 0;
		protected int minY = 50;
		protected List measures;
		
		public TempLine(){
			this.measures = new ArrayList();
		}
		
		protected void addMeasure(int index){
			this.measures.add(new Integer(index));
			this.lastIndex = index;
		}
	}
}