#include "cata_catch.h"
#include "map.h"

#include <memory>
#include <vector>

#include "avatar.h"
#include "coordinates.h"
#include "enums.h"
#include "itype.h"
#include "game.h"
#include "game_constants.h"
#include "map_helpers.h"
#include "point.h"
#include "submap.h"
#include "type_id.h"

TEST_CASE( "map_coordinate_conversion_functions" )
{
    map &here = get_map();

    const on_out_of_scope restore_vertical_shift( [&here]() {
        here.vertical_shift( 0 );
    } );

    tripoint test_point =
        GENERATE( tripoint_zero, tripoint_south, tripoint_east, tripoint_above, tripoint_below );
    tripoint_bub_ms test_bub( test_point );
    int z = GENERATE( 0, 1, -1, OVERMAP_HEIGHT, -OVERMAP_DEPTH );

    // Make sure we're not in the 'easy' case where abs_sub is zero
    if( here.get_abs_sub().x() == 0 ) {
        here.shift( point_east );
    }
    if( here.get_abs_sub().y() == 0 ) {
        here.shift( point_south );
    }
    here.vertical_shift( z );

    CAPTURE( here.get_abs_sub() );

    REQUIRE( here.get_abs_sub().x() != 0 );
    REQUIRE( here.get_abs_sub().y() != 0 );
    REQUIRE( here.get_abs_sub().z() == z );

    point_abs_ms map_origin_ms = project_to<coords::ms>( here.get_abs_sub().xy() );

    tripoint_abs_ms test_abs = map_origin_ms + test_point;

    if( test_abs.z() > OVERMAP_HEIGHT || test_abs.z() < -OVERMAP_DEPTH ) {
        return;
    }

    CAPTURE( test_bub );
    CAPTURE( test_abs );

    // Verify consistency between different implementations
    CHECK( here.getabs( test_bub ) == here.getabs( test_bub.raw() ) );
    CHECK( here.getglobal( test_bub ) == here.getglobal( test_bub.raw() ) );
    CHECK( here.getlocal( test_abs ) == here.getlocal( test_abs.raw() ) );
    CHECK( here.bub_from_abs( test_abs ) == here.bub_from_abs( test_abs.raw() ) );

    CHECK( here.getabs( test_bub ) == here.getglobal( test_bub ).raw() );
    CHECK( here.getlocal( test_abs ) == here.bub_from_abs( test_abs ).raw() );

    // Verify round-tripping
    CHECK( here.getglobal( here.bub_from_abs( test_abs ) ) == test_abs );
    CHECK( here.bub_from_abs( here.getglobal( test_point ) ).raw() == test_point );
}

TEST_CASE( "destroy_grabbed_furniture" )
{
    clear_map();
    avatar &player_character = get_avatar();
    GIVEN( "Furniture grabbed by the player" ) {
        const tripoint test_origin( 60, 60, 0 );
        map &here = get_map();
        player_character.setpos( test_origin );
        const tripoint grab_point = test_origin + tripoint_east;
        here.furn_set( grab_point, furn_id( "f_chair" ) );
        player_character.grab( object_type::FURNITURE, tripoint_east );
        REQUIRE( player_character.get_grab_type() == object_type::FURNITURE );
        WHEN( "The furniture grabbed by the player is destroyed" ) {
            here.destroy( grab_point );
            THEN( "The player's grab is released" ) {
                CHECK( player_character.get_grab_type() == object_type::NONE );
                CHECK( player_character.grab_point == tripoint_zero );
            }
        }
    }
}

TEST_CASE( "map_bounds_checking" )
{
    // FIXME: There are issues with vehicle caching between maps, because
    // vehicles are stored in the global MAPBUFFER which all maps refer to.  To
    // work around the problem we clear the map of vehicles, but this is an
    // inelegant solution.
    clear_map();
    map m;
    tripoint_abs_sm point_away_from_real_map( get_map().get_abs_sub() + point( MAPSIZE_X, 0 ) );
    m.load( point_away_from_real_map, false );
    for( int x = -1; x <= MAPSIZE_X; ++x ) {
        for( int y = -1; y <= MAPSIZE_Y; ++y ) {
            for( int z = -OVERMAP_DEPTH - 1; z <= OVERMAP_HEIGHT + 1; ++z ) {
                INFO( "( " << x << ", " << y << ", " << z << " )" );
                if( x < 0 || x >= MAPSIZE_X ||
                    y < 0 || y >= MAPSIZE_Y ||
                    z < -OVERMAP_DEPTH || z > OVERMAP_HEIGHT ) {
                    CHECK( !m.ter( tripoint{ x, y, z } ) );
                } else {
                    CHECK( m.ter( tripoint{ x, y, z } ) );
                }
            }
        }
    }
}

