#include "cPiston.h"
#include "cRoot.h"
#include "cWorld.h"
#include "BlockID.h"

cPiston::cPiston( cWorld* a_World )
	:m_World ( a_World )
{

}
//Todo: All piston stuff below.
void cPiston::ExtendPiston( int pistx, int pisty, int pistz )
{
	//cWorld* World = cRoot::Get()->GetWorld();
	char metadata = 0;
	char extmetadata = 0;
	int  FirstFluidBlock = 0;
	char piston;

	piston = m_World->GetBlock( pistx, pisty, pistz );

	if ( piston  == E_BLOCK_STICKY_PISTON ) {
		extmetadata = 8;
	}//if sticky piston than top nibble's bit is set to 1;

	if ( ( piston  == E_BLOCK_STICKY_PISTON ) || ( piston == E_BLOCK_PISTON ) ) {
		metadata = m_World->GetBlockMeta( pistx, pisty, pistz);

		if ((int)metadata < 6) { //piston not extended
			switch  ((int)metadata) {
				case 0:
					FirstFluidBlock = FindFluidBlock ( pistx, pisty-1, pistz, pistx, pisty-16, pistz );
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx, pisty-FirstFluidBlock, pistz );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, (char)metadata + 8 );
					m_World->SetBlock( pistx, pisty-1, pistz, E_BLOCK_PISTON_EXTENSION, extmetadata );
					break;
				case 1:
					FirstFluidBlock = FindFluidBlock (pistx, pisty+1, pistz, pistx, pisty+16, pistz);
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx, pisty+FirstFluidBlock, pistz );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, (char)metadata + 8 );
					m_World->SetBlock( pistx, pisty+1, pistz, E_BLOCK_PISTON_EXTENSION, extmetadata+1 );
					break;
				case 2:
					FirstFluidBlock = FindFluidBlock (pistx, pisty, pistz-1, pistx, pisty, pistz-16);
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx, pisty, pistz-FirstFluidBlock );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, (char)metadata + 8 );
					m_World->SetBlock( pistx, pisty, pistz-1, E_BLOCK_PISTON_EXTENSION, extmetadata+2 );
					break;
				case 3:
					FirstFluidBlock = FindFluidBlock (pistx, pisty, pistz+1, pistx, pisty, pistz+16);
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx, pisty, pistz+FirstFluidBlock );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, (char)metadata + 8 );
					m_World->SetBlock( pistx, pisty, pistz+1, E_BLOCK_PISTON_EXTENSION, extmetadata+3 );
					break;
				case 4:
					FirstFluidBlock = FindFluidBlock (pistx-1, pisty, pistz, pistx-16, pisty, pistz);
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx-FirstFluidBlock, pisty, pistz );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, metadata + 8 );
					m_World->SetBlock( pistx-1, pisty, pistz, E_BLOCK_PISTON_EXTENSION, extmetadata+4 );
					break;
				case 5:
					FirstFluidBlock = FindFluidBlock (pistx+1, pisty, pistz, pistx+16, pisty, pistz);
					if (FirstFluidBlock > 0) {
						ChainMove ( pistx, pisty, pistz, pistx+FirstFluidBlock, pisty, pistz );
					}
					m_World->FastSetBlock( pistx, pisty, pistz, piston, metadata + 8 );
					m_World->SetBlock( pistx+1, pisty, pistz, E_BLOCK_PISTON_EXTENSION, extmetadata+5 );
					break;
			}
		}
	}
}

