#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDRightBreast.h"

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

G4MIRDRightBreast::G4MIRDRightBreast() {}

G4MIRDRightBreast::~G4MIRDRightBreast() {}

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

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

    G4double ax = 4.95 * cm;
    G4double by = 4.35 * cm;
    G4double cz = 4.15 * cm;

    G4Ellipsoid* oneRightBreast = new G4Ellipsoid("OneRightBreast", ax, by, cz);

    G4double dx = 20. * cm;
    G4double dy = 10. * cm;
    G4double dz = 35. * cm;

    G4EllipticalTube* Trunk = new G4EllipticalTube("Trunk", dx, dy, dz);

    G4RotationMatrix* rm_relative = new G4RotationMatrix();
    rm_relative->rotateX(90. * degree);

    G4SubtractionSolid* breast = new G4SubtractionSolid("RightBreast", oneRightBreast, Trunk, rm_relative,
                                                        G4ThreeVector(10. * cm, 0.0 * cm, -8.66 * cm));

    G4LogicalVolume* logicRightBreast = new G4LogicalVolume(breast, soft, "logical" + volumeName, 0, 0, 0);

    // Define rotation and position here!
    G4RotationMatrix* rm = new G4RotationMatrix();
    rm->rotateX(90. * degree);
    rm->rotateY(0. * degree);
    rm->rotateZ(-16. * degree);
    G4VPhysicalVolume* physRightBreast =
        new G4PVPlacement(rm, G4ThreeVector(-10. * cm, 9.1 * cm, 52. * cm), "physicalRightBreast", logicRightBreast,
                          mother, false, 0, true);

    // Visualization Attributes
    // G4VisAttributes* RightBreastVisAtt = new G4VisAttributes(G4Colour(1.0,0.41,0.71));
    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);
    G4VisAttributes* RightBreastVisAtt = new G4VisAttributes(colour);
    RightBreastVisAtt->SetForceSolid(wireFrame);
    logicRightBreast->SetVisAttributes(RightBreastVisAtt);

    G4cout << "RightBreast created !!!!!!" << G4endl;

    // Testing RightBreast Volume
    G4double RightBreastVol = logicRightBreast->GetSolid()->GetCubicVolume();
    G4cout << "Volume of RightBreast = " << RightBreastVol / cm3 << " cm^3" << G4endl;

    // Testing RightBreast Material
    G4String RightBreastMat = logicRightBreast->GetMaterial()->GetName();
    G4cout << "Material of RightBreast = " << RightBreastMat << G4endl;

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

    // Testing Mass
    G4double RightBreastMass = (RightBreastVol)*RightBreastDensity;
    G4cout << "Mass of RightBreast = " << RightBreastMass / gram << " g" << G4endl;

    return physRightBreast;
}