#include "Framework/ASoA.h"

#include "Framework/TableBuilder.h"

#include "Framework/AnalysisDataModel.h"

#include <catch_amalgamated.hpp>

using namespace o2::framework;

using namespace arrow;

using namespace o2::soa;

using namespace o2::aod;

namespace o2::aod

{

namespace col

{

DECLARE_SOA_COLUMN(X, x, float);

DECLARE_SOA_COLUMN(Y, y, float);

DECLARE_SOA_COLUMN(Z, z, float);

DECLARE_SOA_COLUMN(D, d, float);

} // namespace col

DECLARE_SOA_TABLE(XY, "AOD", "XY", col::X, col::Y);

DECLARE_SOA_TABLE(ZD, "AOD", "ZD", col::Z, col::D);

} // namespace o2::aod

TEST_CASE("TestJoinedTablesContains")

{

TableBuilder XYBuilder;

// FIXME: using full tracks, instead of stored because of unbound dynamic

// column (normalized phi)

auto xyWriter = XYBuilder.cursor<XY>();

xyWriter(0, 0, 0);

auto tXY = XYBuilder.finalize();

TableBuilder ZDBuilder;

auto zdWriter = ZDBuilder.cursor<ZD>();

zdWriter(0, 7, 1);

auto tZD = ZDBuilder.finalize();

using Test = o2::soa::Join<XY, ZD>;

Test tests{{tXY, tZD}, 0};

REQUIRE(tests.asArrowTable()->num_columns() != 0);

REQUIRE(tests.asArrowTable()->num_columns() ==

tXY->num_columns() + tZD->num_columns());

auto tests2 = join(XY{tXY}, ZD{tZD});

static_assert(std::same_as<Test::table_t, decltype(tests2)::table_t>,

"Joined tables should have the same type, regardless how we construct them");

using FullTracks = o2::soa::Join<o2::aod::Tracks, o2::aod::TracksExtra, o2::aod::TracksCov>;

REQUIRE(FullTracks::contains<o2::aod::Tracks>());

REQUIRE(!FullTracks::contains<o2::aod::Collisions>());

}