/*
 * Copyright (C) 2023 Robin Gareus <robin@gareus.org>
 * Copyright (C) 2023-2024 Adrien Gesta-Fline <dev.agfline@posteo.net>
 *
 * This program 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 2 of the License, or
 * (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <fcntl.h>       // O_WRONLY
#include <glib/gstdio.h> // g_unlink()

#include "pbd/basename.h"
#include "pbd/convert.h"
#include "pbd/file_utils.h"

#include "ardour/audio_track.h"
#include "ardour/audioengine.h"
#include "ardour/audioregion.h"
#include "ardour/filename_extensions.h"
#include "ardour/import_status.h"
#include "ardour/playlist.h"
#include "ardour/plugin_manager.h"
#include "ardour/region_factory.h"
#include "ardour/source_factory.h"
#include "ardour/utils.h"

#include "aaf/libaaf.h"
#include "aaf/utils.h"

#include "ardour_ui.h"
#include "public_editor.h"

#include "pbd/i18n.h"

using namespace std;
using namespace PBD;
using namespace ARDOUR;

static void
aaf_debug_callback (struct aafLog* log, void* ctxdata, int libid, int type, const char* srcfile, const char* srcfunc, int lineno, const char* msg, void* user)
{
	const char* eol = "";

	if (libid != LOG_SRC_ID_TRACE && libid != LOG_SRC_ID_DUMP) {
		switch (type) {
			case VERB_SUCCESS:
				PBD::info << string_compose ("[libaaf] %1:%2 in %3(): ", srcfile, lineno, srcfunc);
				break;
			case VERB_ERROR:
				PBD::error << string_compose ("[libaaf] %1:%2 in %3(): ", srcfile, lineno, srcfunc);
				break;
			case VERB_WARNING:
				PBD::warning << string_compose ("[libaaf] %1:%2 in %3(): ", srcfile, lineno, srcfunc);
				break;
				// case VERB_DEBUG:    PBD::debug   << string_compose ("[libaaf] %1:%2 in %3(): ", srcfile, lineno, srcfunc);  break;
		}
	}

	if (libid != LOG_SRC_ID_DUMP) {
		eol = "\n";
	}

	switch (type) {
		case VERB_SUCCESS:
			PBD::info << msg << eol;
			break;
		case VERB_ERROR:
			PBD::error << msg << eol;
			break;
		case VERB_WARNING:
			PBD::warning << msg << eol;
			break;
			// case VERB_DEBUG:    PBD::debug   << msg << eol;  break;
	}

	LOG_BUFFER_RESET (log);
}

static std::shared_ptr<AudioTrack>
get_nth_audio_track (uint32_t nth, std::shared_ptr<RouteList const> routes)
{
	RouteList rl = *(routes);
	rl.sort (Stripable::Sorter ());

	for (auto const& r : rl) {
		std::shared_ptr<AudioTrack> at = std::dynamic_pointer_cast<AudioTrack> (r);
		if (!at) {
			continue;
		}
		if (nth-- == 0) {
			return at;
		}
	}
	return std::shared_ptr<AudioTrack> ();
}

static std::shared_ptr<AudioTrack>
prepare_audio_track (aafiAudioTrack* aafTrack, Session* s)
{
	/* Use existing track */
	std::shared_ptr<AudioTrack> track = get_nth_audio_track ((aafTrack->number - 1), s->get_routes ());

	if (track) {
		return track;
	}

	/* ..or create a new track */

	uint32_t outputs = 2;
	if (s->master_out ()) {
		outputs = max (outputs, s->master_out ()->n_inputs ().n_audio ());
	}

	list<std::shared_ptr<AudioTrack>> at (s->new_audio_track (aafTrack->format, outputs, NULL, 1, aafTrack->name, PresentationInfo::max_order));

	if (at.empty ()) {
		PBD::fatal << "AAF: Could not create new audio track." << endmsg;
		abort (); /*NOTREACHED*/
	}

	return at.back ();
}

