/*
 *  NINTENDO64 SAMPLE PROGRAM
 *
 *  FILE : dd_bank.c
 *
 *  Copyright (C) 1998, NINTENDO Co,Ltd.
 */


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ultra64.h>
#include <PR/leo.h>
#include "dd_audio.h"


/*
 * 関数プロトタイプ宣言
 */
void syserror(char *);

u32  getLines(FILE *);
u32  readInfo(FILE *, u32 *);
u32  readFlag(FILE *, u8 *, u32 *);

void bankfileNew(ALBankFile *, u32);
void _bankNew(ALBank *, u32);
void _instNew(ALInstrument *, u32);
void _soundNew(ALSound *, u32);

void bankfileRecover(ALBankFile *, u32);
void _bankRecover(ALBank *, u32);
void _instRecover(ALInstrument *, u32);
void _soundRecover(ALSound *, u32);


/*
 * 定義
 */
#define DONE_FLAG    0xffffffff


/*
 * メイン
 */
void main(int argc, char *argv[])
{
  u8
    *devList,
    *waveData,
    *tmpFile,
    *tmpFile2,
    separater[] = " /:;";

  u32
    i,
    j,
    k,
    l,
    lines,
    length,
    tmpBase,
    id,
    *baseList,
    *sizeList,
    *addrList,
    *finalList;

  FILE 
    *infofd,
    *tblfd,
    *flagfd,
    *ctlfd,
    *tmpfd;

  ALBankFile
    *bankFile;
  ALBank
    *bank;
  ALInstrument
    *instrument;
  ALSound
    *sound;


  /* 引数のチェック */
  if(argc != 5)
    syserror("Usage:dd_bank <info file> <flag file> <.ctl> <.tbl>");


  /*
   * infoファイル
   */

  /* ファイルのオープン */
  /* infoファイル */
  if(!(infofd = fopen(argv[1], "r")))
    syserror("Error : can't open file : info file");
  /* .tblファイル */
  if(!(tblfd = fopen(argv[4], "rb")))
    syserror("Error : can't open file : .tbl file");

  /* 波形データ数の取得 */
  if(!(lines = getLines(infofd)))
    syserror("Error : bad file : info file");
  fseek(infofd, 0, SEEK_SET);

#ifdef DEBUG
  fprintf(stderr, "\nlines\n\t%ld\n", lines);
#endif

  /* baseList領域の確保 */
  if(!(baseList = malloc(sizeof(u32) * lines)))
    syserror("Error : can't allocate : baseList");

  /* baseListの構築 */
  if(lines != readInfo(infofd, baseList))
    syserror("Error : bad file : info file");

#ifdef DEBUG
  fprintf(stderr, "\nbaseList\n");
  for(i = 0; i < lines; i++)
    fprintf(stderr, "\t%x\n", baseList[i]);
#endif

  /* .tblファイルサイズの取得 */
  fseek(tblfd, 0, SEEK_END);
  length = (u32)ftell(tblfd);
  fseek(tblfd, 0, SEEK_SET);

  if(length < baseList[lines - 1])
    syserror("Error : miss match : info file and .tbl file");

  /* sizeList領域の確保 */
  if(!(sizeList = malloc(sizeof(u32) * lines)))
    syserror("Error : can't allocate : sizeList");

  /* sizeListの構築 */
  for(i = 0; i < (lines - 1); i++)
    sizeList[i] = baseList[i + 1] - baseList[i];
  sizeList[i] = length - baseList[i];

#ifdef DEBUG
  fprintf(stderr, "\nsizeList\n");
  for(i = 0; i < lines; i++)
    fprintf(stderr, "\t%x\n", sizeList[i]);
#endif

  /* infoファイルのクローズ */
  fclose(infofd);


  /*
   * flagファイル
   */

  /* flagファイルのオープン */
  if(!(flagfd = fopen(argv[2], "r")))
    syserror("Error : can't open file : flag file");

  /* devList及びaddrList領域の確保 */
  if(!(devList = malloc(sizeof(u8) * lines)))
    syserror("Error : can't allocate : devList");
  if(!(addrList = malloc(sizeof(u32) * lines)))
    syserror("Error : can't allocate : addrList");

#ifdef DEBUG
  for(i = 0; i < lines; i++)
    addrList[i] = 0;
#endif

  /* devList及びaddrListの構築 */
  if(lines != readFlag(flagfd, devList, addrList))
    syserror("Error : miss match : info file and flag file");

#ifdef DEBUG
  fprintf(stderr, "\ndevList\n");
  for(i = 0; i < lines; i++)
    fprintf(stderr, "\t%d\n", devList[i]);
  fprintf(stderr, "\naddrList\n");
  for(i = 0; i < lines; i++)
    fprintf(stderr, "\t%x\n", addrList[i]);
#endif

  /* flagファイルのクローズ */
  fclose(flagfd);


  /* 
   * .ctlファイルの更新 
   */

  /* .ctlファイルのオープン */
  if(!(ctlfd = fopen(argv[3], "rb+")))
    syserror("Error : can't open file : .ctl file");

  /* finalList領域の確保 */
  if(!(finalList = malloc(sizeof(u32) * lines)))
    syserror("Error : can't allocate : finalList");

  /* finalListの初期化 */
  for(i = 0; i < lines; i++)
    finalList[i] = baseList[i];

  /* bankFile領域の確保 */
  fseek(ctlfd, 0, SEEK_END);
  length = (u32)ftell(ctlfd);
  fseek(ctlfd, 0, SEEK_SET);

  if(!(bankFile = (ALBankFile *)malloc(sizeof(u8) * length)))
    syserror("Error : can't allocate : bankFile");

  /* .ctlファイルのロード */
  fread(bankFile, sizeof(u8), length, ctlfd);

#ifdef DEBUG
  fprintf(stderr, "\nbankCount\n\t%d\n", bankFile->bankCount);
#endif

  /* .ctlファイルの初期化*/
  bankfileNew(bankFile, (u32)bankFile);

  /* finalListの更新 */
  for(i = 0; i < bankFile->bankCount; i++){
    bank = bankFile->bankArray[i];

    /* percussion */
    if(bank->percussion){
      instrument = bank->percussion;

      for(j = 0; j < instrument->soundCount; j++){
	sound = instrument->soundArray[j];

	/* base値更新済か否かチェック */
	if(sound->wavetable->flags)
	  continue;

	tmpBase = (u32)sound->wavetable->base;

	/* 波形データのid取得 */
	for(id = 0; id < lines; id++)
	  if(tmpBase == baseList[id])
	    break;
	if(id >= lines)
	  syserror("Error : miss match : tmpBase");

	/* デバイスのチェック */
	if(!devList[id])
	  continue;

	/* DDROMのbase値の書換え */
	sound->wavetable->base = (u8 *)((addrList[id] + DDROM_WAVEDATA_START) | DDROM_FLAG);
	sound->wavetable->flags = 1;

	/* アドレスの書換え */
	finalList[id] = DONE_FLAG;
	
	for(k = id + 1; k < lines; k++){
	  if(finalList[k] == DONE_FLAG)
	    continue;
	  finalList[k] -= sizeList[id];
	}
      }
    }

    /* others */
    for(j = 0; j < bank->instCount; j++){
      instrument = bank->instArray[j];
      if(!instrument)
	continue;

      for(k = 0; k < instrument->soundCount; k++){
	sound = instrument->soundArray[k];

	/* base値更新済か否かチェック */
	if(sound->wavetable->flags)
	  continue;

	tmpBase = (u32)sound->wavetable->base;

	/* 波形データのid取得 */
	for(id = 0; id < lines; id++)
	  if(tmpBase == baseList[id])
	    break;
	if(id >= lines)
	  syserror("Error : miss match : tmpBase");

	/* デバイスのチェック */
	if(!devList[id])
	  continue;

	/* DDROMのbase値の書換え */
	sound->wavetable->base = (u8 *)((addrList[id] + DDROM_WAVEDATA_START) | DDROM_FLAG);
	sound->wavetable->flags = 1;

	/* アドレスの書換え */
	finalList[id] = DONE_FLAG;

	for(l = id + 1; l < lines; l++){
	  if(finalList[l] == DONE_FLAG)
	    continue;
	  finalList[l] -= sizeList[id];
	}
      }
    }
  }

#ifdef DEBUG
  fprintf(stderr, "\nfinalList\n");
  for(i = 0; i < lines; i++)
    fprintf(stderr, "\t%x\n", finalList[i]);
#endif

  /* .ctlファイル内base値の更新 */
  for(i = 0; i < bankFile->bankCount; i++){
    bank = bankFile->bankArray[i];

    /* percussion */
    if(bank->percussion){
      instrument = bank->percussion;

      for(j = 0; j < instrument->soundCount; j++){
	sound = instrument->soundArray[j];

	/* base値更新済か否かチェック */
	if(sound->wavetable->flags)
	  continue;

	tmpBase = (u32)sound->wavetable->base;
	
	/* 波形データのidの取得 */
	for(id = 0; id < lines; id++)
	  if(tmpBase == baseList[id])
	    break;
	if(id >= lines)
	  continue;

	sound->wavetable->base = (u8 *)finalList[id];
	sound->wavetable->flags = 1;
      }
    }

    /* others */
    for(j = 0; j < bank->instCount; j++){
      instrument = bank->instArray[j];
      if(!instrument)
	continue;

      for(k = 0; k < instrument->soundCount; k++){
	sound = instrument->soundArray[k];

	/* base値更新済か否かチェック */
	if(sound->wavetable->flags)
	  continue;

	tmpBase = (u32)sound->wavetable->base;
	
	/* 波形データのidの取得 */
	for(id = 0; id < lines; id++)
	  if(tmpBase == baseList[id])
	    break;
	if(id >= lines)
	  continue;

	sound->wavetable->base = (u8 *)finalList[id];
	sound->wavetable->flags = 1;
      }
    }
  }

  /* .ctlファイルの復元 */
  bankfileRecover(bankFile, (u32)bankFile);

  /* .ctlファイルのストア */
  fseek(ctlfd, 0, SEEK_SET);
  fwrite(bankFile, sizeof(u8), length, ctlfd);

   /* .ctlファイルのクローズ */
  fclose(ctlfd);


  /* 
   * .tblファイルの更新 
   */

  /* 一時ファイルの作成 */
  if(!(tmpFile = tmpnam(tmpFile)))
    syserror("Error : can't create file : tmpFile");

  tmpFile2= strtok(tmpFile, separater);
  while(tmpFile2){
    tmpFile = tmpFile2;
    tmpFile2 = strtok(NULL, separater);
  }

  if(!(tmpfd = fopen(tmpFile, "wb")))
    syserror("Error : can't create file : tmpFile");
  
  /* waveData領域の確保 */
  length = sizeList[0];
  for(i = 1; i < lines; i++)
    if(length < sizeList[i])
      length = sizeList[i];

  if(!(waveData = (u8 *)malloc(sizeof(u8) * length)))
    syserror("Error : can't allocate : waveData");

  /* .tblファイルの前詰め */
  for(i = 0; i < lines; i++){
    if(finalList[i] == DONE_FLAG)
      continue;

    fseek(tblfd, baseList[i], SEEK_SET);
    fread(waveData, sizeof(u8), sizeList[i], tblfd);
    fwrite(waveData, sizeof(u8), sizeList[i], tmpfd);
  }

  /* .tblファイル及び一時ファイルのクローズ */
  fclose(tblfd);
  fclose(tmpfd);

  /* ファイル名の変更 */
  if(remove(argv[4]))
    syserror("Error : can't remove : .tbl file");
  if(rename(tmpFile, argv[4]))
    syserror("Error : can't rename : .tbl file");


  /* 各領域の解放 */
  free(baseList);
  free(sizeList);
  free(devList);
  free(addrList);
  free(finalList);
  free(bankFile);
  free(waveData);
}


