#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDBrain.h"

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

G4MIRDBrain::G4MIRDBrain() {}

G4MIRDBrain::~G4MIRDBrain() {}

G4VPhysicalVolume* G4MIRDBrain::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");
    delete material;

    G4double ax = 6. * cm;
    G4double by = 9. * cm;
    G4double cz = 6.5 * cm;

    G4Ellipsoid* brain = new G4Ellipsoid("Brain", ax, by, cz);

    G4LogicalVolume* logicBrain = new G4LogicalVolume(brain, soft, "logical" + volumeName, 0, 0, 0);

    // Define rotation and position here!
    G4VPhysicalVolume* physBrain = new G4PVPlacement(0, G4ThreeVector(0. * cm, 0. * cm, 8.75 * cm), "physicalBrain",
                                                     logicBrain, mother, false, 0, true);

    // Visualization Attributes
    // G4VisAttributes* BrainVisAtt = new G4VisAttributes(G4Colour(0.41,0.41,0.41));
    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);

    G4VisAttributes* BrainVisAtt = new G4VisAttributes(colour);
    BrainVisAtt->SetForceSolid(wireFrame);
    BrainVisAtt->SetLineWidth(0.7 * mm);
    logicBrain->SetVisAttributes(BrainVisAtt);

    // Testing Brain Volume
    G4double BrainVol = logicBrain->GetSolid()->GetCubicVolume();
    G4cout << "Volume of Brain = " << BrainVol / cm3 << " cm^3" << G4endl;

    // Testing Brain Material
    G4String BrainMat = logicBrain->GetMaterial()->GetName();
    G4cout << "Material of Brain = " << BrainMat << G4endl;

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

    // Testing Mass
    G4double BrainMass = (BrainVol)*BrainDensity;
    G4cout << "Mass of Brain = " << BrainMass / gram << " g" << G4endl;

    return physBrain;
}