#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDUrinaryBladder.h"

#include "G4Ellipsoid.hh"
#include "G4LogicalVolume.hh"
#include "G4Material.hh"
#include "G4PVPlacement.hh"
#include "G4SDManager.hh"
#include "G4SubtractionSolid.hh"
#include "G4SystemOfUnits.hh"
#include "G4VPhysicalVolume.hh"
#include "G4VisAttributes.hh"
#include "globals.hh"

G4MIRDUrinaryBladder::G4MIRDUrinaryBladder() {}

G4MIRDUrinaryBladder::~G4MIRDUrinaryBladder() {}

G4VPhysicalVolume* G4MIRDUrinaryBladder::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.958 * cm;
    G4double by = 3.458 * cm;
    G4double cz = 3.458 * cm;

    G4Ellipsoid* bladder = new G4Ellipsoid("bladder_out", ax, by, cz);

    ax = 4.706 * cm;
    by = 3.206 * cm;
    cz = 3.206 * cm;
    G4Ellipsoid* inner = new G4Ellipsoid("innerBladder", ax, by, cz);

    G4SubtractionSolid* totalBladder = new G4SubtractionSolid("bladder", bladder, inner);

    G4LogicalVolume* logicUrinaryBladder = new G4LogicalVolume(totalBladder, soft, "logical" + volumeName, 0, 0, 0);

    // Define rotation and position here!
    G4VPhysicalVolume* physUrinaryBladder =
        new G4PVPlacement(0, G4ThreeVector(0 * cm, -4.5 * cm, -27. * cm), "physicalUrinaryBladder", logicUrinaryBladder,
                          mother, false, 0, true);

    // Visualization Attributes
    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);
    G4VisAttributes* UrinaryBladderVisAtt = new G4VisAttributes(colour);
    // G4VisAttributes* UrinaryBladderVisAtt = new G4VisAttributes(G4Colour(0.85,0.65,0.125));

    UrinaryBladderVisAtt->SetForceSolid(wireFrame);
    logicUrinaryBladder->SetVisAttributes(UrinaryBladderVisAtt);

    G4cout << "UrinaryBladder created !!!!!!" << G4endl;

    // Testing UrinaryBladder Volume
    G4double UrinaryBladderVol = logicUrinaryBladder->GetSolid()->GetCubicVolume();
    G4cout << "Volume of UrinaryBladder = " << UrinaryBladderVol / cm3 << " cm^3" << G4endl;

    // Testing UrinaryBladder Material
    G4String UrinaryBladderMat = logicUrinaryBladder->GetMaterial()->GetName();
    G4cout << "Material of UrinaryBladder = " << UrinaryBladderMat << G4endl;

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

    // Testing Mass
    G4double UrinaryBladderMass = (UrinaryBladderVol)*UrinaryBladderDensity;
    G4cout << "Mass of UrinaryBladder = " << UrinaryBladderMass / gram << " g" << G4endl;

    return physUrinaryBladder;
}