static bool
import_sndfile_as_region (Session* s, struct aafiAudioEssencePointer* aafAudioEssencePtrList, SrcQuality quality, timepos_t& pos, SourceList** oneClipSources, ImportStatus& status, vector<std::shared_ptr<Region>>& regions)
{
	SourceList* sources = NULL;

	if (aafAudioEssencePtrList->user) {
		sources = (SourceList*)aafAudioEssencePtrList->user;
	} else {
		sources = new SourceList;

		/* Import the source */
		status.clear ();

		status.current                 = 1;
		status.total                   = 1;
		status.freeze                  = false;
		status.quality                 = quality;
		status.replace_existing_source = false;
		status.split_midi_channels     = false;
		status.import_markers          = false;
		status.done                    = false;
		status.cancel                  = false;

		int channelCount = 0;

		aafiAudioEssencePointer* aafAudioEssencePtr = NULL;
		AAFI_foreachEssencePointer (aafAudioEssencePtrList, aafAudioEssencePtr)
		{
			if (aafAudioEssencePtr->essenceFile->usable_file_path)
				status.paths.push_back (aafAudioEssencePtr->essenceFile->usable_file_path);
			else
				status.paths.push_back (aafAudioEssencePtr->essenceFile->original_file_path);

			channelCount++;
			PBD::info << string_compose ("AAF: Preparing to import clip channel %1: %2\n", channelCount, aafAudioEssencePtr->essenceFile->unique_name);
		}

		s->import_files (status);

		status.progress = 1.0;
		sources->clear ();

		/* FIXME: There is no way to tell if cancel button was pressed
		 * or if the file failed to import, just that one of these occurred.
		 * We want status.cancel to reflect the user's choice only
		 */
		if (status.cancel && status.current > 1) {
			/* Succeeded to import file, assume user hit cancel */
			return false;
		} else if (status.cancel && status.current == 1) {
			/* Failed to import file, assume user did not hit cancel */
			status.cancel = false;
			return false;
		}

		for (int i = 0; i < channelCount; i++) {
			PropertyList proplist;
			sources->push_back (status.sources.at (i));

			proplist.add (ARDOUR::Properties::start, 0);
			proplist.add (ARDOUR::Properties::length, timecnt_t ((*sources)[0]->length (), pos));
			proplist.add (ARDOUR::Properties::name, aafAudioEssencePtrList->essenceFile->unique_name);
			proplist.add (ARDOUR::Properties::layer, 0);
			proplist.add (ARDOUR::Properties::whole_file, true);
			proplist.add (ARDOUR::Properties::external, true);

			RegionFactory::create (*sources, proplist);
		}

		/* build peakfiles */
		for (SourceList::iterator x = sources->begin (); x != sources->end (); ++x) {
			SourceFactory::setup_peakfile (*x, true);
		}

		aafAudioEssencePtrList->user = sources;
	}

	*oneClipSources = sources;

	/* Put the source on a region */
	string region_name;

	/* take all the sources we have and package them up as a region */
	region_name = region_name_from_path (status.paths.front (), (sources->size () > 1), false);

	/* we checked in import_sndfiles() that there were not too many */
	while (RegionFactory::region_by_name (region_name)) {
		region_name = bump_name_once (region_name, '.');
	}

	return true;
}

static std::shared_ptr<Region>
create_region (vector<std::shared_ptr<Region>> source_regions, aafiAudioClip* aafAudioClip, SourceList& oneClipSources, aafPosition_t clipOffset, aafRational_t samplerate_r)
{
	string unique_file_name = aafAudioClip->essencePointerList->essenceFile->unique_name; // XXX

	aafPosition_t clipPos       = aafi_convertUnit (aafAudioClip->pos, aafAudioClip->track->edit_rate, &samplerate_r);
	aafPosition_t clipLen       = aafi_convertUnit (aafAudioClip->len, aafAudioClip->track->edit_rate, &samplerate_r);
	aafPosition_t essenceOffset = aafi_convertUnit (aafAudioClip->essence_offset, aafAudioClip->track->edit_rate, &samplerate_r);

	PropertyList proplist;

	proplist.add (ARDOUR::Properties::start, essenceOffset);
	proplist.add (ARDOUR::Properties::length, clipLen);
	proplist.add (ARDOUR::Properties::name, unique_file_name);
	proplist.add (ARDOUR::Properties::layer, 0);
	proplist.add (ARDOUR::Properties::whole_file, false);
	proplist.add (ARDOUR::Properties::external, true);

	/* NOTE: region position is set when calling add_region() */

	std::shared_ptr<Region> region = RegionFactory::create (oneClipSources, proplist);

	for (SourceList::iterator source = oneClipSources.begin (); source != oneClipSources.end (); ++source) {
		/* position displayed in Ardour source list */
		(*source)->set_natural_position (timepos_t (clipPos + clipOffset));

		for (vector<std::shared_ptr<Region>>::iterator region = source_regions.begin (); region != source_regions.end (); ++region) {
			if ((*region)->source (0) == *source) {
				/* Enable "Move to Original Position" */
				(*region)->set_position (timepos_t (clipPos + clipOffset - essenceOffset));
			}
		}
	}

	return region;
}