TEST_CASE( "tinymap_bounds_checking" )
{
    // FIXME: There are issues with vehicle caching between maps, because
    // vehicles are stored in the global MAPBUFFER which all maps refer to.  To
    // work around the problem we clear the map of vehicles, but this is an
    // inelegant solution.
    clear_map();
    tinymap m;
    tripoint_abs_sm point_away_from_real_map( get_map().get_abs_sub() + point( MAPSIZE_X, 0 ) );
    m.load( point_away_from_real_map, false );
    for( int x = -1; x <= SEEX * 2; ++x ) {
        for( int y = -1; y <= SEEY * 2; ++y ) {
            for( int z = -OVERMAP_DEPTH - 1; z <= OVERMAP_HEIGHT + 1; ++z ) {
                INFO( "( " << x << ", " << y << ", " << z << " )" );
                if( x < 0 || x >= SEEX * 2 ||
                    y < 0 || y >= SEEY * 2 ||
                    z < -OVERMAP_DEPTH || z > OVERMAP_HEIGHT ) {
                    CHECK( !m.ter( tripoint{ x, y, z } ) );
                } else {
                    CHECK( m.ter( tripoint{ x, y, z } ) );
                }
            }
        }
    }
}

void map::check_submap_active_item_consistency()
{
    process_items();
    for( int z = -OVERMAP_DEPTH; z < OVERMAP_HEIGHT; ++z ) {
        for( int x = 0; x < MAPSIZE; ++x ) {
            for( int y = 0; y < MAPSIZE; ++y ) {
                tripoint p( x, y, z );
                submap *s = get_submap_at_grid( p );
                REQUIRE( s != nullptr );
                bool submap_has_active_items = !s->active_items.empty();
                bool cache_has_active_items = submaps_with_active_items.count( p + abs_sub.xy() ) != 0;
                CAPTURE( abs_sub.xy(), p, p + abs_sub.xy() );
                CHECK( submap_has_active_items == cache_has_active_items );
            }
        }
    }
    for( const tripoint_abs_sm &p : submaps_with_active_items ) {
        tripoint_rel_sm rel = p - abs_sub.xy();
        half_open_rectangle<point_rel_sm> map( point_rel_sm(), point_rel_sm( MAPSIZE, MAPSIZE ) );
        CAPTURE( rel );
        CHECK( map.contains( rel.xy() ) );
    }
}

TEST_CASE( "place_player_can_safely_move_multiple_submaps" )
{
    // Regression test for the situation where game::place_player would misuse
    // map::shift if the resulting shift exceeded a single submap, leading to a
    // broken active item cache.
    g->place_player( tripoint_zero );
    get_map().check_submap_active_item_consistency();
}

TEST_CASE( "inactive_container_with_active_contents", "[active_item][map]" )
{
    map &here = get_map();
    clear_map( -OVERMAP_DEPTH, OVERMAP_HEIGHT );
    CAPTURE( here.get_abs_sub(), here.get_submaps_with_active_items() );
    REQUIRE( here.get_submaps_with_active_items().empty() );
    here.check_submap_active_item_consistency();
    tripoint const test_loc;
    tripoint_abs_sm const test_loc_sm = project_to<coords::sm>( here.getglobal( test_loc ) );

    item bottle_plastic( "bottle_plastic" );
    REQUIRE( !bottle_plastic.needs_processing() );
    item disinfectant( "disinfectant" );
    REQUIRE( disinfectant.needs_processing() );

    ret_val<void> const ret =
        bottle_plastic.put_in( disinfectant, pocket_type::CONTAINER );
    REQUIRE( ret.success() );

    item &bp = here.add_item( test_loc, bottle_plastic );
    here.update_submaps_with_active_items();
    item_location bp_loc( map_cursor( test_loc ), &bp );
    item_location dis_loc( bp_loc, &bp.only_item() );

    REQUIRE( here.get_submaps_with_active_items().count( test_loc_sm ) != 0 );
    here.check_submap_active_item_consistency();

    bool from_container = GENERATE( true, false );
    CAPTURE( from_container );

    if( from_container ) {
        dis_loc.remove_item();
    } else {
        bp_loc.remove_item();
    }
    here.process_items();
    CHECK( here.get_submaps_with_active_items().count( test_loc_sm ) == 0 );
    here.check_submap_active_item_consistency();
}