void cPiston::RetractPiston( int pistx, int pisty, int pistz )
{
	char metadata = m_World->GetBlockMeta( pistx, pisty, pistz);
	char tempmeta;
	char tempblock;

	if ( (int)m_World->GetBlock( pistx, pisty, pistz ) == E_BLOCK_PISTON ) {
			if (metadata > 5) { //piston retracted
			metadata -= 8;//set the piston to retracted state.

			m_World->FastSetBlock( pistx, pisty, pistz, m_World->GetBlock( pistx, pisty, pistz ), metadata  );
			switch  (metadata) {
			case 0:
				if ( m_World->GetBlock( pistx, pisty-1, pistz ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx, pisty-1, pistz, 0, 0 );
				}
				break;

			case 1:
				if ( m_World->GetBlock( pistx, pisty+1, pistz ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx, pisty+1, pistz, 0, 0 );
				}
				break;

			case 2:
				if ( m_World->GetBlock( pistx, pisty, pistz-1 ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx, pisty, pistz-1, 0, 0 );
				}
				break;

			case 3:
				if ( m_World->GetBlock( pistx, pisty, pistz+1 ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx, pisty, pistz+1, 0, 0 );
				}
				break;

			case 4:
				if ( m_World->GetBlock( pistx-1, pisty, pistz ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx-1, pisty, pistz, 0, 0 );
				}
				break;

			case 5:
				if ( m_World->GetBlock( pistx+1, pisty, pistz ) ==  E_BLOCK_PISTON_EXTENSION ) {
					m_World->SetBlock( pistx+1, pisty, pistz, 0, 0 );
				}
				break;

			}
		}
	}


	if ( (int)m_World->GetBlock( pistx, pisty, pistz ) == E_BLOCK_STICKY_PISTON ) {
		if (metadata > 5) { //piston retracted
			metadata -= 8;//set the piston to retracted state.
			m_World->FastSetBlock( pistx, pisty, pistz, m_World->GetBlock( pistx, pisty, pistz ), metadata  );
			switch  (metadata) {
				case 0:
					if ( m_World->GetBlock( pistx, pisty-1, pistz ) ==  E_BLOCK_PISTON_EXTENSION ) {
						tempblock =  m_World->GetBlock( pistx, pisty-2, pistz );
						tempmeta = m_World->GetBlockMeta( pistx, pisty-2, pistz );
						m_World->SetBlock( pistx, pisty-1, pistz, 0, 0 );
						m_World->SetBlock( pistx, pisty-1, pistz, tempblock, tempmeta );
						m_World->SetBlock( pistx, pisty-2, pistz, 0, 0 );
					}
					break;

				case 1:
					if ( m_World->GetBlock( pistx, pisty+1, pistz ) == E_BLOCK_PISTON_EXTENSION ) {
						tempblock = m_World->GetBlock( pistx, pisty+2, pistz );
						tempmeta = m_World->GetBlockMeta( pistx, pisty+2, pistz );
						m_World->SetBlock( pistx, pisty+1, pistz, 0, 0 );
						m_World->SetBlock( pistx, pisty+1, pistz, tempblock , tempmeta );
						m_World->SetBlock( pistx, pisty+2, pistz, 0, 0 );
					}
					break;
				case 2:
					if ( m_World->GetBlock( pistx, pisty, pistz-1) ==  E_BLOCK_PISTON_EXTENSION ) {
						tempblock = m_World->GetBlock( pistx, pisty, pistz-2 );
						tempmeta = m_World->GetBlockMeta( pistx, pisty, pistz-2 );
						m_World->SetBlock( pistx, pisty, pistz-1, 0, 0 );
						m_World->SetBlock( pistx, pisty, pistz-1, tempblock, tempmeta );
						m_World->SetBlock( pistx, pisty, pistz-2, 0, 0 );
					}
					break;
				case 3:
					if ( m_World->GetBlock( pistx, pisty, pistz+1) ==  E_BLOCK_PISTON_EXTENSION ) {
						tempblock = m_World->GetBlock( pistx, pisty, pistz+2 );
						tempmeta = m_World->GetBlockMeta( pistx, pisty, pistz+2 );
						m_World->SetBlock( pistx, pisty, pistz+1, 0, 0 );
						m_World->SetBlock( pistx, pisty, pistz+1, tempblock, tempmeta );
						m_World->SetBlock( pistx, pisty, pistz+2, 0, 0 );
					}
					break;
				case 4:
					if ( m_World->GetBlock( pistx-1, pisty, pistz) ==  E_BLOCK_PISTON_EXTENSION ) {
						tempblock = m_World->GetBlock( pistx-2, pisty, pistz );
						tempmeta = m_World->GetBlockMeta( pistx-2, pisty, pistz );
						m_World->SetBlock( pistx-1, pisty, pistz, 0, 0 );
						m_World->SetBlock( pistx-1, pisty, pistz, tempblock, tempmeta );
						m_World->SetBlock( pistx-2, pisty, pistz, 0, 0 );
					}
					break;
				case 5:
					if ( m_World->GetBlock( pistx+1, pisty, pistz) == E_BLOCK_PISTON_EXTENSION ) {
						tempblock = m_World->GetBlock( pistx+2, pisty, pistz );
						tempmeta = m_World->GetBlockMeta( pistx+2, pisty, pistz );
						m_World->SetBlock( pistx+1, pisty, pistz, 0, 0 );
						m_World->SetBlock( pistx+1, pisty, pistz, tempblock, tempmeta );
						m_World->SetBlock( pistx+2, pisty, pistz, 0, 0 );
					}
					break;
			}
		}
	}
}

