You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
k9copy/libk9copy/k9script.cpp

392 lines
10 KiB

//
// C++ Implementation: k9script
//
// Description:
//
//
// Author: Jean-Michel PETIT <k9copy@free.fr>, (C) 2006
//
// Copyright: See COPYING file that comes with this distribution
//
//
#include "k9script.h"
#include "k9dvd.h"
#include "k9dvdtitle.h"
#include "bswap.h"
#include "dvdread.h"
#include <stdlib.h>
k9Script::k9Script(ifo_handle_t *_ifo, k9DVD *_dvd )
{
m_ifo=_ifo;
m_dvd=_dvd;
}
k9Script::~k9Script()
{}
vm_cmd_t *k9Script::GOTO(char line,char register1,char value)
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0;cmd[1]=0xA1;cmd[2]=0;cmd[3]=register1;cmd[4]=0;
cmd[5]=value;
cmd[6]=0;
cmd[7]=line;
return &m_cmd;
}
vm_cmd_t *k9Script::GOTO(char line)
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0;cmd[1]=0x01;cmd[2]=0;cmd[3]=0;cmd[4]=0;
cmd[5]=0;
cmd[6]=0;
cmd[7]=line;
return &m_cmd;
}
vm_cmd_t *k9Script::JUMPTT(char title,char register1,char register2)
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x30;cmd[1]=0x22;cmd[2]=0;cmd[3]=0;cmd[4]=0;
cmd[5]=title;
cmd[6]=register1;
cmd[7]=register2;
return &m_cmd;
}
vm_cmd_t *k9Script::JUMPVMGM_MENU()
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x30;cmd[1]=0x06;cmd[2]=0;cmd[3]=0;cmd[4]=0;
cmd[5]=0x42;
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
vm_cmd_t *k9Script::CALLVMGM_MENU()
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x30;cmd[1]=0x08;cmd[2]=0;cmd[3]=0;cmd[4]=0x01;
cmd[5]=0x42;
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
vm_cmd_t *k9Script::EXIT(char register1,char register2)
{
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x30;cmd[1]=0x21;cmd[2]=0;cmd[3]=0;cmd[4]=0;
cmd[5]=0;cmd[6]=register1;cmd[7]=register2;
return &m_cmd;
}
vm_cmd_t *k9Script::setGPRM(char numReg,uint16_t value) {
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x71;cmd[1]=0;cmd[2]=0;
cmd[3]=numReg;
B2N_16(value);
memcpy(cmd+4,&value,sizeof(uint16_t));
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
vm_cmd_t *k9Script::setGPRMREG(char numReg,uint16_t value) {
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x61;cmd[1]=0;cmd[2]=0;
cmd[3]=numReg;
B2N_16(value);
memcpy(cmd+4,&value,sizeof(uint16_t));
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
vm_cmd_t *k9Script::setSTN(char numAudio,char numSubpicture) {
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x51;cmd[1]=0;cmd[2]=0;
cmd[3]=numAudio==0 ? 0 : 0x80+numAudio-1;
cmd[4]=numSubpicture==0 ? 0 : 0xC0+numSubpicture-1;
cmd[5]=0;
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
vm_cmd_t *k9Script::setSTN(char numAngle) {
uchar* cmd=(uchar*) &m_cmd;
cmd[0]=0x51;cmd[1]=0;cmd[2]=0;
cmd[3]=0;
cmd[4]=0;
cmd[5]=numAngle==0 ? 0 : 0x80+numAngle-1;
cmd[6]=0;
cmd[7]=0;
return &m_cmd;
}
void k9Script::updatePGCIUT() {
//first, free the old PGCIUT
ifoFree_PGCI_UT(m_ifo);
m_ifo->pgci_ut = (pgci_ut_t*) malloc(sizeof(pgci_ut_t));
pgci_ut_t* pgci_ut = m_ifo->pgci_ut;
pgci_ut->zero_1=0;
pgci_ut->nr_of_lus=1;
pgci_ut->lu = (pgci_lu_t*) malloc(sizeof(pgci_lu_t));
pgci_lu_t *lu =pgci_ut->lu;
uchar * lng=(uchar*) &(lu->lang_code);
lng[0]='n'; lng[1]='e';
lu->lang_extension=0;
lu->exists=0x80;
pgcit_t *pgcit=(pgcit_t*) malloc(sizeof(pgcit_t));
lu->pgcit=pgcit;
memset(pgcit,0,sizeof(pgcit_t));
pgcit->nr_of_pgci_srp=1;
pgci_srp_t * srp=(pgci_srp_t*)malloc(sizeof(pgci_srp_t));
memset(srp,0,sizeof(pgci_srp_t));
pgcit->pgci_srp=srp;
srp->entry_id=0x82;
pgc_t *pgc=(pgc_t*)malloc(sizeof(pgc_t));
srp->pgc=pgc;
memset(pgc,0,sizeof(pgc_t));
pgc_command_tbl_t * command_tbl=(pgc_command_tbl_t*) malloc(sizeof(pgc_command_tbl_t));
pgc->command_tbl=command_tbl;
pgc->command_tbl_offset=1;
memset(command_tbl,0,sizeof(pgc_command_tbl_t));
if (m_ifo->vmgi_mat != NULL)
updatePGCIUT_VMG(command_tbl);
else
updatePGCIUT_VTS(command_tbl);
}
void k9Script::updateFPPGC() {
ifoFree_FP_PGC(m_ifo);
pgc_t*pgc=(pgc_t*)malloc(sizeof(pgc_t));
m_ifo->first_play_pgc=pgc;
memset(pgc,0,sizeof(pgc_t));
pgc_command_tbl_t * command_tbl=(pgc_command_tbl_t*) malloc(sizeof(pgc_command_tbl_t));
pgc->command_tbl=command_tbl;
pgc->command_tbl_offset=1;
memset(command_tbl,0,sizeof(pgc_command_tbl_t));
// set GRPM(0)= First Title
addPreCmd(command_tbl,JUMPTT(m_dvd->getstart()->getnumTitle(),0,0));
// jump VMGM Menu
//addPreCmd(command_tbl,JUMPVMGM_MENU());
}
void k9Script::updatePGC(pgc_t *_pgc ,int numVTS,int numPGC) {
k9DVDTitle *title=NULL;
for (int i=0; i <m_dvd->gettitleCount();i++) {
title=m_dvd->gettitle(i);
if (title->getVTS()== numVTS && title->getTTN()==numPGC && title->getIndexed())
break;
}
pgc_command_tbl_t *command_tbl=_pgc->command_tbl;
if (command_tbl ==NULL) {
command_tbl=(pgc_command_tbl_t*) malloc(sizeof(pgc_command_tbl_t));
_pgc->command_tbl=command_tbl;
_pgc->command_tbl_offset=1;
memset(command_tbl,0,sizeof(pgc_command_tbl_t));
}
if (numPGC==0)
return;
for (int i=0; i < command_tbl->nr_of_pre;i++) {
//replace all JUMPs by a goto to the last line of cell commands ( link to next title)
char *cmd=(char*)&(command_tbl->pre_cmds[i]);
if (cmd[0]==0x30) {
char NOP[8]={0,0,0,0,0,0,0,0};
memcpy(cmd,NOP,8);
}
}
if (title->isSelected()) {
char numSubP=0,numAudio=0;
if (title->getDefSubtitle() !=NULL)
numSubP=title->getDefSubtitle()->getnum();
if (title->getDefAudio() !=NULL)
numAudio=title->getDefAudio()->getID();
if (numSubP+numAudio >0) {
char *c=(char*)command_tbl->pre_cmds;
bool binsert=false;
if (c==NULL)
binsert=true;
else if ( *c==0x51)
binsert=true;
if (binsert)
memcpy(command_tbl->pre_cmds,setSTN( numAudio,numSubP),8);
else
insertPreCmd(command_tbl,setSTN( numAudio,numSubP));
}
}
for (int i=0; i < command_tbl->nr_of_cell;i++) {
//replace all JUMPs by a goto to the last line of cell commands ( link to next title)
JumpVmg(&(command_tbl->cell_cmds[i]));
}
for (int i=0; i < command_tbl->nr_of_post;i++) {
//replace all JUMPs by a goto to the last line of cell commands ( link to next title)
JumpVmg(&(command_tbl->post_cmds[i]));
}
bool insertCall=true;
if (command_tbl->nr_of_post !=0) {
vm_cmd_t *cmd=&command_tbl->post_cmds[command_tbl->nr_of_post -1];
if (memcmp(cmd,CALLVMGM_MENU(),8)==0)
insertCall=false;
}
if (insertCall)
for (int i=0; i <m_dvd->gettitleCount();i++) {
k9DVDTitle *title=m_dvd->gettitle(i);
if (title->getVTS()== numVTS && title->getTTN()==numPGC && title->isSelected() && title->getIndexed()) {
addPostCmd(command_tbl,CALLVMGM_MENU());
break;
}
}
}
// replace the actual jump by a jump to he vmgm menu
void k9Script::JumpVmg(vm_cmd_t * command) {
char *cmd=(char*) command;
if (cmd[0]==0x30) {
cmd[1]=cmd[1] & 0xF0 + 0x08; //0x20 =conditionnal jump
cmd[2]=cmd[3]=0;
cmd[4]=0x01;
cmd[5]=0x42;
}
}
void k9Script::updatePGCIUT_VTS(pgc_command_tbl_t *command_tbl) {
addPreCmd( command_tbl, JUMPVMGM_MENU());
}
void k9Script::updatePGCIUT_VMG(pgc_command_tbl_t *command_tbl)
{
command_tbl->nr_of_pre=0;
command_tbl->pre_cmds=NULL;
//initialisation of the registers
for (int i=0;i<=15;i++)
if (i!=1)
addPreCmd( command_tbl,setGPRM( i,0));
addTitles(command_tbl);
}
void k9Script::addTitles(pgc_command_tbl_t *command_tbl) {
char NOP[8]={0,0,0,0,0,0,0,0};
//Set GPRM(1) = current title (SPRM(4))
addPreCmd(command_tbl,setGPRMREG(1,0x84));
int nbSelected=0;
for (int i=0;i<m_dvd->gettitleCount();i++) {
k9DVDTitle *title=m_dvd->gettitle(i);
if (title->isSelected() && title->getIndexed())
nbSelected++;
}
int a=0;
for (int i=0;i<m_dvd->gettitleCount();i++) {
k9DVDTitle *title=m_dvd->gettitle(i);
if (title->isSelected() && title->getIndexed()) {
//SI GPRM1=numTitle, GOTO i*3
addPreCmd(command_tbl,GOTO(16+nbSelected+a*3+2,1,title->getnumTitle()));
a++;
}
}
addPreCmd( command_tbl,EXIT(0,0));
for (int i=0;i<m_dvd->gettitleCount();i++) {
k9DVDTitle *title=m_dvd->gettitle(i);
if (title->isSelected() && title->getIndexed()) {
//initialization of register 1
addPreCmd(command_tbl,setGPRM(1,0));
//set default subtitle and audio stream
addPreCmd(command_tbl,(vm_cmd_t*)NOP);
//jump to title
if (title->getnextTitle()!=NULL)
addPreCmd(command_tbl,JUMPTT(title->getnextTitle()->getnumTitle(),0,0));
else
addPreCmd(command_tbl,EXIT(0,0));
}
}
}
void k9Script::addPreCmd(pgc_command_tbl_t *command_tbl,vm_cmd_t *cmd) {
command_tbl->nr_of_pre++;
if (command_tbl->pre_cmds == NULL)
command_tbl->pre_cmds=(vm_cmd_t*)malloc(sizeof(vm_cmd_t));
else
command_tbl->pre_cmds=(vm_cmd_t*) realloc(command_tbl->pre_cmds,sizeof(vm_cmd_t)*command_tbl->nr_of_pre);
memcpy(&(command_tbl->pre_cmds[command_tbl->nr_of_pre-1]),cmd,sizeof(vm_cmd_t));
}
void k9Script::insertPreCmd(pgc_command_tbl_t *command_tbl,vm_cmd_t *cmd) {
command_tbl->nr_of_pre++;
if (command_tbl->pre_cmds == NULL)
command_tbl->pre_cmds=(vm_cmd_t*)malloc(sizeof(vm_cmd_t));
else
command_tbl->pre_cmds=(vm_cmd_t*) realloc(command_tbl->pre_cmds,sizeof(vm_cmd_t)*command_tbl->nr_of_pre);
for (int i=command_tbl->nr_of_pre-2;i>=0;i--) {
memcpy(&(command_tbl->pre_cmds[i+1]),&(command_tbl->pre_cmds[i]),sizeof(vm_cmd_t));
char *cmd=(char*)&(command_tbl->pre_cmds[i+1]);
if (cmd[0]==0 && cmd[1]&0x01==0x01)
cmd[7]++;
}
memcpy(&(command_tbl->pre_cmds[0]),cmd,sizeof(vm_cmd_t));
}
void k9Script::addPostCmd(pgc_command_tbl_t *command_tbl,vm_cmd_t *cmd) {
command_tbl->nr_of_post++;
if (command_tbl->post_cmds == NULL)
command_tbl->post_cmds=(vm_cmd_t*)malloc(sizeof(vm_cmd_t));
else
command_tbl->post_cmds=(vm_cmd_t*) realloc(command_tbl->post_cmds,sizeof(vm_cmd_t)*command_tbl->nr_of_post);
memcpy(&(command_tbl->post_cmds[command_tbl->nr_of_post-1]),cmd,sizeof(vm_cmd_t));
}
void k9Script::addCellCmd(pgc_command_tbl_t *command_tbl,vm_cmd_t *cmd) {
command_tbl->nr_of_cell++;
if (command_tbl->cell_cmds == NULL)
command_tbl->cell_cmds=(vm_cmd_t*)malloc(sizeof(vm_cmd_t));
else
command_tbl->cell_cmds=(vm_cmd_t*) realloc(command_tbl->cell_cmds,sizeof(vm_cmd_t)*command_tbl->nr_of_cell);
memcpy(&(command_tbl->cell_cmds[command_tbl->nr_of_cell-1]),cmd,sizeof(vm_cmd_t));
}