/* This file is part of MAUS: http://micewww.pp.rl.ac.uk:8080/projects/maus
 *
 * MAUS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MAUS is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MAUS.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#include <iostream>

#include "Interface/Squeak.hh"
#include "Utils/Exception.hh"
#include "Interface/MICERun.hh"

#include "src/common_cpp/Utils/JsonWrapper.hh"
#include "src/common_cpp/Simulation/MAUSSteppingAction.hh"
#include "src/common_cpp/Simulation/VirtualPlanes.hh"
#include "src/common_cpp/Simulation/MAUSGeant4Manager.hh"


namespace MAUS {

MAUSSteppingAction::MAUSSteppingAction() : _steps(), _keepSteps(false),
    _maxNSteps(0) {
  Json::Value& conf = *MICERun::getInstance()->jsonConfiguration;
  _maxNSteps = JsonWrapper::GetProperty
               (conf, "maximum_number_of_steps", JsonWrapper::intValue).asInt();
  _keepSteps = JsonWrapper::GetProperty
               (conf, "keep_steps", JsonWrapper::booleanValue).asBool();
}

void MAUSSteppingAction::UserSteppingAction(const G4Step * aStep) {
    // record all step points if keep steps flag is set
    if (_keepSteps) {
        if (_steps.size() == 0) {
            _steps.append(StepToJson(aStep, true));
        }
        _steps.append(StepToJson(aStep, false));
    }

    // kill particles that are looping
    G4Track* aTrack = aStep->GetTrack();
    if (aTrack->GetCurrentStepNumber() > _maxNSteps) {
        std::stringstream ss;
        ss << aTrack->GetCurrentStepNumber() <<  " steps greater than maximum "
           << _maxNSteps << " - suspected of looping";
        MAUSGeant4Manager::GetInstance()->GetTracking()
                  ->SetKillReason(aTrack, ss.str());
        aTrack->SetTrackStatus(fStopAndKill);
    }

    // find and append any virtual planes
    MAUSGeant4Manager::GetInstance()->
                         GetVirtualPlanes()->VirtualPlanesSteppingAction(aStep);
}

Json::Value MAUSSteppingAction::StepToJson
                                     (const G4Step* aStep, bool prestep) const {
    G4Track* aTrack = aStep->GetTrack();
    G4StepPoint* point = aStep->GetPostStepPoint();
    if (prestep) point = aStep->GetPreStepPoint();
    if (!aTrack || !point || !aStep)
        throw(Exception(Exception::recoverable,
                     "Failed to resolve step point",
                     "MAUSSteppingAction::StepPointToJson"));
    Json::Value step;
    Json::Value position;
    position["x"] = point->GetPosition().x();
    position["y"] = point->GetPosition().y();
    position["z"] = point->GetPosition().z();
    step["position"] = position;

    Json::Value momentum;
    momentum["x"] = point->GetMomentum().x();
    momentum["y"] = point->GetMomentum().y();
    momentum["z"] = point->GetMomentum().z();
    step["momentum"] = momentum;

    Json::Value spin;
    spin["x"] = point->GetPolarization().x();
    spin["y"] = point->GetPolarization().y();
    spin["z"] = point->GetPolarization().z();
    step["spin"] = spin;

    double f_pos[4] = {point->GetPosition().x(), point->GetPosition().y(),
                       point->GetPosition().z(), point->GetGlobalTime()};
    double field[6] = {0., 0., 0., 0., 0., 0.};
    MAUSGeant4Manager::GetInstance()->GetField()->GetFieldValue(f_pos, field);
    Json::Value bfield;
    bfield["x"] = field[0];
    bfield["y"] = field[1];
    bfield["z"] = field[2];
    step["b_field"] = bfield;

    Json::Value efield;
    efield["x"] = field[3];
    efield["y"] = field[4];
    efield["z"] = field[5];
    step["e_field"] = efield;

    step["energy"] = point->GetTotalEnergy();
    step["proper_time"] = point->GetProperTime();
    step["path_length"] = aTrack->GetTrackLength();
    step["energy_deposited"] = aStep->GetTotalEnergyDeposit();
    step["time"] = point->GetGlobalTime();
    // G4StepPoint GetPhysicalVolume is inlined
    // point->GetTouchable()->GetPhysicalVolume() with no NULL check so we do it
    // here
    if (point->GetTouchable() != NULL && point->GetPhysicalVolume() != NULL) {
        step["volume"] = point->GetPhysicalVolume()->GetName();
    } else {
        step["volume"] = "";
    }
    if (point->GetMaterial() != NULL) {
        step["material"] = point->GetMaterial()->GetName();
    } else {
        step["material"] = "";
    }
    return step;
}

void  MAUSSteppingAction::SetSteps(Json::Value steps) {
    if (!steps.isArray())
        throw(Exception(Exception::recoverable,
              "Attempting to set steps to non-array type",
              "MAUSSteppingAction::SetSteps()"));
    _steps = steps;
}


}  //  ends MAUS namespace