1/*
2 * Copyright (C) 2007 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package android.media;
18
19import android.content.ContentValues;
20import android.provider.MediaStore.Audio;
21import android.provider.MediaStore.Images;
22import android.provider.MediaStore.Video;
23import android.media.DecoderCapabilities;
24import android.media.DecoderCapabilities.VideoDecoder;
25import android.media.DecoderCapabilities.AudioDecoder;
26import android.mtp.MtpConstants;
27
28import java.util.HashMap;
29import java.util.Iterator;
30import java.util.List;
31
32/**
33 * MediaScanner helper class.
34 *
35 * {@hide}
36 */
37public class MediaFile {
38
39    // Audio file types
40    public static final int FILE_TYPE_MP3     = 1;
41    public static final int FILE_TYPE_M4A     = 2;
42    public static final int FILE_TYPE_WAV     = 3;
43    public static final int FILE_TYPE_AMR     = 4;
44    public static final int FILE_TYPE_AWB     = 5;
45    public static final int FILE_TYPE_WMA     = 6;
46    public static final int FILE_TYPE_OGG     = 7;
47    public static final int FILE_TYPE_AAC     = 8;
48    public static final int FILE_TYPE_MKA     = 9;
49    public static final int FILE_TYPE_FLAC    = 10;
50    private static final int FIRST_AUDIO_FILE_TYPE = FILE_TYPE_MP3;
51    private static final int LAST_AUDIO_FILE_TYPE = FILE_TYPE_FLAC;
52
53    // MIDI file types
54    public static final int FILE_TYPE_MID     = 11;
55    public static final int FILE_TYPE_SMF     = 12;
56    public static final int FILE_TYPE_IMY     = 13;
57    private static final int FIRST_MIDI_FILE_TYPE = FILE_TYPE_MID;
58    private static final int LAST_MIDI_FILE_TYPE = FILE_TYPE_IMY;
59
60    // Video file types
61    public static final int FILE_TYPE_MP4     = 21;
62    public static final int FILE_TYPE_M4V     = 22;
63    public static final int FILE_TYPE_3GPP    = 23;
64    public static final int FILE_TYPE_3GPP2   = 24;
65    public static final int FILE_TYPE_WMV     = 25;
66    public static final int FILE_TYPE_ASF     = 26;
67    public static final int FILE_TYPE_MKV     = 27;
68    public static final int FILE_TYPE_MP2TS   = 28;
69    public static final int FILE_TYPE_AVI     = 29;
70    public static final int FILE_TYPE_WEBM    = 30;
71    private static final int FIRST_VIDEO_FILE_TYPE = FILE_TYPE_MP4;
72    private static final int LAST_VIDEO_FILE_TYPE = FILE_TYPE_WEBM;
73
74    // More video file types
75    public static final int FILE_TYPE_MP2PS   = 200;
76    private static final int FIRST_VIDEO_FILE_TYPE2 = FILE_TYPE_MP2PS;
77    private static final int LAST_VIDEO_FILE_TYPE2 = FILE_TYPE_MP2PS;
78
79    // Image file types
80    public static final int FILE_TYPE_JPEG    = 31;
81    public static final int FILE_TYPE_GIF     = 32;
82    public static final int FILE_TYPE_PNG     = 33;
83    public static final int FILE_TYPE_BMP     = 34;
84    public static final int FILE_TYPE_WBMP    = 35;
85    public static final int FILE_TYPE_WEBP    = 36;
86    private static final int FIRST_IMAGE_FILE_TYPE = FILE_TYPE_JPEG;
87    private static final int LAST_IMAGE_FILE_TYPE = FILE_TYPE_WEBP;
88
89    // Playlist file types
90    public static final int FILE_TYPE_M3U      = 41;
91    public static final int FILE_TYPE_PLS      = 42;
92    public static final int FILE_TYPE_WPL      = 43;
93    public static final int FILE_TYPE_HTTPLIVE = 44;
94
95    private static final int FIRST_PLAYLIST_FILE_TYPE = FILE_TYPE_M3U;
96    private static final int LAST_PLAYLIST_FILE_TYPE = FILE_TYPE_HTTPLIVE;
97
98    // Drm file types
99    public static final int FILE_TYPE_FL      = 51;
100    private static final int FIRST_DRM_FILE_TYPE = FILE_TYPE_FL;
101    private static final int LAST_DRM_FILE_TYPE = FILE_TYPE_FL;
102
103    // Other popular file types
104    public static final int FILE_TYPE_TEXT          = 100;
105    public static final int FILE_TYPE_HTML          = 101;
106    public static final int FILE_TYPE_PDF           = 102;
107    public static final int FILE_TYPE_XML           = 103;
108    public static final int FILE_TYPE_MS_WORD       = 104;
109    public static final int FILE_TYPE_MS_EXCEL      = 105;
110    public static final int FILE_TYPE_MS_POWERPOINT = 106;
111    public static final int FILE_TYPE_ZIP           = 107;
112
113    public static class MediaFileType {
114        public final int fileType;
115        public final String mimeType;
116
117        MediaFileType(int fileType, String mimeType) {
118            this.fileType = fileType;
119            this.mimeType = mimeType;
120        }
121    }
122
123    private static final HashMap<String, MediaFileType> sFileTypeMap
124            = new HashMap<String, MediaFileType>();
125    private static final HashMap<String, Integer> sMimeTypeMap
126            = new HashMap<String, Integer>();
127    // maps file extension to MTP format code
128    private static final HashMap<String, Integer> sFileTypeToFormatMap
129            = new HashMap<String, Integer>();
130    // maps mime type to MTP format code
131    private static final HashMap<String, Integer> sMimeTypeToFormatMap
132            = new HashMap<String, Integer>();
133    // maps MTP format code to mime type
134    private static final HashMap<Integer, String> sFormatToMimeTypeMap
135            = new HashMap<Integer, String>();
136
137    static void addFileType(String extension, int fileType, String mimeType) {
138        sFileTypeMap.put(extension, new MediaFileType(fileType, mimeType));
139        sMimeTypeMap.put(mimeType, Integer.valueOf(fileType));
140    }
141
142    static void addFileType(String extension, int fileType, String mimeType, int mtpFormatCode) {
143        addFileType(extension, fileType, mimeType);
144        sFileTypeToFormatMap.put(extension, Integer.valueOf(mtpFormatCode));
145        sMimeTypeToFormatMap.put(mimeType, Integer.valueOf(mtpFormatCode));
146        sFormatToMimeTypeMap.put(mtpFormatCode, mimeType);
147    }
148
149    private static boolean isWMAEnabled() {
150        List<AudioDecoder> decoders = DecoderCapabilities.getAudioDecoders();
151        int count = decoders.size();
152        for (int i = 0; i < count; i++) {
153            AudioDecoder decoder = decoders.get(i);
154            if (decoder == AudioDecoder.AUDIO_DECODER_WMA) {
155                return true;
156            }
157        }
158        return false;
159    }
160
161    private static boolean isWMVEnabled() {
162        List<VideoDecoder> decoders = DecoderCapabilities.getVideoDecoders();
163        int count = decoders.size();
164        for (int i = 0; i < count; i++) {
165            VideoDecoder decoder = decoders.get(i);
166            if (decoder == VideoDecoder.VIDEO_DECODER_WMV) {
167                return true;
168            }
169        }
170        return false;
171    }
172
173    static {
174        addFileType("MP3", FILE_TYPE_MP3, "audio/mpeg", MtpConstants.FORMAT_MP3);
175        addFileType("MPGA", FILE_TYPE_MP3, "audio/mpeg", MtpConstants.FORMAT_MP3);
176        addFileType("M4A", FILE_TYPE_M4A, "audio/mp4", MtpConstants.FORMAT_MPEG);
177        addFileType("WAV", FILE_TYPE_WAV, "audio/x-wav", MtpConstants.FORMAT_WAV);
178        addFileType("AMR", FILE_TYPE_AMR, "audio/amr");
179        addFileType("AWB", FILE_TYPE_AWB, "audio/amr-wb");
180        if (isWMAEnabled()) {
181            addFileType("WMA", FILE_TYPE_WMA, "audio/x-ms-wma", MtpConstants.FORMAT_WMA);
182        }
183        addFileType("OGG", FILE_TYPE_OGG, "audio/ogg", MtpConstants.FORMAT_OGG);
184        addFileType("OGG", FILE_TYPE_OGG, "application/ogg", MtpConstants.FORMAT_OGG);
185        addFileType("OGA", FILE_TYPE_OGG, "application/ogg", MtpConstants.FORMAT_OGG);
186        addFileType("AAC", FILE_TYPE_AAC, "audio/aac", MtpConstants.FORMAT_AAC);
187        addFileType("AAC", FILE_TYPE_AAC, "audio/aac-adts", MtpConstants.FORMAT_AAC);
188        addFileType("MKA", FILE_TYPE_MKA, "audio/x-matroska");
189
190        addFileType("MID", FILE_TYPE_MID, "audio/midi");
191        addFileType("MIDI", FILE_TYPE_MID, "audio/midi");
192        addFileType("XMF", FILE_TYPE_MID, "audio/midi");
193        addFileType("RTTTL", FILE_TYPE_MID, "audio/midi");
194        addFileType("SMF", FILE_TYPE_SMF, "audio/sp-midi");
195        addFileType("IMY", FILE_TYPE_IMY, "audio/imelody");
196        addFileType("RTX", FILE_TYPE_MID, "audio/midi");
197        addFileType("OTA", FILE_TYPE_MID, "audio/midi");
198        addFileType("MXMF", FILE_TYPE_MID, "audio/midi");
199
200        addFileType("MPEG", FILE_TYPE_MP4, "video/mpeg", MtpConstants.FORMAT_MPEG);
201        addFileType("MPG", FILE_TYPE_MP4, "video/mpeg", MtpConstants.FORMAT_MPEG);
202        addFileType("MP4", FILE_TYPE_MP4, "video/mp4", MtpConstants.FORMAT_MPEG);
203        addFileType("M4V", FILE_TYPE_M4V, "video/mp4", MtpConstants.FORMAT_MPEG);
204        addFileType("3GP", FILE_TYPE_3GPP, "video/3gpp",  MtpConstants.FORMAT_3GP_CONTAINER);
205        addFileType("3GPP", FILE_TYPE_3GPP, "video/3gpp", MtpConstants.FORMAT_3GP_CONTAINER);
206        addFileType("3G2", FILE_TYPE_3GPP2, "video/3gpp2", MtpConstants.FORMAT_3GP_CONTAINER);
207        addFileType("3GPP2", FILE_TYPE_3GPP2, "video/3gpp2", MtpConstants.FORMAT_3GP_CONTAINER);
208        addFileType("MKV", FILE_TYPE_MKV, "video/x-matroska");
209        addFileType("WEBM", FILE_TYPE_WEBM, "video/webm");
210        addFileType("TS", FILE_TYPE_MP2TS, "video/mp2ts");
211        addFileType("AVI", FILE_TYPE_AVI, "video/avi");
212
213        if (isWMVEnabled()) {
214            addFileType("WMV", FILE_TYPE_WMV, "video/x-ms-wmv", MtpConstants.FORMAT_WMV);
215            addFileType("ASF", FILE_TYPE_ASF, "video/x-ms-asf");
216        }
217
218        addFileType("JPG", FILE_TYPE_JPEG, "image/jpeg", MtpConstants.FORMAT_EXIF_JPEG);
219        addFileType("JPEG", FILE_TYPE_JPEG, "image/jpeg", MtpConstants.FORMAT_EXIF_JPEG);
220        addFileType("GIF", FILE_TYPE_GIF, "image/gif", MtpConstants.FORMAT_GIF);
221        addFileType("PNG", FILE_TYPE_PNG, "image/png", MtpConstants.FORMAT_PNG);
222        addFileType("BMP", FILE_TYPE_BMP, "image/x-ms-bmp", MtpConstants.FORMAT_BMP);
223        addFileType("WBMP", FILE_TYPE_WBMP, "image/vnd.wap.wbmp");
224        addFileType("WEBP", FILE_TYPE_WEBP, "image/webp");
225
226        addFileType("M3U", FILE_TYPE_M3U, "audio/x-mpegurl", MtpConstants.FORMAT_M3U_PLAYLIST);
227        addFileType("M3U", FILE_TYPE_M3U, "application/x-mpegurl", MtpConstants.FORMAT_M3U_PLAYLIST);
228        addFileType("PLS", FILE_TYPE_PLS, "audio/x-scpls", MtpConstants.FORMAT_PLS_PLAYLIST);
229        addFileType("WPL", FILE_TYPE_WPL, "application/vnd.ms-wpl", MtpConstants.FORMAT_WPL_PLAYLIST);
230        addFileType("M3U8", FILE_TYPE_HTTPLIVE, "application/vnd.apple.mpegurl");
231        addFileType("M3U8", FILE_TYPE_HTTPLIVE, "audio/mpegurl");
232        addFileType("M3U8", FILE_TYPE_HTTPLIVE, "audio/x-mpegurl");
233
234        addFileType("FL", FILE_TYPE_FL, "application/x-android-drm-fl");
235
236        addFileType("TXT", FILE_TYPE_TEXT, "text/plain", MtpConstants.FORMAT_TEXT);
237        addFileType("HTM", FILE_TYPE_HTML, "text/html", MtpConstants.FORMAT_HTML);
238        addFileType("HTML", FILE_TYPE_HTML, "text/html", MtpConstants.FORMAT_HTML);
239        addFileType("PDF", FILE_TYPE_PDF, "application/pdf");
240        addFileType("DOC", FILE_TYPE_MS_WORD, "application/msword", MtpConstants.FORMAT_MS_WORD_DOCUMENT);
241        addFileType("XLS", FILE_TYPE_MS_EXCEL, "application/vnd.ms-excel", MtpConstants.FORMAT_MS_EXCEL_SPREADSHEET);
242        addFileType("PPT", FILE_TYPE_MS_POWERPOINT, "application/mspowerpoint", MtpConstants.FORMAT_MS_POWERPOINT_PRESENTATION);
243        addFileType("FLAC", FILE_TYPE_FLAC, "audio/flac", MtpConstants.FORMAT_FLAC);
244        addFileType("ZIP", FILE_TYPE_ZIP, "application/zip");
245        addFileType("MPG", FILE_TYPE_MP2PS, "video/mp2p");
246        addFileType("MPEG", FILE_TYPE_MP2PS, "video/mp2p");
247    }
248
249    public static boolean isAudioFileType(int fileType) {
250        return ((fileType >= FIRST_AUDIO_FILE_TYPE &&
251                fileType <= LAST_AUDIO_FILE_TYPE) ||
252                (fileType >= FIRST_MIDI_FILE_TYPE &&
253                fileType <= LAST_MIDI_FILE_TYPE));
254    }
255
256    public static boolean isVideoFileType(int fileType) {
257        return (fileType >= FIRST_VIDEO_FILE_TYPE &&
258                fileType <= LAST_VIDEO_FILE_TYPE)
259            || (fileType >= FIRST_VIDEO_FILE_TYPE2 &&
260                fileType <= LAST_VIDEO_FILE_TYPE2);
261    }
262
263    public static boolean isImageFileType(int fileType) {
264        return (fileType >= FIRST_IMAGE_FILE_TYPE &&
265                fileType <= LAST_IMAGE_FILE_TYPE);
266    }
267
268    public static boolean isPlayListFileType(int fileType) {
269        return (fileType >= FIRST_PLAYLIST_FILE_TYPE &&
270                fileType <= LAST_PLAYLIST_FILE_TYPE);
271    }
272
273    public static boolean isDrmFileType(int fileType) {
274        return (fileType >= FIRST_DRM_FILE_TYPE &&
275                fileType <= LAST_DRM_FILE_TYPE);
276    }
277
278    public static MediaFileType getFileType(String path) {
279        int lastDot = path.lastIndexOf(".");
280        if (lastDot < 0)
281            return null;
282        return sFileTypeMap.get(path.substring(lastDot + 1).toUpperCase());
283    }
284
285    public static boolean isMimeTypeMedia(String mimeType) {
286        int fileType = getFileTypeForMimeType(mimeType);
287        return isAudioFileType(fileType) || isVideoFileType(fileType)
288                || isImageFileType(fileType) || isPlayListFileType(fileType);
289    }
290
291    // generates a title based on file name
292    public static String getFileTitle(String path) {
293        // extract file name after last slash
294        int lastSlash = path.lastIndexOf('/');
295        if (lastSlash >= 0) {
296            lastSlash++;
297            if (lastSlash < path.length()) {
298                path = path.substring(lastSlash);
299            }
300        }
301        // truncate the file extension (if any)
302        int lastDot = path.lastIndexOf('.');
303        if (lastDot > 0) {
304            path = path.substring(0, lastDot);
305        }
306        return path;
307    }
308
309    public static int getFileTypeForMimeType(String mimeType) {
310        Integer value = sMimeTypeMap.get(mimeType);
311        return (value == null ? 0 : value.intValue());
312    }
313
314    public static String getMimeTypeForFile(String path) {
315        MediaFileType mediaFileType = getFileType(path);
316        return (mediaFileType == null ? null : mediaFileType.mimeType);
317    }
318
319    public static int getFormatCode(String fileName, String mimeType) {
320        if (mimeType != null) {
321            Integer value = sMimeTypeToFormatMap.get(mimeType);
322            if (value != null) {
323                return value.intValue();
324            }
325        }
326        int lastDot = fileName.lastIndexOf('.');
327        if (lastDot > 0) {
328            String extension = fileName.substring(lastDot + 1).toUpperCase();
329            Integer value = sFileTypeToFormatMap.get(extension);
330            if (value != null) {
331                return value.intValue();
332            }
333        }
334        return MtpConstants.FORMAT_UNDEFINED;
335    }
336
337    public static String getMimeTypeForFormatCode(int formatCode) {
338        return sFormatToMimeTypeMap.get(formatCode);
339    }
340}
341