X-Git-Url: http://gitweb.fperrin.net/?p=GpsPrune.git;a=blobdiff_plain;f=src%2Ftim%2Fprune%2Fload%2FJpegLoader.java;fp=src%2Ftim%2Fprune%2Fload%2FJpegLoader.java;h=3e22eba029ccb76ce341a2fffaea6327669a28f1;hp=0000000000000000000000000000000000000000;hb=ce6f2161b8596f7018d6a76bff79bc9e571f35fd;hpb=2d8cb72e84d5cc1089ce77baf1e34ea3ea2f8465

diff --git a/src/tim/prune/load/JpegLoader.java b/src/tim/prune/load/JpegLoader.java
new file mode 100644
index 0000000..3e22eba
--- /dev/null
+++ b/src/tim/prune/load/JpegLoader.java
@@ -0,0 +1,384 @@
+package tim.prune.load;
+
+import java.io.File;
+import java.util.TreeSet;
+
+import javax.swing.BoxLayout;
+import javax.swing.JCheckBox;
+import javax.swing.JFileChooser;
+import javax.swing.JFrame;
+import javax.swing.JPanel;
+
+import tim.prune.App;
+import tim.prune.I18nManager;
+import tim.prune.config.Config;
+import tim.prune.data.Altitude;
+import tim.prune.data.DataPoint;
+import tim.prune.data.LatLonRectangle;
+import tim.prune.data.Latitude;
+import tim.prune.data.Longitude;
+import tim.prune.data.Photo;
+import tim.prune.data.Timestamp;
+import tim.prune.data.TimestampLocal;
+import tim.prune.data.TimestampUtc;
+import tim.prune.data.UnitSetLibrary;
+import tim.prune.function.Cancellable;
+import tim.prune.jpeg.InternalExifLibrary;
+import tim.prune.jpeg.JpegData;
+
+/**
+ * Class to manage the loading of Jpegs and dealing with the GPS data from them
+ */
+public class JpegLoader implements Runnable, Cancellable
+{
+	private App _app = null;
+	private JFrame _parentFrame = null;
+	private JFileChooser _fileChooser = null;
+	private GenericFileFilter _fileFilter = null;
+	private JCheckBox _subdirCheckbox = null;
+	private JCheckBox _noExifCheckbox = null;
+	private JCheckBox _outsideAreaCheckbox = null;
+	private MediaLoadProgressDialog _progressDialog = null;
+	private int[] _fileCounts = null;
+	private boolean _cancelled = false;
+	private LatLonRectangle _trackRectangle = null;
+	private TreeSet<Photo> _photos = null;
+
+
+	/**
+	 * Constructor
+	 * @param inApp Application object to inform of photo load
+	 * @param inParentFrame parent frame to reference for dialogs
+	 */
+	public JpegLoader(App inApp, JFrame inParentFrame)
+	{
+		_app = inApp;
+		_parentFrame = inParentFrame;
+		_fileFilter = new JpegFileFilter();
+	}
+
+
+	/**
+	 * Open the GUI to select options and start the load
+	 * @param inRectangle track rectangle
+	 */
+	public void openDialog(LatLonRectangle inRectangle)
+	{
+		// Create file chooser if necessary
+		if (_fileChooser == null)
+		{
+			_fileChooser = new JFileChooser();
+			_fileChooser.setMultiSelectionEnabled(true);
+			_fileChooser.setFileSelectionMode(JFileChooser.FILES_AND_DIRECTORIES);
+			_fileChooser.setFileFilter(_fileFilter);
+			_fileChooser.setDialogTitle(I18nManager.getText("menu.file.addphotos"));
+			_subdirCheckbox = new JCheckBox(I18nManager.getText("dialog.jpegload.subdirectories"));
+			_subdirCheckbox.setSelected(true);
+			_noExifCheckbox = new JCheckBox(I18nManager.getText("dialog.jpegload.loadjpegswithoutcoords"));
+			_noExifCheckbox.setSelected(true);
+			_outsideAreaCheckbox = new JCheckBox(I18nManager.getText("dialog.jpegload.loadjpegsoutsidearea"));
+			_outsideAreaCheckbox.setSelected(true);
+			JPanel panel = new JPanel();
+			panel.setLayout(new BoxLayout(panel, BoxLayout.Y_AXIS));
+			panel.add(_subdirCheckbox);
+			panel.add(_noExifCheckbox);
+			panel.add(_outsideAreaCheckbox);
+			_fileChooser.setAccessory(panel);
+			// start from directory in config if already set by other operations
+			String configDir = Config.getConfigString(Config.KEY_PHOTO_DIR);
+			if (configDir == null) {configDir = Config.getConfigString(Config.KEY_TRACK_DIR);}
+			if (configDir != null) {_fileChooser.setCurrentDirectory(new File(configDir));}
+		}
+		// enable/disable track checkbox
+		_trackRectangle = inRectangle;
+		_outsideAreaCheckbox.setEnabled(_trackRectangle != null && !_trackRectangle.isEmpty());
+		// Show file dialog to choose file / directory(ies)
+		if (_fileChooser.showOpenDialog(_parentFrame) == JFileChooser.APPROVE_OPTION)
+		{
+			// Bring up dialog before starting
+			_progressDialog = new MediaLoadProgressDialog(_parentFrame, this);
+			_progressDialog.show();
+			// start thread for processing
+			new Thread(this).start();
+		}
+	}
+
+	/** Cancel */
+	public void cancel() {
+		_cancelled = true;
+	}
+
+
+	/**
+	 * Run method for performing tasks in separate thread
+	 */
+	public void run()
+	{
+		// Initialise arrays, errors, summaries
+		_fileCounts = new int[3]; // files, jpegs, gps
+		_photos = new TreeSet<Photo>(new MediaSorter());
+		File[] files = _fileChooser.getSelectedFiles();
+		// Loop recursively over selected files/directories to count files
+		int numFiles = countFileList(files, true, _subdirCheckbox.isSelected());
+		// Set up the progress bar for this number of files
+		_progressDialog.showProgress(0, numFiles);
+		_cancelled = false;
+
+		// Process the files recursively and build lists of photos
+		processFileList(files, true, _subdirCheckbox.isSelected());
+		_progressDialog.close();
+		if (_cancelled) {return;}
+
+		if (_fileCounts[0] == 0)
+		{
+			// No files found at all
+			_app.showErrorMessage("error.jpegload.dialogtitle", "error.jpegload.nofilesfound");
+		}
+		else if (_fileCounts[1] == 0)
+		{
+			// No jpegs found
+			_app.showErrorMessage("error.jpegload.dialogtitle", "error.jpegload.nojpegsfound");
+		}
+		else if (!_noExifCheckbox.isSelected() && _fileCounts[2] == 0)
+		{
+			// Need coordinates but no gps information found
+			_app.showErrorMessage("error.jpegload.dialogtitle", "error.jpegload.nogpsfound");
+		}
+		else
+		{
+			// Found some photos to load - pass information back to app
+			_app.informPhotosLoaded(_photos);
+		}
+	}
+
+
+	/**
+	 * Process a list of files and/or directories
+	 * @param inFiles array of file/directories
+	 * @param inFirstDir true if first directory
+	 * @param inDescend true to descend to subdirectories
+	 */
+	private void processFileList(File[] inFiles, boolean inFirstDir, boolean inDescend)
+	{
+		if (inFiles == null) return;
+		// Loop over elements in array
+		for (int i=0; i<inFiles.length && !_cancelled; i++)
+		{
+			File file = inFiles[i];
+			if (file.exists() && file.canRead())
+			{
+				// Check whether it's a file or a directory
+				if (file.isFile())
+				{
+					processFile(file);
+				}
+				else if (file.isDirectory() && (inFirstDir || inDescend))
+				{
+					// Always process first directory,
+					// only process subdirectories if checkbox selected
+					File[] files = file.listFiles();
+					processFileList(files, false, inDescend);
+				}
+			}
+			// if file doesn't exist or isn't readable - ignore
+		}
+	}
+
+
+	/**
+	 * Process the given file, by attempting to extract its tags
+	 * @param inFile file object to read
+	 */
+	private void processFile(File inFile)
+	{
+		// Update progress bar
+		_fileCounts[0]++; // file found
+		_progressDialog.showProgress(_fileCounts[0], -1);
+
+		// Check whether filename corresponds with accepted filenames
+		if (!_fileFilter.acceptFilename(inFile.getName())) {return;}
+		// If it's a Jpeg, we can use ExifReader to get coords, otherwise we could try exiftool (if it's installed)
+
+		if (inFile.exists() && inFile.canRead()) {
+			_fileCounts[1]++; // jpeg found
+		}
+		Photo photo = createPhoto(inFile);
+		if (photo.getDataPoint() != null) {
+			_fileCounts[2]++; // photo has coordinates
+		}
+		// Check the criteria for adding the photo - check whether the photo has coordinates and if so if they're within the rectangle
+		if ( (photo.getDataPoint() != null || _noExifCheckbox.isSelected())
+			&& (photo.getDataPoint() == null || !_outsideAreaCheckbox.isEnabled()
+				|| _outsideAreaCheckbox.isSelected() || _trackRectangle.containsPoint(photo.getDataPoint())))
+		{
+			_photos.add(photo);
+		}
+	}
+
+	/**
+	 * Create a Photo object for the given file, including reading exif information
+	 * @param inFile file object
+	 * @return Photo object
+	 */
+	public static Photo createPhoto(File inFile)
+	{
+		// Create Photo object
+		Photo photo = new Photo(inFile);
+		// Try to get information out of exif
+		JpegData jpegData = new InternalExifLibrary().getJpegData(inFile);
+		Timestamp timestamp = null;
+		if (jpegData != null)
+		{
+			if (jpegData.isGpsValid())
+			{
+				timestamp = createTimestamp(jpegData.getGpsDatestamp(), jpegData.getGpsTimestamp());
+				// Make DataPoint and attach to Photo
+				DataPoint point = createDataPoint(jpegData);
+				point.setPhoto(photo);
+				point.setSegmentStart(true);
+				photo.setDataPoint(point);
+				photo.setOriginalStatus(Photo.Status.TAGGED);
+			}
+			// Use exif timestamp if gps timestamp not available
+			if (timestamp == null && jpegData.getOriginalTimestamp() != null) {
+				timestamp = createTimestamp(jpegData.getOriginalTimestamp());
+			}
+			if (timestamp == null && jpegData.getDigitizedTimestamp() != null) {
+				timestamp = createTimestamp(jpegData.getDigitizedTimestamp());
+			}
+			photo.setExifThumbnail(jpegData.getThumbnailImage());
+			// Also extract orientation tag for setting rotation state of photo
+			photo.setRotation(jpegData.getRequiredRotation());
+			// Set bearing, if any
+			photo.setBearing(jpegData.getBearing());
+		}
+		// Use file timestamp if exif timestamp isn't available
+		if (timestamp == null) {
+			timestamp = new TimestampUtc(inFile.lastModified());
+		}
+		// Apply timestamp to photo and its point (if any)
+		photo.setTimestamp(timestamp);
+		if (photo.getDataPoint() != null) {
+			// photo.getDataPoint().setFieldValue(Field.TIMESTAMP, timestamp.getText(Timestamp.Format.ISO8601), false);
+		}
+		return photo;
+	}
+
+
+	/**
+	 * Recursively count the selected Files so we can draw a progress bar
+	 * @param inFiles file list
+	 * @param inFirstDir true if first directory
+	 * @param inDescend true to descend to subdirectories
+	 * @return count of the files selected
+	 */
+	private int countFileList(File[] inFiles, boolean inFirstDir, boolean inDescend)
+	{
+		int fileCount = 0;
+		if (inFiles != null)
+		{
+			// Loop over elements in array
+			for (int i=0; i<inFiles.length; i++)
+			{
+				File file = inFiles[i];
+				if (file.exists() && file.canRead())
+				{
+					// Store first directory in config for later
+					if (i == 0 && inFirstDir) {
+						File workingDir = file.isDirectory()?file:file.getParentFile();
+						Config.setConfigString(Config.KEY_PHOTO_DIR, workingDir.getAbsolutePath());
+					}
+					// Check whether it's a file or a directory
+					if (file.isFile())
+					{
+						fileCount++;
+					}
+					else if (file.isDirectory() && (inFirstDir || inDescend))
+					{
+						fileCount += countFileList(file.listFiles(), false, inDescend);
+					}
+				}
+			}
+		}
+		return fileCount;
+	}
+
+
+	/**
+	 * Create a DataPoint object from the given jpeg data
+	 * @param inData Jpeg data including coordinates
+	 * @return DataPoint object for Track
+	 */
+	private static DataPoint createDataPoint(JpegData inData)
+	{
+		// Create model objects from jpeg data
+		double latval = getCoordinateDoubleValue(inData.getLatitude(),
+			inData.getLatitudeRef() == 'N' || inData.getLatitudeRef() == 'n');
+		Latitude latitude = new Latitude(latval, Latitude.FORMAT_DEG_MIN_SEC);
+		double lonval = getCoordinateDoubleValue(inData.getLongitude(),
+			inData.getLongitudeRef() == 'E' || inData.getLongitudeRef() == 'e');
+		Longitude longitude = new Longitude(lonval, Longitude.FORMAT_DEG_MIN_SEC);
+		Altitude altitude = null;
+		if (inData.hasAltitude()) {
+			altitude = new Altitude(inData.getAltitude(), UnitSetLibrary.UNITS_METRES);
+		}
+		return new DataPoint(latitude, longitude, altitude);
+	}
+
+
+	/**
+	 * Convert an array of 3 doubles (deg-min-sec) into a double coordinate value
+	 * @param inValues array of three doubles for deg-min-sec
+	 * @param isPositive true for positive hemisphere, for positive double value
+	 * @return double value of coordinate, either positive or negative
+	 */
+	private static double getCoordinateDoubleValue(double[] inValues, boolean isPositive)
+	{
+		if (inValues == null || inValues.length != 3) return 0.0;
+		double value = inValues[0]        // degrees
+			+ inValues[1] / 60.0          // minutes
+			+ inValues[2] / 60.0 / 60.0;  // seconds
+		// make sure it's the correct sign
+		value = Math.abs(value);
+		if (!isPositive) value = -value;
+		return value;
+	}
+
+
+	/**
+	 * Use the given int values to create a timestamp
+	 * @param inDate ints describing date
+	 * @param inTime ints describing time
+	 * @return Timestamp object corresponding to inputs
+	 */
+	private static Timestamp createTimestamp(int[] inDate, int[] inTime)
+	{
+		if (inDate == null || inTime == null || inDate.length != 3 || inTime.length != 3) {
+			return null;
+		}
+		return new TimestampLocal(inDate[0], inDate[1], inDate[2],
+			inTime[0], inTime[1], inTime[2]);
+	}
+
+
+	/**
+	 * Use the given String value to create a timestamp
+	 * @param inStamp timestamp from exif
+	 * @return Timestamp object corresponding to input
+	 */
+	private static Timestamp createTimestamp(String inStamp)
+	{
+		Timestamp stamp = null;
+		try
+		{
+			stamp = new TimestampLocal(Integer.parseInt(inStamp.substring(0, 4)),
+				Integer.parseInt(inStamp.substring(5, 7)),
+				Integer.parseInt(inStamp.substring(8, 10)),
+				Integer.parseInt(inStamp.substring(11, 13)),
+				Integer.parseInt(inStamp.substring(14, 16)),
+				Integer.parseInt(inStamp.substring(17)));
+		}
+		catch (NumberFormatException nfe) {}
+		return stamp;
+	}
+}