#include "G4HumanPhantomColour.h"
#include "G4HumanPhantomMaterial.h"
#include "G4MIRDRibCage.h"

#include "G4Box.hh"
#include "G4EllipticalTube.hh"
#include "G4PVPlacement.hh"
#include "G4PVReplica.hh"
#include "G4SDManager.hh"
#include "G4SubtractionSolid.hh"
#include "G4SystemOfUnits.hh"
#include "G4VisAttributes.hh"
#include "globals.hh"

G4MIRDRibCage::G4MIRDRibCage() {}

G4MIRDRibCage::~G4MIRDRibCage() {}

G4VPhysicalVolume* G4MIRDRibCage::Construct(const G4String& volumeName, G4VPhysicalVolume* mother,
                                            const G4String& colourName, G4bool wireFrame, G4bool) {
    G4HumanPhantomMaterial* material = new G4HumanPhantomMaterial();

    G4cout << "Construct " << volumeName << " with mother volume " << mother->GetName() << G4endl;

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

    delete material;

    G4double dx = 17. * cm;          // a2
    G4double dy = 9.8 * cm;          // b2
    G4double thickness = 32.4 * cm;  // z2/2 of cage

    G4EllipticalTube* outCage = new G4EllipticalTube("outCage", dx, dy, thickness / 2.);

    dx = 16.4 * cm;          // a1
    dy = 9.2 * cm;           // b1
    G4double dz = 34. * cm;  // z2/2

    G4EllipticalTube* inCage = new G4EllipticalTube("inCage", dx, dy, dz / 2.);

    G4SubtractionSolid* cage =
        new G4SubtractionSolid("Cage", outCage, inCage, 0, G4ThreeVector(0. * cm, 0. * cm, 0. * cm));

    G4LogicalVolume* logicRibCage = new G4LogicalVolume(cage, soft, "logicalCage", 0, 0, 0);

    G4VPhysicalVolume* physRibCage = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, thickness / 2. + 0.1 * cm),
                                                       // with respect to the trunk
                                                       "physicalRibCage", logicRibCage, mother, false, 0, true);

    G4double xx = 17. * cm;
    G4double yy = 9.8 * cm;
    G4double ribThickness = 1.4 * cm;
    G4EllipticalTube* rib_out = new G4EllipticalTube("rib_out", xx, yy, ribThickness / 2.);

    xx = 16.5 * cm;
    yy = 9.3 * cm;
    G4double zz = 1.5 * cm;
    G4EllipticalTube* rib_in = new G4EllipticalTube("rib_in", xx, yy, zz / 2.);
    G4SubtractionSolid* rib = new G4SubtractionSolid("rib", rib_out, rib_in);

    G4LogicalVolume* logicRib = new G4LogicalVolume(rib, skeleton, "logical" + volumeName, 0, 0, 0);

    physRib1 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-32.2 * cm / 2. + 0.8 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib2 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-32.2 * cm / 2. + 0.8 * cm + 2.8 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib3 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 5.6 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib4 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 8.4 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib5 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 11.2 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib6 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 14. * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib7 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 16.8 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib8 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 19.6 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib9 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 22.4 * cm)),
                                 // with respect to the trunk
                                 "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib10 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 25.2 * cm)),
                                  // with respect to the trunk
                                  "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib11 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 28. * cm)),
                                  // with respect to the trunk
                                  "physicalRib", logicRib, physRibCage, false, 0, true);

    physRib12 = new G4PVPlacement(0, G4ThreeVector(0.0, 0.0, (-thickness / 2. + 0.8 * cm + 30.8 * cm)),
                                  // with respect to the trunk
                                  "physicalRib", logicRib, physRibCage, false, 0, true);

    // Visualization Attributes
    logicRibCage->SetVisAttributes(G4VisAttributes::GetInvisible());

    // G4VisAttributes* RibCageVisAtt = new G4VisAttributes(G4Colour(0.46,0.53,0.6));

    G4HumanPhantomColour* colourPointer = new G4HumanPhantomColour();
    G4Colour colour = colourPointer->GetColour(colourName);
    G4VisAttributes* RibCageVisAtt = new G4VisAttributes(colour);
    RibCageVisAtt->SetForceSolid(wireFrame);
    logicRib->SetVisAttributes(RibCageVisAtt);

    G4cout << "RibCage created !!!!!!" << G4endl;
    // Testing Pelvis Volume
    G4double RibCageVol = logicRib->GetSolid()->GetCubicVolume();
    G4cout << "Volume of RibCage = " << ((RibCageVol)*12.) / cm3 << " cm^3" << G4endl;

    // Testing RibCage Material
    G4String RibCageMat = logicRib->GetMaterial()->GetName();
    G4cout << "Material of RibCage = " << RibCageMat << G4endl;

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

    // Testing Mass
    G4double RibCageMass = (RibCageVol)*RibCageDensity * 12;  // 12 is the total number of ribs;
    G4cout << "Mass of RibCage = " << (RibCageMass) / gram << " g" << G4endl;

    return physRibCage;
}