#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDLowerLargeIntestine.h"

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

G4MIRDLowerLargeIntestine::G4MIRDLowerLargeIntestine() {}

G4MIRDLowerLargeIntestine::~G4MIRDLowerLargeIntestine() {}

G4VPhysicalVolume* G4MIRDLowerLargeIntestine::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 dx = 1.88 * cm;  // a
    G4double dy = 2.13 * cm;  // b
    G4double dz = 7.64 * cm;  //(z1-z2)/2

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

    G4double rmin = 0.0 * cm;
    G4double rmax = 1.88 * cm;  // a
    G4double rtor = 5.72 * cm;  // R1
    G4double startphi = 0. * degree;
    G4double deltaphi = 90. * degree;

    G4Torus* SigmoidColonUpLowerLargeIntestine =
        new G4Torus("SigmoidColonUpLowerLargeIntestine", rmin, rmax, rtor, startphi, deltaphi);

    rtor = 3. * cm;  // R2
    G4VSolid* SigmoidColonDownLowerLargeIntestine =
        new G4Torus("SigmoidColonDownLowerLargeIntestine", rmin, rmax, rtor, startphi, deltaphi);

    G4RotationMatrix* relative_rm = new G4RotationMatrix();
    relative_rm->rotateY(180. * degree);
    relative_rm->rotateZ(90. * degree);

    G4UnionSolid* SigmoidColonLowerLargeIntestine =
        new G4UnionSolid("SigmoidColonLowerLargeIntestine", SigmoidColonUpLowerLargeIntestine,
                         SigmoidColonDownLowerLargeIntestine, relative_rm, G4ThreeVector(0.0, 8.72 * cm, 0.0));
    // R1 + R2

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

    G4UnionSolid* LowerLargeIntestine =
        new G4UnionSolid("LowerLargeIntestine", DescendingColonLowerLargeIntestine, SigmoidColonLowerLargeIntestine,
                         relative_rm_2, G4ThreeVector(-5.72 * cm, 0.0 * cm, -7.64 * cm));  // -rtor,0, -dz

    G4LogicalVolume* logicLowerLargeIntestine =
        new G4LogicalVolume(LowerLargeIntestine, soft, "logical" + volumeName, 0, 0, 0);

    G4VPhysicalVolume* physLowerLargeIntestine =
        new G4PVPlacement(0,  // R1+ R2, -2.36 (y0), z0
                          G4ThreeVector(8.72 * cm, -2.36 * cm, -18.64 * cm), "physicalLowerLargeIntestine",
                          logicLowerLargeIntestine, mother, false, 0, true);

    // Visualization Attributes
    // G4VisAttributes* LowerLargeIntestineVisAtt = new G4VisAttributes(G4Colour(1.0,1.0,0.0));
    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);
    G4VisAttributes* LowerLargeIntestineVisAtt = new G4VisAttributes(colour);
    LowerLargeIntestineVisAtt->SetForceSolid(wireFrame);
    logicLowerLargeIntestine->SetVisAttributes(LowerLargeIntestineVisAtt);

    G4cout << "LowerLargeIntestine created !!!!!!" << G4endl;

    // Testing LowerLargeIntestine Volume
    G4double LowerLargeIntestineVol = logicLowerLargeIntestine->GetSolid()->GetCubicVolume();
    G4cout << "Volume of LowerLargeIntestine = " << LowerLargeIntestineVol / cm3 << " cm^3" << G4endl;

    // Testing LowerLargeIntestine Material
    G4String LowerLargeIntestineMat = logicLowerLargeIntestine->GetMaterial()->GetName();
    G4cout << "Material of LowerLargeIntestine = " << LowerLargeIntestineMat << G4endl;

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

    // Testing Mass
    G4double LowerLargeIntestineMass = (LowerLargeIntestineVol)*LowerLargeIntestineDensity;
    G4cout << "Mass of LowerLargeIntestine = " << LowerLargeIntestineMass / gram << " g" << G4endl;

    return physLowerLargeIntestine;
}