#===============================================
# TkVNL
#===============================================
package TkCASTEP;
use Crystal::CASTEP;
use TkPlotModule;
@ISA = qw(CASTEP TkPlotModule);

#公開したいサブルーチン
#@EXPORT = qw(DelSpace Reduce01 MakePath RegExpQuote);

use strict;

use Utils;
use MyTk::GraphFrameArray;
use GraphData;

#============================================================
#　変数等取得関数
#============================================================

#============================================================
#　コンストラクタ、デストラクタ
#============================================================
sub new
{
	my ($module, $app) = @_;
#	my ($module, $app, $canvas) = @_;
	my $this = {};
	bless $this;
	$this->SetApplication($app);
#	$this->SetCanvas($canvas);
	return $this;
}

sub DESTROY
{
	my $this = shift;
	$this->SUPER::DESTROY(@_);
}

#============================================================
#　継承クラスで定義しなおす関数
#============================================================
sub CheckFileType
{
	my ($path) = @_;
	return CASTEP::CheckFileType($path);
}

sub CreateGraphFrame
{
	my ($this, $canvas) = @_;
	$canvas = $this->Canvas();
#	$this->SetCanvas($canvas) if($canvas);
#	$canvas = $this->Canvas() unless($canvas);

	my $App = $this->App();
	my $font = $App->{'GraphFrameFont'};
	my @font = split(/,/, $font) if($font);
	my $w = $canvas->width();
	my $h = $canvas->height();

	my $FileType = $this->FileType();
	return undef unless($FileType);
	my $pDataArray = $this->DataArray();
	return unless($pDataArray);

# Canvas, GraphFrameの初期化
	my $GraphFrameArray = $this->{'GraphFrameArray'}
			= new GraphFrameArray($this->mw());
#			= new GraphFrameArray($this->mw(), $canvas);
	$GraphFrameArray->SetCanvasSize($w, $h);

	if($FileType eq "CASTEP Band Output file") {
		for(my $i = 0 ; ; $i++) {
			my $Data  = $pDataArray->GetGraphData($i);
			last unless($Data);
			$GraphFrameArray->AddGraphFrame();
			my $GraphFrame0  = $GraphFrameArray->GetGraphFrame($i);
			my $FramePosStr0 = $App->{"GraphFramePosition"};
			my $XScale0 = $GraphFrame0->GetXScale(0);
			my $YScale0 = $GraphFrame0->GetYScale(0);
			$GraphFrame0->SetPositionByStr($FramePosStr0);
			$GraphFrame0->SetViewRange(0, 0, 1, 1);
			$GraphFrame0->SetSmoothLine(1);
			if($i > 0) {
				$XScale0->SetScaleStringVisible(0);
				$XScale0->SetScaleVisible(0);
				$XScale0->SetCaptionVisible(0);
				$YScale0->SetScaleStringVisible(0);
				$YScale0->SetScaleVisible(0);
				$YScale0->SetCaptionVisible(0);
			}
		}
	}
	else {
		$GraphFrameArray->AddGraphFrame();
		my $GraphFrame0  = $GraphFrameArray->GetGraphFrame(0);
		my $FramePosStr0 = $App->{"GraphFramePosition"};
		my $XScale0 = $GraphFrame0->GetXScale(0);
		my $YScale0 = $GraphFrame0->GetYScale(0);
		$GraphFrame0->SetPositionByStr($FramePosStr0);
		$GraphFrame0->SetXCaption("Energy");
		$GraphFrame0->SetYCaption('PDOS');
		$GraphFrame0->SetViewRange(0, 0, 1, 1);
	}
}

sub AssignGraphData
{
	my ($this) = @_;

	my $GraphFrameArray = $this->GetGraphFrameArray();
	return undef unless($GraphFrameArray);
	my $GraphFrame0  = $GraphFrameArray->GetGraphFrame(0);
	return undef unless($GraphFrame0);
	my $FileType   = $this->FileType();
	my $pDataArray = $this->DataArray();
	return unless($pDataArray);
	my $Data  = $pDataArray->GetGraphData(0);
	return unless($Data);
	my $nData = $Data->nData();
	my $title = $Data->Title();

	my $linecolor = "auto";
	if($FileType eq "CASTEP Band Output file") {
		my @GraphFrame;
		for(my $id = 0 ; ; $id++) {
			$GraphFrame[$id] = $GraphFrameArray->GetGraphFrame($id);
			last unless($GraphFrame[$id]);
			my $Data  = $pDataArray->GetGraphData($id);

			for(my $i = 0 ; ; $i++) {
				my $pX = $Data->GetDataArray("x$i");
				my $pY = $Data->GetDataArray("y$i");
				last unless($pY);
				my $pNe = $Data->{"Ne$i"};
				if($pNe) {
					if($pNe->[0] < 0.5) {
						$linecolor = 'grey'  if($id == 0);
						$linecolor = 'green' if($id == 1);
					}
					elsif($pNe->[0] < 1.5) {
						$linecolor = 'blue'   if($id == 0);
						$linecolor = 'orange' if($id == 1);
					}
					else {
						$linecolor = 'black' if($id == 0);
						$linecolor = 'red'   if($id == 1);
					}
				}
				else {
					$linecolor = 'black' if($id == 0);
					$linecolor = 'red'   if($id == 1);
				}
				my $nData = $GraphFrame[$id]->AddGraphData($pX, $pY,
					1, $linecolor, "", 6, "red", 0, "red", "XAutoSkip",
					$Data->{"x0_Name"}, $Data->{"y${i}_Name"});
				my $data = $GraphFrame[$id]->GetPlotData($nData-1);
			}
			$GraphFrame[$id]->CalMinMax();
			$GraphFrame[$id]->AdjustViewRange(0.05, 0.05, 0.05, 0.05);
		}
		if($GraphFrame[1]) {
			$GraphFrame[0]->AddSynchronousFrame('x', $GraphFrame[1]);
			$GraphFrame[0]->AddSynchronousFrame('y', $GraphFrame[1]);
			$GraphFrame[1]->AddSynchronousFrame('x', $GraphFrame[0]);
			$GraphFrame[1]->AddSynchronousFrame('y', $GraphFrame[0]);
		}
	}
	else {
		for(my $i = 0 ; ; $i++) {
#print "Assign $i\n";
			my $pX = $Data->GetDataArray("x$i");
			my $pY = $Data->GetDataArray("y$i");
			last unless($pY);
			my $nData = $GraphFrame0->AddGraphData($pX, $pY,
				1, $linecolor, "", 6, "red", 0, "red", "XAutoSkip",
				$Data->{"x0_Name"}, $Data->{"y${i}_Name"});
			my $data = $GraphFrame0->GetPlotData($nData-1);
		}
		$GraphFrame0->CalMinMax();
		$GraphFrame0->AdjustViewRange(0.05, 0.05, 0.05, 0.05);
	}

	if($FileType eq "CASTEP Band Output file") {
		$GraphFrame0->SetXCaption($Data->{"x0_Name"});
		$GraphFrame0->SetYCaption('Energy');
	}
	else {
		$GraphFrame0->SetXCaption($Data->{"x0_Name"});
		$GraphFrame0->SetYCaption($Data->{"y0_Name"});
	}

	$this->AdjustViewRange();
}

