#include "stdafx.h"
#include "widgets/vehicleparams.h"
#include "simulation.h"
#include "driveruipanels.h"
#include "Driver.h"
#include "Train.h"

ui::vehicleparams_panel::vehicleparams_panel(const std::string &vehicle)
    : ui_panel(std::string(STR("Vehicle parameters")) + ": " + vehicle, false), m_vehicle_name(vehicle)
{

}

void screen_window_callback(ImGuiSizeCallbackData *data) {
	auto config = static_cast<const global_settings::pythonviewport_config*>(data->UserData);
	data->DesiredSize.y = data->DesiredSize.x * (float)config->size.y / (float)config->size.x;
}

void ui::vehicleparams_panel::render_contents()
{
	TDynamicObject *vehicle_ptr = simulation::Vehicles.find(m_vehicle_name);
	if (!vehicle_ptr) {
		is_open = false;
		return;
	}

	TTrain *train_ptr = simulation::Trains.find(m_vehicle_name);
	if (train_ptr) {
		const TTrain::screen_map &screens = train_ptr->get_screens();

		for (const auto &viewport : Global.python_viewports) {
			for (auto const &entry : screens) {
				if (std::get<std::string>(entry) != viewport.surface)
					continue;

				std::string window_name = STR("Screen") + "##" + viewport.surface;
				ImGui::SetNextWindowSizeConstraints(ImVec2(200, 200), ImVec2(2500, 2500), screen_window_callback,
				                                    const_cast<global_settings::pythonviewport_config*>(&viewport));
				if (ImGui::Begin(window_name.c_str())) {
					float aspect = (float)viewport.size.y / viewport.size.x;

					glm::mat3 proj = glm::translate(glm::scale(glm::mat3(), 1.0f / viewport.scale), viewport.offset);

					glm::vec2 uv0 = glm::vec2(proj * glm::vec3(0.0f, 1.0f, 1.0f));
					glm::vec2 uv1 = glm::vec2(proj * glm::vec3(1.0f, 0.0f, 1.0f));

					ImVec2 size = ImGui::GetContentRegionAvail();

					ImGui::Image(reinterpret_cast<void*>(std::get<1>(entry)->shared_tex), size, ImVec2(uv0.x, uv0.y), ImVec2(uv1.x, uv1.y));
				}
				ImGui::End();
			}
		}
	}

	TDynamicObject &vehicle = *vehicle_ptr;
	TMoverParameters &mover = *vehicle.MoverParameters;

	std::vector<text_line> lines;
	std::array<char, 1024> buffer;

	auto const isdieselenginepowered { ( mover.EngineType == TEngineType::DieselElectric ) || ( mover.EngineType == TEngineType::DieselEngine ) };
	auto const isdieselinshuntmode { mover.ShuntMode && mover.EngineType == TEngineType::DieselElectric };

	std::snprintf(
	    buffer.data(), buffer.size(),
	    STR_C("Devices: %c%c%c%c%c%c%c%c%c%c%c%c%c%c%s%s\nPower transfers: %.0f@%.0f%s%s%s%.0f@%.0f"),
	    // devices
	    ( mover.Battery ? 'B' : '.' ),
	    ( mover.Mains ? 'M' : '.' ),
	    ( mover.FuseFlag ? '!' : '.' ),
	    ( mover.PantRearUp ? ( mover.PantRearVolt > 0.0 ? 'O' : 'o' ) : '.' ),
	    ( mover.PantFrontUp ? ( mover.PantFrontVolt > 0.0 ? 'P' : 'p' ) : '.' ),
	    ( mover.PantPressLockActive ? '!' : ( mover.PantPressSwitchActive ? '*' : '.' ) ),
	    ( mover.WaterPump.is_active ? 'W' : ( false == mover.WaterPump.breaker ? '-' : ( mover.WaterPump.is_enabled ? 'w' : '.' ) ) ),
	    ( true == mover.WaterHeater.is_damaged ? '!' : ( mover.WaterHeater.is_active ? 'H' : ( false == mover.WaterHeater.breaker ? '-' : ( mover.WaterHeater.is_enabled ? 'h' : '.' ) ) ) ),
	    ( mover.FuelPump.is_active ? 'F' : ( mover.FuelPump.is_enabled ? 'f' : '.' ) ),
	    ( mover.OilPump.is_active ? 'O' : ( mover.OilPump.is_enabled ? 'o' : '.' ) ),
	    ( false == mover.ConverterAllowLocal ? '-' : ( mover.ConverterAllow ? ( mover.ConverterFlag ? 'X' : 'x' ) : '.' ) ),
	    ( mover.ConvOvldFlag ? '!' : '.' ),
	    ( mover.CompressorFlag ? 'C' : ( false == mover.CompressorAllowLocal ? '-' : ( ( mover.CompressorAllow || mover.CompressorStart == start_t::automatic ) ? 'c' : '.' ) ) ),
	    ( mover.CompressorGovernorLock ? '!' : '.' ),
	    "",
	    std::string( isdieselenginepowered ? STR(" oil pressure: ") + to_string( mover.OilPump.pressure, 2 )  : "" ).c_str(),
	    // power transfers
	    mover.Couplers[ end::front ].power_high.voltage,
	    mover.Couplers[ end::front ].power_high.current,
	    std::string( mover.Couplers[ end::front ].power_high.is_local ? "" : "-" ).c_str(),
	    std::string( vehicle.DirectionGet() ? ":<<:" : ":>>:" ).c_str(),
	    std::string( mover.Couplers[ end::rear ].power_high.is_local ? "" : "-" ).c_str(),
	    mover.Couplers[ end::rear ].power_high.voltage,
	    mover.Couplers[ end::rear ].power_high.current );

	ImGui::TextUnformatted(buffer.data());

	std::snprintf(
	    buffer.data(), buffer.size(),
	    STR_C("Controllers:\n master: %d(%d), secondary: %s\nEngine output: %.1f, current: %.0f\nRevolutions:\n engine: %.0f, motors: %.0f\n engine fans: %.0f, motor fans: %.0f+%.0f, cooling fans: %.0f+%.0f"),
	    // controllers
	    mover.MainCtrlPos,
	    mover.MainCtrlActualPos,
	    std::string( isdieselinshuntmode ? to_string( mover.AnPos, 2 ) + STR(" (shunt mode)") : std::to_string( mover.ScndCtrlPos ) + "(" + std::to_string( mover.ScndCtrlActualPos ) + ")" ).c_str(),
	    // engine
	    mover.EnginePower,
	    std::abs( mover.TrainType == dt_EZT ? mover.ShowCurrent( 0 ) : mover.Im ),
	    // revolutions
	    std::abs( mover.enrot ) * 60,
	    std::abs( mover.nrot ) * mover.Transmision.Ratio * 60,
	    mover.RventRot * 60,
	    std::abs( mover.MotorBlowers[end::front].revolutions ),
	    std::abs( mover.MotorBlowers[end::rear].revolutions ),
	    mover.dizel_heat.rpmw,
	    mover.dizel_heat.rpmw2 );

	ImGui::TextUnformatted(buffer.data());

	if( isdieselenginepowered ) {
		std::snprintf(
		    buffer.data(), buffer.size(),
		    STR_C("\nTemperatures:\n engine: %.2f, oil: %.2f, water: %.2f%c%.2f"),
		    mover.dizel_heat.Ts,
		    mover.dizel_heat.To,
		    mover.dizel_heat.temperatura1,
		    ( mover.WaterCircuitsLink ? '-' : '|' ),
		    mover.dizel_heat.temperatura2 );
		ImGui::TextUnformatted(buffer.data());
	}

	std::string brakedelay;
	{
		std::vector<std::pair<int, std::string>> delays {
			{ bdelay_G, "G" },
			{ bdelay_P, "P" },
			{ bdelay_R, "R" },
			{ bdelay_M, "+Mg" } };

		for( auto const &delay : delays ) {
			if( ( mover.BrakeDelayFlag & delay.first ) == delay.first ) {
				brakedelay += delay.second;
			}
		}
	}

	std::snprintf(
	    buffer.data(), buffer.size(),
	    STR_C("Brakes:\n train: %.2f, independent: %.2f, mode: %d, delay: %s, load flag: %d\nBrake cylinder pressures:\n train: %.2f, independent: %.2f, status: 0x%.2x\nPipe pressures:\n brake: %.2f (hat: %.2f), main: %.2f, control: %.2f\nTank pressures:\n auxiliary: %.2f, main: %.2f, control: %.2f"),
	    // brakes
	    mover.fBrakeCtrlPos,
	    mover.LocalBrakePosA,
	    mover.BrakeOpModeFlag,
	    brakedelay.c_str(),
	    mover.LoadFlag,
	    // cylinders
	    mover.BrakePress,
	    mover.LocBrakePress,
	    mover.Hamulec->GetBrakeStatus(),
	    // pipes
	    mover.PipePress,
	    mover.BrakeCtrlPos2,
	    mover.ScndPipePress,
	    mover.CntrlPipePress,
	    // tanks
	    mover.Hamulec->GetBRP(),
	    mover.Compressor,
	    mover.Hamulec->GetCRP() );

	ImGui::TextUnformatted(buffer.data());

	if( mover.EnginePowerSource.SourceType == TPowerSource::CurrentCollector ) {
		std::snprintf(
		    buffer.data(), buffer.size(),
		    STR_C(" pantograph: %.2f%cMT"),
		    mover.PantPress,
		    ( mover.bPantKurek3 ? '-' : '|' ) );
		ImGui::TextUnformatted(buffer.data());
	}

	std::snprintf(
	    buffer.data(), buffer.size(),
	    STR_C("Forces:\n tractive: %.1f, brake: %.1f, friction: %.2f%s\nAcceleration:\n tangential: %.2f, normal: %.2f (path radius: %s)\nVelocity: %.2f, distance traveled: %.2f\nPosition: [%.2f, %.2f, %.2f]"),
	    // forces
	    mover.Ft * 0.001f * ( mover.ActiveCab ? mover.ActiveCab : vehicle.ctOwner ? vehicle.ctOwner->Controlling()->ActiveCab : 1 ) + 0.001f,
	    mover.Fb * 0.001f,
	    mover.Adhesive( mover.RunningTrack.friction ),
	    ( mover.SlippingWheels ? " (!)" : "" ),
	    // acceleration
	    mover.AccSVBased,
	    mover.AccN + 0.001f,
	    std::string( std::abs( mover.RunningShape.R ) > 10000.0 ? "~0" : to_string( mover.RunningShape.R, 0 ) ).c_str(),
	    // velocity
	    vehicle.GetVelocity(),
	    mover.DistCounter,
	    // position
	    vehicle.GetPosition().x,
	    vehicle.GetPosition().y,
	    vehicle.GetPosition().z );

	ImGui::TextUnformatted(buffer.data());

	if (ImGui::Button(STR_C("Radiostop")))
		m_relay.post(user_command::radiostop, 0.0, 0.0, GLFW_PRESS, 0, vehicle_ptr->GetPosition());
	ImGui::SameLine();

	if (ImGui::Button(STR_C("Stop and repair")))
		m_relay.post(user_command::resetconsist, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	ImGui::SameLine();

	if (ImGui::Button(STR_C("Reset position"))) {
		std::string payload = vehicle_ptr->name() + '%' + vehicle_ptr->initial_track->name();
		m_relay.post(user_command::consistteleport, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &payload);
		m_relay.post(user_command::resetconsist, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	}

	if (ImGui::Button(STR_C("Refill main tank")))
		m_relay.post(user_command::fillcompressor, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	ImGui::SameLine();

	if (ImGui::Button(STR_C("Rupture main pipe")))
		m_relay.post(user_command::pullalarmchain, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	ImGui::SameLine();

	ImGui::Button(STR_C("independent brake releaser"));
	if (ImGui::IsItemClicked())
		m_relay.post(user_command::consistreleaser, 0.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	if (ImGui::IsItemDeactivated())
		m_relay.post(user_command::consistreleaser, 0.0, 0.0, GLFW_RELEASE, 0, glm::vec3(0.0f), &vehicle_ptr->name());

	if (vehicle_ptr->MoverParameters->V < 0.01) {
		if (ImGui::Button(STR_C("Move +500m")))
			m_relay.post(user_command::dynamicmove, 500.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
		ImGui::SameLine();

		if (ImGui::Button(STR_C("Move -500m")))
			m_relay.post(user_command::dynamicmove, -500.0, 0.0, GLFW_PRESS, 0, glm::vec3(0.0f), &vehicle_ptr->name());
	}
}