/*
 * エラー出力
 */
void syserror(char *out)
{
  fprintf(stderr, "%s\n", out);
  exit(1);
}
    

/*
 * infoファイルより波形データ数を取得
 */
u32 getLines(FILE *infofd)
{
  u32
    count = 0;
  char
    str[128],
    *strtmp,
    separater[] = " \t\n:;=()";
  
  while(fgets(str, 128, infofd)){
    strtmp = strtok(str, separater);
    if(strtmp)
      if(!strcmp(strtmp, "base"))
	count ++;
  }
  
  return count;
}


/*
 * infoファイルよりbaseアドレスを取得
 */
u32 readInfo(FILE *infofd, u32 *base)
{
  u32
    count = 0;
  char
    str[128],
    *strtmp,
    separater[] = " \t\n:;=()";

  while(fgets(str, 128, infofd)){
    strtmp = strtok(str, separater);

    if(strtmp)
      if(!strcmp(strtmp, "base")){
	strtmp = strtok(NULL, separater);

	if(!strtmp)
	  syserror("Error : bad file : info file");

	base[count++] = (u32)strtol(strtmp, NULL, 16);
      }
  }
  
  return count;
}


/*
 * flagファイルよりデバイス名(dev)及びアドレス(addr)を取得
 */
u32  readFlag(FILE *flagfd, u8 *dev, u32 *addr)
{
  u32
    count = 0;
  char
    str[128],
    *strtmp,
    separater[] = " \t\n:;=()";

  while(fgets(str, 128, flagfd)){
    if(str[0] == '#')
      continue;

    strtmp = strtok(str, separater);

    if(strtmp)
      if(!strcmp(strtmp, "DDROM")){
	strtmp = strtok(NULL, separater);
	
	if(!strtmp)
	  syserror("Error : bad file : flag file");
	
	dev[count] = 1;
	addr[count++] = (u32)strtol(strtmp, NULL, 16);
      }
      else if(!strcmp(strtmp, "OTHER"))
	dev[count++] = 0;
      else
	fprintf(stderr, "Warning : not supported device type : flag file\n");
  }
  
  return count;
}