static void
set_region_gain (aafiAudioClip* aafAudioClip, std::shared_ptr<Region> region, Session* s)
{
	if (aafAudioClip->gain && aafAudioClip->gain->flags & AAFI_AUDIO_GAIN_CONSTANT) {
		std::dynamic_pointer_cast<AudioRegion> (region)->set_scale_amplitude (aafRationalToFloat (aafAudioClip->gain->value[0]));
	}

	if (aafAudioClip->automation) {
		aafiAudioGain*                  level = aafAudioClip->automation;
		std::shared_ptr<AudioRegion>    ar    = std::dynamic_pointer_cast<AudioRegion> (region);
		std::shared_ptr<AutomationList> al    = ar->envelope ();

		for (unsigned int i = 0; i < level->pts_cnt; ++i) {
			al->fast_simple_add (timepos_t (aafRationalToFloat (level->time[i]) * region->length ().samples ()), aafRationalToFloat (level->value[i]));
		}
	}
}

static FadeShape
aaf_fade_interpol_to_ardour_fade_shape (aafiInterpolation_e interpol)
{
	switch (interpol & AAFI_INTERPOL_MASK) {
		case AAFI_INTERPOL_NONE:
			return FadeConstantPower;
		case AAFI_INTERPOL_LINEAR:
			return FadeLinear;
		case AAFI_INTERPOL_LOG:
			return FadeConstantPower;
		case AAFI_INTERPOL_CONSTANT:
			return FadeConstantPower;
		case AAFI_INTERPOL_POWER:
			return FadeConstantPower;
		case AAFI_INTERPOL_BSPLINE:
			return FadeConstantPower;
		default:
			return FadeConstantPower;
	}
}

static void
set_region_fade (aafiAudioClip* aafAudioClip, std::shared_ptr<Region> region, aafRational_t* samplerate)
{
	if (aafAudioClip == NULL) {
		return;
	}

	aafiTransition* fadein  = aafi_getFadeIn (aafAudioClip);
	aafiTransition* fadeout = aafi_getFadeOut (aafAudioClip);
	aafiTransition* xfade   = (aafAudioClip->timelineItem->prev) ? aafi_timelineItemToCrossFade (aafAudioClip->timelineItem->prev) : NULL;

	if (xfade) {
		if (fadein == NULL) {
			fadein = xfade;
		} else {
			PBD::warning << "Clip has both fadein and crossfade : crossfade will be ignored." << endmsg;
		}
	}

	FadeShape   fade_shape;
	samplecnt_t fade_len;

	if (fadein != NULL) {
		fade_shape = aaf_fade_interpol_to_ardour_fade_shape ((aafiInterpolation_e)(fadein->flags & AAFI_INTERPOL_MASK));
		fade_len   = aafi_convertUnit (fadein->len, aafAudioClip->track->edit_rate, samplerate);

		std::dynamic_pointer_cast<AudioRegion> (region)->set_fade_in (fade_shape, fade_len);
	}

	if (fadeout != NULL) {
		fade_shape = aaf_fade_interpol_to_ardour_fade_shape ((aafiInterpolation_e)(fadeout->flags & AAFI_INTERPOL_MASK));
		fade_len   = aafi_convertUnit (fadeout->len, aafAudioClip->track->edit_rate, samplerate);

		std::dynamic_pointer_cast<AudioRegion> (region)->set_fade_out (fade_shape, fade_len);
	}
}