sub AdjustViewRange
{
	my ($this) = @_;
	my $GraphFrameArray = $this->GetGraphFrameArray();
	my $pGraphFrame = $GraphFrameArray->GetpGraphFrameArray();
	my $FileType   = $this->FileType();

	if($FileType eq "CASTEP Band Output file") {
		my ($xmin, $xmax) = $pGraphFrame->[0]->GetViewXRange();
		my ($ymin, $ymax) = $pGraphFrame->[0]->GetViewYRange();
		for(my $id = 1 ; ; $id++) {
			my $GraphFrame = $GraphFrameArray->GetGraphFrame($id);
			last unless($GraphFrame);
			$GraphFrame->SetViewRange($xmin, $ymin, $xmax, $ymax);
		}
	}
}

sub Draw
{
	my ($this, $canvas) = @_;
	$canvas = $this->Canvas();
#	$this->SetCanvas($canvas) if($canvas);
#	$canvas = $this->Canvas() unless($canvas);
	$this->mw()->RefleshCanvas();
#	$canvas->ClearAll();
	my $mw = $this->mw();
	my $FileType   = $this->FileType();
	my $GraphFrameArray = $this->GetGraphFrameArray();
	my $pGraphFrame = $GraphFrameArray->GetpGraphFrameArray();

	my @GraphFrame = @$pGraphFrame;
	my $App = $this->App();
	my $font = $App->{'GraphFrameFont'};
	my @font = split(/,/, $font) if($font);
	my $w = $canvas->width();
	my $h = $canvas->height();

	if($font) {
		$canvas->SetFont(\@font);
		$GraphFrameArray->SetFont(\@font);
	}
	$GraphFrameArray->SetCanvasSize($w, $h);

	$this->mw()->Balloon()->detach($canvas);

	$this->mw()->WriteStatusBar("Drawing...");
	$GraphFrameArray->Draw($canvas);

#バンド構造図の縦線を描画
	if($FileType eq "CASTEP Band Output file") {
		my $pDataArray = $this->DataArray();
		my $pBandBoundaryDistances = $pDataArray->{'BandBoundaryDistances'};
		my $pBoundaryPositions = $this->{'DataArray'}->{'BandBoundaryPositions'};
		my $GraphFrame0  = $GraphFrameArray->GetGraphFrame(0);
		my ($x0, $y0, $x1, $y1) = $GraphFrame0->GetPosition();
		my ($vx0, $vx1)         = $GraphFrame0->GetViewXRange();
		my %TagHash;
		for(my $i = 0 ; $i < @$pBandBoundaryDistances ; $i++) {
			my $b = $pBandBoundaryDistances->[$i];
			next if($b < $vx0 or $vx1 < $b);
			my ($x, $y) = $GraphFrame0->ValueToPosition($b, 0);
			my $str = $pBoundaryPositions->[$i];
			my $tag = "Boundary$i";
			$b = Utils::Round($b, 4);
			$TagHash{$tag} = "($str) x=$b";
			$canvas->DrawLine($x, $y0, $x, $y1, 1, "black", $tag);
		}
		$this->mw()->Balloon()->attach($canvas, 
			-balloonposition => 'mouse', -msg => \%TagHash);
	}
	$this->mw()->WriteStatusBar("Finish Drawing");
}

sub ConfigureIniFileVariables
{
	my ($this) = @_;
	my $App = $this->App();
	$App->ConfigureIniFileVariables();

	my $Style = $App->Args()->GetGetArg("style");
	$Style = "General" unless($Style);

	$App->AddIniFileVariable("\\$Style\\GraphFrame_Position",        "GraphFramePosition",     "r0.20,0.80,0.80,0.10");
#	$App->AddIniFileVariable("\\$Style\\GraphFrameDOS0_Position",    "GraphFrameDOS0Position", "r0.20,0.80,0.80,0.45");
#	$App->AddIniFileVariable("\\$Style\\GraphFrameDOS1_Position",    "GraphFrameDOS1Position", "r0.20,0.45,0.80,0.10");

	$App->ReadSetting();
}

1;