| /* |
| * Copyright (C) 2010 Bluecherry, LLC www.bluecherrydvr.com |
| * Copyright (C) 2010 Ben Collins <bcollins@bluecherry.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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
| */ |
| |
| #include <linux/kernel.h> |
| #include <linux/module.h> |
| #include <linux/kthread.h> |
| #include <linux/freezer.h> |
| #include <media/v4l2-ioctl.h> |
| #include <media/v4l2-common.h> |
| #include <media/videobuf-dma-sg.h> |
| #include "solo6x10.h" |
| #include "tw28.h" |
| #include "jpeg.h" |
| |
| #define MIN_VID_BUFFERS 4 |
| #define FRAME_BUF_SIZE (128 * 1024) |
| #define MP4_QS 16 |
| |
| static int solo_enc_thread(void *data); |
| |
| extern unsigned video_nr; |
| |
| struct solo_enc_fh { |
| struct solo_enc_dev *enc; |
| u32 fmt; |
| u16 rd_idx; |
| u8 enc_on; |
| enum solo_enc_types type; |
| struct videobuf_queue vidq; |
| struct list_head vidq_active; |
| struct task_struct *kthread; |
| struct p2m_desc desc[SOLO_NR_P2M_DESC]; |
| }; |
| |
| static const u32 solo_user_ctrls[] = { |
| V4L2_CID_BRIGHTNESS, |
| V4L2_CID_CONTRAST, |
| V4L2_CID_SATURATION, |
| V4L2_CID_HUE, |
| V4L2_CID_SHARPNESS, |
| 0 |
| }; |
| |
| static const u32 solo_mpeg_ctrls[] = { |
| V4L2_CID_MPEG_VIDEO_ENCODING, |
| V4L2_CID_MPEG_VIDEO_GOP_SIZE, |
| 0 |
| }; |
| |
| static const u32 solo_private_ctrls[] = { |
| V4L2_CID_MOTION_ENABLE, |
| V4L2_CID_MOTION_THRESHOLD, |
| 0 |
| }; |
| |
| static const u32 solo_fmtx_ctrls[] = { |
| V4L2_CID_RDS_TX_RADIO_TEXT, |
| 0 |
| }; |
| |
| static const u32 *solo_ctrl_classes[] = { |
| solo_user_ctrls, |
| solo_mpeg_ctrls, |
| solo_fmtx_ctrls, |
| solo_private_ctrls, |
| NULL |
| }; |
| |
| static int solo_is_motion_on(struct solo_enc_dev *solo_enc) |
| { |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| u8 ch = solo_enc->ch; |
| |
| if (solo_dev->motion_mask & (1 << ch)) |
| return 1; |
| return 0; |
| } |
| |
| static void solo_motion_toggle(struct solo_enc_dev *solo_enc, int on) |
| { |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| u8 ch = solo_enc->ch; |
| |
| spin_lock(&solo_enc->lock); |
| |
| if (on) |
| solo_dev->motion_mask |= (1 << ch); |
| else |
| solo_dev->motion_mask &= ~(1 << ch); |
| |
| /* Do this regardless of if we are turning on or off */ |
| solo_reg_write(solo_enc->solo_dev, SOLO_VI_MOT_CLEAR, |
| 1 << solo_enc->ch); |
| solo_enc->motion_detected = 0; |
| |
| solo_reg_write(solo_dev, SOLO_VI_MOT_ADR, |
| SOLO_VI_MOTION_EN(solo_dev->motion_mask) | |
| (SOLO_MOTION_EXT_ADDR(solo_dev) >> 16)); |
| |
| if (solo_dev->motion_mask) |
| solo_irq_on(solo_dev, SOLO_IRQ_MOTION); |
| else |
| solo_irq_off(solo_dev, SOLO_IRQ_MOTION); |
| |
| spin_unlock(&solo_enc->lock); |
| } |
| |
| /* Should be called with solo_enc->lock held */ |
| static void solo_update_mode(struct solo_enc_dev *solo_enc) |
| { |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| assert_spin_locked(&solo_enc->lock); |
| |
| solo_enc->interlaced = (solo_enc->mode & 0x08) ? 1 : 0; |
| solo_enc->bw_weight = max(solo_dev->fps / solo_enc->interval, 1); |
| |
| switch (solo_enc->mode) { |
| case SOLO_ENC_MODE_CIF: |
| solo_enc->width = solo_dev->video_hsize >> 1; |
| solo_enc->height = solo_dev->video_vsize; |
| break; |
| case SOLO_ENC_MODE_D1: |
| solo_enc->width = solo_dev->video_hsize; |
| solo_enc->height = solo_dev->video_vsize << 1; |
| solo_enc->bw_weight <<= 2; |
| break; |
| default: |
| WARN(1, "mode is unknown\n"); |
| } |
| } |
| |
| /* Should be called with solo_enc->lock held */ |
| static int solo_enc_on(struct solo_enc_fh *fh) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| u8 ch = solo_enc->ch; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| u8 interval; |
| |
| assert_spin_locked(&solo_enc->lock); |
| |
| if (fh->enc_on) |
| return 0; |
| |
| solo_update_mode(solo_enc); |
| |
| /* Make sure to bw check on first reader */ |
| if (!atomic_read(&solo_enc->readers)) { |
| if (solo_enc->bw_weight > solo_dev->enc_bw_remain) |
| return -EBUSY; |
| else |
| solo_dev->enc_bw_remain -= solo_enc->bw_weight; |
| } |
| |
| fh->enc_on = 1; |
| fh->rd_idx = solo_enc->solo_dev->enc_wr_idx; |
| |
| if (fh->type == SOLO_ENC_TYPE_EXT) |
| solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(ch), 1); |
| |
| if (atomic_inc_return(&solo_enc->readers) > 1) |
| return 0; |
| |
| /* Disable all encoding for this channel */ |
| solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(ch), 0); |
| |
| /* Common for both std and ext encoding */ |
| solo_reg_write(solo_dev, SOLO_VE_CH_INTL(ch), |
| solo_enc->interlaced ? 1 : 0); |
| |
| if (solo_enc->interlaced) |
| interval = solo_enc->interval - 1; |
| else |
| interval = solo_enc->interval; |
| |
| /* Standard encoding only */ |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP(ch), solo_enc->gop); |
| solo_reg_write(solo_dev, SOLO_VE_CH_QP(ch), solo_enc->qp); |
| solo_reg_write(solo_dev, SOLO_CAP_CH_INTV(ch), interval); |
| |
| /* Extended encoding only */ |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP_E(ch), solo_enc->gop); |
| solo_reg_write(solo_dev, SOLO_VE_CH_QP_E(ch), solo_enc->qp); |
| solo_reg_write(solo_dev, SOLO_CAP_CH_INTV_E(ch), interval); |
| |
| /* Enables the standard encoder */ |
| solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(ch), solo_enc->mode); |
| |
| /* Settle down Beavis... */ |
| mdelay(10); |
| |
| return 0; |
| } |
| |
| static void solo_enc_off(struct solo_enc_fh *fh) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| if (!fh->enc_on) |
| return; |
| |
| if (fh->kthread) { |
| kthread_stop(fh->kthread); |
| fh->kthread = NULL; |
| } |
| |
| solo_dev->enc_bw_remain += solo_enc->bw_weight; |
| fh->enc_on = 0; |
| |
| if (atomic_dec_return(&solo_enc->readers) > 0) |
| return; |
| |
| solo_reg_write(solo_dev, SOLO_CAP_CH_SCALE(solo_enc->ch), 0); |
| solo_reg_write(solo_dev, SOLO_CAP_CH_COMP_ENA_E(solo_enc->ch), 0); |
| } |
| |
| static int solo_start_fh_thread(struct solo_enc_fh *fh) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| |
| fh->kthread = kthread_run(solo_enc_thread, fh, SOLO6X10_NAME "_enc"); |
| |
| /* Oops, we had a problem */ |
| if (IS_ERR(fh->kthread)) { |
| spin_lock(&solo_enc->lock); |
| solo_enc_off(fh); |
| spin_unlock(&solo_enc->lock); |
| |
| return PTR_ERR(fh->kthread); |
| } |
| |
| return 0; |
| } |
| |
| static void enc_reset_gop(struct solo_dev *solo_dev, u8 ch) |
| { |
| BUG_ON(ch >= solo_dev->nr_chans); |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP(ch), 1); |
| solo_dev->v4l2_enc[ch]->reset_gop = 1; |
| } |
| |
| static int enc_gop_reset(struct solo_dev *solo_dev, u8 ch, u8 vop) |
| { |
| BUG_ON(ch >= solo_dev->nr_chans); |
| if (!solo_dev->v4l2_enc[ch]->reset_gop) |
| return 0; |
| if (vop) |
| return 1; |
| solo_dev->v4l2_enc[ch]->reset_gop = 0; |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP(ch), |
| solo_dev->v4l2_enc[ch]->gop); |
| return 0; |
| } |
| |
| static void enc_write_sg(struct scatterlist *sglist, void *buf, int size) |
| { |
| struct scatterlist *sg; |
| u8 *src = buf; |
| |
| for (sg = sglist; sg && size > 0; sg = sg_next(sg)) { |
| u8 *p = sg_virt(sg); |
| size_t len = sg_dma_len(sg); |
| int i; |
| |
| for (i = 0; i < len && size; i++) |
| p[i] = *(src++); |
| } |
| } |
| |
| static int enc_get_mpeg_dma_sg(struct solo_dev *solo_dev, |
| struct p2m_desc *desc, |
| struct scatterlist *sglist, int skip, |
| unsigned int off, unsigned int size) |
| { |
| int ret; |
| |
| if (off > SOLO_MP4E_EXT_SIZE(solo_dev)) |
| return -EINVAL; |
| |
| if (off + size <= SOLO_MP4E_EXT_SIZE(solo_dev)) { |
| return solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_MP4E, |
| desc, 0, sglist, skip, |
| SOLO_MP4E_EXT_ADDR(solo_dev) + off, size); |
| } |
| |
| /* Buffer wrap */ |
| ret = solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_MP4E, desc, 0, |
| sglist, skip, SOLO_MP4E_EXT_ADDR(solo_dev) + off, |
| SOLO_MP4E_EXT_SIZE(solo_dev) - off); |
| |
| ret |= solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_MP4E, desc, 0, |
| sglist, skip + SOLO_MP4E_EXT_SIZE(solo_dev) - off, |
| SOLO_MP4E_EXT_ADDR(solo_dev), |
| size + off - SOLO_MP4E_EXT_SIZE(solo_dev)); |
| |
| return ret; |
| } |
| |
| static int enc_get_mpeg_dma_t(struct solo_dev *solo_dev, |
| dma_addr_t buf, unsigned int off, |
| unsigned int size) |
| { |
| int ret; |
| |
| if (off > SOLO_MP4E_EXT_SIZE(solo_dev)) |
| return -EINVAL; |
| |
| if (off + size <= SOLO_MP4E_EXT_SIZE(solo_dev)) { |
| return solo_p2m_dma_t(solo_dev, SOLO_P2M_DMA_ID_MP4E, 0, buf, |
| SOLO_MP4E_EXT_ADDR(solo_dev) + off, size); |
| } |
| |
| /* Buffer wrap */ |
| ret = solo_p2m_dma_t(solo_dev, SOLO_P2M_DMA_ID_MP4E, 0, buf, |
| SOLO_MP4E_EXT_ADDR(solo_dev) + off, |
| SOLO_MP4E_EXT_SIZE(solo_dev) - off); |
| |
| ret |= solo_p2m_dma_t(solo_dev, SOLO_P2M_DMA_ID_MP4E, 0, |
| buf + SOLO_MP4E_EXT_SIZE(solo_dev) - off, |
| SOLO_MP4E_EXT_ADDR(solo_dev), |
| size + off - SOLO_MP4E_EXT_SIZE(solo_dev)); |
| |
| return ret; |
| } |
| |
| static int enc_get_mpeg_dma(struct solo_dev *solo_dev, void *buf, |
| unsigned int off, unsigned int size) |
| { |
| int ret; |
| |
| dma_addr_t dma_addr = pci_map_single(solo_dev->pdev, buf, size, |
| PCI_DMA_FROMDEVICE); |
| ret = enc_get_mpeg_dma_t(solo_dev, dma_addr, off, size); |
| pci_unmap_single(solo_dev->pdev, dma_addr, size, PCI_DMA_FROMDEVICE); |
| |
| return ret; |
| } |
| |
| static int enc_get_jpeg_dma_sg(struct solo_dev *solo_dev, |
| struct p2m_desc *desc, |
| struct scatterlist *sglist, int skip, |
| unsigned int off, unsigned int size) |
| { |
| int ret; |
| |
| if (off > SOLO_JPEG_EXT_SIZE(solo_dev)) |
| return -EINVAL; |
| |
| if (off + size <= SOLO_JPEG_EXT_SIZE(solo_dev)) { |
| return solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_JPEG, |
| desc, 0, sglist, skip, |
| SOLO_JPEG_EXT_ADDR(solo_dev) + off, size); |
| } |
| |
| /* Buffer wrap */ |
| ret = solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_JPEG, desc, 0, |
| sglist, skip, SOLO_JPEG_EXT_ADDR(solo_dev) + off, |
| SOLO_JPEG_EXT_SIZE(solo_dev) - off); |
| |
| ret |= solo_p2m_dma_sg(solo_dev, SOLO_P2M_DMA_ID_JPEG, desc, 0, |
| sglist, skip + SOLO_JPEG_EXT_SIZE(solo_dev) - off, |
| SOLO_JPEG_EXT_ADDR(solo_dev), |
| size + off - SOLO_JPEG_EXT_SIZE(solo_dev)); |
| |
| return ret; |
| } |
| |
| /* Returns true of __chk is within the first __range bytes of __off */ |
| #define OFF_IN_RANGE(__off, __range, __chk) \ |
| ((__off <= __chk) && ((__off + __range) >= __chk)) |
| |
| static void solo_jpeg_header(struct solo_enc_dev *solo_enc, |
| struct videobuf_dmabuf *vbuf) |
| { |
| struct scatterlist *sg; |
| void *src = jpeg_header; |
| size_t copied = 0; |
| size_t to_copy = sizeof(jpeg_header); |
| |
| for (sg = vbuf->sglist; sg && copied < to_copy; sg = sg_next(sg)) { |
| size_t this_copy = min(sg_dma_len(sg), |
| (unsigned int)(to_copy - copied)); |
| u8 *p = sg_virt(sg); |
| |
| memcpy(p, src + copied, this_copy); |
| |
| if (OFF_IN_RANGE(copied, this_copy, SOF0_START + 5)) |
| p[(SOF0_START + 5) - copied] = |
| 0xff & (solo_enc->height >> 8); |
| if (OFF_IN_RANGE(copied, this_copy, SOF0_START + 6)) |
| p[(SOF0_START + 6) - copied] = 0xff & solo_enc->height; |
| if (OFF_IN_RANGE(copied, this_copy, SOF0_START + 7)) |
| p[(SOF0_START + 7) - copied] = |
| 0xff & (solo_enc->width >> 8); |
| if (OFF_IN_RANGE(copied, this_copy, SOF0_START + 8)) |
| p[(SOF0_START + 8) - copied] = 0xff & solo_enc->width; |
| |
| copied += this_copy; |
| } |
| } |
| |
| static int solo_fill_jpeg(struct solo_enc_fh *fh, struct solo_enc_buf *enc_buf, |
| struct videobuf_buffer *vb, |
| struct videobuf_dmabuf *vbuf) |
| { |
| struct solo_dev *solo_dev = fh->enc->solo_dev; |
| int size = enc_buf->jpeg_size; |
| |
| /* Copy the header first (direct write) */ |
| solo_jpeg_header(fh->enc, vbuf); |
| |
| vb->size = size + sizeof(jpeg_header); |
| |
| /* Grab the jpeg frame */ |
| return enc_get_jpeg_dma_sg(solo_dev, fh->desc, vbuf->sglist, |
| sizeof(jpeg_header), |
| enc_buf->jpeg_off, size); |
| } |
| |
| static inline int vop_interlaced(__le32 *vh) |
| { |
| return (__le32_to_cpu(vh[0]) >> 30) & 1; |
| } |
| |
| static inline u32 vop_size(__le32 *vh) |
| { |
| return __le32_to_cpu(vh[0]) & 0xFFFFF; |
| } |
| |
| static inline u8 vop_hsize(__le32 *vh) |
| { |
| return (__le32_to_cpu(vh[1]) >> 8) & 0xFF; |
| } |
| |
| static inline u8 vop_vsize(__le32 *vh) |
| { |
| return __le32_to_cpu(vh[1]) & 0xFF; |
| } |
| |
| /* must be called with *bits % 8 = 0 */ |
| static void write_bytes(u8 **out, unsigned *bits, const u8 *src, unsigned count) |
| { |
| memcpy(*out, src, count); |
| *out += count; |
| *bits += count * 8; |
| } |
| |
| static void write_bits(u8 **out, unsigned *bits, u32 value, unsigned count) |
| { |
| |
| value <<= 32 - count; // shift to the right |
| |
| while (count--) { |
| **out <<= 1; |
| **out |= !!(value & (1 << 31)); /* MSB */ |
| value <<= 1; |
| if (++(*bits) % 8 == 0) |
| (*out)++; |
| } |
| } |
| |
| static void write_ue(u8 **out, unsigned *bits, unsigned value) /* H.264 only */ |
| { |
| uint32_t max = 0, cnt = 0; |
| |
| while (value > max) { |
| max = (max + 2) * 2 - 2; |
| cnt++; |
| } |
| write_bits(out, bits, 1, cnt + 1); |
| write_bits(out, bits, ~(max - value), cnt); |
| } |
| |
| static void write_se(u8 **out, unsigned *bits, int value) /* H.264 only */ |
| { |
| if (value <= 0) |
| write_ue(out, bits, -value * 2); |
| else |
| write_ue(out, bits, value * 2 - 1); |
| } |
| |
| static void write_mpeg4_end(u8 **out, unsigned *bits) |
| { |
| write_bits(out, bits, 0, 1); |
| /* align on 32-bit boundary */ |
| if (*bits % 32) |
| write_bits(out, bits, 0xFFFFFFFF, 32 - *bits % 32); |
| } |
| |
| static void write_h264_end(u8 **out, unsigned *bits, int align) |
| { |
| write_bits(out, bits, 1, 1); |
| while ((*bits) % 8) |
| write_bits(out, bits, 0, 1); |
| if (align) |
| while ((*bits) % 32) |
| write_bits(out, bits, 0, 1); |
| } |
| |
| static void mpeg4_write_vol(u8 **out, struct solo_dev *solo_dev, |
| __le32 *vh, unsigned fps, unsigned interval) |
| { |
| static const u8 hdr[] = { |
| 0, 0, 1, 0x00 /* video_object_start_code */, |
| 0, 0, 1, 0x20 /* video_object_layer_start_code */ |
| }; |
| unsigned bits = 0; |
| unsigned width = vop_hsize(vh) << 4; |
| unsigned height = vop_vsize(vh) << 4; |
| unsigned interlaced = vop_interlaced(vh); |
| |
| write_bytes(out, &bits, hdr, sizeof(hdr)); |
| write_bits(out, &bits, 0, 1); /* random_accessible_vol */ |
| write_bits(out, &bits, 0x04, 8); /* video_object_type_indication: main */ |
| write_bits(out, &bits, 1, 1); /* is_object_layer_identifier */ |
| write_bits(out, &bits, 2, 4); /* video_object_layer_verid: table V2-39 */ |
| write_bits(out, &bits, 0, 3); /* video_object_layer_priority */ |
| if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) |
| write_bits(out, &bits, 3, 4); /* aspect_ratio_info, assuming 4:3 */ |
| else |
| write_bits(out, &bits, 2, 4); |
| write_bits(out, &bits, 1, 1); /* vol_control_parameters */ |
| write_bits(out, &bits, 1, 2); /* chroma_format: 4:2:0 */ |
| write_bits(out, &bits, 1, 1); /* low_delay */ |
| write_bits(out, &bits, 0, 1); /* vbv_parameters */ |
| write_bits(out, &bits, 0, 2); /* video_object_layer_shape: rectangular */ |
| write_bits(out, &bits, 1, 1); /* marker_bit */ |
| write_bits(out, &bits, fps, 16); /* vop_time_increment_resolution */ |
| write_bits(out, &bits, 1, 1); /* marker_bit */ |
| write_bits(out, &bits, 1, 1); /* fixed_vop_rate */ |
| write_bits(out, &bits, interval, 15); /* fixed_vop_time_increment */ |
| write_bits(out, &bits, 1, 1); /* marker_bit */ |
| write_bits(out, &bits, width, 13); /* video_object_layer_width */ |
| write_bits(out, &bits, 1, 1); /* marker_bit */ |
| write_bits(out, &bits, height, 13); /* video_object_layer_height */ |
| write_bits(out, &bits, 1, 1); /* marker_bit */ |
| write_bits(out, &bits, interlaced, 1); /* interlaced */ |
| write_bits(out, &bits, 1, 1); /* obmc_disable */ |
| write_bits(out, &bits, 0, 2); /* sprite_enable */ |
| write_bits(out, &bits, 0, 1); /* not_8_bit */ |
| write_bits(out, &bits, 1, 0); /* quant_type */ |
| write_bits(out, &bits, 0, 1); /* load_intra_quant_mat */ |
| write_bits(out, &bits, 0, 1); /* load_nonintra_quant_mat */ |
| write_bits(out, &bits, 0, 1); /* quarter_sample */ |
| write_bits(out, &bits, 1, 1); /* complexity_estimation_disable */ |
| write_bits(out, &bits, 1, 1); /* resync_marker_disable */ |
| write_bits(out, &bits, 0, 1); /* data_partitioned */ |
| write_bits(out, &bits, 0, 1); /* newpred_enable */ |
| write_bits(out, &bits, 0, 1); /* reduced_resolution_vop_enable */ |
| write_bits(out, &bits, 0, 1); /* scalability */ |
| write_mpeg4_end(out, &bits); |
| } |
| |
| static void h264_write_vol(u8 **out, struct solo_dev *solo_dev, __le32 *vh) |
| { |
| static const u8 sps[] = { |
| 0, 0, 0, 1 /* start code */, 0x67, 66 /* profile_idc */, |
| 0 /* constraints */, 30 /* level_idc */ |
| }; |
| static const u8 pps[] = { |
| 0, 0, 0, 1 /* start code */, 0x68 |
| }; |
| |
| unsigned bits = 0; |
| unsigned mbs_w = vop_hsize(vh); |
| unsigned mbs_h = vop_vsize(vh); |
| |
| write_bytes(out, &bits, sps, sizeof(sps)); |
| write_ue(out, &bits, 0); /* seq_parameter_set_id */ |
| write_ue(out, &bits, 5); /* log2_max_frame_num_minus4 */ |
| write_ue(out, &bits, 0); /* pic_order_cnt_type */ |
| write_ue(out, &bits, 6); /* log2_max_pic_order_cnt_lsb_minus4 */ |
| write_ue(out, &bits, 1); /* max_num_ref_frames */ |
| write_bits(out, &bits, 0, 1); /* gaps_in_frame_num_value_allowed_flag */ |
| write_ue(out, &bits, mbs_w - 1); /* pic_width_in_mbs_minus1 */ |
| write_ue(out, &bits, mbs_h - 1); /* pic_height_in_map_units_minus1 */ |
| write_bits(out, &bits, 1, 1); /* frame_mbs_only_flag */ |
| write_bits(out, &bits, 1, 1); /* direct_8x8_frame_field_flag */ |
| write_bits(out, &bits, 0, 1); /* frame_cropping_flag */ |
| write_bits(out, &bits, 0, 1); /* vui_parameters_present_flag */ |
| write_h264_end(out, &bits, 0); |
| |
| write_bytes(out, &bits, pps, sizeof(pps)); |
| write_ue(out, &bits, 0); /* pic_parameter_set_id */ |
| write_ue(out, &bits, 0); /* seq_parameter_set_id */ |
| write_bits(out, &bits, 0, 1); /* entropy_coding_mode_flag */ |
| write_bits(out, &bits, 0, 1); /* bottom_field_pic_order_in_frame_present_flag */ |
| write_ue(out, &bits, 0); /* num_slice_groups_minus1 */ |
| write_ue(out, &bits, 0); /* num_ref_idx_l0_default_active_minus1 */ |
| write_ue(out, &bits, 0); /* num_ref_idx_l1_default_active_minus1 */ |
| write_bits(out, &bits, 0, 1); /* weighted_pred_flag */ |
| write_bits(out, &bits, 0, 2); /* weighted_bipred_idc */ |
| write_se(out, &bits, 0); /* pic_init_qp_minus26 */ |
| write_se(out, &bits, 0); /* pic_init_qs_minus26 */ |
| write_se(out, &bits, 2); /* chroma_qp_index_offset */ |
| write_bits(out, &bits, 0, 1); /* deblocking_filter_control_present_flag */ |
| write_bits(out, &bits, 1, 1); /* constrained_intra_pred_flag */ |
| write_bits(out, &bits, 0, 1); /* redundant_pic_cnt_present_flag */ |
| write_h264_end(out, &bits, 1); |
| } |
| |
| static int solo_fill_mpeg(struct solo_enc_fh *fh, struct solo_enc_buf *enc_buf, |
| struct videobuf_buffer *vb, |
| struct videobuf_dmabuf *vbuf) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| #define VH_WORDS 16 |
| #define MAX_VOL_HEADER_LENGTH 64 |
| |
| __le32 vh[VH_WORDS]; |
| int ret; |
| int frame_size, frame_off; |
| int skip = 0; |
| |
| if (WARN_ON_ONCE(enc_buf->size <= sizeof(vh))) |
| return -EINVAL; |
| |
| /* First get the hardware vop header (not real mpeg) */ |
| ret = enc_get_mpeg_dma(solo_dev, vh, enc_buf->off, sizeof(vh)); |
| if (WARN_ON_ONCE(ret)) |
| return ret; |
| |
| if (WARN_ON_ONCE(vop_size(vh) > enc_buf->size)) |
| return -EINVAL; |
| |
| vb->width = vop_hsize(vh) << 4; |
| vb->height = vop_vsize(vh) << 4; |
| vb->size = vop_size(vh); |
| |
| /* If this is a key frame, add extra m4v header */ |
| if (!enc_buf->vop) { |
| u8 header[MAX_VOL_HEADER_LENGTH], *out = header; |
| |
| if (solo_dev->flags & FLAGS_6110) |
| h264_write_vol(&out, solo_dev, vh); |
| else |
| mpeg4_write_vol(&out, solo_dev, vh, |
| solo_dev->fps * 1000, |
| solo_enc->interval * 1000); |
| skip = out - header; |
| enc_write_sg(vbuf->sglist, header, skip); |
| /* Adjust the dma buffer past this header */ |
| vb->size += skip; |
| } |
| |
| /* Now get the actual mpeg payload */ |
| frame_off = (enc_buf->off + sizeof(vh)) % SOLO_MP4E_EXT_SIZE(solo_dev); |
| frame_size = enc_buf->size - sizeof(vh); |
| |
| ret = enc_get_mpeg_dma_sg(solo_dev, fh->desc, vbuf->sglist, |
| skip, frame_off, frame_size); |
| WARN_ON_ONCE(ret); |
| |
| return ret; |
| } |
| |
| static void solo_enc_fillbuf(struct solo_enc_fh *fh, |
| struct videobuf_buffer *vb) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct solo_enc_buf *enc_buf = NULL; |
| struct videobuf_dmabuf *vbuf; |
| int ret; |
| int error = 1; |
| u16 idx = fh->rd_idx; |
| |
| while (idx != solo_dev->enc_wr_idx) { |
| struct solo_enc_buf *ebuf = &solo_dev->enc_buf[idx]; |
| |
| idx = (idx + 1) % SOLO_NR_RING_BUFS; |
| |
| if (ebuf->ch != solo_enc->ch) |
| continue; |
| |
| if (fh->fmt == V4L2_PIX_FMT_MPEG) { |
| if (fh->type == ebuf->type) { |
| enc_buf = ebuf; |
| break; |
| } |
| } else { |
| /* For mjpeg, keep reading to the newest frame */ |
| enc_buf = ebuf; |
| } |
| } |
| |
| fh->rd_idx = idx; |
| |
| if (WARN_ON_ONCE(!enc_buf)) |
| goto buf_err; |
| |
| if ((fh->fmt == V4L2_PIX_FMT_MPEG && |
| vb->bsize < enc_buf->size) || |
| (fh->fmt == V4L2_PIX_FMT_MJPEG && |
| vb->bsize < (enc_buf->jpeg_size + sizeof(jpeg_header)))) { |
| WARN_ON_ONCE(1); |
| goto buf_err; |
| } |
| |
| vbuf = videobuf_to_dma(vb); |
| if (WARN_ON_ONCE(!vbuf)) |
| goto buf_err; |
| |
| if (fh->fmt == V4L2_PIX_FMT_MPEG) |
| ret = solo_fill_mpeg(fh, enc_buf, vb, vbuf); |
| else |
| ret = solo_fill_jpeg(fh, enc_buf, vb, vbuf); |
| |
| if (!ret) |
| error = 0; |
| |
| buf_err: |
| if (error) { |
| vb->state = VIDEOBUF_ERROR; |
| } else { |
| vb->field_count++; |
| vb->ts = enc_buf->ts; |
| vb->state = VIDEOBUF_DONE; |
| } |
| |
| wake_up(&vb->done); |
| |
| return; |
| } |
| |
| static void solo_enc_thread_try(struct solo_enc_fh *fh) |
| { |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct videobuf_buffer *vb; |
| |
| for (;;) { |
| spin_lock(&solo_enc->lock); |
| |
| if (fh->rd_idx == solo_dev->enc_wr_idx) |
| break; |
| |
| if (list_empty(&fh->vidq_active)) |
| break; |
| |
| vb = list_first_entry(&fh->vidq_active, |
| struct videobuf_buffer, queue); |
| |
| if (!waitqueue_active(&vb->done)) |
| break; |
| |
| list_del(&vb->queue); |
| |
| spin_unlock(&solo_enc->lock); |
| |
| solo_enc_fillbuf(fh, vb); |
| } |
| |
| assert_spin_locked(&solo_enc->lock); |
| spin_unlock(&solo_enc->lock); |
| } |
| |
| static int solo_enc_thread(void *data) |
| { |
| struct solo_enc_fh *fh = data; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| DECLARE_WAITQUEUE(wait, current); |
| |
| set_freezable(); |
| add_wait_queue(&solo_enc->thread_wait, &wait); |
| |
| for (;;) { |
| long timeout = schedule_timeout_interruptible(HZ); |
| if (timeout == -ERESTARTSYS || kthread_should_stop()) |
| break; |
| solo_enc_thread_try(fh); |
| try_to_freeze(); |
| } |
| |
| remove_wait_queue(&solo_enc->thread_wait, &wait); |
| |
| return 0; |
| } |
| |
| void solo_motion_isr(struct solo_dev *solo_dev) |
| { |
| u32 status; |
| int i; |
| |
| solo_reg_write(solo_dev, SOLO_IRQ_STAT, SOLO_IRQ_MOTION); |
| |
| status = solo_reg_read(solo_dev, SOLO_VI_MOT_STATUS); |
| |
| for (i = 0; i < solo_dev->nr_chans; i++) { |
| struct solo_enc_dev *solo_enc = solo_dev->v4l2_enc[i]; |
| |
| BUG_ON(solo_enc == NULL); |
| |
| if (solo_enc->motion_detected) |
| continue; |
| if (!(status & (1 << i))) |
| continue; |
| |
| solo_enc->motion_detected = 1; |
| } |
| } |
| |
| void solo_enc_v4l2_isr(struct solo_dev *solo_dev) |
| { |
| struct solo_enc_buf *enc_buf; |
| u32 mpeg_current, mpeg_next, mpeg_size; |
| u32 jpeg_current, jpeg_next, jpeg_size; |
| u32 reg_mpeg_size; |
| u8 cur_q, vop_type; |
| u8 ch; |
| enum solo_enc_types enc_type; |
| |
| solo_reg_write(solo_dev, SOLO_IRQ_STAT, SOLO_IRQ_ENCODER); |
| |
| cur_q = ((solo_reg_read(solo_dev, SOLO_VE_STATE(11)) & 0xF) + 1) % MP4_QS; |
| |
| reg_mpeg_size = ((solo_reg_read(solo_dev, SOLO_VE_STATE(0)) & 0xFFFFF) + 64 + 8) & ~7; |
| |
| while (solo_dev->enc_idx != cur_q) { |
| mpeg_current = solo_reg_read(solo_dev, |
| SOLO_VE_MPEG4_QUE(solo_dev->enc_idx)); |
| jpeg_current = solo_reg_read(solo_dev, |
| SOLO_VE_JPEG_QUE(solo_dev->enc_idx)); |
| solo_dev->enc_idx = (solo_dev->enc_idx + 1) % MP4_QS; |
| mpeg_next = solo_reg_read(solo_dev, |
| SOLO_VE_MPEG4_QUE(solo_dev->enc_idx)); |
| jpeg_next = solo_reg_read(solo_dev, |
| SOLO_VE_JPEG_QUE(solo_dev->enc_idx)); |
| |
| ch = (mpeg_current >> 24) & 0x1f; |
| if (ch >= SOLO_MAX_CHANNELS) { |
| ch -= SOLO_MAX_CHANNELS; |
| enc_type = SOLO_ENC_TYPE_EXT; |
| } else |
| enc_type = SOLO_ENC_TYPE_STD; |
| |
| vop_type = (mpeg_current >> 29) & 3; |
| |
| mpeg_current &= 0x00ffffff; |
| mpeg_next &= 0x00ffffff; |
| jpeg_current &= 0x00ffffff; |
| jpeg_next &= 0x00ffffff; |
| |
| mpeg_size = (SOLO_MP4E_EXT_SIZE(solo_dev) + |
| mpeg_next - mpeg_current) % |
| SOLO_MP4E_EXT_SIZE(solo_dev); |
| |
| jpeg_size = (SOLO_JPEG_EXT_SIZE(solo_dev) + |
| jpeg_next - jpeg_current) % |
| SOLO_JPEG_EXT_SIZE(solo_dev); |
| |
| /* XXX I think this means we had a ring overflow? */ |
| if (mpeg_current > mpeg_next && mpeg_size != reg_mpeg_size) { |
| enc_reset_gop(solo_dev, ch); |
| continue; |
| } |
| |
| /* When resetting the GOP, skip frames until I-frame */ |
| if (enc_gop_reset(solo_dev, ch, vop_type)) |
| continue; |
| |
| enc_buf = &solo_dev->enc_buf[solo_dev->enc_wr_idx]; |
| |
| enc_buf->vop = vop_type; |
| enc_buf->ch = ch; |
| enc_buf->off = mpeg_current; |
| enc_buf->size = mpeg_size; |
| enc_buf->jpeg_off = jpeg_current; |
| enc_buf->jpeg_size = jpeg_size; |
| enc_buf->type = enc_type; |
| |
| do_gettimeofday(&enc_buf->ts); |
| |
| solo_dev->enc_wr_idx = (solo_dev->enc_wr_idx + 1) % |
| SOLO_NR_RING_BUFS; |
| |
| wake_up_interruptible(&solo_dev->v4l2_enc[ch]->thread_wait); |
| } |
| |
| return; |
| } |
| |
| static int solo_enc_buf_setup(struct videobuf_queue *vq, unsigned int *count, |
| unsigned int *size) |
| { |
| *size = FRAME_BUF_SIZE; |
| |
| if (*count < MIN_VID_BUFFERS) |
| *count = MIN_VID_BUFFERS; |
| |
| return 0; |
| } |
| |
| static int solo_enc_buf_prepare(struct videobuf_queue *vq, |
| struct videobuf_buffer *vb, |
| enum v4l2_field field) |
| { |
| struct solo_enc_fh *fh = vq->priv_data; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| |
| vb->size = FRAME_BUF_SIZE; |
| if (vb->baddr != 0 && vb->bsize < vb->size) |
| return -EINVAL; |
| |
| /* These properties only change when queue is idle */ |
| vb->width = solo_enc->width; |
| vb->height = solo_enc->height; |
| vb->field = field; |
| |
| if (vb->state == VIDEOBUF_NEEDS_INIT) { |
| int rc = videobuf_iolock(vq, vb, NULL); |
| if (rc < 0) { |
| struct videobuf_dmabuf *dma = videobuf_to_dma(vb); |
| videobuf_dma_unmap(vq->dev, dma); |
| videobuf_dma_free(dma); |
| vb->state = VIDEOBUF_NEEDS_INIT; |
| return rc; |
| } |
| } |
| vb->state = VIDEOBUF_PREPARED; |
| |
| return 0; |
| } |
| |
| static void solo_enc_buf_queue(struct videobuf_queue *vq, |
| struct videobuf_buffer *vb) |
| { |
| struct solo_enc_fh *fh = vq->priv_data; |
| |
| vb->state = VIDEOBUF_QUEUED; |
| list_add_tail(&vb->queue, &fh->vidq_active); |
| wake_up_interruptible(&fh->enc->thread_wait); |
| } |
| |
| static void solo_enc_buf_release(struct videobuf_queue *vq, |
| struct videobuf_buffer *vb) |
| { |
| struct videobuf_dmabuf *dma = videobuf_to_dma(vb); |
| |
| videobuf_dma_unmap(vq->dev, dma); |
| videobuf_dma_free(dma); |
| vb->state = VIDEOBUF_NEEDS_INIT; |
| } |
| |
| static struct videobuf_queue_ops solo_enc_video_qops = { |
| .buf_setup = solo_enc_buf_setup, |
| .buf_prepare = solo_enc_buf_prepare, |
| .buf_queue = solo_enc_buf_queue, |
| .buf_release = solo_enc_buf_release, |
| }; |
| |
| static unsigned int solo_enc_poll(struct file *file, |
| struct poll_table_struct *wait) |
| { |
| struct solo_enc_fh *fh = file->private_data; |
| |
| return videobuf_poll_stream(file, &fh->vidq, wait); |
| } |
| |
| static int solo_enc_mmap(struct file *file, struct vm_area_struct *vma) |
| { |
| struct solo_enc_fh *fh = file->private_data; |
| |
| return videobuf_mmap_mapper(&fh->vidq, vma); |
| } |
| |
| static int solo_enc_open(struct file *file) |
| { |
| struct solo_enc_dev *solo_enc = video_drvdata(file); |
| struct solo_enc_fh *fh; |
| |
| fh = kzalloc(sizeof(*fh), GFP_KERNEL); |
| if (fh == NULL) |
| return -ENOMEM; |
| |
| fh->enc = solo_enc; |
| file->private_data = fh; |
| INIT_LIST_HEAD(&fh->vidq_active); |
| fh->fmt = V4L2_PIX_FMT_MPEG; |
| fh->type = SOLO_ENC_TYPE_STD; |
| |
| videobuf_queue_sg_init(&fh->vidq, &solo_enc_video_qops, |
| &solo_enc->solo_dev->pdev->dev, |
| &solo_enc->lock, |
| V4L2_BUF_TYPE_VIDEO_CAPTURE, |
| V4L2_FIELD_INTERLACED, |
| sizeof(struct videobuf_buffer), fh, NULL); |
| |
| return 0; |
| } |
| |
| static ssize_t solo_enc_read(struct file *file, char __user *data, |
| size_t count, loff_t *ppos) |
| { |
| struct solo_enc_fh *fh = file->private_data; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| |
| /* Make sure the encoder is on */ |
| if (!fh->enc_on) { |
| int ret; |
| |
| spin_lock(&solo_enc->lock); |
| ret = solo_enc_on(fh); |
| spin_unlock(&solo_enc->lock); |
| if (ret) |
| return ret; |
| |
| ret = solo_start_fh_thread(fh); |
| if (ret) |
| return ret; |
| } |
| |
| return videobuf_read_stream(&fh->vidq, data, count, ppos, 0, |
| file->f_flags & O_NONBLOCK); |
| } |
| |
| static int solo_enc_release(struct file *file) |
| { |
| struct solo_enc_fh *fh = file->private_data; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| |
| videobuf_stop(&fh->vidq); |
| videobuf_mmap_free(&fh->vidq); |
| |
| spin_lock(&solo_enc->lock); |
| solo_enc_off(fh); |
| spin_unlock(&solo_enc->lock); |
| |
| kfree(fh); |
| |
| return 0; |
| } |
| |
| static int solo_enc_querycap(struct file *file, void *priv, |
| struct v4l2_capability *cap) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| strcpy(cap->driver, SOLO6X10_NAME); |
| snprintf(cap->card, sizeof(cap->card), "Softlogic 6x10 Enc %d", |
| solo_enc->ch); |
| snprintf(cap->bus_info, sizeof(cap->bus_info), "PCI %s", |
| pci_name(solo_dev->pdev)); |
| cap->version = SOLO6X10_VER_NUM; |
| cap->capabilities = V4L2_CAP_VIDEO_CAPTURE | |
| V4L2_CAP_READWRITE | |
| V4L2_CAP_STREAMING; |
| return 0; |
| } |
| |
| static int solo_enc_enum_input(struct file *file, void *priv, |
| struct v4l2_input *input) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| if (input->index) |
| return -EINVAL; |
| |
| snprintf(input->name, sizeof(input->name), "Encoder %d", |
| solo_enc->ch + 1); |
| input->type = V4L2_INPUT_TYPE_CAMERA; |
| |
| if (solo_dev->video_type == SOLO_VO_FMT_TYPE_NTSC) |
| input->std = V4L2_STD_NTSC_M; |
| else |
| input->std = V4L2_STD_PAL_B; |
| |
| if (!tw28_get_video_status(solo_dev, solo_enc->ch)) |
| input->status = V4L2_IN_ST_NO_SIGNAL; |
| |
| return 0; |
| } |
| |
| static int solo_enc_set_input(struct file *file, void *priv, unsigned int index) |
| { |
| if (index) |
| return -EINVAL; |
| |
| return 0; |
| } |
| |
| static int solo_enc_get_input(struct file *file, void *priv, |
| unsigned int *index) |
| { |
| *index = 0; |
| |
| return 0; |
| } |
| |
| static int solo_enc_enum_fmt_cap(struct file *file, void *priv, |
| struct v4l2_fmtdesc *f) |
| { |
| switch (f->index) { |
| case 0: |
| f->pixelformat = V4L2_PIX_FMT_MPEG; |
| strcpy(f->description, "MPEG-4 AVC"); |
| break; |
| case 1: |
| f->pixelformat = V4L2_PIX_FMT_MJPEG; |
| strcpy(f->description, "MJPEG"); |
| break; |
| default: |
| return -EINVAL; |
| } |
| |
| f->flags = V4L2_FMT_FLAG_COMPRESSED; |
| |
| return 0; |
| } |
| |
| static int solo_enc_try_fmt_cap(struct file *file, void *priv, |
| struct v4l2_format *f) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct v4l2_pix_format *pix = &f->fmt.pix; |
| |
| if (pix->pixelformat != V4L2_PIX_FMT_MPEG && |
| pix->pixelformat != V4L2_PIX_FMT_MJPEG) |
| return -EINVAL; |
| |
| /* We cannot change width/height in mid read */ |
| if (atomic_read(&solo_enc->readers) > 0) { |
| if (pix->width != solo_enc->width || |
| pix->height != solo_enc->height) |
| return -EBUSY; |
| } |
| |
| if (pix->width < solo_dev->video_hsize || |
| pix->height < solo_dev->video_vsize << 1) { |
| /* Default to CIF 1/2 size */ |
| pix->width = solo_dev->video_hsize >> 1; |
| pix->height = solo_dev->video_vsize; |
| } else { |
| /* Full frame */ |
| pix->width = solo_dev->video_hsize; |
| pix->height = solo_dev->video_vsize << 1; |
| } |
| |
| if (pix->field == V4L2_FIELD_ANY) |
| pix->field = V4L2_FIELD_INTERLACED; |
| else if (pix->field != V4L2_FIELD_INTERLACED) |
| pix->field = V4L2_FIELD_INTERLACED; |
| |
| /* Just set these */ |
| pix->colorspace = V4L2_COLORSPACE_SMPTE170M; |
| pix->sizeimage = FRAME_BUF_SIZE; |
| |
| return 0; |
| } |
| |
| static int solo_enc_set_fmt_cap(struct file *file, void *priv, |
| struct v4l2_format *f) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct v4l2_pix_format *pix = &f->fmt.pix; |
| int ret; |
| |
| spin_lock(&solo_enc->lock); |
| |
| ret = solo_enc_try_fmt_cap(file, priv, f); |
| if (ret) { |
| spin_unlock(&solo_enc->lock); |
| return ret; |
| } |
| |
| if (pix->width == solo_dev->video_hsize) |
| solo_enc->mode = SOLO_ENC_MODE_D1; |
| else |
| solo_enc->mode = SOLO_ENC_MODE_CIF; |
| |
| /* This does not change the encoder at all */ |
| fh->fmt = pix->pixelformat; |
| |
| if (pix->priv) |
| fh->type = SOLO_ENC_TYPE_EXT; |
| ret = solo_enc_on(fh); |
| |
| spin_unlock(&solo_enc->lock); |
| |
| if (ret) |
| return ret; |
| |
| return solo_start_fh_thread(fh); |
| } |
| |
| static int solo_enc_get_fmt_cap(struct file *file, void *priv, |
| struct v4l2_format *f) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct v4l2_pix_format *pix = &f->fmt.pix; |
| |
| pix->width = solo_enc->width; |
| pix->height = solo_enc->height; |
| pix->pixelformat = fh->fmt; |
| pix->field = solo_enc->interlaced ? V4L2_FIELD_INTERLACED : |
| V4L2_FIELD_NONE; |
| pix->sizeimage = FRAME_BUF_SIZE; |
| pix->colorspace = V4L2_COLORSPACE_SMPTE170M; |
| |
| return 0; |
| } |
| |
| static int solo_enc_reqbufs(struct file *file, void *priv, |
| struct v4l2_requestbuffers *req) |
| { |
| struct solo_enc_fh *fh = priv; |
| |
| return videobuf_reqbufs(&fh->vidq, req); |
| } |
| |
| static int solo_enc_querybuf(struct file *file, void *priv, |
| struct v4l2_buffer *buf) |
| { |
| struct solo_enc_fh *fh = priv; |
| |
| return videobuf_querybuf(&fh->vidq, buf); |
| } |
| |
| static int solo_enc_qbuf(struct file *file, void *priv, struct v4l2_buffer *buf) |
| { |
| struct solo_enc_fh *fh = priv; |
| |
| return videobuf_qbuf(&fh->vidq, buf); |
| } |
| |
| static int solo_enc_dqbuf(struct file *file, void *priv, |
| struct v4l2_buffer *buf) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| int ret; |
| |
| /* Make sure the encoder is on */ |
| if (!fh->enc_on) { |
| spin_lock(&solo_enc->lock); |
| ret = solo_enc_on(fh); |
| spin_unlock(&solo_enc->lock); |
| if (ret) |
| return ret; |
| |
| ret = solo_start_fh_thread(fh); |
| if (ret) |
| return ret; |
| } |
| |
| ret = videobuf_dqbuf(&fh->vidq, buf, file->f_flags & O_NONBLOCK); |
| if (ret) |
| return ret; |
| |
| /* Signal motion detection */ |
| if (solo_is_motion_on(solo_enc)) { |
| buf->flags |= V4L2_BUF_FLAG_MOTION_ON; |
| if (solo_enc->motion_detected) { |
| buf->flags |= V4L2_BUF_FLAG_MOTION_DETECTED; |
| solo_reg_write(solo_enc->solo_dev, SOLO_VI_MOT_CLEAR, |
| 1 << solo_enc->ch); |
| solo_enc->motion_detected = 0; |
| } |
| } |
| |
| /* Check for key frame on mpeg data */ |
| if (fh->fmt == V4L2_PIX_FMT_MPEG) { |
| struct videobuf_dmabuf *vbuf = |
| videobuf_to_dma(fh->vidq.bufs[buf->index]); |
| |
| if (vbuf) { |
| u8 *p = sg_virt(vbuf->sglist); |
| if (p[3] == 0x00) |
| buf->flags |= V4L2_BUF_FLAG_KEYFRAME; |
| else |
| buf->flags |= V4L2_BUF_FLAG_PFRAME; |
| } |
| } |
| |
| return 0; |
| } |
| |
| static int solo_enc_streamon(struct file *file, void *priv, |
| enum v4l2_buf_type i) |
| { |
| struct solo_enc_fh *fh = priv; |
| |
| if (i != V4L2_BUF_TYPE_VIDEO_CAPTURE) |
| return -EINVAL; |
| |
| return videobuf_streamon(&fh->vidq); |
| } |
| |
| static int solo_enc_streamoff(struct file *file, void *priv, |
| enum v4l2_buf_type i) |
| { |
| struct solo_enc_fh *fh = priv; |
| |
| if (i != V4L2_BUF_TYPE_VIDEO_CAPTURE) |
| return -EINVAL; |
| |
| return videobuf_streamoff(&fh->vidq); |
| } |
| |
| static int solo_enc_s_std(struct file *file, void *priv, v4l2_std_id *i) |
| { |
| return 0; |
| } |
| |
| static int solo_enum_framesizes(struct file *file, void *priv, |
| struct v4l2_frmsizeenum *fsize) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_dev *solo_dev = fh->enc->solo_dev; |
| |
| if (fsize->pixel_format != V4L2_PIX_FMT_MPEG) |
| return -EINVAL; |
| |
| switch (fsize->index) { |
| case 0: |
| fsize->discrete.width = solo_dev->video_hsize >> 1; |
| fsize->discrete.height = solo_dev->video_vsize; |
| break; |
| case 1: |
| fsize->discrete.width = solo_dev->video_hsize; |
| fsize->discrete.height = solo_dev->video_vsize << 1; |
| break; |
| default: |
| return -EINVAL; |
| } |
| |
| fsize->type = V4L2_FRMSIZE_TYPE_DISCRETE; |
| |
| return 0; |
| } |
| |
| static int solo_enum_frameintervals(struct file *file, void *priv, |
| struct v4l2_frmivalenum *fintv) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_dev *solo_dev = fh->enc->solo_dev; |
| |
| if (fintv->pixel_format != V4L2_PIX_FMT_MPEG || fintv->index) |
| return -EINVAL; |
| |
| fintv->type = V4L2_FRMIVAL_TYPE_STEPWISE; |
| |
| fintv->stepwise.min.numerator = solo_dev->fps; |
| fintv->stepwise.min.denominator = 1; |
| |
| fintv->stepwise.max.numerator = solo_dev->fps; |
| fintv->stepwise.max.denominator = 15; |
| |
| fintv->stepwise.step.numerator = 1; |
| fintv->stepwise.step.denominator = 1; |
| |
| return 0; |
| } |
| |
| static int solo_g_parm(struct file *file, void *priv, |
| struct v4l2_streamparm *sp) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct v4l2_captureparm *cp = &sp->parm.capture; |
| |
| cp->capability = V4L2_CAP_TIMEPERFRAME; |
| cp->timeperframe.numerator = solo_enc->interval; |
| cp->timeperframe.denominator = solo_dev->fps; |
| cp->capturemode = 0; |
| /* XXX: Shouldn't we be able to get/set this from videobuf? */ |
| cp->readbuffers = 2; |
| |
| return 0; |
| } |
| |
| static int solo_s_parm(struct file *file, void *priv, |
| struct v4l2_streamparm *sp) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| struct v4l2_captureparm *cp = &sp->parm.capture; |
| |
| spin_lock(&solo_enc->lock); |
| |
| if (atomic_read(&solo_enc->readers) > 0) { |
| spin_unlock(&solo_enc->lock); |
| return -EBUSY; |
| } |
| |
| if ((cp->timeperframe.numerator == 0) || |
| (cp->timeperframe.denominator == 0)) { |
| /* reset framerate */ |
| cp->timeperframe.numerator = 1; |
| cp->timeperframe.denominator = solo_dev->fps; |
| } |
| |
| if (cp->timeperframe.denominator != solo_dev->fps) |
| cp->timeperframe.denominator = solo_dev->fps; |
| |
| if (cp->timeperframe.numerator > 15) |
| cp->timeperframe.numerator = 15; |
| |
| solo_enc->interval = cp->timeperframe.numerator; |
| |
| cp->capability = V4L2_CAP_TIMEPERFRAME; |
| |
| solo_enc->gop = max(solo_dev->fps / solo_enc->interval, 1); |
| solo_update_mode(solo_enc); |
| |
| spin_unlock(&solo_enc->lock); |
| |
| return 0; |
| } |
| |
| static int solo_queryctrl(struct file *file, void *priv, |
| struct v4l2_queryctrl *qc) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| qc->id = v4l2_ctrl_next(solo_ctrl_classes, qc->id); |
| if (!qc->id) |
| return -EINVAL; |
| |
| switch (qc->id) { |
| case V4L2_CID_BRIGHTNESS: |
| case V4L2_CID_CONTRAST: |
| case V4L2_CID_SATURATION: |
| case V4L2_CID_HUE: |
| return v4l2_ctrl_query_fill(qc, 0x00, 0xff, 1, 0x80); |
| case V4L2_CID_SHARPNESS: |
| return v4l2_ctrl_query_fill(qc, 0x00, 0x0f, 1, 0x00); |
| case V4L2_CID_MPEG_VIDEO_ENCODING: |
| return v4l2_ctrl_query_fill( |
| qc, V4L2_MPEG_VIDEO_ENCODING_MPEG_1, |
| V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC, 1, |
| V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC); |
| case V4L2_CID_MPEG_VIDEO_GOP_SIZE: |
| return v4l2_ctrl_query_fill(qc, 1, 255, 1, solo_dev->fps); |
| #ifdef PRIVATE_CIDS |
| case V4L2_CID_MOTION_THRESHOLD: |
| qc->flags |= V4L2_CTRL_FLAG_SLIDER; |
| qc->type = V4L2_CTRL_TYPE_INTEGER; |
| qc->minimum = 0; |
| qc->maximum = 0xffff; |
| qc->step = 1; |
| qc->default_value = SOLO_DEF_MOT_THRESH; |
| strlcpy(qc->name, "Motion Detection Threshold", |
| sizeof(qc->name)); |
| return 0; |
| case V4L2_CID_MOTION_ENABLE: |
| qc->type = V4L2_CTRL_TYPE_BOOLEAN; |
| qc->minimum = 0; |
| qc->maximum = qc->step = 1; |
| qc->default_value = 0; |
| strlcpy(qc->name, "Motion Detection Enable", sizeof(qc->name)); |
| return 0; |
| #else |
| case V4L2_CID_MOTION_THRESHOLD: |
| return v4l2_ctrl_query_fill(qc, 0, 0xffff, 1, |
| SOLO_DEF_MOT_THRESH); |
| case V4L2_CID_MOTION_ENABLE: |
| return v4l2_ctrl_query_fill(qc, 0, 1, 1, 0); |
| #endif |
| case V4L2_CID_RDS_TX_RADIO_TEXT: |
| qc->type = V4L2_CTRL_TYPE_STRING; |
| qc->minimum = 0; |
| qc->maximum = OSD_TEXT_MAX; |
| qc->step = 1; |
| qc->default_value = 0; |
| strlcpy(qc->name, "OSD Text", sizeof(qc->name)); |
| return 0; |
| } |
| |
| return -EINVAL; |
| } |
| |
| static int solo_querymenu(struct file *file, void *priv, |
| struct v4l2_querymenu *qmenu) |
| { |
| struct v4l2_queryctrl qctrl; |
| int err; |
| |
| qctrl.id = qmenu->id; |
| err = solo_queryctrl(file, priv, &qctrl); |
| if (err) |
| return err; |
| |
| return v4l2_ctrl_query_menu(qmenu, &qctrl, NULL); |
| } |
| |
| static int solo_g_ctrl(struct file *file, void *priv, |
| struct v4l2_control *ctrl) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| switch (ctrl->id) { |
| case V4L2_CID_BRIGHTNESS: |
| case V4L2_CID_CONTRAST: |
| case V4L2_CID_SATURATION: |
| case V4L2_CID_HUE: |
| case V4L2_CID_SHARPNESS: |
| return tw28_get_ctrl_val(solo_dev, ctrl->id, solo_enc->ch, |
| &ctrl->value); |
| case V4L2_CID_MPEG_VIDEO_ENCODING: |
| ctrl->value = V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC; |
| break; |
| case V4L2_CID_MPEG_VIDEO_GOP_SIZE: |
| ctrl->value = solo_enc->gop; |
| break; |
| case V4L2_CID_MOTION_THRESHOLD: |
| ctrl->value = solo_enc->motion_thresh; |
| break; |
| case V4L2_CID_MOTION_ENABLE: |
| ctrl->value = solo_is_motion_on(solo_enc); |
| break; |
| default: |
| return -EINVAL; |
| } |
| |
| return 0; |
| } |
| |
| static int solo_s_ctrl(struct file *file, void *priv, |
| struct v4l2_control *ctrl) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| struct solo_dev *solo_dev = solo_enc->solo_dev; |
| |
| switch (ctrl->id) { |
| case V4L2_CID_BRIGHTNESS: |
| case V4L2_CID_CONTRAST: |
| case V4L2_CID_SATURATION: |
| case V4L2_CID_HUE: |
| case V4L2_CID_SHARPNESS: |
| return tw28_set_ctrl_val(solo_dev, ctrl->id, solo_enc->ch, |
| ctrl->value); |
| case V4L2_CID_MPEG_VIDEO_ENCODING: |
| if (ctrl->value != V4L2_MPEG_VIDEO_ENCODING_MPEG_4_AVC) |
| return -ERANGE; |
| break; |
| case V4L2_CID_MPEG_VIDEO_GOP_SIZE: |
| if (ctrl->value < 1 || ctrl->value > 255) |
| return -ERANGE; |
| solo_enc->gop = ctrl->value; |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP(solo_enc->ch), |
| solo_enc->gop); |
| solo_reg_write(solo_dev, SOLO_VE_CH_GOP_E(solo_enc->ch), |
| solo_enc->gop); |
| break; |
| case V4L2_CID_MOTION_THRESHOLD: |
| /* TODO accept value on lower 16-bits and use high |
| * 16-bits to assign the value to a specific block */ |
| if (ctrl->value < 0 || ctrl->value > 0xffff) |
| return -ERANGE; |
| solo_enc->motion_thresh = ctrl->value; |
| solo_set_motion_threshold(solo_dev, solo_enc->ch, ctrl->value); |
| break; |
| case V4L2_CID_MOTION_ENABLE: |
| solo_motion_toggle(solo_enc, ctrl->value); |
| break; |
| default: |
| return -EINVAL; |
| } |
| |
| return 0; |
| } |
| |
| static int solo_s_ext_ctrls(struct file *file, void *priv, |
| struct v4l2_ext_controls *ctrls) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| int i; |
| |
| for (i = 0; i < ctrls->count; i++) { |
| struct v4l2_ext_control *ctrl = (ctrls->controls + i); |
| int err; |
| |
| switch (ctrl->id) { |
| case V4L2_CID_RDS_TX_RADIO_TEXT: |
| if (ctrl->size - 1 > OSD_TEXT_MAX) |
| err = -ERANGE; |
| else { |
| err = copy_from_user(solo_enc->osd_text, |
| ctrl->string, |
| OSD_TEXT_MAX); |
| solo_enc->osd_text[OSD_TEXT_MAX] = '\0'; |
| if (!err) |
| err = solo_osd_print(solo_enc); |
| } |
| break; |
| default: |
| err = -EINVAL; |
| } |
| |
| if (err < 0) { |
| ctrls->error_idx = i; |
| return err; |
| } |
| } |
| |
| return 0; |
| } |
| |
| static int solo_g_ext_ctrls(struct file *file, void *priv, |
| struct v4l2_ext_controls *ctrls) |
| { |
| struct solo_enc_fh *fh = priv; |
| struct solo_enc_dev *solo_enc = fh->enc; |
| int i; |
| |
| for (i = 0; i < ctrls->count; i++) { |
| struct v4l2_ext_control *ctrl = (ctrls->controls + i); |
| int err; |
| |
| switch (ctrl->id) { |
| case V4L2_CID_RDS_TX_RADIO_TEXT: |
| if (ctrl->size < OSD_TEXT_MAX) { |
| ctrl->size = OSD_TEXT_MAX; |
| err = -ENOSPC; |
| } else { |
| err = copy_to_user(ctrl->string, |
| solo_enc->osd_text, |
| OSD_TEXT_MAX); |
| } |
| break; |
| default: |
| err = -EINVAL; |
| } |
| |
| if (err < 0) { |
| ctrls->error_idx = i; |
| return err; |
| } |
| } |
| |
| return 0; |
| } |
| |
| static const struct v4l2_file_operations solo_enc_fops = { |
| .owner = THIS_MODULE, |
| .open = solo_enc_open, |
| .release = solo_enc_release, |
| .read = solo_enc_read, |
| .poll = solo_enc_poll, |
| .mmap = solo_enc_mmap, |
| .ioctl = video_ioctl2, |
| }; |
| |
| static const struct v4l2_ioctl_ops solo_enc_ioctl_ops = { |
| .vidioc_querycap = solo_enc_querycap, |
| .vidioc_s_std = solo_enc_s_std, |
| /* Input callbacks */ |
| .vidioc_enum_input = solo_enc_enum_input, |
| .vidioc_s_input = solo_enc_set_input, |
| .vidioc_g_input = solo_enc_get_input, |
| /* Video capture format callbacks */ |
| .vidioc_enum_fmt_vid_cap = solo_enc_enum_fmt_cap, |
| .vidioc_try_fmt_vid_cap = solo_enc_try_fmt_cap, |
| .vidioc_s_fmt_vid_cap = solo_enc_set_fmt_cap, |
| .vidioc_g_fmt_vid_cap = solo_enc_get_fmt_cap, |
| /* Streaming I/O */ |
| .vidioc_reqbufs = solo_enc_reqbufs, |
| .vidioc_querybuf = solo_enc_querybuf, |
| .vidioc_qbuf = solo_enc_qbuf, |
| .vidioc_dqbuf = solo_enc_dqbuf, |
| .vidioc_streamon = solo_enc_streamon, |
| .vidioc_streamoff = solo_enc_streamoff, |
| /* Frame size and interval */ |
| .vidioc_enum_framesizes = solo_enum_framesizes, |
| .vidioc_enum_frameintervals = solo_enum_frameintervals, |
| /* Video capture parameters */ |
| .vidioc_s_parm = solo_s_parm, |
| .vidioc_g_parm = solo_g_parm, |
| /* Controls */ |
| .vidioc_queryctrl = solo_queryctrl, |
| .vidioc_querymenu = solo_querymenu, |
| .vidioc_g_ctrl = solo_g_ctrl, |
| .vidioc_s_ctrl = solo_s_ctrl, |
| .vidioc_g_ext_ctrls = solo_g_ext_ctrls, |
| .vidioc_s_ext_ctrls = solo_s_ext_ctrls, |
| }; |
| |
| static struct video_device solo_enc_template = { |
| .name = SOLO6X10_NAME, |
| .fops = &solo_enc_fops, |
| .ioctl_ops = &solo_enc_ioctl_ops, |
| .minor = -1, |
| .release = video_device_release, |
| |
| .tvnorms = V4L2_STD_NTSC_M | V4L2_STD_PAL_B, |
| .current_norm = V4L2_STD_NTSC_M, |
| }; |
| |
| static struct solo_enc_dev *solo_enc_alloc(struct solo_dev *solo_dev, u8 ch) |
| { |
| struct solo_enc_dev *solo_enc; |
| int ret; |
| |
| solo_enc = kzalloc(sizeof(*solo_enc), GFP_KERNEL); |
| if (!solo_enc) |
| return ERR_PTR(-ENOMEM); |
| |
| solo_enc->vfd = video_device_alloc(); |
| if (!solo_enc->vfd) { |
| kfree(solo_enc); |
| return ERR_PTR(-ENOMEM); |
| } |
| |
| solo_enc->solo_dev = solo_dev; |
| solo_enc->ch = ch; |
| |
| *solo_enc->vfd = solo_enc_template; |
| solo_enc->vfd->parent = &solo_dev->pdev->dev; |
| ret = video_register_device(solo_enc->vfd, VFL_TYPE_GRABBER, |
| video_nr); |
| if (ret < 0) { |
| video_device_release(solo_enc->vfd); |
| kfree(solo_enc); |
| return ERR_PTR(ret); |
| } |
| |
| video_set_drvdata(solo_enc->vfd, solo_enc); |
| |
| snprintf(solo_enc->vfd->name, sizeof(solo_enc->vfd->name), |
| "%s-enc (%i/%i)", SOLO6X10_NAME, solo_dev->vfd->num, |
| solo_enc->vfd->num); |
| |
| if (video_nr != -1) |
| video_nr++; |
| |
| spin_lock_init(&solo_enc->lock); |
| init_waitqueue_head(&solo_enc->thread_wait); |
| atomic_set(&solo_enc->readers, 0); |
| |
| solo_enc->qp = SOLO_DEFAULT_QP; |
| solo_enc->gop = solo_dev->fps; |
| solo_enc->interval = 1; |
| solo_enc->mode = SOLO_ENC_MODE_CIF; |
| solo_enc->motion_thresh = SOLO_DEF_MOT_THRESH; |
| |
| spin_lock(&solo_enc->lock); |
| solo_update_mode(solo_enc); |
| spin_unlock(&solo_enc->lock); |
| |
| return solo_enc; |
| } |
| |
| static void solo_enc_free(struct solo_enc_dev *solo_enc) |
| { |
| if (solo_enc == NULL) |
| return; |
| |
| video_unregister_device(solo_enc->vfd); |
| kfree(solo_enc); |
| } |
| |
| int solo_enc_v4l2_init(struct solo_dev *solo_dev) |
| { |
| int i; |
| |
| for (i = 0; i < solo_dev->nr_chans; i++) { |
| solo_dev->v4l2_enc[i] = solo_enc_alloc(solo_dev, i); |
| if (IS_ERR(solo_dev->v4l2_enc[i])) |
| break; |
| } |
| |
| if (i != solo_dev->nr_chans) { |
| int ret = PTR_ERR(solo_dev->v4l2_enc[i]); |
| while (i--) |
| solo_enc_free(solo_dev->v4l2_enc[i]); |
| return ret; |
| } |
| |
| /* D1@MAX-FPS * 4 */ |
| solo_dev->enc_bw_remain = solo_dev->fps * 4 * 4; |
| |
| dev_info(&solo_dev->pdev->dev, "Encoders as /dev/video%d-%d\n", |
| solo_dev->v4l2_enc[0]->vfd->num, |
| solo_dev->v4l2_enc[solo_dev->nr_chans - 1]->vfd->num); |
| |
| return 0; |
| } |
| |
| void solo_enc_v4l2_exit(struct solo_dev *solo_dev) |
| { |
| int i; |
| |
| solo_irq_off(solo_dev, SOLO_IRQ_MOTION); |
| |
| for (i = 0; i < solo_dev->nr_chans; i++) |
| solo_enc_free(solo_dev->v4l2_enc[i]); |
| } |