/*
 * ALBankFileの初期化
 */
void bankfileNew(ALBankFile *bf, u32 offset)
{
  u32 i;

  for(i = 0; i < bf->bankCount; i++){
    bf->bankArray[i] = (ALBank *)((u32)bf->bankArray[i] + offset);

#ifdef DEBUG
  fprintf(stderr, "  instCount\n\t%d\n", bf->bankArray[i]->instCount);
#endif

    _bankNew(bf->bankArray[i], offset);
  }
}


/*
 * ALBankの初期化
 */
void _bankNew(ALBank *bk, u32 offset)
{
  u32 i;

  if(bk->flags)
    return;
  bk->flags = 1;

  if(bk->percussion){
    bk->percussion = (ALInstrument *)((u32)bk->percussion + offset);
    
#ifdef DEBUG
    fprintf(stderr, "    soundCount\n\t%d\n", bk->percussion->soundCount);
#endif
    
    _instNew(bk->percussion, offset);
  }

  for(i = 0; i < bk->instCount; i++){
    if(!bk->instArray[i])
      continue;

    bk->instArray[i] = (ALInstrument *)((u32)bk->instArray[i] + offset);

#ifdef DEBUG
      fprintf(stderr, "    soundCount\n\t%d\n", bk->instArray[i]->soundCount);
#endif

    _instNew(bk->instArray[i], offset);
  }
}