static void
set_session_timecode (AAF_Iface* aafi, Session* s)
{
	using namespace Timecode;

	uint16_t       aafFPS = aafi->Timecode->fps;
	TimecodeFormat ardourtc;

	/*
	 *  Fractional timecodes are never explicitly set into tc->fps, so we deduce
	 *  them based on edit_rate value.
	 */

	switch (aafFPS) {
		case 24:
			if (aafi->Timecode->edit_rate->numerator == 24000 &&
			    aafi->Timecode->edit_rate->denominator == 1001) {
				ardourtc = timecode_23976;
			} else {
				ardourtc = timecode_24;
			}
			break;

		case 25:
			if (aafi->Timecode->edit_rate->numerator == 25000 &&
			    aafi->Timecode->edit_rate->denominator == 1001) {
				ardourtc = timecode_24976;
			} else {
				ardourtc = timecode_25;
			}
			break;

		case 30:
			if (aafi->Timecode->edit_rate->numerator == 30000 &&
			    aafi->Timecode->edit_rate->denominator == 1001) {
				if (aafi->Timecode->drop) {
					ardourtc = timecode_2997drop;
				} else {
					ardourtc = timecode_2997;
				}
			} else {
				if (aafi->Timecode->drop) {
					ardourtc = timecode_30drop;
				} else {
					ardourtc = timecode_30;
				}
			}
			break;

		case 60:
			if (aafi->Timecode->edit_rate->numerator == 60000 &&
			    aafi->Timecode->edit_rate->denominator == 1001) {
				ardourtc = timecode_5994;
			} else {
				ardourtc = timecode_60;
			}
			break;

		default:
			PBD::error << string_compose ("Unknown AAF timecode fps : %1 (%2/%3).", aafFPS, aafi->Timecode->edit_rate->numerator, aafi->Timecode->edit_rate->denominator) << endmsg;
			return;
	}

	s->config.set_timecode_format (ardourtc);
}

/* Create and open Sesssion from AAF
 * return > 0 if file is not a [valid] AAF
 * return < 0 if session creation failed.
 * return 0 on success. path and snapshot are set.
 */
