//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
//
// 
// John Allison  3rd April 2001
// Model which knows how to draw axes.

#include "G4AxesModel.hh"

#include "G4ModelingParameters.hh"
#include "G4VGraphicsScene.hh"
#include "G4VisAttributes.hh"
#include "G4Polyline.hh"
#include "G4Colour.hh"
#include "G4UnitsTable.hh"
#include "G4ArrowModel.hh"
#include "G4TextModel.hh"

G4AxesModel::~G4AxesModel ()
{
  delete fZAnnotationModel;
  delete fZLabelModel;
  delete fZAxisModel;
  delete fYAnnotationModel;
  delete fYLabelModel;
  delete fYAxisModel;
  delete fXAnnotationModel;
  delete fXLabelModel;
  delete fXAxisModel;
}

G4AxesModel::G4AxesModel
(G4double x0, G4double y0, G4double z0, G4double length,
 G4double arrowWidth, const G4String& colourString,
 const G4String& description,
 G4bool withAnnotation,
 G4double textSize,
 const G4Transform3D& transform)
{
  Construct
  (x0,y0,z0,length,arrowWidth,
   colourString,description,withAnnotation,textSize,
   transform);
}

// Alternative short constructor
G4AxesModel::G4AxesModel
(G4double x0, G4double y0, G4double z0, G4double length,
 const G4Transform3D& transform)
{
  G4double arrowWidth = 1.;
  const G4String& colourString = "auto";
  const G4String& description = "";
  G4bool withAnnotation = true;
  G4double textSize = 10.;
  Construct
  (x0,y0,z0,length,arrowWidth,
   colourString,description,withAnnotation,textSize,
   transform);
}

void G4AxesModel::Construct
(G4double x0, G4double y0, G4double z0, G4double length,
 G4double arrowWidth, const G4String& colourString,
 const G4String& description,
 G4bool withAnnotation,
 G4double textSize,
 const G4Transform3D& transform)
{
  fXAxisModel = nullptr;
  fXLabelModel = nullptr;
  fXAnnotationModel = nullptr;
  fYAxisModel = nullptr;
  fYLabelModel = nullptr;
  fYAnnotationModel = nullptr;
  fZAxisModel = nullptr;
  fZLabelModel = nullptr;
  fZAnnotationModel = nullptr;

  fTransform = transform;

  fType = "Axes";
  fGlobalTag = fType;
  fGlobalDescription = fType + ": " + description;
  fExtent = G4VisExtent
    (x0, x0+length, y0, y0+length, z0, z0+length);

  G4Colour colour(1,1,1,1);  // Default white and opaque (unless "auto").
  G4bool autoColour = false;
  if (colourString == "auto") autoColour = true;
  else {
    if (!G4Colour::GetColour(colourString, colour)) {
      G4ExceptionDescription ed;
      ed << "Colour \"" << colourString
	 << "\" not found.  Defaulting to white and opaque.";
      G4Exception
	("G4AxesModel::G4AxesModel",
	 "modeling0011", JustWarning, ed);
    }
  }

  G4String annotation = G4BestUnit(length,"Length");
  G4int lineSegmentsPerCircle = 6;

  G4Colour xColour(colour);
  if (autoColour) xColour = G4Colour::Red();
  fXAxisModel = new G4ArrowModel
    (x0, y0, z0, x0+length, y0, z0, arrowWidth,
     xColour, "x-axis: " + description,
     lineSegmentsPerCircle, fTransform);
  if (withAnnotation) {
    G4VisAttributes va(xColour);
    G4Text lmtext("x",G4Point3D(x0+1.05*length,y0,z0));
    lmtext.SetScreenSize(textSize);
    lmtext.SetOffset(0.5*textSize,0.5*textSize);
    lmtext.SetLayout(G4Text::centre);
    lmtext.SetVisAttributes(va);
    fXLabelModel = new G4TextModel(lmtext,fTransform);
    G4Text amtext(annotation,G4Point3D(x0+0.8*length,y0,z0));
    amtext.SetScreenSize(textSize);
    amtext.SetOffset(-1.5*textSize,-1.5*textSize);
    amtext.SetLayout(G4Text::centre);
    amtext.SetVisAttributes(va);
    fXAnnotationModel = new G4TextModel(amtext,fTransform);
  }

  G4Colour yColour(colour);
  if (autoColour) yColour = G4Colour::Green();
  fYAxisModel = new G4ArrowModel
    (x0, y0, z0, x0, y0+length, z0, arrowWidth,
     yColour, "y-axis: " + description,
     lineSegmentsPerCircle, fTransform);
  if (withAnnotation) {
    G4VisAttributes va(yColour);
    G4Text lmtext("y",G4Point3D(x0,y0+1.05*length,z0));
    lmtext.SetScreenSize(textSize);
    lmtext.SetOffset(0.5*textSize,0.5*textSize);
    lmtext.SetLayout(G4Text::centre);
    lmtext.SetVisAttributes(va);
    fYLabelModel = new G4TextModel(lmtext,fTransform);
    G4Text amtext(annotation,G4Point3D(x0,y0+0.8*length,z0));
    amtext.SetScreenSize(textSize);
    amtext.SetOffset(-1.5*textSize,-1.5*textSize);
    amtext.SetLayout(G4Text::centre);
    amtext.SetVisAttributes(va);
    fYAnnotationModel = new G4TextModel(amtext,fTransform);
  }

  G4Colour zColour(colour);
  if (autoColour) zColour = G4Colour::Blue();
  fZAxisModel = new G4ArrowModel
    (x0, y0, z0, x0, y0, z0+length, arrowWidth,
     zColour, "z-axis: " + description,
     lineSegmentsPerCircle, fTransform);
  if (withAnnotation) {
    G4VisAttributes va(zColour);
    G4Text lmtext("z",G4Point3D(x0,y0,z0+1.05*length));
    lmtext.SetScreenSize(textSize);
    lmtext.SetOffset(0.5*textSize,0.5*textSize);
    lmtext.SetLayout(G4Text::centre);
    lmtext.SetVisAttributes(va);
    fZLabelModel = new G4TextModel(lmtext,fTransform);
    G4Text amtext(annotation,G4Point3D(x0,y0,z0+0.8*length));
    amtext.SetScreenSize(textSize);
    amtext.SetOffset(-1.5*textSize,-1.5*textSize);
    amtext.SetLayout(G4Text::centre);
    amtext.SetVisAttributes(va);
    fZAnnotationModel = new G4TextModel(amtext,fTransform);
  }
}

void G4AxesModel::DescribeYourselfTo (G4VGraphicsScene& sceneHandler)
{
  if (fXAxisModel)       fXAxisModel->      DescribeYourselfTo(sceneHandler);
  if (fXLabelModel)      fXLabelModel->     DescribeYourselfTo(sceneHandler);
  if (fXAnnotationModel) fXAnnotationModel->DescribeYourselfTo(sceneHandler);

  if (fYAxisModel)       fYAxisModel->      DescribeYourselfTo(sceneHandler);
  if (fYLabelModel)      fYLabelModel->     DescribeYourselfTo(sceneHandler);
  if (fYAnnotationModel) fYAnnotationModel->DescribeYourselfTo(sceneHandler);

  if (fZAxisModel)       fZAxisModel->      DescribeYourselfTo(sceneHandler);
  if (fZLabelModel)      fZLabelModel->     DescribeYourselfTo(sceneHandler);
  if (fZAnnotationModel) fZAnnotationModel->DescribeYourselfTo(sceneHandler);
}