/*
 * ALInstrumentの初期化
 */
void _instNew(ALInstrument *it, u32 offset)
{
  u32 i;
  
  if(it->flags)
    return;
  it->flags = 1;

  for(i = 0; i < it->soundCount; i++){
    it->soundArray[i] = (ALSound *)((u32)it->soundArray[i] + offset);
    _soundNew(it->soundArray[i], offset);
  }
}


/*
 * ALSoundの初期化
 */
void _soundNew(ALSound *sd, u32 offset)
{
  if(sd->flags)
    return;
  sd->flags = 1;

  sd->wavetable = (ALWaveTable *)((u32)sd->wavetable + offset);

#ifdef DEBUG
	fprintf(stderr, "      base\n\t%x\n", sd->wavetable->base);
#endif
}


/*
 * ALBankFileの復元
 */
void bankfileRecover(ALBankFile *bf, u32 offset)
{
  u32 i;

  for(i = 0; i < bf->bankCount; i++){
    _bankRecover(bf->bankArray[i], offset);
    bf->bankArray[i] = (ALBank *)((u32)bf->bankArray[i] - offset);
  }
}


/*
 * ALBankの復元
 */
void _bankRecover(ALBank *bk, u32 offset)
{
  u32 i;

  if(!bk->flags)
    return;
  bk->flags = 0;

  if(bk->percussion){
    _instRecover(bk->percussion, offset);
    bk->percussion = (ALInstrument *)((u32)bk->percussion - offset);
  }

  for(i = 0; i < bk->instCount; i++){
    if(!bk->instArray[i])
      continue;

    _instRecover(bk->instArray[i], offset);
    bk->instArray[i] = (ALInstrument *)((u32)bk->instArray[i] - offset);
  }
}


/*
 * ALInstrumentの復元
 */
void _instRecover(ALInstrument *it, u32 offset)
{
  u32 i;
  
  if(!it->flags)
    return;
  it->flags = 0;

  for(i = 0; i < it->soundCount; i++){
    _soundRecover(it->soundArray[i], offset);
    it->soundArray[i] = (ALSound *)((u32)it->soundArray[i] - offset);
  }
}


/*
 * ALSoundの復元
 */
void _soundRecover(ALSound *sd, u32 offset)
{
  if(!sd->flags)
    return;
  sd->flags = 0;
  sd->wavetable->flags = 0;

  sd->wavetable = (ALWaveTable *)((u32)sd->wavetable - offset);
}