#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDHeart.h"

#include "G4Ellipsoid.hh"
#include "G4LogicalVolume.hh"
#include "G4Material.hh"
#include "G4PVPlacement.hh"
#include "G4RotationMatrix.hh"
#include "G4SDManager.hh"
#include "G4Sphere.hh"
#include "G4SystemOfUnits.hh"
#include "G4ThreeVector.hh"
#include "G4UnionSolid.hh"
#include "G4VPhysicalVolume.hh"
#include "G4VisAttributes.hh"
#include "globals.hh"

G4MIRDHeart::G4MIRDHeart() {}

G4MIRDHeart::~G4MIRDHeart() {}

G4VPhysicalVolume* G4MIRDHeart::Construct(const G4String& volumeName, G4VPhysicalVolume* mother,
                                          const G4String& colourName, G4bool wireFrame, G4bool) {
    G4cout << "Construct " << volumeName << " with mother " << mother->GetName() << G4endl;

    G4HumanPhantomMaterial* material = new G4HumanPhantomMaterial();

    G4Material* soft = material->GetMaterial("soft_tissue");

    G4double ax = 4.00 * cm;
    G4double by = 4.00 * cm;
    G4double cz = 7.00 * cm;
    G4double zcut1 = -7.00 * cm;
    G4double zcut2 = 0.0 * cm;

    G4Ellipsoid* heart1 = new G4Ellipsoid("Heart1", ax, by, cz, zcut1, zcut2);

    G4double rmin = 0. * cm;
    G4double rmax = 3.99 * cm;
    G4double startphi = 0. * degree;
    G4double deltaphi = 360. * degree;
    G4double starttheta = 0. * degree;
    G4double deltatheta = 90. * degree;

    G4Sphere* heart2 = new G4Sphere("Heart2", rmin, rmax, startphi, deltaphi, starttheta, deltatheta);

    G4UnionSolid* heart = new G4UnionSolid("Heart", heart1, heart2);

    G4LogicalVolume* logicHeart = new G4LogicalVolume(heart, soft, "logical" + volumeName, 0, 0, 0);

    // Define rotation and position here!
    G4RotationMatrix* rm = new G4RotationMatrix();
    rm->rotateY(25. * degree);
    G4VPhysicalVolume* physHeart = new G4PVPlacement(rm, G4ThreeVector(0.0, -3.0 * cm, 15.32 * cm), "physicalHeart",
                                                     logicHeart, mother, false, 0, true);

    // Visualization Attributes
    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);
    G4VisAttributes* HeartVisAtt = new G4VisAttributes(colour);
    HeartVisAtt->SetForceSolid(wireFrame);
    logicHeart->SetVisAttributes(HeartVisAtt);

    G4cout << "Heart created !!!!!!" << G4endl;

    // Testing Heart Volume
    G4double HeartVol = logicHeart->GetSolid()->GetCubicVolume();
    G4cout << "Volume of Heart = " << HeartVol / cm3 << " cm^3" << G4endl;

    // Testing Heart Material
    G4String HeartMat = logicHeart->GetMaterial()->GetName();
    G4cout << "Material of Heart = " << HeartMat << G4endl;

    // Testing Density
    G4double HeartDensity = logicHeart->GetMaterial()->GetDensity();
    G4cout << "Density of Material = " << HeartDensity * cm3 / g << " g/cm^3" << G4endl;

    // Testing Mass
    G4double HeartMass = (HeartVol)*HeartDensity;
    G4cout << "Mass of Heart = " << HeartMass / gram << " g" << G4endl;

    return physHeart;
}