int
ARDOUR_UI::new_session_from_aaf (string const& aaf, string const& target_dir, string& path, string& snapshot)
{
	if (PBD::downcase (aaf).find (advanced_authoring_format_suffix) == string::npos) {
		return 1;
	}

	if (_session) {
		if (unload_session (false)) {
			/* unload cancelled by user */
			return 1;
		}
	}

	/* Possible libaaf log to external file : part 1/2 */
	// string logfile = Glib::build_filename (g_get_tmp_dir (), "aaf-import-XXXXXX.log");
	// int logfd = g_mkstemp_full (&logfile[0], O_WRONLY, 0700);
	//
	// fprintf(stderr, "Logfile: %s\n",&logfile[0] );
	//
	// if (logfd < 0) {
	// 	error << _("AAF: Could not prepare log file") << endmsg;
	// 	fprintf(stderr, "AAF: Could not prepare log file\n" );
	// 	return -1;
	// }
	//
	// FILE *logfp = fdopen( logfd, "w" );
	//
	// if (!logfp) {
	// 	error << _("AAF: Could not prepare log file") << endmsg;
	// 	fprintf(stderr, "AAF: Could not prepare log file\n" );
	// 	return -1;
	// }

	AAF_Iface* aafi = aafi_alloc (NULL);

	if (!aafi) {
		error << "AAF: Could not init AAF library." << endmsg;
		return -1;
	}

	/* protools options must be set (there is no sens not setting them with Ardour) */
	uint32_t aaf_protools_options = (AAFI_PROTOOLS_OPT_REPLACE_CLIP_FADES | AAFI_PROTOOLS_OPT_REMOVE_SAMPLE_ACCURATE_EDIT);

	aafi_set_option_int (aafi, "trace", 1);
	aafi_set_option_int (aafi, "protools", aaf_protools_options);
	// aafi_set_option_str (aafi, "media_location", media_location_path.c_str ());

	aafi_set_debug (aafi, VERB_DEBUG, 0, NULL, &aaf_debug_callback, this);

	if (aafi_load_file (aafi, aaf.c_str ())) {
		error << "AAF: Could not load AAF file." << endmsg;
		aafi_release (&aafi);
		return -1;
	}

	snapshot = legalize_for_universal_path (basename_nosuffix (aaf));
	path     = Glib::build_filename (target_dir, snapshot);

	if (Glib::file_test (path, Glib::FILE_TEST_EXISTS)) {
		error << string_compose (_ ("AAF: Destination '%1' already exists."), path) << endmsg;
		snapshot = ""; // XXX?
		path     = "";
		aafi_release (&aafi);
		return -1;
	}

	/* Create media cache */
	GError* err = NULL;
	char*   td  = g_dir_make_tmp ("aaf-cache-XXXXXX", &err);

	if (!td) {
		error << string_compose (_ ("AAF: Could not prepare media cache: %1"), err->message) << endmsg;
		aafi_release (&aafi);
		return -1;
	}

	const string media_cache_path = PBD::canonical_path (td);
	g_free (td);
	g_clear_error (&err);

	/* all systems go. create sessions */
	BusProfile bus_profile;
	bus_profile.master_out_channels = 2;

	aafRational_t samplerate_r;

	samplerate_r.numerator   = aafi->Audio->samplerate;
	samplerate_r.denominator = 1;

	std::string restore_backend;
	if (!AudioEngine::instance ()->running ()) {
		AudioEngine* e  = AudioEngine::instance ();
		restore_backend = e->current_backend_name ();
		e->set_backend ("None (Dummy)", "", "");
		e->start ();
		PluginManager::instance ().refresh (true);
		attach_to_engine ();
	}
	if (!AudioEngine::instance ()->running ()) {
		PBD::error << _ ("AAF: Could not start [dummy] engine for AAF import .") << endmsg;
		return -1;
	}

	build_session_stage_two (path, snapshot, "", bus_profile, false, Temporal::AudioTime, aafi->Audio->samplerate);

	if (!_session) {
		aafi_release (&aafi);
		PBD::remove_directory (media_cache_path);
		if (!restore_backend.empty ()) {
			AudioEngine::instance ()->stop ();
			AudioEngine::instance ()->set_backend (restore_backend, "", "");
		}
		error << _ ("AAF: Could not create new session for AAF import .") << endmsg;
		return -1;
	}

	/* Possible libaaf log to external file : part 2/2
	 * Moving log file from temp/ to session/
	 */

	// string newlogfile = Glib::build_filename (path, "aaf-import.log");
	//
	// if (!PBD::copy_file (logfile, newlogfile)) {
	// // if (g_rename (logfile.c_str(), newlogfile.c_str()) != 0) {
	// 	error << string_compose (_("Could not copy logfile from \"%1\" to \"%2\": %3"),
	// 				 logfile, newlogfile, strerror (errno)) << endmsg;
	// 	fprintf(stderr, "Could not copy logfile from \"%s\" to \"%s\": %s\n", logfile.c_str(), newlogfile.c_str(), strerror (errno) );
	// } else {
	// 	fprintf(stderr, "Copied logfile from \"%s\" to \"%s\"\n", logfile.c_str(), newlogfile.c_str() );
	// 	g_unlink(logfile.c_str ());
	// 	logfile = newlogfile;
	// 	fprintf(stderr, "New logfile : \"%s\"\n", logfile.c_str() );
	// }

	switch (aafi->Audio->samplesize) {
		case 16:
			_session->config.set_native_file_data_format (ARDOUR::FormatInt16);
			break;
		case 24:
			_session->config.set_native_file_data_format (ARDOUR::FormatInt24);
			break;
		case 32:
			_session->config.set_native_file_data_format (ARDOUR::FormatFloat);
			break;
		default:
			break;
	}

	/* Import Sources */

	SourceList*                     oneClipSources;
	ARDOUR::ImportStatus            import_status;
	vector<std::shared_ptr<Region>> source_regions;
	timepos_t                       pos = timepos_t::max (Temporal::AudioTime);

	aafiAudioTrack*          aafAudioTrack      = NULL;
	aafiTimelineItem*        aafAudioItem       = NULL;
	aafiAudioClip*           aafAudioClip       = NULL;
	aafiAudioEssencePointer* aafAudioEssencePtr = NULL;

	aafPosition_t sessionStart = aafi_convertUnit (aafi->compositionStart, aafi->compositionStart_editRate, &samplerate_r);

	AAFI_foreachAudioTrack (aafi, aafAudioTrack)
	{
		std::shared_ptr<AudioTrack> track = prepare_audio_track (aafAudioTrack, _session);

		AAFI_foreachTrackItem (aafAudioTrack, aafAudioItem)
		{
			aafAudioClip = aafi_timelineItemToAudioClip (aafAudioItem);

			if (!aafAudioClip) {
				continue;
			}

			if (aafAudioClip->essencePointerList == NULL) {
				error << _ ("AAF: Clip has no essence.") << endmsg;
				continue;
			}

			int   essenceError = 0;
			char* essenceName  = aafAudioClip->essencePointerList->essenceFile->name;

			AAFI_foreachEssencePointer (aafAudioClip->essencePointerList, aafAudioEssencePtr)
			{
				struct aafiAudioEssenceFile* audioEssenceFile = aafAudioEssencePtr->essenceFile;

				if (!audioEssenceFile) {
					PBD::error << string_compose (_ ("AAF: Could not create new region for clip '%1': Missing audio essence"), audioEssenceFile->unique_name) << endmsg;
					essenceError++;
					continue;
				}

				if (audioEssenceFile->is_embedded) {
					if (aafi_extractAudioEssenceFile (aafi, audioEssenceFile, AAFI_EXTRACT_DEFAULT, media_cache_path.c_str (), 0, 0, NULL, NULL) < 0) {
						PBD::error << string_compose ("AAF: Could not extract audio file '%1' from AAF.", audioEssenceFile->unique_name) << endmsg;
						essenceError++;
						continue;
					}
				} else if (!audioEssenceFile->is_embedded && !audioEssenceFile->usable_file_path) {
					PBD::error << string_compose ("AAF: Could not locate external audio file: '%1'", audioEssenceFile->original_file_path) << endmsg;
					essenceError++;
					continue;
				}
			}

			if (essenceError) {
				PBD::error << string_compose ("AAF: Error parsing audio essence pointerlist : %1\n", essenceName);
				continue;
			}

			if (!import_sndfile_as_region (_session, aafAudioClip->essencePointerList, SrcBest, pos, &oneClipSources, import_status, source_regions)) {
				PBD::error << string_compose ("AAF: Could not import '%1' to session.", essenceName) << endmsg;
				continue;
			} else {
				AAFI_foreachEssencePointer (aafAudioClip->essencePointerList, aafAudioEssencePtr)
				{
					if (aafAudioEssencePtr->essenceFile->is_embedded) {
						g_unlink (aafAudioEssencePtr->essenceFile->usable_file_path);
					}
				}
			}

			if (!oneClipSources || oneClipSources->size () == 0) {
				error << string_compose (_ ("AAF: Could not create new region for clip '%1': Region has no source"), essenceName) << endmsg;
				continue;
			}

			std::shared_ptr<Region> region = create_region (source_regions, aafAudioClip, *oneClipSources, sessionStart, samplerate_r);

			if (!region) {
				error << string_compose (_ ("AAF: Could not create new region for clip '%1'"), essenceName) << endmsg;
				continue;
			}

			/* converts whatever edit_rate clip is in, to samples */
			aafPosition_t clipPos = aafi_convertUnit (aafAudioClip->pos, aafAudioClip->track->edit_rate, &samplerate_r);

			track->playlist ()->add_region (region, timepos_t (clipPos + sessionStart));
			set_region_gain (aafAudioClip, region, _session);
			set_region_fade (aafAudioClip, region, &samplerate_r);
			if (aafAudioClip->mute) {
				region->set_muted (true);
			}
		}
	}

	// oneClipSources.clear ();

	aafiMarker* marker = NULL;

	AAFI_foreachMarker (aafi, marker)
	{
		aafPosition_t markerStart = sessionStart + aafi_convertUnit (marker->start, marker->edit_rate, &samplerate_r);
		aafPosition_t markerEnd   = sessionStart + aafi_convertUnit ((marker->start + marker->length), marker->edit_rate, &samplerate_r);

		Location* location;

		if (marker->length == 0) {
			location = new Location (*_session, timepos_t (markerStart), timepos_t (markerStart), marker->name, Location::Flags (Location::IsMark));
		} else {
			location = new Location (*_session, timepos_t (markerStart), timepos_t (markerEnd), marker->name, Location::Flags (Location::IsRangeMarker));
		}

		_session->locations ()->add (location, true);
	}

	/* set session range */
	aafRational_t nominal_sample_rate;
	nominal_sample_rate.numerator   = _session->nominal_sample_rate ();
	nominal_sample_rate.denominator = 1;

	samplepos_t start = samplepos_t (aafi_convertUnit (aafi->compositionStart, aafi->compositionStart_editRate, &nominal_sample_rate));
	samplepos_t end   = samplepos_t (aafi_convertUnit (aafi->compositionLength, aafi->compositionLength_editRate, &nominal_sample_rate)) + start;
	_session->maybe_update_session_range (timepos_t (start), timepos_t (end));

	/* set timecode */
	set_session_timecode (aafi, _session);

	the_editor ().access_action ("Editor", "zoom-to-session");

	/* Cleanup */
	import_status.progress = 1.0;
	import_status.done     = true;
	import_status.sources.clear ();
	import_status.all_done = true;

	_session->save_state ("");

	source_regions.clear ();

	PBD::remove_directory (media_cache_path);

	aafi_release (&aafi);

	if (!restore_backend.empty ()) {
		AudioEngine::instance ()->stop ();
		AudioEngine::instance ()->set_backend (restore_backend, "", "");
	}
	return 0;
}