TEST_CASE( "milk_rotting", "[active_item][map]" )
{
    map &here = get_map();
    clear_map( -OVERMAP_DEPTH, OVERMAP_HEIGHT );
    CAPTURE( here.get_abs_sub(), here.get_submaps_with_active_items() );
    here.check_submap_active_item_consistency();
    REQUIRE( here.get_submaps_with_active_items().empty() );
    tripoint const test_loc;
    tripoint_abs_sm const test_loc_sm = project_to<coords::sm>( here.getglobal( test_loc ) );

    restore_on_out_of_scope<std::optional<units::temperature>> restore_temp(
                get_weather().forced_temperature );
    get_weather().forced_temperature = units::from_celsius( 21 );
    REQUIRE( units::to_celsius( get_weather().get_temperature( test_loc ) ) == 21 );

    item almond_milk( "almond_milk" );
    item *bp = nullptr;

    bool const in_container = GENERATE( true, false );
    CAPTURE( in_container );
    bool sealed = false;

    if( in_container ) {
        sealed = GENERATE( true, false );
        item bottle_plastic( "bottle_plastic" );
        ret_val<void> const ret = bottle_plastic.put_in( almond_milk, pocket_type::CONTAINER );
        REQUIRE( ret.success() );

        bp = &here.add_item( test_loc, bottle_plastic );
        REQUIRE( bp->num_item_stacks() == 1 );
        if( sealed ) {
            bp->get_contents().seal_all_pockets();
        }
        REQUIRE( bp->any_pockets_sealed() == sealed );
    } else {
        here.add_item( test_loc, almond_milk );
    }
    CAPTURE( sealed );
    here.check_submap_active_item_consistency();
    REQUIRE( here.get_submaps_with_active_items().count( test_loc_sm ) != 0 );

    time_point const dest = calendar::turn + almond_milk.get_comestible()->spoils * 2;
    while( calendar::turn < dest ) {
        calendar::turn += time_duration::from_seconds( almond_milk.processing_speed() ) + 1_seconds;
        here.process_items();
    }
    if( in_container ) {
        CHECK( bp->num_item_stacks() == static_cast<size_t>( sealed ) );
    } else {
        CHECK( here.i_at( test_loc ).empty() == !sealed );
    }
    here.check_submap_active_item_consistency();
    CHECK( here.get_submaps_with_active_items().count( test_loc_sm ) == static_cast<size_t>( sealed ) );
}

TEST_CASE( "active_monster_drops", "[active_item][map]" )
{
    clear_map();
    get_avatar().setpos( tripoint_zero );
    tripoint start_loc = get_avatar().pos() + tripoint_east;
    map &here = get_map();
    restore_on_out_of_scope<std::optional<units::temperature>> restore_temp(
                get_weather().forced_temperature );
    get_weather().forced_temperature = units::from_celsius( 21 );

    bool const cookie_rotten_before_death = GENERATE( true, false );
    CAPTURE( cookie_rotten_before_death );

    item bag_plastic( "bag_plastic" );
    item cookie( "cookies" );
    REQUIRE( cookie.needs_processing() );
    if( cookie_rotten_before_death ) {
        cookie.set_relative_rot( 10 );
        REQUIRE( cookie.has_rotten_away() );
    }
    REQUIRE( bag_plastic.put_in( cookie, pocket_type::CONTAINER ).success() );

    monster &zombo = spawn_test_monster( "mon_zombie", start_loc, true );
    zombo.no_extra_death_drops = true;
    zombo.inv.emplace_back( bag_plastic );
    calendar::turn += time_duration::from_seconds( cookie.processing_speed() + 1 );
    zombo.die( nullptr );
    REQUIRE( here.i_at( start_loc ).size() == 1 );
    item &dropped_bag = here.i_at( start_loc ).begin()->only_item();

    if( !cookie_rotten_before_death ) {
        item &dropped_cookie = dropped_bag.only_item();
        dropped_cookie.set_relative_rot( 10 );
        REQUIRE( dropped_cookie.has_rotten_away() );
        calendar::turn += time_duration::from_seconds( cookie.processing_speed() + 1 );
        here.process_items(); // only corpse is scheduled here
        here.process_items(); // only cookie is scheduled here
    }
    CHECK( dropped_bag.empty() );
}