void cPiston::ChainMove ( int ax, int ay, int az, int bx, int by, int bz )
{

char lastmeta;
char lastblock;

	if ( ( ax != bx ) && ( ay == by ) && ( az == bz  ) ) { //move x

		if ( ax > bx  ) {

			while ( ax > bx ) {
				lastmeta  = m_World->GetBlockMeta( bx, by, bz);
				lastblock = m_World->GetBlock( bx, by, bz);
				m_World->SetBlock( bx+1, by, bz, lastblock, lastmeta );
				bx += 1;
			}

		} else {

			while ( ax < bx ) {
				lastmeta  = m_World->GetBlockMeta( bx, by, bz);
				lastblock = m_World->GetBlock( bx, by, bz);
				m_World->SetBlock( bx+1, by, bz, lastblock, lastmeta );
				bx -= 1;
			}

		}

		m_World->SetBlock( ax, ay, az, 0, 0 );

	} else if ( ( ax == bx ) && ( ay != by ) && ( az == bz  ) ) { //move y

		if ( ay > by  ) {

				while ( ay > by ) {
					lastmeta  = m_World->GetBlockMeta( bx, by, bz);
					lastblock = m_World->GetBlock( bx, by, bz);
					m_World->SetBlock( bx, by-1, bz, lastblock, lastmeta );
					by += 1 ;
				}

			} else {

				while ( ay < by ) {
					lastmeta  = m_World->GetBlockMeta( bx, by, bz);
					lastblock = m_World->GetBlock( bx, by, bz);
					m_World->SetBlock( bx, by+1, bz, lastblock, lastmeta );
					by -= 1;
				}

			}

		m_World->SetBlock( ax, ay, az, 0, 0 );

	} else if ( ( ax == bx ) && ( ay == by ) && ( az != bz  ) ) { //move z

		if ( az > bz  ) {

			while ( az > bz ) {
				lastmeta  = m_World->GetBlockMeta( bx, by, bz );
				lastblock = m_World->GetBlock( bx, by, bz );
				m_World->SetBlock( bx, by, bz-1, lastblock, lastmeta );
				bz += 1;
			}

		} else {

			while ( az < bz ) {
				lastmeta  = m_World->GetBlockMeta( bx, by, bz );
				lastblock = m_World->GetBlock( bx, by, bz );
				m_World->SetBlock( bx, by, bz+1, lastblock, lastmeta );
				bz -= 1;
			}

		}

		m_World->SetBlock( ax, ay, az, 0, 0 );

	} else {

		//LOG("Bad coords fed into cPiston::ChainMove");

	}


}


int cPiston::FindFluidBlock ( int ax, int ay, int az, int bx, int by, int bz )
{

char thisblock;
int cnt = 0;
	if ( ( ax != bx ) && ( ay == by ) && ( az == bz  ) ) { //check x

		if ( ax < bx  ) {
			while ( ax < bx ) {
				thisblock = m_World->GetBlock( ax, ay, az);
				if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
					return cnt;
				}
				cnt++;
				ax += 1;
			}

		} else {

			while ( ax > bx ) {
				thisblock = m_World->GetBlock( ax, ay, az);
				if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
					return cnt;
				}
				cnt++;
				ax -= 1;
			}

		}

		return cnt;

	} else if ( ( ax == bx ) && ( ay != by ) && ( az == bz  ) ) { //check y

		if ( ay < by  ) {

			while ( ay < by ) {
				thisblock = m_World->GetBlock( ax, ay, az);
				if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
					return cnt;
				}
				cnt++;
				ay += 1;
			}

		} else {

			while ( ay > by ) {
				thisblock = m_World->GetBlock( ax, ay, az);
				if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
					return cnt;
				}
				cnt++;
				ay -= 1;
			}

		}

		return cnt;

	} else if ( ( ax == bx ) && ( ay == by ) && ( az != bz  ) ) { //check z

		if ( az < bz  ) {

			while ( az < bz ) {
				thisblock = m_World->GetBlock( ax, ay, az);
 				if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
					return cnt;
				}
				cnt++;
				az += 1;
			}

		} else {

			while ( az > bz ) {
        thisblock = m_World->GetBlock( ax, ay, az);
        if ( ((int)thisblock == 0) || ((int)thisblock == E_BLOCK_STATIONARY_WATER) || ((int)thisblock == E_BLOCK_WATER) || ((int)thisblock == E_BLOCK_STATIONARY_LAVA) || ((int)thisblock == E_BLOCK_LAVA) ) {
        	return cnt;
				}
				cnt++;
  			az -= 1;
  		}

  	}

		return cnt;

	} else {

	return cnt;

	}
}